Martin Bär commited on
Commit
353745d
·
1 Parent(s): 57aba38

Remove unnecessary imports

Browse files
Files changed (2) hide show
  1. app.py +0 -3
  2. basic_agent.py +1 -7
app.py CHANGED
@@ -3,9 +3,6 @@ import gradio as gr
3
  import requests
4
  import pandas as pd
5
 
6
- from llama_index.embeddings.huggingface import HuggingFaceEmbedding
7
- from llama_index.core import Settings
8
-
9
  from basic_agent import BasicAgent
10
 
11
  # (Keep Constants as is)
 
3
  import requests
4
  import pandas as pd
5
 
 
 
 
6
  from basic_agent import BasicAgent
7
 
8
  # (Keep Constants as is)
basic_agent.py CHANGED
@@ -56,7 +56,7 @@ class BasicAgent:
56
  can_handoff_to=["WikiAgent", "WebAgent", "StatsAgent", "AudioAgent", "ImageAgent"],
57
  )
58
 
59
- # Wikipedia tool does not return the tables from the page...
60
  wiki_spec = WikipediaToolSpec()
61
  wiki_search_tool = wiki_spec.to_tool_list()[1]
62
 
@@ -163,12 +163,6 @@ class BasicAgent:
163
  print(f"\n{'='*50}")
164
  print(f"🤖 Agent: {current_agent}")
165
  print(f"{'='*50}\n")
166
-
167
- # if isinstance(event, AgentStream):
168
- # if event.delta:
169
- # print(event.delta, end="", flush=True)
170
- # elif isinstance(event, AgentInput):
171
- # print("📥 Input:", event.input)
172
  elif isinstance(event, AgentOutput):
173
  if event.response.content:
174
  print("📤 Output:", event.response.content)
 
56
  can_handoff_to=["WikiAgent", "WebAgent", "StatsAgent", "AudioAgent", "ImageAgent"],
57
  )
58
 
59
+ # TODO Wikipedia tool does not return the tables from the page...
60
  wiki_spec = WikipediaToolSpec()
61
  wiki_search_tool = wiki_spec.to_tool_list()[1]
62
 
 
163
  print(f"\n{'='*50}")
164
  print(f"🤖 Agent: {current_agent}")
165
  print(f"{'='*50}\n")
 
 
 
 
 
 
166
  elif isinstance(event, AgentOutput):
167
  if event.response.content:
168
  print("📤 Output:", event.response.content)