model modification

agent.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool, Tool, VisitWebpageTool
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool, Tool, VisitWebpageTool, HfApiModel
 import os
 import json
 import requests
@@ -128,9 +128,9 @@ class ImageAnalysisTool(Tool):
 
 
 def final_answer_formatting(answer, question):
-    model = LiteLLMModel(
-        model_id="gemini/gemini-2.0-flash",
-        api_key=os.getenv("GOOGLE_API_KEY"),
+    client = InferenceClient(
+        provider="nebius",
+        api_key=os.getenv("HF_API_KEY"),
     )
 
     prompt = f"""
@@ -140,18 +140,27 @@ def final_answer_formatting(answer, question):
     Answer: {answer}
     """
 
-    messages = [
-    {"role": "user", "content": [{"type": "text", "text": prompt}]}
-    ]
+    completion = client.chat.completions.create(
+        model="mistralai/Mistral-Small-3.1-24B-Instruct-2503",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": prompt
+                    }
+                ]
+            }
+        ],
+    )
 
-    output = model(messages).content
-    return output
+    return completion.choices[0].message.content
 
 
 web_agent = CodeAgent(
-    model=LiteLLMModel(
-        model_id="gemini/gemini-2.0-flash",
-        api_key=os.getenv("GOOGLE_API_KEY"),
+    model=HfApiModel(
+        model_id='Qwen/Qwen3-32B'
     ),
     tools=[
         WikipediaSearchTool(),
@@ -169,9 +178,8 @@ web_agent = CodeAgent(
 )
 
 audio_agent = CodeAgent(
-    model=LiteLLMModel(
-        model_id="gemini/gemini-2.0-flash",
-        api_key=os.getenv("GOOGLE_API_KEY"),
+    model=HfApiModel(
+        model_id='Qwen/Qwen3-32B'
     ),
     tools=[AudioToTextTool()],
     add_base_tools=False,
@@ -183,9 +191,8 @@ audio_agent = CodeAgent(
 
 manager_agent = CodeAgent(
     name="manager_agent",
-    model=LiteLLMModel(
-        model_id="gemini/gemini-2.5-flash-preview-04-17",
-        api_key=os.getenv("GOOGLE_API_KEY"),
+    model=HfApiModel(
+        model_id='Qwen/Qwen3-235B-A22B'
     ),
     tools=[getFile(), LoadXlsxFileTool(), LoadTextFileTool(), ImageAnalysisTool()],
     managed_agents=[web_agent, audio_agent],

f918266a-b3e0-4914-865d-4faa564f1aef.py

@@ -0,0 +1,35 @@
+from random import randint
+import time
+
+class UhOh(Exception):
+    pass
+
+class Hmm:
+    def __init__(self):
+        self.value = randint(-100, 100)
+
+    def Yeah(self):
+        if self.value == 0:
+            return True
+        else:
+            raise UhOh()
+
+def Okay():
+    while True:
+        yield Hmm()
+
+def keep_trying(go, first_try=True):
+    maybe = next(go)
+    try:
+        if maybe.Yeah():
+            return maybe.value
+    except UhOh:
+        if first_try:
+            print("Working...")
+            print("Please wait patiently...")
+        time.sleep(0.1)
+        return keep_trying(go, first_try=False)
+
+if __name__ == "__main__":
+    go = Okay()
+    print(f"{keep_trying(go)}")

requirements.txt

@@ -4,4 +4,6 @@ pandas
 openpyxl
 huggingface-hub
 smolagents
-python-dotenv
+python-dotenv
+litellm
+wikipedia-api