Update features/insight_and_tasks/agents/task_extraction_model.py

features/insight_and_tasks/agents/task_extraction_model.py

@@ -30,28 +30,37 @@ def create_example_structure():
         "current_quarter_info": "Q2 2025, 24 days remaining",
         "okrs": [
             {
-                "objective": "Improve LinkedIn employer branding performance",
-                "data_subject": "linkedin_performance",
+                "objective_description": "Improve LinkedIn employer branding performance",
+                "objective_timeline": "short",
+                "objective_owner": "marketing",
                 "key_results": [
                     {
-                        "description": "Increase monthly follower growth by 50%",
+                        "key_result_description": "Increase monthly follower growth by 50%",
+                        "target_metric": "+50 followers",
                         "target_value": "50% increase",
                         "current_value": "22 followers/month",
+                        "key_result_type": "performance",
+                        "data_subject": "posts",
                         "tasks": [
                             {
-                                "title": "Increase posting frequency",
-                                "description": "Post 2-3 times per week consistently",
-                                "task_type": "content_creation",
+                                "task_description": "Increase posting frequency",
+                                "objective_deliverable": "Post 2-3 times per week consistently",
+                                "task_category": "content_creation",
                                 "priority": "high",
-                                "effort_level": "medium",
+                                "effort": "medium",
                                 "timeline": "this_quarter",
-                                "data_subject": "linkedin_performance"
+                                "data_subject": "linkedin_performance",
+                                "responsible_party": "marketing",
+                                "success_criteria_metrics": "post 3 times",
+                                "dependencies_prerequisites": "a linkedin account",
+                                "why_proposed": "because previous data show low post frequency and low follwoer gains"
                             }
                         ]
                     }
                 ]
             }
-        ]
+        ],
+        "overall_strategic_focus": "increase followers"
     }
 
 # --- Helper Function for Date Calculations ---
@@ -124,7 +133,7 @@ def extract_tasks_from_text(user_text_input: str, api_key: str) -> TaskExtractio
 
 Your output MUST be a valid JSON object that strictly conforms to the 'TaskExtractionOutput' schema provided.
 EXAMPLE STRUCTURE (follow this pattern exactly):
-{json.dumps(example_structure, indent=2)}
+{json.dumps(create_example_structure, indent=2)}
 
 CURRENT CONTEXTUAL INFORMATION (CRITICAL - Use these exact values in your output where specified):
 - Current Quarter: Q{quarter}