methunraj commited on
Commit
c9b74e0
·
1 Parent(s): 0daea93

refactor: remove thinking_budget parameter from Gemini model initialization in data arranger agent

Browse files
Files changed (1) hide show
  1. workflow/financial_workflow.py +1 -1
workflow/financial_workflow.py CHANGED
@@ -81,7 +81,7 @@ class FinancialDocumentWorkflow(Workflow):
81
 
82
  # Data Arranger Agent - Organizes data into categories for Excel
83
  data_arranger: Agent = Agent(
84
- model=Gemini(id=settings.DATA_ARRANGER_MODEL,thinking_budget=settings.DATA_ARRANGER_MODEL_THINKING_BUDGET,api_key=settings.GOOGLE_API_KEY),
85
  description="Financial data organization and analysis expert",
86
  instructions=prompt_loader.load_instructions_as_list("agents/data_arranger"),
87
  tools=[FileTools()], # FileTools for saving arranged data
 
81
 
82
  # Data Arranger Agent - Organizes data into categories for Excel
83
  data_arranger: Agent = Agent(
84
+ model=Gemini(id=settings.DATA_ARRANGER_MODEL,api_key=settings.GOOGLE_API_KEY),
85
  description="Financial data organization and analysis expert",
86
  instructions=prompt_loader.load_instructions_as_list("agents/data_arranger"),
87
  tools=[FileTools()], # FileTools for saving arranged data