Update features/insight_and_tasks/agents/task_extraction_agent.py

features/insight_and_tasks/agents/task_extraction_agent.py

@@ -48,7 +48,7 @@ class TaskExtractionAgent:
             current_date: The current date to use for quarter calculations. Defaults to today.
         """
         self.api_key = api_key # Store if needed by LlmAgent or other components
-        self.model_name = "gemini-2.5-pro-preview-03-25" #model_name or DEFAULT_AGENT_MODEL
+        self.model_name = model_name or DEFAULT_AGENT_MODEL
         self.current_date = current_date or datetime.utcnow().date() # Use date object for consistency
 
         # LlmAgent is initialized with dynamic instruction and output schema