Update research_tools/web_search.py

research_tools/web_search.py

@@ -3,7 +3,7 @@ Web Search Tool using DuckDuckGo via smolagents with Mistral
 """
 from .base_tool import BaseTool
 from typing import Optional
-from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, VisitWebpageTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, VisitWebpageTool, OpenAIServerModel
 import os
 
 class WebSearchTool(BaseTool):
@@ -15,14 +15,12 @@ class WebSearchTool(BaseTool):
         
         try:
             # Create custom Mistral model for CodeAgent
-            from smolagents.models import OpenAIModel
-            
             mistral_key = os.getenv("MISTRAL_API_KEY")
             if mistral_key:
-                mistral_model = OpenAIModel(
+                mistral_model = OpenAIServerModel(
                     api_key=mistral_key,
-                    base_url="https://api.mistral.ai/v1",
-                    model_name="mistral-large-latest"
+                    api_base="https://api.mistral.ai/v1",
+                    model_id="mistral/mistral-large-latest"
                 )
             else:
                 # Fallback to InferenceClientModel if no Mistral key