diff --git a/pkg/entity/persistence/bot.py b/pkg/entity/persistence/bot.py index 355e1208..77e13994 100644 --- a/pkg/entity/persistence/bot.py +++ b/pkg/entity/persistence/bot.py @@ -7,11 +7,12 @@ class Bot(Base): """机器人""" __tablename__ = 'bots' - uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True) + uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True, unique=True) name = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) description = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) adapter = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) adapter_config = sqlalchemy.Column(sqlalchemy.JSON, nullable=False) enable = sqlalchemy.Column(sqlalchemy.Boolean, nullable=False, default=False) + use_pipeline_uuid = sqlalchemy.Column(sqlalchemy.String(255), nullable=True) created_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False, server_default=sqlalchemy.func.now()) updated_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False, server_default=sqlalchemy.func.now(), onupdate=sqlalchemy.func.now()) diff --git a/pkg/entity/persistence/model.py b/pkg/entity/persistence/model.py index a77ffd27..13700f25 100644 --- a/pkg/entity/persistence/model.py +++ b/pkg/entity/persistence/model.py @@ -8,7 +8,7 @@ class LLMModel(Base): """LLM 模型""" __tablename__ = 'llm_models' - uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True) + uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True, unique=True) name = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) description = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) requester = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) diff --git a/pkg/entity/persistence/pipeline.py b/pkg/entity/persistence/pipeline.py index 6d1b499d..369cd57a 100644 --- a/pkg/entity/persistence/pipeline.py +++ b/pkg/entity/persistence/pipeline.py @@ -7,7 +7,7 @@ class LegacyPipeline(Base): """旧版流水线""" __tablename__ = 'legacy_pipelines' - uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True) + uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True, unique=True) name = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) description = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) created_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False, server_default=sqlalchemy.func.now()) @@ -22,7 +22,7 @@ class PipelineRunRecord(Base): """流水线运行记录""" __tablename__ = 'pipeline_run_records' - uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True) + uuid = sqlalchemy.Column(sqlalchemy.String(255), primary_key=True, unique=True) pipeline_uuid = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) status = sqlalchemy.Column(sqlalchemy.String(255), nullable=False) created_at = sqlalchemy.Column(sqlalchemy.DateTime, nullable=False, server_default=sqlalchemy.func.now())