Spaces:
Sleeping
Sleeping
use function agent
Browse files
app.py
CHANGED
|
@@ -14,6 +14,10 @@ from llama_index.core.agent.workflow import (
|
|
| 14 |
AgentStream,
|
| 15 |
ReActAgent,
|
| 16 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
|
| 19 |
# (Keep Constants as is)
|
|
@@ -53,8 +57,8 @@ class BasicAgent:
|
|
| 53 |
model="models/gemini-2.0-flash",
|
| 54 |
api_key=gemini_api_key,
|
| 55 |
)
|
| 56 |
-
agent =
|
| 57 |
-
|
| 58 |
llm=llm,
|
| 59 |
system_prompt=SYSTEM_PROMPT,
|
| 60 |
)
|
|
|
|
| 14 |
AgentStream,
|
| 15 |
ReActAgent,
|
| 16 |
)
|
| 17 |
+
from llama_index.core.agent.workflow import FunctionAgent
|
| 18 |
+
from llama_index.core.tools.tool_spec.load_and_search import (
|
| 19 |
+
LoadAndSearchToolSpec,
|
| 20 |
+
)
|
| 21 |
|
| 22 |
|
| 23 |
# (Keep Constants as is)
|
|
|
|
| 57 |
model="models/gemini-2.0-flash",
|
| 58 |
api_key=gemini_api_key,
|
| 59 |
)
|
| 60 |
+
agent = FunctionAgent(
|
| 61 |
+
tools=LoadAndSearchToolSpec.from_defaults([ReverseTextTool,RunPythonFileTool,download_server,wiki_tool,GetChessPosition]).to_tool_list(),
|
| 62 |
llm=llm,
|
| 63 |
system_prompt=SYSTEM_PROMPT,
|
| 64 |
)
|