Michael commited on
Commit
6f8b054
·
1 Parent(s): f7753f2

add search tool

Browse files
Files changed (2) hide show
  1. app.py +4 -2
  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