fix thumbnail issue (#2917)

### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)
infiniflow · Oct 21, 2024 · 2ced25c · 2ced25c
1 parent 1935c3b
commit 2ced25c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/db/services/document_service.py b/api/db/services/document_service.py
@@ -317,7 +317,7 @@ def get_doc_id_by_doc_name(cls, doc_name):
     @classmethod
     @DB.connection_context()
     def get_thumbnails(cls, docids):
-        fields = [cls.model.id, cls.model.thumbnail]
+        fields = [cls.model.id, cls.model.kb_id, cls.model.thumbnail]
         return list(cls.model.select(
             *fields).where(cls.model.id.in_(docids)).dicts())