diff --git a/api/chat_routes.py b/api/chat_routes.py index 2bbb380..b0c32d4 100644 --- a/api/chat_routes.py +++ b/api/chat_routes.py @@ -2351,7 +2351,7 @@ def rag(): sub_queries = intent.sub_queries search_result = search_hybrid( - message, + retrieval_query, top_k=RAG_SEARCH_TOP_K, candidates=RAG_SEARCH_CANDIDATES, allowed_collections=collections, diff --git a/core/engine.py b/core/engine.py index 2749f30..8491074 100644 --- a/core/engine.py +++ b/core/engine.py @@ -559,7 +559,6 @@ class RAGEngine: top_dist = result['distances'][0][0] if result.get('distances') and result['distances'][0] else 1.0 top_score = 1.0 - top_dist if top_score >= CACHE_MIN_SCORE: - doc_ids = result.get('ids', [[]])[0] if result.get('ids') else [] cache.set_query_result(query, kb_name, result) result['_debug'] = _debug return result @@ -581,7 +580,6 @@ class RAGEngine: top_dist = result['distances'][0][0] if result.get('distances') and result['distances'][0] else 1.0 top_score = 1.0 - top_dist if top_score >= CACHE_MIN_SCORE: - doc_ids = result.get('ids', [[]])[0] if result.get('ids') else [] cache.set_query_result(query, kb_name, result) result['_debug'] = _debug return result @@ -617,8 +615,6 @@ class RAGEngine: top_dist = result['distances'][0][0] if result.get('distances') and result['distances'][0] else 1.0 top_score = 1.0 - top_dist if top_score >= CACHE_MIN_SCORE: - # 传递 doc_ids 实现细粒度缓存失效 - doc_ids = result.get('ids', [[]])[0] if result.get('ids') else [] cache.set_query_result(query, kb_name, result) _debug['timing']['total_ms'] = int((time.time() - _overall_start) * 1000) result['_debug'] = _debug @@ -810,8 +806,6 @@ class RAGEngine: top_dist = fused_results['distances'][0][0] if fused_results.get('distances') and fused_results['distances'][0] else 1.0 top_score = 1.0 - top_dist # 距离转相似度 if top_score >= CACHE_MIN_SCORE: # 置信度阈值 - # 传递 doc_ids 实现细粒度缓存失效 - doc_ids = fused_results.get('ids', [[]])[0] if fused_results.get('ids') else [] cache.set_query_result(query, kb_name, fused_results) fused_results['_debug'] = _debug