tt

app.py

@@ -1,9 +1,25 @@
 import os
 import gradio as gr
 import requests
-from smolagents import CodeAgent, InferenceClientModel, tool,ToolCallingAgent , PythonInterpreterTool,DuckDuckGoSearchTool,VisitWebpageTool
+from smolagents import CodeAgent, InferenceClientModel, tool,ToolCallingAgent ,LiteLLMModel, PythonInterpreterTool,DuckDuckGoSearchTool,VisitWebpageTool
 import inspect
 import pandas as pd
+import litellm
+
+key = os.environ['access_token']
+key_mistral = os.environ['mistral_token']
+
+
+# litellm._turn_on_debug()  
+
+# client = Mistral(api_key="RYZqsXtCJ7fXiZqG4bVtIGqSEP3L8Pxb")
+
+# 2. Wrapper SmolAgents
+mistral_model = LiteLLMModel(
+    model_id="mistral/mistral-small-latest",
+    api_key=key_mistral
+)
+
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -19,7 +35,7 @@ class BasicAgent:
     model_id="Qwen/Qwen3-32B",
     token=key)
         interpreter = PythonInterpreterTool()  
-        self.agentic = ToolCallingAgent(model=model,tools = [interpreter,DuckDuckGoSearchTool(),VisitWebpageTool()])
+        self.agentic = ToolCallingAgent(model=mistral_model,tools = [interpreter,DuckDuckGoSearchTool(),VisitWebpageTool()])
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")

requirements.txt

@@ -3,3 +3,4 @@ requests
 smolagents
 mistralai
 duckduckgo-search
+smolagents[litellm]