Spaces:
Runtime error
Runtime error
Gonzalo Lope
commited on
Commit
·
ee1ba96
1
Parent(s):
3eb5975
fixed imports and added a wikipedia and reverse string tool
Browse files- requirements.txt +2 -1
- smolagents_test_new.py +12 -3
requirements.txt
CHANGED
|
@@ -1,3 +1,4 @@
|
|
| 1 |
gradio
|
| 2 |
requests
|
| 3 |
-
smolagents
|
|
|
|
|
|
| 1 |
gradio
|
| 2 |
requests
|
| 3 |
+
smolagents
|
| 4 |
+
duckduckgo-search
|
smolagents_test_new.py
CHANGED
|
@@ -1,11 +1,20 @@
|
|
| 1 |
from huggingface_hub import login
|
| 2 |
-
from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
|
|
|
|
| 3 |
|
| 4 |
|
| 5 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
|
| 7 |
-
|
| 8 |
-
agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=InferenceClientModel())
|
| 9 |
|
| 10 |
|
| 11 |
if __name__ == "__main__":
|
|
|
|
| 1 |
from huggingface_hub import login
|
| 2 |
+
from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, WikipediaSearchTool
|
| 3 |
+
from smolagents import tool
|
| 4 |
|
| 5 |
|
| 6 |
|
| 7 |
+
@tool
|
| 8 |
+
def reverse_string(input_string: str) -> str:
|
| 9 |
+
"""This is a tool that given a string, returns the reversed string.
|
| 10 |
+
|
| 11 |
+
Args:
|
| 12 |
+
input_string (str): The string to reverse.
|
| 13 |
+
|
| 14 |
+
"""
|
| 15 |
+
return input_string[::-1]
|
| 16 |
|
| 17 |
+
agent = CodeAgent(tools=[DuckDuckGoSearchTool(),reverse_string,WikipediaSearchTool()], model=InferenceClientModel())
|
|
|
|
| 18 |
|
| 19 |
|
| 20 |
if __name__ == "__main__":
|