adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

knowledge_retrieval/bug/fix

Browse Source
This commit is contained in:
lixinyue
2026-02-04 15:51:13 +08:00
parent 41550d4a41
commit 514c19a247
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
api/app/core/rag/nlp/search.py
View File

@@ -28,7 +28,9 @@ from app.core.rag.common.float_utils import get_float
from app.core.rag.common.constants import PAGERANK_FLD, TAG_FLD from app.core.rag.common.constants import PAGERANK_FLD, TAG_FLD
from app.core.rag.llm.chat_model import Base from app.core.rag.llm.chat_model import Base
from app.core.rag.llm.embedding_model import OpenAIEmbed from app.core.rag.llm.embedding_model import OpenAIEmbed
import logging
logger = logging.getLogger(__name__)
def knowledge_retrieval( def knowledge_retrieval(
query: str, query: str,
@@ -171,10 +173,16 @@ def knowledge_retrieval(
return rerank(db=db, reranker_id=reranker_id, query=query, docs=all_results, top_k=reranker_top_k) return rerank(db=db, reranker_id=reranker_id, query=query, docs=all_results, top_k=reranker_top_k)
except Exception as rerank_error: except Exception as rerank_error:
# If reranker fails, log warning and continue with original results # If reranker fails, log warning and continue with original results
print(f"Failed to rerank documents: {str(rerank_error)}") logger.warning(
print(f"Continuing with original retrieval results (count: {len(all_results)})") "Reranker failed, falling back to original results",
extra={
# use graph "reranker_id": reranker_id,
"query": query,
"doc_count": len(all_results),
"error": str(e),
},
)
if use_graph: if use_graph:
try: try:
from app.core.rag.common.settings import kg_retriever from app.core.rag.common.settings import kg_retriever