z

tools.py

@@ -16,7 +16,28 @@ from typing import Optional, Tuple
 from huggingface_hub import list_models
 from pathlib import Path
 import json 
+from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_community.document_loaders import UnstructuredExcelLoader
 
+def excel_search(excel_path: str ) -> str:
+    """Given the path to an excel file, return the excel file contents as raw text.
+    
+    Args:
+        excel_path: The path to where the excel file is stored."""
+    loader = UnstructuredExcelLoader(excel_path, mode="elements")
+    raw_excel_docs = loader.load()
+    formatted_search_docs = "\n\n---\n\n".join(
+        [
+            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
+            for doc in raw_excel_docs
+        ])
+    return {"excel_results": formatted_search_docs}
+
+excel_search_tool = Tool(
+    name="excel_search",
+    func=excel_search,
+    description="Searches for and retrieves information in an Excel Document using the given path."
+)
 def wiki_search(query: str) -> str:
     """Search Wikipedia for a query and return maximum 2 results.