feat: add thinking budget parameter to Data Arranger agent model configuration

config/settings.py

@@ -36,7 +36,7 @@ class Settings:
     COORDINATOR_MODEL = os.getenv("COORDINATOR_MODEL", "gemini-2.5-pro")
     PROMPT_ENGINEER_MODEL = os.getenv("PROMPT_ENGINEER_MODEL", "gemini-2.5-pro")
     DATA_EXTRACTOR_MODEL = os.getenv("DATA_EXTRACTOR_MODEL", "gemini-2.5-pro")
-    DATA_ARRANGER_MODEL = os.getenv("DATA_ARRANGER_MODEL", "gemini-2.0-flash")
+    DATA_ARRANGER_MODEL = os.getenv("DATA_ARRANGER_MODEL", "gemini-2.5-pro")
     CODE_GENERATOR_MODEL = os.getenv("CODE_GENERATOR_MODEL", "gemini-2.0-flash")
 
     COORDINATOR_MODEL_THINKING_BUDGET=2048

workflow/financial_workflow.py

@@ -80,7 +80,7 @@ class FinancialDocumentWorkflow(Workflow):
     
     # Data Arranger Agent - Organizes data into categories for Excel
     data_arranger: Agent = Agent(
-        model=Gemini(id=settings.DATA_ARRANGER_MODEL,api_key=settings.GOOGLE_API_KEY),
+        model=Gemini(id=settings.DATA_ARRANGER_MODEL,thinking_budget=settings.DATA_ARRANGER_MODEL_THINKING_BUDGET,api_key=settings.GOOGLE_API_KEY),
         description="Financial data organization and analysis expert", 
         instructions=prompt_loader.load_instructions_as_list("agents/data_arranger"),
         tools=[FileTools()],  # FileTools for saving arranged data