AgentCourseFinalProject / agent_course_tools.py
Saptadip Sarkar
fix: added tools
5164dc5
raw
history blame
632 Bytes
from llama_index.tools.wikipedia import WikipediaToolSpec
from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
from llama_index.core.tools.tool_spec.load_and_search import (
LoadAndSearchToolSpec,
)
wiki_spec = WikipediaToolSpec()
wiki_tool = LoadAndSearchToolSpec.from_defaults(wiki_spec.to_tool_list()[1]).to_tool_list()
ddgo_spec = DuckDuckGoSearchToolSpec()
ddgo_tool = LoadAndSearchToolSpec.from_defaults(ddgo_spec.to_tool_list()[1]).to_tool_list()
# Create the Agent with load/search tools
#agent = OpenAIAgent.from_tools(
# LoadAndSearchToolSpec.from_defaults(wiki_tool).to_tool_list(), verbose=True
#)