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),