Wiki and arxiv tools added

app.py

@@ -11,6 +11,8 @@ from llama_index.llms.groq import Groq
 from llama_index.core.agent.workflow import FunctionAgent
 
 from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
+from llama_index.tools.wikipedia import WikipediaToolSpec
+from llama_index.tools.arxiv import ArxivToolSpec
 from llama_index.core.tools import FunctionTool
 
 # (Keep Constants as is)
@@ -21,12 +23,14 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         api_key = os.getenv("GROQ_API_KEY")
-        tool_spec = DuckDuckGoSearchToolSpec()
-        search_tool = FunctionTool.from_defaults(tool_spec.duckduckgo_full_search)
+        duck_spec = DuckDuckGoSearchToolSpec()
+        search_tool = FunctionTool.from_defaults(duck_spec.duckduckgo_full_search)
+        wiki_tool = WikipediaToolSpec().to_tool_list()
+        arxiv_tool = ArxivToolSpec().to_tool_list()
         llm = Groq(model="deepseek-r1-distill-llama-70b", api_key=api_key)
-        self.agent = FunctionAgent(tools=[search_tool], llm=llm, system_prompt='''
+        self.agent = FunctionAgent(tools=[search_tool, wiki_tool, arxiv_tool], llm=llm, system_prompt='''
         You are a general AI assistant.
-        I will ask you a question. You can use the search tool to help you with an answer.
+        I will ask you a question. You can use the tools at your disposal to help you with an answer.
         Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
         YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
         If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.