abstract tools

basic_agent.py

@@ -1,18 +1,10 @@
-from duckduckgo_search import DDGS
-
-# DuckDuckGo Search Tool
-def search_duckduckgo(query: str, max_results: int = 3) -> list[str]:
-    with DDGS() as ddgs:
-        results = ddgs.text(query, safesearch="Moderate")
-        return [r["body"] for r in results[:max_results]]
+from tools import search_duckduckgo
 
 class BasicAgent:
     def __call__(self, question: str) -> str:
         try:
-            results = search_duckduckgo(question)
-            if results:
-                return results[0]
-            else:
-                return "I couldn't find an answer."
+            result = search_duckduckgo.invoke({"query": question})
+            return result
         except Exception as e:
             return f"Error searching: {e}"
+