diff --git a/api/db/db_models.py b/api/db/db_models.py index 555f734455..433ed78afe 100644 --- a/api/db/db_models.py +++ b/api/db/db_models.py @@ -899,7 +899,7 @@ class Document(DataBaseModel): created_by = CharField(max_length=32, null=False, help_text="who created it", index=True) name = CharField(max_length=255, null=True, help_text="file name", index=True) location = CharField(max_length=255, null=True, help_text="where dose it store", index=True) - size = IntegerField(default=0, index=True) + size = BigIntegerField(default=0, index=True) token_num = IntegerField(default=0, index=True) chunk_num = IntegerField(default=0, index=True) progress = FloatField(default=0, index=True) @@ -924,7 +924,7 @@ class File(DataBaseModel): created_by = CharField(max_length=32, null=False, help_text="who created it", index=True) name = CharField(max_length=255, null=False, help_text="file name or folder name", index=True) location = CharField(max_length=255, null=True, help_text="where dose it store", index=True) - size = IntegerField(default=0, index=True) + size = BigIntegerField(default=0, index=True) type = CharField(max_length=32, null=False, help_text="file extension", index=True) source_type = CharField(max_length=128, null=False, default="", help_text="where dose this document come from", index=True) @@ -1648,6 +1648,8 @@ def migrate_db(): alter_db_add_column(migrator, "memory", "tenant_llm_id", IntegerField(null=True, help_text="id in tenant_llm", index=True)) alter_db_add_column(migrator, "user_canvas_version", "release", BooleanField(null=False, help_text="is released", default=False, index=True)) alter_db_add_column(migrator, "api_4_conversation", "version_title", CharField(max_length=255, null=True, help_text="canvas version title when session created", index=False)) + alter_db_column_type(migrator, "document", "size", BigIntegerField(default=0, index=True)) + alter_db_column_type(migrator, "file", "size", BigIntegerField(default=0, index=True)) logging.disable(logging.NOTSET) # this is after re-enabling logging to allow logging changed user emails migrate_add_unique_email(migrator)