Update agents.py

agents.py

@@ -1,57 +1,61 @@
 from crewai import Agent, Task, Crew
 
-data_analyst_agent = Agent(
-    role="Data Analyst",
-    goal="Monitor and analyze market data in real-time "
-         "to identify trends and predict market movements.",
-    backstory="Specializing in financial markets, this agent "
-              "uses statistical modeling and machine learning "
-              "to provide crucial insights. With a knack for data, "
-              "the Data Analyst Agent is the cornerstone for "
-              "informing trading decisions.",
-    verbose=True,
-    allow_delegation=True,
-    tools = [scrape_tool, search_tool]
-)
+def get_data_analyst_agent():
+    return Agent(
+        role="Data Analyst",
+        goal="Monitor and analyze market data in real-time "
+             "to identify trends and predict market movements.",
+        backstory="Specializing in financial markets, this agent "
+                  "uses statistical modeling and machine learning "
+                  "to provide crucial insights. With a knack for data, "
+                  "the Data Analyst Agent is the cornerstone for "
+                  "informing trading decisions.",
+        verbose=True,
+        allow_delegation=True,
+        tools = [scrape_tool, search_tool]
+    )
 
-trading_strategy_agent = Agent(
-    role="Trading Strategy Developer",
-    goal="Develop and test various trading strategies based "
-         "on insights from the Data Analyst Agent.",
-    backstory="Equipped with a deep understanding of financial "
-              "markets and quantitative analysis, this agent "
-              "devises and refines trading strategies. It evaluates "
-              "the performance of different approaches to determine "
-              "the most profitable and risk-averse options.",
-    verbose=True,
-    allow_delegation=True,
-    tools = [scrape_tool, search_tool]
-)
+def get_trading_strategy_agent():
+    return Agent(
+        role="Trading Strategy Developer",
+        goal="Develop and test various trading strategies based "
+             "on insights from the Data Analyst Agent.",
+        backstory="Equipped with a deep understanding of financial "
+                  "markets and quantitative analysis, this agent "
+                  "devises and refines trading strategies. It evaluates "
+                  "the performance of different approaches to determine "
+                  "the most profitable and risk-averse options.",
+        verbose=True,
+        allow_delegation=True,
+        tools = [scrape_tool, search_tool]
+    )
 
-execution_agent = Agent(
-    role="Trade Advisor",
-    goal="Suggest optimal trade execution strategies "
-         "based on approved trading strategies.",
-    backstory="This agent specializes in analyzing the timing, price, "
-              "and logistical details of potential trades. By evaluating "
-              "these factors, it provides well-founded suggestions for "
-              "when and how trades should be executed to maximize "
-              "efficiency and adherence to strategy.",
-    verbose=True,
-    allow_delegation=True,
-    tools = [scrape_tool, search_tool]
-)
+def get_execution_agent():
+    return Agent(
+        role="Trade Advisor",
+        goal="Suggest optimal trade execution strategies "
+             "based on approved trading strategies.",
+        backstory="This agent specializes in analyzing the timing, price, "
+                  "and logistical details of potential trades. By evaluating "
+                  "these factors, it provides well-founded suggestions for "
+                  "when and how trades should be executed to maximize "
+                  "efficiency and adherence to strategy.",
+        verbose=True,
+        allow_delegation=True,
+        tools = [scrape_tool, search_tool]
+    )
 
-risk_management_agent = Agent(
-    role="Risk Advisor",
-    goal="Evaluate and provide insights on the risks "
-         "associated with potential trading activities.",
-    backstory="Armed with a deep understanding of risk assessment models "
-              "and market dynamics, this agent scrutinizes the potential "
-              "risks of proposed trades. It offers a detailed analysis of "
-              "risk exposure and suggests safeguards to ensure that "
-              "trading activities align with the firm’s risk tolerance.",
-    verbose=True,
-    allow_delegation=True,
-    tools = [scrape_tool, search_tool]
-)
+def get_risk_management_agent():
+    return Agent(
+        role="Risk Advisor",
+        goal="Evaluate and provide insights on the risks "
+             "associated with potential trading activities.",
+        backstory="Armed with a deep understanding of risk assessment models "
+                  "and market dynamics, this agent scrutinizes the potential "
+                  "risks of proposed trades. It offers a detailed analysis of "
+                  "risk exposure and suggests safeguards to ensure that "
+                  "trading activities align with the firm’s risk tolerance.",
+        verbose=True,
+        allow_delegation=True,
+        tools = [scrape_tool, search_tool]
+    )