From 8a3b5d1d76242a1b2513f010ad7d95355ff3b4a2 Mon Sep 17 00:00:00 2001 From: "kwrobel.eth" Date: Fri, 4 Jul 2025 13:46:31 +0200 Subject: [PATCH] Fix a small typo in count of used fragments (#8673) ### What problem does this PR solve? Fix a small typo in count of used fragments. ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --------- Co-authored-by: Kevin Hu --- rag/prompts.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rag/prompts.py b/rag/prompts.py index ff8083916..ac04846da 100644 --- a/rag/prompts.py +++ b/rag/prompts.py @@ -104,6 +104,7 @@ def kb_prompt(kbinfos, max_tokens): from api.db.services.document_service import DocumentService knowledges = [ck["content_with_weight"] for ck in kbinfos["chunks"]] + kwlg_len = len(knowledges) used_token_count = 0 chunks_num = 0 for i, c in enumerate(knowledges): @@ -111,7 +112,7 @@ def kb_prompt(kbinfos, max_tokens): chunks_num += 1 if max_tokens * 0.97 < used_token_count: knowledges = knowledges[:i] - logging.warning(f"Not all the retrieval into prompt: {i + 1}/{len(knowledges)}") + logging.warning(f"Not all the retrieval into prompt: {len(knowledges)}/{kwlg_len}") break docs = DocumentService.get_by_ids([ck["doc_id"] for ck in kbinfos["chunks"][:chunks_num]])