Kevin Hu
commited on
Commit
·
cf6b668
1
Parent(s):
3fa1285
fix: keyerror issue (#3512)
Browse files### What problem does this PR solve?
#3511
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
- api/apps/sdk/session.py +2 -2
api/apps/sdk/session.py
CHANGED
@@ -165,11 +165,11 @@ def completion(tenant_id, chat_id):
|
|
165 |
"document_id": chunk["doc_id"],
|
166 |
"document_name": chunk["docnm_kwd"],
|
167 |
"dataset_id": chunk["kb_id"],
|
168 |
-
"image_id": chunk
|
169 |
"similarity": chunk["similarity"],
|
170 |
"vector_similarity": chunk["vector_similarity"],
|
171 |
"term_similarity": chunk["term_similarity"],
|
172 |
-
"positions": chunk
|
173 |
}
|
174 |
chunk_list.append(new_chunk)
|
175 |
reference["chunks"] = chunk_list
|
|
|
165 |
"document_id": chunk["doc_id"],
|
166 |
"document_name": chunk["docnm_kwd"],
|
167 |
"dataset_id": chunk["kb_id"],
|
168 |
+
"image_id": chunk.get("img_id", ""),
|
169 |
"similarity": chunk["similarity"],
|
170 |
"vector_similarity": chunk["vector_similarity"],
|
171 |
"term_similarity": chunk["term_similarity"],
|
172 |
+
"positions": chunk.get("positions", []),
|
173 |
}
|
174 |
chunk_list.append(new_chunk)
|
175 |
reference["chunks"] = chunk_list
|