Update agent.py

agent.py

@@ -1,5 +1,5 @@
 import os
-from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel, OpenAIServerModel
 
 @tool
 def add(a: int, b: int) -> int:
@@ -55,7 +55,11 @@ def divide(a: int, b: int) -> float:
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        llm = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
+        llm = OpenAIServerModel(
+            model_id="gemini-2.0-flash",
+            # Google Gemini OpenAI-compatible API base URL
+            api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
+            api_key=GEMINI_API_KEY,)
         
         self.agent = CodeAgent(
             tools=[add,subtract,multiply,divide,DuckDuckGoSearchTool()],