From 4d77c429bd83880f065c809af24c39598fbee1c9 Mon Sep 17 00:00:00 2001 From: carry <2641257231@qq.com> Date: Wed, 9 Apr 2025 18:18:29 +0800 Subject: [PATCH] =?UTF-8?q?refactor(schema):=20=E6=9B=B4=E6=96=B0=20datase?= =?UTF-8?q?t=20=E6=A8=A1=E5=9E=8B=E5=B9=B6=E4=B8=BA=20doc=20=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E6=B7=BB=E5=8A=A0=E7=89=88=E6=9C=AC=E5=AD=97=E6=AE=B5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在 doc 模型中添加 version 字段，用于表示文档版本 - 将 dataset 模型中的 source_doc 字段类型从 list[doc] 改为 doc，简化数据结构 --- schema/dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/schema/dataset.py b/schema/dataset.py index 4835e2f..a972545 100644 --- a/schema/dataset.py +++ b/schema/dataset.py @@ -7,6 +7,7 @@ class doc(BaseModel): name: str = Field(default="", description="文档名称") path: str = Field(default="", description="文档路径") markdown_files: list[str] = Field(default_factory=list, description="文档路径列表") + version: Optional[str] = Field(default="", description="文档版本") class Q_A(BaseModel): question: str = Field(default="", min_length=1,description="问题") @@ -20,7 +21,7 @@ class dataset(BaseModel): id: Optional[int] = Field(default=None, description="数据集ID") name: Optional[str] = Field(default=None, description="数据集名称") model_id: Optional[list[str]] = Field(default=None, description="数据集使用的模型ID") - source_doc: Optional[list[doc]] = Field(default=None, description="数据集来源文档") + source_doc: Optional[doc] = Field(default=None, description="数据集来源文档") description: Optional[str] = Field(default="", description="数据集描述") created_at: datetime = Field( default_factory=lambda: datetime.now(timezone.utc),