Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 11:57:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: add DuckDuckGo search functionality using duckduckgo_search library

Browse Source
This commit is contained in:
Yash-1511
2024-06-11 19:49:08 +05:30
parent 06976c4551
commit 83f9475584
3 changed files with 50 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
backend/apps/rag/main.py
View File

@@ -70,6 +70,7 @@ from apps.rag.search.searxng import search_searxng
from apps.rag.search.serper import search_serper
from apps.rag.search.serpstack import search_serpstack
from apps.rag.search.serply import search_serply
from apps.rag.search.duckduckgo import search_duckduckgo
from utils.misc import (
calculate_sha256,
@@ -820,6 +821,8 @@ def search_web(engine: str, query: str) -> list[SearchResult]:
)
else:
raise Exception("No SERPLY_API_KEY found in environment variables")
elif engine == "duckduckgo":
return search_duckduckgo(query, app.state.config.RAG_WEB_SEARCH_RESULT_COUNT)
else:
raise Exception("No search engine API key found in environment variables")