Merge pull request #3380 from Yash-1511/main

feat: add jina_search as new websearch provider
This commit is contained in:
Timothy Jaeryang Baek authored and GitHub committed 2024-06-22 15:19:38 -07:00
commit fd96c9c68d
3 files changed
+46 -1

No files matched your search

+3
View File
@@ -77,6 +77,7 @@ 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 apps.rag.search.tavily import search_tavily
from apps.rag.search.jina_search import search_jina
from utils.misc import (
calculate_sha256,
@@ -856,6 +857,8 @@ def search_web(engine: str, query: str) -> list[SearchResult]:
)
else:
raise Exception("No TAVILY_API_KEY found in environment variables")
elif engine == "jina":
return search_jina(query, app.state.config.RAG_WEB_SEARCH_RESULT_COUNT)
else:
raise Exception("No search engine API key found in environment variables")