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

Merge pull request #3029 from Yash-1511/main

Browse Source 
feat: add DuckDuckGo search functionality using duckduckgo_search library
This commit is contained in:
Timothy Jaeryang Baek
2024-06-11 09:53:26 -07:00
committed by GitHub
parent a2631ed38e 07b08ef67e
commit d709038b5b
4 changed files with 52 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -71,6 +71,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,
@@ -821,6 +822,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")