Update tasks.py

tasks.py

@@ -1,51 +1,62 @@
 from crewai import Task
 
-from agents import get_content_planner_agent, get_content_writer_agent, get_editor_agent
+from agents import get_data_analyst_agent, get_trading_strategy_agent, get_execution_agent, get_risk_management_agent
 
-def get_content_planner_task():
+def get_data_analysis_task():
     return Task(
         description=(
-            "1. Prioritize the latest trends, key players, "
-                "and noteworthy news on {topic}.\n"
-            "2. Identify the target audience, considering "
-                "their interests and pain points.\n"
-            "3. Develop a detailed content outline including "
-                "an introduction, key points, and a call to action.\n"
-            "4. Include SEO keywords and relevant data or sources."
-        ),
-        expected_output="A comprehensive content plan document "
-            "with an outline, audience analysis, "
-            "SEO keywords, and resources.",
-        agent=get_content_planner_agent()
+            "Continuously monitor and analyze market data for "
+            "the selected stock ({stock_selection}). "
+            "Use statistical modeling and machine learning to "
+            "identify trends and predict market movements."
+        ),
+        expected_output=(
+            "Insights and alerts about significant market "
+            "opportunities or threats for {stock_selection}."
+        ),
+        agent=get_data_analyst_agent(),
     )
 
-def get_content_writer_task():
+def get_trading_strategy_task():
     return Task(
         description=(
-            "1. Use the content plan to craft a compelling "
-                "blog post on {topic}.\n"
-            "2. Incorporate SEO keywords naturally.\n"
-    		"3. Sections/Subtitles are properly named "
-                "in an engaging manner.\n"
-            "4. Ensure the post is structured with an "
-                "engaging introduction, insightful body, "
-                "and a summarizing conclusion.\n"
-            "5. Proofread for grammatical errors and "
-                "alignment with the brand's voice.\n"
-        ),
-        expected_output="A well-written blog post "
-            "in markdown format, ready for publication, "
-            "each section should have 2 or 3 paragraphs.",
-        agent=get_content_writer_agent()
+            "Develop and refine trading strategies based on "
+            "the insights from the Data Analyst and "
+            "user-defined risk tolerance ({risk_tolerance}). "
+            "Consider trading preferences ({trading_strategy_preference})."
+        ),
+        expected_output=(
+            "A set of potential trading strategies for {stock_selection} "
+            "that align with the user's risk tolerance."
+        ),
+        agent=get_trading_strategy_agent(),
     )
 
-def get_editor_task():
+def get_execution_task():
     return Task(
-        description=("Proofread the given blog post for "
-                     "grammatical errors and "
-                     "alignment with the brand's voice."),
-        expected_output="A well-written blog post in markdown format, "
-                        "ready for publication, "
-                        "each section should have 2 or 3 paragraphs.",
-        agent=get_editor_agent()
+        description=(
+            "Analyze approved trading strategies to determine the "
+            "best execution methods for {stock_selection}, "
+            "considering current market conditions and optimal pricing."
+        ),
+        expected_output=(
+            "Detailed execution plans suggesting how and when to "
+            "execute trades for {stock_selection}."
+        ),
+        agent=get_execution_agent(),
+    )
+
+def get_risk_management_task():
+    return Task(
+        description=(
+            "Evaluate the risks associated with the proposed trading "
+            "strategies and execution plans for {stock_selection}. "
+            "Provide a detailed analysis of potential risks "
+            "and suggest mitigation strategies."
+        ),
+        expected_output=(
+            "A comprehensive risk analysis report detailing potential "
+            "risks and mitigation recommendations for {stock_selection}."
+        ),
+        agent=get_risk_management_agent(),
     )