simple smolagent with duck duck go search tool

app.py

@@ -5,8 +5,12 @@ import inspect
 import pandas as pd
 import time
 from google import genai
+from smolagents import CodeAgent, ToolCallingAgent, DuckDuckGoSearchTool, LiteLLMModel, PythonInterpreterTool, tool, TOOL_CALLING_SYSTEM_PROMPT,DuckDuckGoSearchTool
 
-client = genai.Client(api_key=os.getenv("API_KEY"))
+
+#client = genai.Client(api_key=os.getenv("API_KEY"))
+model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-exp",
+                     api_key=os.getenv("API_KEY"))
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -20,13 +24,23 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
 
-        response = client.models.generate_content(
-    model="gemini-2.5-flash-preview-04-17", contents=question
+#         response = client.models.generate_content(
+#     model="gemini-2.5-flash-preview-04-17", contents=question
+# )
+#         gemini_flash_answer = response.text
+
+        agent = CodeAgent(
+    tools=[DuckDuckGoSearchTool()],
+    model=model,
+    #additional_authorized_imports=["helium"],
+    #step_callbacks=[save_screenshot],
+    max_steps=20,
+    verbosity_level=2,
 )
-        gemini_flash_answer = response.text
+        answer = agent.run(question)
         #fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {gemini_flash_answer}")
-        return gemini_flash_answer
+        print(f"Agent returning fixed answer: {answer}")
+        return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """

requirements.txt

@@ -1,3 +1,4 @@
 gradio
 requests
-google-genai
+google-genai
+smolagents