Upload agent

agent.json

@@ -8,12 +8,14 @@
         "final_answer"
     ],
     "model": {
-        "class": "HfApiModel",
+        "class": "LiteLLMModel",
         "data": {
-            "last_input_token_count": null,
-            "last_output_token_count": null,
-            "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct",
-            "provider": null
+            "temperature": 0.2,
+            "max_tokens": 500,
+            "last_input_token_count": 14218,
+            "last_output_token_count": 211,
+            "model_id": "anthropic/claude-3-haiku-20240307",
+            "api_base": null
         }
     },
     "managed_agents": {},
@@ -43,22 +45,22 @@
     "name": null,
     "description": null,
     "requirements": [
+        "duckduckgo_search",
         "markdownify",
         "requests",
-        "smolagents",
-        "duckduckgo_search"
+        "smolagents"
     ],
     "authorized_imports": [
-        "unicodedata",
-        "stat",
+        "statistics",
         "re",
-        "time",
-        "itertools",
-        "collections",
         "random",
+        "stat",
+        "collections",
+        "time",
+        "unicodedata",
+        "math",
         "datetime",
-        "queue",
-        "statistics",
-        "math"
+        "itertools",
+        "queue"
     ]
 }

app.py

@@ -1,6 +1,6 @@
 import yaml
 import os
-from smolagents import GradioUI, CodeAgent, HfApiModel
+from smolagents import GradioUI, CodeAgent, LiteLLMModel
 
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -14,9 +14,11 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
 
-model = HfApiModel(
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-provider=None,
+model = LiteLLMModel(
+temperature=0.2,
+max_tokens=500,
+model_id='anthropic/claude-3-haiku-20240307',
+api_base=None,
 )
 
 web_search = WebSearch()

requirements.txt

@@ -1,4 +1,4 @@
+duckduckgo_search
 markdownify
 requests
 smolagents
-duckduckgo_search