Update agent.py

agent.py

@@ -1,5 +1,60 @@
 from llama_index.core.tools import FunctionTool
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
+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
 
-llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
+llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-VL-32B-Instruct")
+
+@tool
+def web_search(query: str) -> list:
+    """Search Tavily for a query and return a maximum 3 results as LlamaIndex Documents.
+
+    Args:
+        query: The search query.
+    """
+    search_docs = TavilySearchResults(max_results=3).invoke(query=query)
+    
+    llama_docs = []
+    for doc in search_docs:
+        source = doc.metadata.get("source", "")
+        page = doc.metadata.get("page", "")
+        content = doc.page_content
+        full_metadata = {
+            "source": source,
+            "page": page
+        }
+        llama_docs.append(Document(text=content, metadata=full_metadata))
+
+    return llama_docs
+
+@tool
+def wiki_search(query: str) -> list:
+    """Search Wikipedia for a query and return maximum 2 results as LlamaIndex Documents.
+
+    Args:
+        query: The search query.
+    """
+    # Perform Wikipedia search using WikipediaLoader
+    search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
+    
+    # Create a list of LlamaIndex Document objects
+    llama_docs = []
+    for doc in search_docs:
+        source = doc.metadata.get("source", "")
+        page = doc.metadata.get("page", "")
+        content = doc.page_content
+        full_metadata = {
+            "source": source,
+            "page": page
+        }
+        llama_docs.append(Document(text=content, metadata=full_metadata))
+
+    return llama_docs
+
+agent = AgentWorkflow.from_tools_or_functions(
+    [web_search],
+    llm=llm
+)