Upload 5 files

tools/citation_tool.py

@@ -1,16 +1,10 @@
 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()})"
-        except:
-            return "Could not format citation correctly."
+def cite(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()})"
+    except:
+        return "Could not format citation correctly."

tools/fallback_tool.py

@@ -1,12 +1,6 @@
 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?"
+def fallback(_: 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

@@ -1,16 +1,10 @@
 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}"
-        except Exception as e:
-            return f"Error: {str(e)}"
+def python(code: str) -> str:
+    """Evaluates a math expression using sandboxed eval."""
+    try:
+        result = str(eval(code, {"__builtins__": {}}))
+        return f"Answer: {result}"
+    except Exception as e:
+        return f"Error: {str(e)}"

tools/search_tool.py

@@ -2,15 +2,9 @@ from smolagents import tool
 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 = [f"{r['title']} - {r['href']}" for r in results[:3]]
-            return "\n".join(output) if output else "No relevant results found."
+def web_search(query: str) -> str:
+    """Returns top search results for a given query."""
+    with DDGS() as ddgs:
+        results = ddgs.text(query)
+        output = [f"{r['title']} - {r['href']}" for r in results[:3]]
+        return "\n".join(output) if output else "No relevant results found."

tools/summarizer_tool.py

@@ -4,15 +4,9 @@ from transformers import pipeline
 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)
-        return result[0]['summary_text']
+def summarize(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)
+    return result[0]['summary_text']