Update agent.py

agent.py

@@ -6,7 +6,7 @@ import pandas as pd
 from dotenv import load_dotenv
 from smolagents import (CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool,
                         LiteLLMModel, PythonInterpreterTool,
-                        WikipediaSearchTool)
+                        WikipediaSearchTool, OpenAIClient)
 from smolagents.tools import Tool
 from tabulate import tabulate
 
@@ -14,8 +14,14 @@ from tabulate import tabulate
 load_dotenv()
 
 # Initialize the model
-model = LiteLLMModel(
-    model_id=os.getenv("GEMINI_MODEL"), api_key=os.getenv("KEY")
+# model = LiteLLMModel(
+#     model_id=os.getenv("GEMINI_MODEL"), api_key=os.getenv("KEY")
+# )
+
+client = OpenAIClient(
+    base_url="https://router.huggingface.co/v1",
+    api_key=os.environ["HF_TOKEN"],
+    model="openai/gpt-oss-120b:novita"
 )
 
 
@@ -93,7 +99,7 @@ class GaiaAgent:
         ]
 
         self.agent = CodeAgent(
-            model=model,
+            llm_client=client,
             tools=tools,
             add_base_tools=True,
             additional_authorized_imports=["pandas", "numpy", "csv", "subprocess"],