iamTangsang commited on
Commit
742160b
·
verified ·
1 Parent(s): fccde42

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -194,7 +194,7 @@ class BasicAgent:
194
  TextTransformer(),
195
  FileAttachmentQueryTool(model_name=GEMINI_MODEL_NAME),
196
  ]
197
- self.agent = CodeAgent(
198
  model=model,
199
  tools=tools,
200
  add_base_tools=False,
 
194
  TextTransformer(),
195
  FileAttachmentQueryTool(model_name=GEMINI_MODEL_NAME),
196
  ]
197
+ self.agent = ToolCallingAgent(
198
  model=model,
199
  tools=tools,
200
  add_base_tools=False,