Просмотр исходного кода

Fix/6615 40 varchar limit on model name (#6623)

crazywoola 9 месяцев назад
Родитель
Сommit
f324374b95
2 измененных файлов с 42 добавлено и 1 удалено
  1. 41 0
      api/migrations/versions/53bf8af60645_update_model.py
  2. 1 1
      api/models/dataset.py

+ 41 - 0
api/migrations/versions/53bf8af60645_update_model.py

@@ -0,0 +1,41 @@
+"""update model
+
+Revision ID: 53bf8af60645
+Revises: 8e5588e6412e
+Create Date: 2024-07-24 08:06:55.291031
+
+"""
+import sqlalchemy as sa
+from alembic import op
+
+import models as models
+
+# revision identifiers, used by Alembic.
+revision = '53bf8af60645'
+down_revision = '8e5588e6412e'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('embeddings', schema=None) as batch_op:
+        batch_op.alter_column('provider_name',
+               existing_type=sa.VARCHAR(length=40),
+               type_=sa.String(length=255),
+               existing_nullable=False,
+               existing_server_default=sa.text("''::character varying"))
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('embeddings', schema=None) as batch_op:
+        batch_op.alter_column('provider_name',
+               existing_type=sa.String(length=255),
+               type_=sa.VARCHAR(length=40),
+               existing_nullable=False,
+               existing_server_default=sa.text("''::character varying"))
+
+    # ### end Alembic commands ###

+ 1 - 1
api/models/dataset.py

@@ -640,7 +640,7 @@ class Embedding(db.Model):
     hash = db.Column(db.String(64), nullable=False)
     embedding = db.Column(db.LargeBinary, nullable=False)
     created_at = db.Column(db.DateTime, nullable=False, server_default=db.text('CURRENT_TIMESTAMP(0)'))
-    provider_name = db.Column(db.String(40), nullable=False,
+    provider_name = db.Column(db.String(255), nullable=False,
                               server_default=db.text("''::character varying"))
 
     def set_embedding(self, embedding_data: list[float]):