Update agent.py

agent.py

@@ -5,6 +5,7 @@ from langchain_community.document_loaders import WikipediaLoader
 from llama_index.schema import Document
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_core.tools import tool
+from llama_index.agents import AgentWorkflow
 
 llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-VL-32B-Instruct")
 
@@ -55,6 +56,6 @@ def wiki_search(query: str) -> list:
     return llama_docs
 
 agent = AgentWorkflow.from_tools_or_functions(
-    [web_search],
+    [web_search, wiki_search],  # Add wiki_search here if needed
     llm=llm
 )