From f3ae4a3bae0423d6ec9fd3fa8a45c1d198fa83a2 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Tue, 25 Mar 2025 15:57:03 +0800
Subject: [PATCH] Fix: img_id errror. (#6504)

### What problem does this PR solve?

#6499

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 api/apps/sdk/doc.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/api/apps/sdk/doc.py b/api/apps/sdk/doc.py
index ef7b49bf7..c393b862a 100644
--- a/api/apps/sdk/doc.py
+++ b/api/apps/sdk/doc.py
@@ -934,7 +934,7 @@ def list_chunks(tenant_id, dataset_id, document_id):
             "important_keywords":chunk.get("important_kwd",[]),
             "questions":chunk.get("question_kwd",[]),
             "dataset_id":chunk.get("kb_id",chunk.get("dataset_id")),
-            "image_id":chunk["img_id"],
+            "image_id":chunk.get("img_id"),
             "available":bool(chunk.get("available_int",1)),
             "positions":chunk.get("position_int",[]),
         }