Add Excel tool

agent.py

@@ -13,6 +13,7 @@ from supabase.client import Client, create_client
 from tools.math_tools import add, subtract, multiply, divide, modulus, power, sqrt
 from tools.search_tools import search_wikipedia, web_search
 from tools.image_video_tools import query_image
+from tools.file_tools import analyze_excel_file
 
 
 # embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") #  dim=768
@@ -40,7 +41,7 @@ def build_graph():
         max_retries=2,
         google_api_key=os.getenv("GOOGLE_API_KEY")  # Get API key from environment variable
         )
-    tools = [add, subtract, multiply, divide, modulus, power, sqrt, web_search, search_wikipedia, query_image]
+    tools = [add, subtract, multiply, divide, modulus, power, sqrt, web_search, search_wikipedia, query_image, analyze_excel_file]
 
     llm_with_tools = llm.bind_tools(tools)
 

tools/file_tools.py

@@ -0,0 +1,29 @@
+import pandas as pd
+from langchain_core.tools import tool
+
+@tool
+def analyze_excel_file(file_path: str, query: str) -> str:
+    """
+    Analyzes an Excel file using pandas and answer a question about it.
+    Args:
+        file_path (str): the path to the Excel file.
+        query (str): Question about the data
+    """
+    try:
+        # Read the Excel file
+        df = pd.read_excel(file_path)
+
+        # Run various analyses based on the query
+        result = (
+            f"Excel file loaded with {len(df)} rows and {len(df.columns)} columns.\n"
+        )
+        result += f"Columns: {', '.join(df.columns)}\n\n"
+
+        # Add summary statistics
+        result += "Summary statistics:\n"
+        result += str(df.describe())
+
+        return result
+
+    except Exception as e:
+        return f"Error analyzing Excel file: {str(e)}"