Switched to OpenAI model

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import ToolCallingAgent, HfApiModel, DuckDuckGoSearchTool
+from smolagents import OpenAIServerModel, ToolCallingAgent, HfApiModel, DuckDuckGoSearchTool
 
 
 # (Keep Constants as is)
@@ -16,8 +16,15 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         # Initialize the agent with a model and tools
-        agent = ToolCallingAgent(tools=[DuckDuckGoSearchTool()], model=HfApiModel())
+        
+        model = OpenAIServerModel(
+            model_id=os.environ["MODEL_ID"],
+            api_base="https://api.openai.com/v1",
+            api_key=os.environ["OPENAI_API_KEY"],
+        )
+        agent = ToolCallingAgent(tools=[DuckDuckGoSearchTool()], model=model)
         self.agent = agent
+        print(f"Agent initialized with model ID: {os.environ['MODEL_ID']}")
         print(f"Agent initialized with tools: {self.agent.tools}")
 
     def __call__(self, question: str) -> str: