fix: 修复新增sftmodel的报错 (#353)

dataelement · Feb 26, 2024 · 34448d9 · 34448d9
2 parents b91eebb + 761b114
commit 34448d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/backend/bisheng/database/models/sft_model.py b/src/backend/bisheng/database/models/sft_model.py
@@ -9,7 +9,7 @@
 
 # 可用于训练的model列表
 class SftModelBase(SQLModelSerializable):
-    id: str = Field(default=None, nullable=False, primary_key=True, description='唯一ID')
+    id: int = Field(default=None, nullable=False, primary_key=True, description='唯一ID')
     create_time: Optional[datetime] = Field(sa_column=Column(
         DateTime, nullable=False, index=True, server_default=text('CURRENT_TIMESTAMP')))
     update_time: Optional[datetime] = Field(sa_column=Column(