Update tools and CodeAgent

app.py

@@ -21,6 +21,12 @@ search_tool = DuckDuckGoSearchTool()
 def rag_search(query: str) -> str:
     """
     Retrieve relevant information from the local FAISS knowledge base.
+
+    Args:
+        query: The question or search query to retrieve relevant documents for.
+
+    Returns:
+        A string containing the most relevant question-answer pairs from the knowledge base.
     """
 
     agent = GLOBAL_AGENT
@@ -40,16 +46,22 @@ def rag_search(query: str) -> str:
         question = item.get("Question", "")
         answer = item.get("Final answer", "")
 
-        results.append(f"Question: {question}\nAnswer: {answer}")
+        results.append(
+            f"Question: {question}\nAnswer: {answer}"
+        )
 
     return "\n\n".join(results)
 
-
 @tool
 def calculator(expression: str) -> str:
     """
-    Evaluate mathematical expressions.
-    Example: 5*23+12
+    Evaluate a mathematical expression.
+
+    Args:
+        expression: A mathematical expression such as "5*23+12".
+
+    Returns:
+        The computed result as a string.
     """
 
     try:
@@ -62,6 +74,12 @@ def calculator(expression: str) -> str:
 def web_search(query: str) -> str:
     """
     Search the web for up-to-date information.
+
+    Args:
+        query: The search query.
+
+    Returns:
+        A short text snippet of the web search results.
     """
 
     try:
@@ -73,8 +91,17 @@ def web_search(query: str) -> str:
 
 
 @tool
-def image_reader(agent, image_path=None):
-    """Placeholder multimodal tool."""
+def image_reader(image_path: str) -> str:
+    """
+    Analyze an image.
+
+    Args:
+        image_path: Path to the image file.
+
+    Returns:
+        A placeholder message since image analysis is not implemented.
+    """
+
     return "IMAGE_ANALYSIS_NOT_IMPLEMENTED"