Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, WikipediaSearchTool
+from smolagents import LiteLLMModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -14,21 +14,19 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        model = OpenAIServerModel(model_id="deepseek-chat",api_key="sk-04da63f756c9468182d278f52e33ad15", api_base="https://api.deepseek.com")
-
-        search_tool = DuckDuckGoSearchTool()
-        wiki_search = WikipediaSearchTool()
 
+        model = LiteLLMModel(
+            model_id="gemini/gemini-2.5-flash-preview-04-17",
+            )
         self.agent = CodeAgent(
-            model = model,
-            tools=[
-                search_tool,
-        		wiki_search,
-            ]
+            model=model,
+            tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
+            add_base_tools=True,
         )
-
+        print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
-        return self.agent.run(question) # type: ignore
+        answer = self.agent.run(question)
+        return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """