new models and search

agent.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool, Tool, VisitWebpageTool, HfApiModel
+from smolagents import CodeAgent, LiteLLMModel, GoogleSearchTool, WikipediaSearchTool, Tool, VisitWebpageTool, HfApiModel
 import os
 import json
 import requests
@@ -159,10 +159,13 @@ def final_answer_formatting(answer, question):
 
 
 web_agent = CodeAgent(
-    model=HfApiModel(),
+    model=LiteLLMModel(
+        model="mistral/mistral-large-latest",
+        api_key=os.getenv("MISTRAL_API_KEY"),
+    ),
     tools=[
         WikipediaSearchTool(),
-        DuckDuckGoSearchTool(), 
+        GoogleSearchTool(provider="serpapi"), 
         VisitWebpageTool()
     ],
     add_base_tools=False,

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

@@ -1,35 +0,0 @@
-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)}")