Import source: axon/services/sources/ddg

Import the existing local project snapshot. This commit organizes existing files; it does not represent newly developed functionality.
2026-09-18 23:06:37 +03:30
parent 08579794c1
commit e2fbe90654
1 changed files with 48 additions and 0 deletions
@@ -0,0 +1,48 @@
+import asyncio
+import logging
+
+try:
+    from ddgs import DDGS
+except ImportError:
+    from duckduckgo_search import DDGS
+
+from axon.models import SearchResult
+
+logger = logging.getLogger(__name__)
+
+_TIMELIMIT = {"day": "d", "week": "w", "month": "m", "year": "y"}
+
+
+async def search_ddg(
+    query: str,
+    count: int = 10,
+    topic: str = "general",
+    time_range: str | None = None,
+) -> list[SearchResult]:
+    timelimit = _TIMELIMIT.get(time_range) if time_range else None
+
+    def _blocking_search() -> list[dict]:
+        with DDGS() as ddgs:
+            if topic == "news":
+                return list(ddgs.news(query, max_results=count, timelimit=timelimit))
+            return list(ddgs.text(query, max_results=count, timelimit=timelimit))
+
+    try:
+        raw = await asyncio.to_thread(_blocking_search)
+        results = []
+        for item in raw:
+            url = item.get("href") or item.get("url", "")
+            results.append(
+                SearchResult(
+                    title=item.get("title", ""),
+                    url=url,
+                    content=item.get("body", ""),
+                    score=0.5,
+                    published_date=item.get("date"),
+                    favicon=None,
+                )
+            )
+        return results
+    except Exception as exc:
+        logger.error("DuckDuckGo search error: %s", exc)
+        return []