Bandook01 commited on
Commit
9fd4531
·
1 Parent(s): 711208d

use function agent

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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 = AgentWorkflow.from_tools_or_functions(
57
- tools_or_functions=[ReverseTextTool,RunPythonFileTool,download_server,wiki_tool,GetChessPosition],
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
  )