Update agent.py

agent.py

@@ -3,7 +3,7 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientMod
 from dotenv import load_dotenv
 load_dotenv()
 
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
 @tool
 def add(a: int, b: int) -> int:
@@ -60,10 +60,9 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         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,)
+        model_id="gpt-4o",
+        api_base="https://api.openai.com/v1",
+        api_key=os.environ["OPENAI_API_KEY"],)
         
         self.agent = CodeAgent(
             tools=[add,subtract,multiply,divide,DuckDuckGoSearchTool()],