From b4ae257e21a221310fd6d0b2b0f8a4ff787cc77c Mon Sep 17 00:00:00 2001
From: jiaoja <jiaoja@asiainfo.com>
Date: Tue, 12 Nov 2024 15:36:25 +0800
Subject: [PATCH] Fixes for field errors and missing fields in SQLAlchemy model
 '__repr__' methods

---
 .../chatchat/server/db/models/knowledge_base_model.py        | 2 +-
 .../chatchat/server/db/models/knowledge_file_model.py        | 5 ++++-
 .../chatchat/server/db/models/knowledge_metadata_model.py    | 4 ++--
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/libs/chatchat-server/chatchat/server/db/models/knowledge_base_model.py b/libs/chatchat-server/chatchat/server/db/models/knowledge_base_model.py
index 1ef7f0aa18..dd8c6a2714 100644
--- a/libs/chatchat-server/chatchat/server/db/models/knowledge_base_model.py
+++ b/libs/chatchat-server/chatchat/server/db/models/knowledge_base_model.py
@@ -22,7 +22,7 @@ class KnowledgeBaseModel(Base):
     create_time = Column(DateTime, default=func.now(), comment="创建时间")
 
     def __repr__(self):
-        return f"<KnowledgeBase(id='{self.id}', kb_name='{self.kb_name}',kb_intro='{self.kb_info} vs_type='{self.vs_type}', embed_model='{self.embed_model}', file_count='{self.file_count}', create_time='{self.create_time}')>"
+        return f"<KnowledgeBase(id='{self.id}', kb_name='{self.kb_name}',kb_info='{self.kb_info} vs_type='{self.vs_type}', embed_model='{self.embed_model}', file_count='{self.file_count}', create_time='{self.create_time}')>"
 
 
 # 创建一个对应的 Pydantic 模型
diff --git a/libs/chatchat-server/chatchat/server/db/models/knowledge_file_model.py b/libs/chatchat-server/chatchat/server/db/models/knowledge_file_model.py
index 67fdff27ea..101b7ba247 100644
--- a/libs/chatchat-server/chatchat/server/db/models/knowledge_file_model.py
+++ b/libs/chatchat-server/chatchat/server/db/models/knowledge_file_model.py
@@ -23,7 +23,10 @@ class KnowledgeFileModel(Base):
     create_time = Column(DateTime, default=func.now(), comment="创建时间")
 
     def __repr__(self):
-        return f"<KnowledgeFile(id='{self.id}', file_name='{self.file_name}', file_ext='{self.file_ext}', kb_name='{self.kb_name}', document_loader_name='{self.document_loader_name}', text_splitter_name='{self.text_splitter_name}', file_version='{self.file_version}', create_time='{self.create_time}')>"
+        return (f"<KnowledgeFile(id='{self.id}', file_name='{self.file_name}', file_ext='{self.file_ext}', kb_name='{self.kb_name}',"
+                f" document_loader_name='{self.document_loader_name}', text_splitter_name='{self.text_splitter_name}', "
+                f" file_version='{self.file_version}', file_mtime='{self.file_mtime}', file_size='{self.file_size}',"
+                f" custom_docs='{self.custom_docs}', docs_count='{self.docs_count}', create_time='{self.create_time}')>")
 
 
 class FileDocModel(Base):
diff --git a/libs/chatchat-server/chatchat/server/db/models/knowledge_metadata_model.py b/libs/chatchat-server/chatchat/server/db/models/knowledge_metadata_model.py
index 3f4b982a83..f0f38639ae 100644
--- a/libs/chatchat-server/chatchat/server/db/models/knowledge_metadata_model.py
+++ b/libs/chatchat-server/chatchat/server/db/models/knowledge_metadata_model.py
@@ -5,7 +5,7 @@
 
 class SummaryChunkModel(Base):
     """
-    chunk summary模型，用于存储file_doc中每个doc_id的chunk 片段，
+    chunk summary模型，用于存储file_doc中每个doc_id的chunk片段，
     数据来源:
         用户输入: 用户上传文件，可填写文件的描述，生成的file_doc中的doc_id，存入summary_chunk中
         程序自动切分 对file_doc表meta_data字段信息中存储的页码信息，按每页的页码切分，自定义prompt生成总结文本，将对应页码关联的doc_id存入summary_chunk中
@@ -27,5 +27,5 @@ class SummaryChunkModel(Base):
     def __repr__(self):
         return (
             f"<SummaryChunk(id='{self.id}', kb_name='{self.kb_name}', summary_context='{self.summary_context}',"
-            f" doc_ids='{self.doc_ids}', metadata='{self.metadata}')>"
+            f" summary_id='{self.summary_id}', doc_ids='{self.doc_ids}', metadata='{self.metadata}')>"
         )