From 6d771a450e216a7b5330531467452190258e0e6f Mon Sep 17 00:00:00 2001 From: ulleo Date: Mon, 21 Jul 2025 17:29:20 +0800 Subject: [PATCH] fix: chat record save error --- .../versions/031_modify_chat_record.py | 33 +++++++++++++++++++ backend/apps/chat/models/chat_model.py | 2 +- 2 files changed, 34 insertions(+), 1 deletion(-) create mode 100644 backend/alembic/versions/031_modify_chat_record.py diff --git a/backend/alembic/versions/031_modify_chat_record.py b/backend/alembic/versions/031_modify_chat_record.py new file mode 100644 index 00000000..aa391e26 --- /dev/null +++ b/backend/alembic/versions/031_modify_chat_record.py @@ -0,0 +1,33 @@ +"""031_modify_chat_record + +Revision ID: bd2ed188b5bd +Revises: c1d7ac00b3a8 +Create Date: 2025-07-21 17:27:55.985821 + +""" +from alembic import op +import sqlalchemy as sa +import sqlmodel.sql.sqltypes +from sqlalchemy.dialects import postgresql + +# revision identifiers, used by Alembic. +revision = 'bd2ed188b5bd' +down_revision = 'c1d7ac00b3a8' +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.alter_column('chat_record', 'engine_type', + existing_type=sa.VARCHAR(length=64), + nullable=True) + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.alter_column('chat_record', 'engine_type', + existing_type=sa.VARCHAR(length=64), + nullable=False) + # ### end Alembic commands ### diff --git a/backend/apps/chat/models/chat_model.py b/backend/apps/chat/models/chat_model.py index 199fae5a..a83c6b07 100644 --- a/backend/apps/chat/models/chat_model.py +++ b/backend/apps/chat/models/chat_model.py @@ -36,7 +36,7 @@ class ChatRecord(SQLModel, table=True): finish_time: datetime = Field(sa_column=Column(DateTime(timezone=False), nullable=True)) create_by: int = Field(sa_column=Column(BigInteger, nullable=True)) datasource: int = Field(sa_column=Column(BigInteger, nullable=True)) - engine_type: str = Field(max_length=64) + engine_type: str = Field(max_length=64, nullable=True) question: str = Field(sa_column=Column(Text, nullable=True)) sql_answer: str = Field(sa_column=Column(Text, nullable=True)) sql: str = Field(sa_column=Column(Text, nullable=True))