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
|