Update agent.py

agent.py

@@ -12,7 +12,7 @@ from typing import Any
 from dotenv import load_dotenv
 from google.generativeai import types, configure
 
-from smolagents import InferenceClientModel, LiteLLMModel, ToolCallingAgent, Tool, DuckDuckGoSearchTool
+from smolagents import InferenceClientModel, LiteLLMModel, CodeAgent, ToolCallingAgent, Tool, DuckDuckGoSearchTool
 
 # Load environment and configure Gemini
 load_dotenv()
@@ -194,7 +194,7 @@ class BasicAgent:
             TextTransformer(),
             FileAttachmentQueryTool(model_name=GEMINI_MODEL_NAME),
         ]
-        self.agent = ToolCallingAgent(
+        self.agent = CodeAgent(
             model=model,
             tools=tools,
             add_base_tools=False,