Michael commited on
Commit ·
6f8b054
1
Parent(s): f7753f2
add search tool
Browse files- app.py +4 -2
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -3,7 +3,7 @@ import os
|
|
| 3 |
import gradio as gr
|
| 4 |
import pandas as pd
|
| 5 |
import requests
|
| 6 |
-
from smolagents import CodeAgent, OpenAIModel
|
| 7 |
|
| 8 |
# Read any API keys
|
| 9 |
openai_api_key = os.environ["OPENAI_API_KEY"]
|
|
@@ -24,8 +24,10 @@ class BasicAgent:
|
|
| 24 |
api_key=openai_api_key,
|
| 25 |
)
|
| 26 |
|
|
|
|
|
|
|
| 27 |
self.agent = CodeAgent(
|
| 28 |
-
tools=[],
|
| 29 |
model=model,
|
| 30 |
)
|
| 31 |
|
|
|
|
| 3 |
import gradio as gr
|
| 4 |
import pandas as pd
|
| 5 |
import requests
|
| 6 |
+
from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIModel
|
| 7 |
|
| 8 |
# Read any API keys
|
| 9 |
openai_api_key = os.environ["OPENAI_API_KEY"]
|
|
|
|
| 24 |
api_key=openai_api_key,
|
| 25 |
)
|
| 26 |
|
| 27 |
+
search_tool = DuckDuckGoSearchTool()
|
| 28 |
+
|
| 29 |
self.agent = CodeAgent(
|
| 30 |
+
tools=[search_tool],
|
| 31 |
model=model,
|
| 32 |
)
|
| 33 |
|
requirements.txt
CHANGED
|
@@ -1,3 +1,4 @@
|
|
| 1 |
gradio
|
| 2 |
requests
|
| 3 |
-
smolagents[openai]
|
|
|
|
|
|
| 1 |
gradio
|
| 2 |
requests
|
| 3 |
+
smolagents[openai]
|
| 4 |
+
ddgs
|