Update langgraph_agent.py

langgraph_agent.py

@@ -6,6 +6,7 @@ from typing import Dict, List, Union
 
 from PIL import Image as PILImage
 from huggingface_hub import InferenceClient
+
 from langgraph.graph import START, StateGraph, MessagesState
 from langgraph.prebuilt import tools_condition, ToolNode
 from langchain_openai import ChatOpenAI
@@ -15,33 +16,38 @@ from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_core.tools import tool
 
-# Correct import for GoogleSearchAPIWrapper
 from langchain_google_community import GoogleSearchAPIWrapper
 
 @tool
 def multiply(a: int, b: int) -> int:
+    """Multiply two integers."""
     return a * b
 
 @tool
 def add(a: int, b: int) -> int:
+    """Add two integers."""
     return a + b
 
 @tool
 def subtract(a: int, b: int) -> int:
+    """Subtract the second integer from the first."""
     return a - b
 
 @tool
 def divide(a: int, b: int) -> float:
+    """Divide first integer by second; error if divisor is zero."""
     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 first integer by second."""
     return a % b
 
 @tool
 def wiki_search(query: str) -> dict:
+    """Search Wikipedia for a query and return up to 2 documents."""
     try:
         docs = WikipediaLoader(query=query, load_max_docs=2, lang="en").load()
         if not docs:
@@ -55,13 +61,12 @@ def wiki_search(query: str) -> dict:
         print(f"Error in wiki_search tool: {e}")
         return {"wiki_results": f"Error occurred while searching Wikipedia for '{query}'. Details: {str(e)}"}
 
-# Instantiate GoogleSearchAPIWrapper
 search = GoogleSearchAPIWrapper()
 
 @tool
 def google_web_search(query: str) -> str:
+    """Perform a web search (via Google Custom Search) and return results."""
     try:
-        # Use the run method of the GoogleSearchAPIWrapper instance
         return search.run(query)
     except Exception as e:
         print(f"Error in google_web_search tool: {e}")
@@ -69,6 +74,7 @@ def google_web_search(query: str) -> str:
 
 @tool
 def arvix_search(query: str) -> dict:
+    """Search arXiv for a query and return up to 3 paper excerpts."""
     docs = ArxivLoader(query=query, load_max_docs=3).load()
     formatted = "\n\n---\n\n".join(
         f'<Document source="{d.metadata["source"]}"/>\n{d.page_content[:1000]}'
@@ -85,6 +91,7 @@ else:
 
 @tool
 def read_file_content(file_path: str) -> Dict[str, str]:
+    """Reads the content of a file and returns its primary information. For text/code/excel, returns content. For media, returns a prompt to use specific tools."""
     try:
         _, file_extension = os.path.splitext(file_path)
         file_extension = file_extension.lower()
@@ -110,6 +117,7 @@ def read_file_content(file_path: str) -> Dict[str, str]:
 
 @tool
 def python_interpreter(code: str) -> Dict[str, str]:
+    """Executes Python code and returns its standard output. If there's an error during execution, it returns the error message."""
     old_stdout = io.StringIO()
     with contextlib.redirect_stdout(old_stdout):
         try:
@@ -123,6 +131,7 @@ def python_interpreter(code: str) -> Dict[str, str]:
 
 @tool
 def describe_image(image_path: str) -> Dict[str, str]:
+    """Generates a textual description for an image file (JPEG, JPG, PNG) using an image-to-text model from the Hugging Face Inference API. Requires HF_API_TOKEN environment variable to be set."""
     if not HF_INFERENCE_CLIENT:
         return {"error": "Hugging Face API token not configured for image description. Cannot use this tool."}
     try:
@@ -137,6 +146,7 @@ def describe_image(image_path: str) -> Dict[str, str]:
 
 @tool
 def transcribe_audio(audio_path: str) -> Dict[str, str]:
+    """Transcribes an audio file (e.g., MP3) to text using an automatic speech recognition model from the Hugging Face Inference API. Requires HF_API_TOKEN environment variable to be set."""
     if not HF_INFERENCE_CLIENT:
         return {"error": "Hugging Face API token not configured for audio transcription. Cannot use this tool."}
     try: