methunraj
commited on
Commit
·
c9b74e0
1
Parent(s):
0daea93
refactor: remove thinking_budget parameter from Gemini model initialization in data arranger agent
Browse files
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,
|
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
|