Update agents.py

agents.py

@@ -1,61 +1,62 @@
 from crewai import Agent, Task, Crew
 
-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]
-    )
-
-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]
-    )
-
-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]
-    )
-
-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]
-    )
+class Agents:
+    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]
+        )
+    
+    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]
+        )
+    
+    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]
+        )
+    
+    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]
+        )