trying other inference provider

agent.py

@@ -6,8 +6,8 @@ Facultative
 - wiki_search
 - arxiv_search
 '''
-
-from smolagents import CodeAgent, HfApiModel, tool, DuckDuckGoSearchTool, MLXModel
+import os
+from smolagents import CodeAgent, tool, DuckDuckGoSearchTool, OpenAIServerModel
 
 
 @tool
@@ -92,15 +92,9 @@ def rounder(a:float, n:int) -> float:
 def get_agent() -> CodeAgent:
   search_tool = DuckDuckGoSearchTool()
 
-  model = MLXModel(
-    "mlx-community/Qwen2.5-Coder-32B-Instruct-4bit",
-    {
-        "temperature": 0.7,
-        "top_k": 20,  
-        "top_p": 0.8, 
-        "min_p": 0.05,
-        "num_ctx": 32768,
-    },
-)
+  model = OpenAIServerModel(
+    model_id='codestral-latest',
+    api_base="https://codestral.mistral.ai/v1/",
+    api_key=os.getenv('CODESTRAL_API_KEY'))
 
   return CodeAgent(tools=[add, subtract, multiply, divide, modulus, rounder, search_tool], model=model)

requirements.txt

@@ -1,4 +1,3 @@
 gradio
 requests
-smolagents[mlx-lm]
-duckduckgo-search
+smolagents