Add mathematical operation tools: add, multiply, and divide; implement python code parser tool

agent.py

@@ -16,6 +16,10 @@ from supabase.client import Client, create_client
 from tools.wiki_search import wiki_search
 from tools.web_search import web_search
 from tools.arvix_search import arvix_search
+from tools.python_code_parser import python_code_parser
+from tools.multiply import multiply
+from tools.add import add
+from tools.divide import divide
 # from tools.audio_transcription import audio_transcription
 
 load_dotenv()
@@ -48,6 +52,10 @@ tools = [
     wiki_search,
     web_search,
     arvix_search,
+    python_code_parser,
+    add,
+    multiply,
+    divide,
     # audio_transcription,
 ]
 

tools/add.py

@@ -0,0 +1,12 @@
+from langchain_core.tools import tool
+
+@tool
+def add(a: int, b: int) -> int:
+    """Add two numbers.
+    Args:
+        a: first int
+        b: second int
+    Returns:
+        The sum of a and b.
+    """
+    return a + b

tools/divide.py

@@ -0,0 +1,14 @@
+from langchain_core.tools import tool
+
+@tool
+def divide(a: int, b: int) -> int:
+    """Divide two numbers.
+    Args:
+        a: first int
+        b: second int
+    Returns:
+        The result of dividing a by b.
+    """
+    if b == 0:
+        raise ValueError("Impossible to divide by zero.")
+    return a / b

tools/multiply.py

@@ -0,0 +1,12 @@
+from langchain_core.tools import tool
+
+@tool
+def multiply(a: int, b: int) -> int:
+    """Multiply two numbers.
+    Args:
+        a: first int
+        b: second int
+    Returns:
+        The product of a and b.
+    """
+    return a * b

tools/python_code_parser.py

@@ -0,0 +1,35 @@
+"""Tool to parse Python code."""
+from langchain_core.tools import tool
+from langchain_community.document_loaders.generic import GenericLoader
+from langchain_community.document_loaders.parsers import LanguageParser
+
+@tool
+def python_code_parser(file_path: str) -> str:
+    """
+    Parse Python code to extract function names and their docstrings.
+
+    Args:
+        code: The Python code to parse.
+
+    Returns:
+        A string containing the function names and their docstrings.
+    """
+    if not file_path:
+        return "0"
+
+
+    loader = GenericLoader.from_filesystem(
+        file_path,
+        glob="**/*",
+        suffixes=[".py"],
+        parser=LanguageParser()
+    )
+    search_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 search_docs
+    ])
+
+    return {"audio_results": formatted_search_docs}