adding tools

agent.py

@@ -1,24 +1,34 @@
+import os
 from smolagents import (
     CodeAgent, InferenceClientModel, HfApiModel,
-    DuckDuckGoSearchTool
+    DuckDuckGoSearchTool, LiteLLMModel
 )
 
-# Initialize the Hugging Face model
-# model = InferenceClientModel()
-model = HfApiModel()
 
-tools = [
-    DuckDuckGoSearchTool(),
-]
-
-def create_agent():
+class MyAgent(CodeAgent):
     """
-    Create and return a CodeAgent instance with the specified tools and model.
+    Custom agent class that extends CodeAgent.
     """
-    # Initialize the CodeAgent with the tools and model
-    agent = CodeAgent(
-        tools = tools,
-        model = model
+    tools = [
+            DuckDuckGoSearchTool(),
+             ]
+    model = LiteLLMModel(
+        model_id="gemini/gemini-2.0-flash", api_key=os.getenv("GEMINI_API_KEY", "")
     )
-    return agent
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(
+            tools=self.tools,
+            model=self.model,
+            additional_authorized_imports=[
+                "markdownify",
+                "requests",
+                "pandas",
+                "io",
+            ],
+            name="MyAgent",
+            description = "A custom agent that can search the web and generate code.",
+            add_base_tools=True,
+            max_steps=30,
+        )
 

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 
-from agent import create_agent
+from agent import MyAgent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -32,7 +32,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = create_agent()  # Call the function to create your agent
+        agent = MyAgent()  # Call the function to create your agent
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None

tools.py

@@ -0,0 +1 @@
+from smolagents import tools