devMls
commited on
Commit
·
533604b
1
Parent(s):
bc59ad5
Fix bug over chunks classification by document in the promp (#4156)
Browse filesThe refactor in 0.15 contains a small bug that eliminate the
classification
### What problem does this PR solve
### Type of change
- [X] Bug Fix (non-breaking change which fixes an issue)
api/db/services/dialog_service.py
CHANGED
|
@@ -124,7 +124,7 @@ def kb_prompt(kbinfos, max_tokens):
|
|
| 124 |
for i, ck in enumerate(kbinfos["chunks"]):
|
| 125 |
if i >= chunks_num:
|
| 126 |
break
|
| 127 |
-
doc2chunks["docnm_kwd"].append(ck["content_with_weight"])
|
| 128 |
|
| 129 |
knowledges = []
|
| 130 |
for nm, chunks in doc2chunks.items():
|
|
|
|
| 124 |
for i, ck in enumerate(kbinfos["chunks"]):
|
| 125 |
if i >= chunks_num:
|
| 126 |
break
|
| 127 |
+
doc2chunks[ck["docnm_kwd"]].append(ck["content_with_weight"])
|
| 128 |
|
| 129 |
knowledges = []
|
| 130 |
for nm, chunks in doc2chunks.items():
|