zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-10 01:11:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(chat): add Querit web search provider (#17813)

Browse Source
This commit is contained in:
EthanZhang
2026-08-05 09:54:46 +08:00
committed by GitHub
parent 4d68e154ce
commit bdcd8aadde
32 changed files with 1253 additions and 134 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
rag/advanced_rag/tree_structured_query_decomposition_retrieval.py
View File

@@ -19,7 +19,7 @@ from functools import partial
from api.db.services.llm_service import LLMBundle
from rag.prompts import kb_prompt
from rag.prompts.generator import sufficiency_check, multi_queries_gen
from rag.utils.tavily_conn import Tavily
from rag.utils.web_search_conn import create_web_search_provider
from timeit import default_timer as timer
@@ -49,13 +49,13 @@ class TreeStructuredQueryDecompositionRetrieval:
except Exception as e:
logging.error(f"Knowledge base retrieval error: {e}")
# 2. Web retrieval (if Tavily API is configured)
# 2. Web retrieval (if a web search provider is configured)
try:
if self.internet_enabled and self.prompt_config.get("tavily_api_key"):
tav = Tavily(self.prompt_config["tavily_api_key"])
tav_res = tav.retrieve_chunks(search_query)
kbinfos["chunks"].extend(tav_res["chunks"])
kbinfos["doc_aggs"].extend(tav_res["doc_aggs"])
web_search = create_web_search_provider(self.prompt_config) if self.internet_enabled else None
if web_search:
web_res = web_search.retrieve_chunks(search_query)
kbinfos["chunks"].extend(web_res["chunks"])
kbinfos["doc_aggs"].extend(web_res["doc_aggs"])
except Exception as e:
logging.error(f"Web retrieval error: {e}")