Spaces:
Sleeping
Sleeping
Martin Bär
commited on
Commit
·
353745d
1
Parent(s):
57aba38
Remove unnecessary imports
Browse files- app.py +0 -3
- 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)
|