Upload 5 files

tools/citation_tool.py

@@ -0,0 +1,13 @@
+from smolagents import tool
+
+@tool
+class CiteTool:
+    name = "cite"
+    description = "Add citation to a given answer with a valid URL."
+
+    def use(self, input: str) -> str:
+        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

@@ -0,0 +1,9 @@
+from smolagents import tool
+
+@tool
+class FallbackTool:
+    name = "fallback"
+    description = "Handle unanswerable or unclear queries."
+
+    def use(self, _: str) -> str:
+        return "I'm sorry, I couldn't find the answer to your question. Could you rephrase or try something else?"

tools/math_tool.py

@@ -0,0 +1,13 @@
+from smolagents import tool
+
+@tool
+class PythonTool:
+    name = "python"
+    description = "Execute Python code to solve math problems."
+
+    def use(self, code: str) -> str:
+        try:
+            result = str(eval(code, {"__builtins__": {}}))
+            return f"Answer: {result}"
+        except Exception as e:
+            return f"Error: {str(e)}"

tools/search_tool.py

@@ -0,0 +1,15 @@
+from smolagents import tool
+from duckduckgo_search import DDGS
+
+@tool
+class WebSearchTool:
+    name = "web_search"
+    description = "Search the web for up-to-date factual information."
+
+    def use(self, query: str) -> str:
+        with DDGS() as ddgs:
+            results = ddgs.text(query)
+            output = []
+            for r in results[:3]:
+                output.append(f"{r['title']} - {r['href']}")
+            return "\n".join(output) if output else "No relevant results found."

tools/summarizer_tool.py

@@ -0,0 +1,15 @@
+from smolagents import tool
+from transformers import pipeline
+
+summarizer = pipeline("summarization")
+
+@tool
+class SummarizerTool:
+    name = "summarize"
+    description = "Summarize a long text into a short paragraph."
+
+    def use(self, input: str) -> str:
+        if len(input) < 50:
+            return input
+        result = summarizer(input, max_length=100, min_length=25, do_sample=False)
+        return result[0]['summary_text']