Update features/insight_and_tasks/agents/task_extraction_agent.py

features/insight_and_tasks/agents/task_extraction_agent.py

@@ -116,88 +116,26 @@ class TaskExtractionAgent:
         OUTPUT FORMAT:
         You MUST return a single JSON object that strictly conforms to the 'TaskExtractionOutput' Pydantic schema.
         This JSON object will contain:
-        
-{{
-  "current_quarter_info": "Q{quarter}, {days_remaining} days remaining",
-  "okrs": [
-    {{
-      "objective_description": "Clear, aspirational objective statement",
-      "key_results": [
+        - 'current_quarter_info': A string exactly like "Q{quarter}, {days_remaining} days remaining". (This is fixed based on the context above).
+        - 'okrs': A list, where each item is an 'OKR' object.
+        - 'overall_strategic_focus': (Optional) A brief summary of the main strategic themes emerging from the OKRs.
+        - 'generation_timestamp': (This will be auto-filled if you conform to the schema, or you can provide an ISO timestamp).
+        Example of a Task (ensure all fields from the Pydantic model are covered):
         {{
-          "key_result_description": "Specific, measurable key result",
-          "target_metric": "Metric name (e.g., 'Engagement Rate')",
-          "target_value": "Target value (e.g., '15% increase')",
-          "tasks": [
-            {{
-              "task_category": "Category name",
-              "task_description": "Specific action to take",
-              "objective_deliverable": "What will be delivered",
-              "effort": "Small|Medium|Large",
-              "timeline": "Immediate|Short-term|Medium-term|Long-term",
-              "responsible_party": "Who is responsible",
-              "success_criteria_metrics": "How success is measured",
-              "dependencies_prerequisites": "What needs to happen first (or null)",
-              "priority": "High|Medium|Low",
-              "priority_justification": "Why this priority level",
-              "why_proposed": "Connection to analysis insights",
-              "task_type": "initiative|tracking",
-              "data_subject": "follower_stats|posts|mentions|general (or null)"
-            }}
-          ]
+            "task_category": "Content Creation",
+            "task_description": "Launch a 3-part blog series on AI in Marketing.",
+            "objective_deliverable": "Objective: Increase thought leadership in AI marketing. Deliverable: 3 published blog posts.",
+            "effort": "Medium",
+            "timeline": "Short-term",
+            "responsible_party": "Content Team Lead",
+            "success_criteria_metrics": "Average 500 views per post, 10+ shares per post.",
+            "dependencies_prerequisites": "Keyword research for AI marketing topics completed.",
+            "priority": "High",
+            "priority_justification": "Addresses key strategic goal of establishing AI expertise.",
+            "why_proposed": "Analysis highlighted a gap in content related to AI, a high-interest area for our target audience.",
+            "task_type": "initiative",
+            "data_subject": "general"
         }}
-      ],
-      "objective_timeline": "Short-term|Medium-term|Long-term",
-      "objective_owner": "Team or role responsible"
-    }}
-  ],
-  "overall_strategic_focus": "Summary of main strategic themes",
-  "generation_timestamp": "{datetime.utcnow().isoformat()}"
-}}
-
-## CRITICAL REQUIREMENTS:
-
-### 1. OKR Structure (MANDATORY):
-- Create 2-4 Objectives maximum
-- Each Objective MUST have 2-4 Key Results
-- Each Key Result MUST have 2-5 Tasks
-- NO EMPTY ARRAYS - every section must contain items
-
-### 2. Task Requirements (ALL FIELDS MANDATORY):
-- task_category: Broad theme (e.g., "Content Strategy", "Audience Growth")
-- task_description: Specific action statement
-- objective_deliverable: Clear deliverable description
-- effort: Must be exactly "Small", "Medium", or "Large"
-- timeline: Must be exactly "Immediate", "Short-term", "Medium-term", or "Long-term"
-- responsible_party: Specific role or team
-- success_criteria_metrics: Measurable success indicators
-- dependencies_prerequisites: Prerequisites or null
-- priority: Must be exactly "High", "Medium", or "Low"
-- priority_justification: Brief explanation for priority
-- why_proposed: Direct link to analysis findings
-- task_type: Must be exactly "initiative" or "tracking"
-- data_subject: For tracking tasks, use "follower_stats", "posts", "mentions", or "general"; for initiatives, use "general" or null
-
-### 3. Key Result Requirements:
-- key_result_description: Specific, measurable outcome
-- target_metric: The metric being measured (required)
-- target_value: The target to achieve (required)
-- tasks: Array of tasks (minimum 2 tasks per key result)
-
-### 4. Timeline Considerations:
-Given {days_remaining} days left in Q{quarter}:
-- "Immediate": 1-2 weeks
-- "Short-term": Rest of current quarter
-- "Medium-term": Next quarter (3-6 months)
-- "Long-term": 6+ months
-
-### 5. Task Type Guidelines:
-- "initiative": New projects, campaigns, process changes
-- "tracking": Monitoring, measurement, ongoing analysis
-
-### 6. Priority Assignment Logic:
-- High: Critical for quarter goals, high impact, urgent
-- Medium: Important but not critical, moderate impact
-- Low: Nice to have, low impact, can be delayed
         Focus on quality, actionability, and strict adherence to the output schema.
         """