Update agent.py

agent.py

@@ -11,14 +11,13 @@ import cmath
 import pandas as pd
 import uuid
 import numpy as np
-# from code_interpreter import CodeInterpreter
+
 from PIL import Image
 import pandasql as psql
 
-# interpreter_instance = CodeInterpreter()
 
 
-"""Langraph"""
+### Langraph ###
 from langgraph.graph import START, StateGraph, MessagesState
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader
@@ -31,18 +30,18 @@ from langchain_huggingface import (
     HuggingFaceEndpoint,
     HuggingFaceEmbeddings,
 )
-# from langchain_community.vectorstores import SupabaseVectorStore
+
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
-# from supabase.client import Client, create_client
+
 from langchain.vectorstores import Chroma
 import os
 os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"
 
 load_dotenv()
 
-### =============== BROWSER TOOLS =============== ###
+### =============== SERACH TOOLS =============== ###
 
 def _format_documents_as_pseudo_xml(documents, result_tag="Document", truncate_chars=None) -> str:
     """
@@ -129,101 +128,6 @@ def arxiv_search(query: str) -> str:
     except Exception as e:
         return f"Error during Arxiv search: {str(e)}"
 
-# from typing import Dict, Any
-# import pandas as pd
-# from langchain_core.tools import tool
-
-
-# def _format_stdout(result: Dict[str, Any]) -> str:
-#     if stdout := result.get("stdout"):
-#         return "\n**Standard Output:**\n```\n" + stdout.strip() + "\n```"
-#     return ""
-
-
-# def _format_stderr(result: Dict[str, Any]) -> str:
-#     if stderr := result.get("stderr"):
-#         return "\n**Standard Error (if any):**\n```\n" + stderr.strip() + "\n```"
-#     return ""
-
-
-# def _format_execution_result(result: Dict[str, Any]) -> str:
-#     if "result" in result and result["result"] is not None:
-#         return "\n**Execution Result:**\n```\n" + str(result["result"]).strip() + "\n```"
-#     return ""
-
-
-# def _format_dataframes(result: Dict[str, Any]) -> str:
-#     if "dataframes" in result and result["dataframes"]:
-#         output = []
-#         for df_info in result["dataframes"]:
-#             df_preview = pd.DataFrame(df_info["head"])
-#             output.append(
-#                 f"\n**DataFrame `{df_info['name']}` (Shape: {df_info['shape']})**\n"
-#                 "First 5 rows:\n```\n" + str(df_preview) + "\n```"
-#             )
-#         return "\n".join(output)
-#     return ""
-
-
-# def _format_plots(result: Dict[str, Any]) -> str:
-#     if "plots" in result and result["plots"]:
-#         return f"\n**Generated {len(result['plots'])} plot(s)** (Image data returned separately)"
-#     return ""
-
-
-# @tool
-# def execute_code_multilang(code: str, language: str = "python") -> Dict[str, Any]:
-#     """
-#     Execute code in multiple languages and return a structured result.
-
-#     Args:
-#         code (str): The source code to execute.
-#         language (str): Language of the code. Supported: "python", "bash", "sql", "c", "java".
-
-#     Returns:
-#         Dict[str, Any]: Execution summary with status, language, raw result, and formatted output.
-#     """
-#     supported_languages = ["python", "bash", "sql", "c", "java"]
-#     language = language.lower()
-
-#     if language not in supported_languages:
-#         return {
-#             "status": "error",
-#             "summary": f"❌ Unsupported language: `{language}`. Supported: {', '.join(supported_languages)}",
-#             "language": language
-#         }
-
-#     try:
-#         result = interpreter_instance.execute_code(code, language=language)
-#     except Exception as e:
-#         return {
-#             "status": "error",
-#             "summary": f"❌ Execution crashed with error: {str(e)}",
-#             "language": language
-#         }
-
-#     status = result.get("status", "error")
-#     output_parts = []
-
-#     if status == "success":
-#         output_parts.append(f"✅ Code executed successfully in **{language.upper()}**")
-#         output_parts.append(_format_stdout(result))
-#         output_parts.append(_format_stderr(result))
-#         output_parts.append(_format_execution_result(result))
-#         output_parts.append(_format_dataframes(result))
-#         output_parts.append(_format_plots(result))
-#     else:
-#         output_parts.append(f"❌ Code execution failed in **{language.upper()}**")
-#         output_parts.append(_format_stderr(result))
-
-#     summary = "\n".join(filter(None, output_parts))
-
-#     return {
-#         "status": status,
-#         "language": language,
-#         "summary": summary,
-#         "raw_result": result,
-#     }
 
 ### =============== MATHEMATICAL TOOLS =============== ###
 
@@ -344,8 +248,6 @@ def validate_file_path(file_path: str) -> None:
         raise ValueError("File is too large to analyze securely.")
 
 
-# ========== Tools ========== #
-
 @tool
 def save_and_read_file(content: str, filename: Optional[str] = None) -> str:
     """
@@ -525,12 +427,6 @@ tools = [
     extract_text_from_image,
     analyze_csv_file,
     analyze_excel_file,
-    # execute_code_multilang,
-    # analyze_image,
-    # transform_image,
-    # draw_on_image,
-    # generate_simple_image,
-    # combine_images,
 ]
 
 
@@ -593,7 +489,7 @@ def build_graph(provider: str = "groq"):
 
 # test
 if __name__ == "__main__":
-    question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
+    question = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
     graph = build_graph(provider="groq")
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({"messages": messages})