Update agent.py

agent.py

@@ -1,19 +1,26 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, WikipediaSearchTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, WikipediaSearchTool, HfApiModel # OpenAIServerModel
 import asyncio
 
 class MagAgent:
     def __init__(self):
         """Initialize the MagAgent with search tools."""
         print("Initializing MagAgent with search tools...")
+        model = HfApiModel(
+            max_tokens=2096,
+            temperature=0.5,
+        #    model_id="gemini/gemini-2.0-flash-lite-001",
+            # api_key = os.environ["GEMINI_KEY"], #userdata.get("GEMINI_KEY")
+            model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/",
+            custom_role_conversions=None,
+        )
         self.agent = CodeAgent(
-            model=OpenAIServerModel(model_id="gpt-4o"), 
+            model= model  # OpenAIServerModel(model_id="gpt-4o"), 
             tools=[
                 DuckDuckGoSearchTool(),
                 WikipediaSearchTool()
             ]
         )
-        print("MagAgent initialized."
-)
+        print("MagAgent initialized.")
 
     async def __call__(self, question: str) -> str:
         """Process a question asynchronously using the MagAgent."""