AliA1997 commited on
Commit
b55a082
·
1 Parent(s): 65e0326

Fixed agent issues.

Browse files
Files changed (2) hide show
  1. app.py +3 -6
  2. requirements.txt +0 -1
app.py CHANGED
@@ -1,16 +1,13 @@
1
  import os
2
  import gradio as gr
3
  from huggingface_hub import login
4
- from smolagents import DuckDuckGoSearchTool, WikipediaSearchTool, ToolCallingAgent, InferenceClientModel
5
 
6
  login(os.environ.get('HF_TOKEN'))
7
 
8
  def create_agent():
9
- tools = [
10
- DuckDuckGoSearchTool(),
11
- WikipediaSearchTool()
12
- ]
13
- model = InferenceClientModel(model_id="gpt-4o-mini", tool_choice="auto")
14
  return ToolCallingAgent(tools=tools, model=model)
15
 
16
 
 
1
  import os
2
  import gradio as gr
3
  from huggingface_hub import login
4
+ from smolagents import DuckDuckGoSearchTool, ToolCallingAgent, InferenceClientModel
5
 
6
  login(os.environ.get('HF_TOKEN'))
7
 
8
  def create_agent():
9
+ tools = [DuckDuckGoSearchTool()]
10
+ model = InferenceClientModel()
 
 
 
11
  return ToolCallingAgent(tools=tools, model=model)
12
 
13
 
requirements.txt CHANGED
@@ -4,7 +4,6 @@ smolagents
4
  gradio>=4.44.0
5
  huggingface_hub>=0.24.0
6
  duckduckgo-search
7
- wikipedia-api
8
  pydub
9
  soundfile
10
  torch
 
4
  gradio>=4.44.0
5
  huggingface_hub>=0.24.0
6
  duckduckgo-search
 
7
  pydub
8
  soundfile
9
  torch