Update features/insight_and_tasks/agents/task_extraction_model.py

features/insight_and_tasks/agents/task_extraction_model.py

@@ -144,7 +144,8 @@ TEXT TO ANALYZE:
     if response_json_text:
         try:
             # Validate and parse the JSON response using the Pydantic model
-            return TaskExtractionOutput.model_validate_json(response_json_text)
+            task_output = TaskExtractionOutput.model_validate_json(response_json_text)
+            return task_output, quarter, year, days_remaining
         except Exception as e: # Catch Pydantic validation errors or JSON parsing errors
             raise ValueError(f"Failed to parse or validate API response: {e}\nRaw response text: {response_json_text}")
     else: