Update agent.py

agent.py

@@ -1,13 +1,13 @@
-import os
-from dotenv import load_dotenv
-load_dotenv()
-
 from langchain_core.tools import tool
 from langgraph.graph import StateGraph, START, MessagesState
 from langgraph.prebuilt import tools_condition, ToolNode
 from langchain_groq import ChatGroq
 from langchain_core.messages import HumanMessage, SystemMessage
 import math
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
 
 groq_api_key = os.getenv("GROQ_API_KEY")
 serpapi_api_key = os.getenv("SERPAPI_API_KEY")
@@ -17,28 +17,34 @@ serpapi_api_key = os.getenv("SERPAPI_API_KEY")
 # -------------------------
 @tool
 def add(a: float, b: float) -> float:
+    """Adds two numbers and returns the sum."""
     return a + b
 
 @tool
 def subtract(a: float, b: float) -> float:
+    """Subtracts second number from first and returns the difference."""
     return a - b
 
 @tool
 def multiply(a: float, b: float) -> float:
+    """Multiplies two numbers and returns the product."""
     return a * b
 
 @tool
 def divide(a: float, b: float) -> float:
+    """Divides first number by second and returns the quotient. Returns infinity if divisor is zero."""
     if b == 0:
         return float('inf')
     return a / b
 
 @tool
 def modulus(a: int, b: int) -> int:
+    """Returns the modulus (remainder) of a divided by b."""
     return a % b
 
 @tool
 def python_eval(code: str) -> str:
+    """Evaluates a Python expression and returns the result or error message."""
     try:
         result = eval(code)
         return f"Result: {result}"
@@ -47,18 +53,22 @@ def python_eval(code: str) -> str:
 
 @tool
 def translate_to_arabic(text: str) -> str:
+    """Returns a placeholder Arabic translation of the input text."""
     return f"Arabic translation of '{text}'"
 
 @tool
 def translate_to_english(text: str) -> str:
+    """Returns a placeholder English translation of the input text."""
     return f"English translation of '{text}'"
 
 @tool
 def summarize_text(text: str) -> str:
+    """Returns a summary (first 100 characters) of the input text."""
     return f"Summary: {text[:100]}..."
 
 @tool
 def analyze_sentiment(text: str) -> str:
+    """Analyzes the sentiment of the text and returns Positive, Negative, or Neutral."""
     if any(word in text.lower() for word in ["good", "great", "excellent", "happy"]):
         return "Sentiment: Positive"
     elif any(word in text.lower() for word in ["bad", "terrible", "sad", "hate"]):
@@ -67,6 +77,7 @@ def analyze_sentiment(text: str) -> str:
 
 @tool
 def speech_to_text_stub(audio: str) -> str:
+    """A placeholder tool to convert audio input to text."""
     return "Converted audio to text: (This is a placeholder result)"
 
 # -------------------------
@@ -110,5 +121,7 @@ def build_deepseek_graph():
     builder.add_conditional_edges("assistant", tools_condition)
     builder.add_edge("tools", "assistant")
 
-    ninu = builder.compile()  
+    ninu = builder.compile()
     return ninu
+
+