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)

Files changed (1) hide show
  1. 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["img_id"],
169
  "similarity": chunk["similarity"],
170
  "vector_similarity": chunk["vector_similarity"],
171
  "term_similarity": chunk["term_similarity"],
172
- "positions": chunk["positions"],
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