fix thumbnail issue

infiniflow · Oct 21, 2024 · b9a8cd6 · b9a8cd6
1 parent 1935c3b
commit b9a8cd6
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())