Update agent.py

agent.py

@@ -25,6 +25,7 @@ print(f"SUPABASE_SERVICE_KEY: {SUPABASE_SERVICE_KEY[:10]}..." if SUPABASE_SERVIC
 
 
 def get_supabase_client():
+    """Create and return a Supabase client from environment variables."""
     if not SUPABASE_URL or not SUPABASE_SERVICE_KEY:
         raise ValueError("Supabase environment variables are missing.")
     return create_client(SUPABASE_URL, SUPABASE_SERVICE_KEY)
@@ -32,38 +33,117 @@ def get_supabase_client():
 
 @tool
 def multiply(a: int, b: int) -> int:
+    """Multiply two integers and return the result.
+
+    Args:
+        a (int): First multiplier.
+        b (int): Second multiplier.
+
+    Returns:
+        int: The product of a and b.
+    """
     return a * b
 
+
 @tool
 def add(a: int, b: int) -> int:
+    """Add two integers and return the sum.
+
+    Args:
+        a (int): First addend.
+        b (int): Second addend.
+
+    Returns:
+        int: The sum of a and b.
+    """
     return a + b
 
+
 @tool
 def subtract(a: int, b: int) -> int:
+    """Subtract b from a and return the difference.
+
+    Args:
+        a (int): Minuend.
+        b (int): Subtrahend.
+
+    Returns:
+        int: The difference a - b.
+    """
     return a - b
 
+
 @tool
-def divide(a: int, b: int) -> int:
+def divide(a: int, b: int) -> float:
+    """Divide a by b and return the quotient.
+
+    Args:
+        a (int): Dividend.
+        b (int): Divisor.
+
+    Raises:
+        ValueError: If b is zero.
+
+    Returns:
+        float: The quotient of a divided by b.
+    """
     if b == 0:
         raise ValueError("Cannot divide by zero.")
     return a / b
 
+
 @tool
 def modulus(a: int, b: int) -> int:
+    """Return the remainder of dividing a by b.
+
+    Args:
+        a (int): Dividend.
+        b (int): Divisor.
+
+    Returns:
+        int: The remainder of a divided by b.
+    """
     return a % b
 
+
 @tool
 def wiki_search(query: str) -> str:
+    """Search Wikipedia for the query and return the combined content of top 2 documents.
+
+    Args:
+        query (str): Search query string.
+
+    Returns:
+        str: Combined content of top 2 Wikipedia documents separated by "---".
+    """
     docs = WikipediaLoader(query=query, load_max_docs=2).load()
     return "\n\n---\n\n".join([doc.page_content for doc in docs])
 
+
 @tool
 def web_search(query: str) -> str:
+    """Search the web using TavilySearchResults and return combined content of top 3 results.
+
+    Args:
+        query (str): Search query string.
+
+    Returns:
+        str: Combined content of top 3 web search results separated by "---".
+    """
     docs = TavilySearchResults(max_results=3).invoke(query=query)
     return "\n\n---\n\n".join([doc.page_content for doc in docs])
 
+
 @tool
 def arvix_search(query: str) -> str:
+    """Search Arxiv for the query and return combined content (up to 1000 chars) of top 3 documents.
+
+    Args:
+        query (str): Search query string.
+
+    Returns:
+        str: Combined content snippet of top 3 Arxiv documents separated by "---".
+    """
     docs = ArxivLoader(query=query, load_max_docs=3).load()
     return "\n\n---\n\n".join([doc.page_content[:1000] for doc in docs])