Update tasks.py

tasks.py

@@ -1,66 +1,51 @@
 from crewai import Task
 
-from agents import get_data_analyst_agent, get_trading_strategy_agent, get_execution_agent, get_risk_management_agent
+from agents import get_planner, get_writer, get_editor
 
-def get_data_analysis_task():
-    # Task for Data Analyst Agent: Analyze Market Data
+def get_plan_task():
     return Task(
         description=(
-            "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(),
+            "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_planner(),
     )
 
-def get_strategy_development_task():
-    # Task for Trading Strategy Agent: Develop Trading Strategies
+def get_write_task():
     return Task(
         description=(
-            "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(),
+            "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_writer(),
     )
 
-def get_execution_planning_task():
-    # Task for Trade Advisor Agent: Plan Trade Execution
+def get_edit_task():
     return Task(
-        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_assessment_task():
-    # Task for Risk Advisor Agent: Assess Trading Risks
-    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(),
+        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()
     )