Upload 5 files

tools/citation_tool.py

@@ -2,10 +2,13 @@ from smolagents import tool
 
 @tool
 class CiteTool:
+    """Tool to format answers with source citations."""
+
     name = "cite"
     description = "Add citation to a given answer with a valid URL."
 
     def use(self, input: str) -> str:
+        """Takes input as 'answer ||| url' and formats a markdown citation."""
         try:
             answer, url = input.split("|||")
             return f"{answer.strip()}\n\nSource: [{url.strip()}]({url.strip()})"

tools/fallback_tool.py

@@ -2,8 +2,11 @@ from smolagents import tool
 
 @tool
 class FallbackTool:
+    """Tool to return a default response if no tool applies."""
+
     name = "fallback"
     description = "Handle unanswerable or unclear queries."
 
     def use(self, _: str) -> str:
+        """Returns a polite fallback message."""
         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,10 +2,13 @@ from smolagents import tool
 
 @tool
 class PythonTool:
+    """Tool to solve math/code problems using Python eval."""
+
     name = "python"
     description = "Execute Python code to solve math problems."
 
     def use(self, code: str) -> str:
+        """Evaluates a math expression using sandboxed eval."""
         try:
             result = str(eval(code, {"__builtins__": {}}))
             return f"Answer: {result}"

tools/search_tool.py

@@ -3,10 +3,13 @@ from duckduckgo_search import DDGS
 
 @tool
 class WebSearchTool:
+    """Tool to search the web for factual answers."""
+
     name = "web_search"
     description = "Search the web for up-to-date factual information."
 
     def use(self, query: str) -> str:
+        """Returns top search results for a given query."""
         with DDGS() as ddgs:
             results = ddgs.text(query)
             output = []

tools/summarizer_tool.py

@@ -5,10 +5,13 @@ summarizer = pipeline("summarization")
 
 @tool
 class SummarizerTool:
+    """Tool to summarize long text into concise form."""
+
     name = "summarize"
     description = "Summarize a long text into a short paragraph."
 
     def use(self, input: str) -> str:
+        """Returns a summarized version of the input text."""
         if len(input) < 50:
             return input
         result = summarizer(input, max_length=100, min_length=25, do_sample=False)