Spaces:
Sleeping
Sleeping
tools updated
Browse files
app.py
CHANGED
|
@@ -25,8 +25,10 @@ class BasicAgent:
|
|
| 25 |
api_key = os.getenv("GROQ_API_KEY")
|
| 26 |
duck_spec = DuckDuckGoSearchToolSpec()
|
| 27 |
search_tool = FunctionTool.from_defaults(duck_spec.duckduckgo_full_search)
|
| 28 |
-
|
| 29 |
-
|
|
|
|
|
|
|
| 30 |
llm = Groq(model="deepseek-r1-distill-llama-70b", api_key=api_key)
|
| 31 |
self.agent = FunctionAgent(tools=[search_tool, wiki_tool, arxiv_tool], llm=llm, system_prompt='''
|
| 32 |
You are a general AI assistant.
|
|
|
|
| 25 |
api_key = os.getenv("GROQ_API_KEY")
|
| 26 |
duck_spec = DuckDuckGoSearchToolSpec()
|
| 27 |
search_tool = FunctionTool.from_defaults(duck_spec.duckduckgo_full_search)
|
| 28 |
+
wiki_spec = WikipediaToolSpec()
|
| 29 |
+
wiki_tool = FunctionTool.from_defaults(wiki_spec.search_data)
|
| 30 |
+
arxiv_spec = ArxivToolSpec()
|
| 31 |
+
arxiv_tool = FunctionTool.from_defaults(arxiv_spec.arxiv_query)
|
| 32 |
llm = Groq(model="deepseek-r1-distill-llama-70b", api_key=api_key)
|
| 33 |
self.agent = FunctionAgent(tools=[search_tool, wiki_tool, arxiv_tool], llm=llm, system_prompt='''
|
| 34 |
You are a general AI assistant.
|