Upload 5 files

tools/citation_tool.py

@@ -2,7 +2,15 @@ from smolagents import tool
 
 @tool
 def cite(input: str) -> str:
-    """Takes input as 'answer ||| url' and formats a markdown citation."""
+    """
+    Takes input as 'answer ||| url' and formats a markdown citation.
+
+    Args:
+        input (str): Combined answer and URL separated by '|||'.
+
+    Returns:
+        str: Formatted citation string with markdown link.
+    """
     try:
         answer, url = input.split("|||")
         return f"{answer.strip()}\n\nSource: [{url.strip()}]({url.strip()})"

tools/fallback_tool.py

@@ -2,5 +2,13 @@ from smolagents import tool
 
 @tool
 def fallback(_: str) -> str:
-    """Returns a polite fallback message."""
+    """
+    Returns a polite fallback message when no specific tool applies.
+
+    Args:
+        _ (str): Unused input string.
+
+    Returns:
+        str: Default fallback response.
+    """
     return "I'm sorry, I couldn't find the answer to your question. Could you rephrase or try something else?"

tools/math_tool.py

@@ -2,7 +2,15 @@ from smolagents import tool
 
 @tool
 def python(code: str) -> str:
-    """Evaluates a math expression using sandboxed eval."""
+    """
+    Evaluates a math expression using sandboxed eval.
+
+    Args:
+        code (str): A simple Python expression to evaluate.
+
+    Returns:
+        str: The result of the evaluation or error message.
+    """
     try:
         result = str(eval(code, {"__builtins__": {}}))
         return f"Answer: {result}"

tools/search_tool.py

@@ -3,7 +3,15 @@ from duckduckgo_search import DDGS
 
 @tool
 def web_search(query: str) -> str:
-    """Returns top search results for a given query."""
+    """
+    Returns top search results for a given query.
+
+    Args:
+        query (str): The search string.
+
+    Returns:
+        str: Top search results from DuckDuckGo.
+    """
     with DDGS() as ddgs:
         results = ddgs.text(query)
         output = [f"{r['title']} - {r['href']}" for r in results[:3]]

tools/summarizer_tool.py

@@ -5,7 +5,15 @@ summarizer = pipeline("summarization")
 
 @tool
 def summarize(input: str) -> str:
-    """Returns a summarized version of the input text."""
+    """
+    Returns a summarized version of the input text.
+
+    Args:
+        input (str): The text to be summarized.
+
+    Returns:
+        str: Concise summary of the input.
+    """
     if len(input) < 50:
         return input
     result = summarizer(input, max_length=100, min_length=25, do_sample=False)