revert

open-webui · tjbck · Jul 6, 2024 · Jun 18, 2024 · Jun 21, 2024 · Jun 24, 2024
commit bfc53b49fd05e033537415db020972f89be1bd22
diff --git a/backend/apps/webui/internal/migrations/004_add_archived.py b/backend/apps/webui/internal/migrations/004_add_archived.py
@@ -1,4 1,4 @@
-"""Peewee migrations -- 009_add_models.py.
 """Peewee migrations -- 002_add_local_sharing.py.
 
 Some examples (model - class or model name)::
 
@@ -37,25 37,10 @@
 def migrate(migrator: Migrator, database: pw.Database, *, fake=False):
  """Write your migrations here."""
 
- @migrator.create_model
- class Model(pw.Model):
- id = pw.TextField(unique=True)
- user_id = pw.TextField()
- base_model_id = pw.TextField(null=True)
-
- name = pw.TextField()
-
- meta = pw.TextField()
- params = pw.TextField()
-
- created_at = pw.BigIntegerField(null=False)
- updated_at = pw.BigIntegerField(null=False)
-
- class Meta:
- table_name = "model"
  migrator.add_fields("chat", archived=pw.BooleanField(default=False))
 
 
 def rollback(migrator: Migrator, database: pw.Database, *, fake=False):
  """Write your rollback migrations here."""
 
- migrator.remove_model("model")
  migrator.remove_fields("chat", "archived")
diff --git a/backend/apps/webui/internal/migrations/005_add_updated_at.py b/backend/apps/webui/internal/migrations/005_add_updated_at.py
@@ -1,4 1,4 @@
-"""Peewee migrations -- 009_add_models.py.
 """Peewee migrations -- 002_add_local_sharing.py.
 
 Some examples (model - class or model name)::
 
@@ -37,25 37,94 @@
 def migrate(migrator: Migrator, database: pw.Database, *, fake=False):
  """Write your migrations here."""
 
- @migrator.create_model
- class Model(pw.Model):
- id = pw.TextField(unique=True)
- user_id = pw.TextField()
- base_model_id = pw.TextField(null=True)
  if isinstance(database, pw.SqliteDatabase):
  migrate_sqlite(migrator, database, fake=fake)
  else:
  migrate_external(migrator, database, fake=fake)
 
- name = pw.TextField()
 
- meta = pw.TextField()
- params = pw.TextField()
 def migrate_sqlite(migrator: Migrator, database: pw.Database, *, fake=False):
  # Adding fields created_at and updated_at to the 'chat' table
  migrator.add_fields(
  "chat",
  created_at=pw.DateTimeField(null=True), # Allow null for transition
  updated_at=pw.DateTimeField(null=True), # Allow null for transition
  )
 
- created_at = pw.BigIntegerField(null=False)
- updated_at = pw.BigIntegerField(null=False)
  # Populate the new fields from an existing 'timestamp' field
  migrator.sql(
  "UPDATE chat SET created_at = timestamp, updated_at = timestamp WHERE timestamp IS NOT NULL"
  )
 
- class Meta:
- table_name = "model"
  # Now that the data has been copied, remove the original 'timestamp' field
  migrator.remove_fields("chat", "timestamp")
 
  # Update the fields to be not null now that they are populated
  migrator.change_fields(
  "chat",
  created_at=pw.DateTimeField(null=False),
  updated_at=pw.DateTimeField(null=False),
  )
 
 
 def migrate_external(migrator: Migrator, database: pw.Database, *, fake=False):
  # Adding fields created_at and updated_at to the 'chat' table
  migrator.add_fields(
  "chat",
  created_at=pw.BigIntegerField(null=True), # Allow null for transition
  updated_at=pw.BigIntegerField(null=True), # Allow null for transition
  )
 
  # Populate the new fields from an existing 'timestamp' field
  migrator.sql(
  "UPDATE chat SET created_at = timestamp, updated_at = timestamp WHERE timestamp IS NOT NULL"
  )
 
  # Now that the data has been copied, remove the original 'timestamp' field
  migrator.remove_fields("chat", "timestamp")
 
  # Update the fields to be not null now that they are populated
  migrator.change_fields(
  "chat",
  created_at=pw.BigIntegerField(null=False),
  updated_at=pw.BigIntegerField(null=False),
  )
 
 
 def rollback(migrator: Migrator, database: pw.Database, *, fake=False):
  """Write your rollback migrations here."""
 
- migrator.remove_model("model")
  if isinstance(database, pw.SqliteDatabase):
  rollback_sqlite(migrator, database, fake=fake)
  else:
  rollback_external(migrator, database, fake=fake)
 
 
 def rollback_sqlite(migrator: Migrator, database: pw.Database, *, fake=False):
  # Recreate the timestamp field initially allowing null values for safe transition
  migrator.add_fields("chat", timestamp=pw.DateTimeField(null=True))
 
  # Copy the earliest created_at date back into the new timestamp field
  # This assumes created_at was originally a copy of timestamp
  migrator.sql("UPDATE chat SET timestamp = created_at")
 
  # Remove the created_at and updated_at fields
  migrator.remove_fields("chat", "created_at", "updated_at")
 
  # Finally, alter the timestamp field to not allow nulls if that was the original setting
  migrator.change_fields("chat", timestamp=pw.DateTimeField(null=False))
 
 
 def rollback_external(migrator: Migrator, database: pw.Database, *, fake=False):
  # Recreate the timestamp field initially allowing null values for safe transition
  migrator.add_fields("chat", timestamp=pw.BigIntegerField(null=True))
 
  # Copy the earliest created_at date back into the new timestamp field
  # This assumes created_at was originally a copy of timestamp
  migrator.sql("UPDATE chat SET timestamp = created_at")
 
  # Remove the created_at and updated_at fields
  migrator.remove_fields("chat", "created_at", "updated_at")
 
  # Finally, alter the timestamp field to not allow nulls if that was the original setting
  migrator.change_fields("chat", timestamp=pw.BigIntegerField(null=False))
diff --git a/backend/apps/webui/internal/migrations/006_migrate_timestamps_and_charfields.py b/backend/apps/webui/internal/migrations/006_migrate_timestamps_and_charfields.py
@@ -1,4 1,4 @@
-"""Peewee migrations -- 009_add_models.py.
 """Peewee migrations -- 006_migrate_timestamps_and_charfields.py.
 
 Some examples (model - class or model name)::
 
@@ -37,25 37,94 @@
 def migrate(migrator: Migrator, database: pw.Database, *, fake=False):
  """Write your migrations here."""
 
- @migrator.create_model
- class Model(pw.Model):
- id = pw.TextField(unique=True)
- user_id = pw.TextField()
- base_model_id = pw.TextField(null=True)
-
- name = pw.TextField()
-
- meta = pw.TextField()
- params = pw.TextField()
-
- created_at = pw.BigIntegerField(null=False)
- updated_at = pw.BigIntegerField(null=False)
-
- class Meta:
- table_name = "model"
  # Alter the tables with timestamps
  migrator.change_fields(
  "chatidtag",
  timestamp=pw.BigIntegerField(),
  )
  migrator.change_fields(
  "document",
  timestamp=pw.BigIntegerField(),
  )
  migrator.change_fields(
  "modelfile",
  timestamp=pw.BigIntegerField(),
  )
  migrator.change_fields(
  "prompt",
  timestamp=pw.BigIntegerField(),
  )
  migrator.change_fields(
  "user",
  timestamp=pw.BigIntegerField(),
  )
  # Alter the tables with varchar to text where necessary
  migrator.change_fields(
  "auth",
  password=pw.TextField(),
  )
  migrator.change_fields(
  "chat",
  title=pw.TextField(),
  )
  migrator.change_fields(
  "document",
  title=pw.TextField(),
  filename=pw.TextField(),
  )
  migrator.change_fields(
  "prompt",
  title=pw.TextField(),
  )
  migrator.change_fields(
  "user",
  profile_image_url=pw.TextField(),
  )
 
 
 def rollback(migrator: Migrator, database: pw.Database, *, fake=False):
  """Write your rollback migrations here."""
 
- migrator.remove_model("model")
  if isinstance(database, pw.SqliteDatabase):
  # Alter the tables with timestamps
  migrator.change_fields(
  "chatidtag",
  timestamp=pw.DateField(),
  )
  migrator.change_fields(
  "document",
  timestamp=pw.DateField(),
  )
  migrator.change_fields(
  "modelfile",
  timestamp=pw.DateField(),
  )
  migrator.change_fields(
  "prompt",
  timestamp=pw.DateField(),
  )
  migrator.change_fields(
  "user",
  timestamp=pw.DateField(),
  )
  migrator.change_fields(
  "auth",
  password=pw.CharField(max_length=255),
  )
  migrator.change_fields(
  "chat",
  title=pw.CharField(),
  )
  migrator.change_fields(
  "document",
  title=pw.CharField(),
  filename=pw.CharField(),
  )
  migrator.change_fields(
  "prompt",
  title=pw.CharField(),
  )
  migrator.change_fields(
  "user",
  profile_image_url=pw.CharField(),
  )
diff --git a/backend/apps/webui/internal/migrations/007_add_user_last_active_at.py b/backend/apps/webui/internal/migrations/007_add_user_last_active_at.py
@@ -1,4 1,4 @@
-"""Peewee migrations -- 009_add_models.py.
 """Peewee migrations -- 002_add_local_sharing.py.
 
 Some examples (model - class or model name)::
 
@@ -37,25 37,43 @@
 def migrate(migrator: Migrator, database: pw.Database, *, fake=False):
  """Write your migrations here."""
 
- @migrator.create_model
- class Model(pw.Model):
- id = pw.TextField(unique=True)
- user_id = pw.TextField()
- base_model_id = pw.TextField(null=True)
  # Adding fields created_at and updated_at to the 'user' table
  migrator.add_fields(
  "user",
  created_at=pw.BigIntegerField(null=True), # Allow null for transition
  updated_at=pw.BigIntegerField(null=True), # Allow null for transition
  last_active_at=pw.BigIntegerField(null=True), # Allow null for transition
  )
 
- name = pw.TextField()
  # Populate the new fields from an existing 'timestamp' field
  migrator.sql(
  'UPDATE "user" SET created_at = timestamp, updated_at = timestamp, last_active_at = timestamp WHERE timestamp IS NOT NULL'
  )
 
-  meta = pw.TextField()
-  params = pw.TextField()
  # Now that the data has been copied, remove the original 'timestamp' field
  migrator.remove_fields("user", "timestamp")
 
- created_at = pw.BigIntegerField(null=False)
- updated_at = pw.BigIntegerField(null=False)
-
- class Meta:
- table_name = "model"
  # Update the fields to be not null now that they are populated
  migrator.change_fields(
  "user",
  created_at=pw.BigIntegerField(null=False),
  updated_at=pw.BigIntegerField(null=False),
  last_active_at=pw.BigIntegerField(null=False),
  )
 
 
 def rollback(migrator: Migrator, database: pw.Database, *, fake=False):
  """Write your rollback migrations here."""
 
- migrator.remove_model("model")
  # Recreate the timestamp field initially allowing null values for safe transition
  migrator.add_fields("user", timestamp=pw.BigIntegerField(null=True))
 
  # Copy the earliest created_at date back into the new timestamp field
  # This assumes created_at was originally a copy of timestamp
  migrator.sql('UPDATE "user" SET timestamp = created_at')
 
  # Remove the created_at and updated_at fields
  migrator.remove_fields("user", "created_at", "updated_at", "last_active_at")
 
  # Finally, alter the timestamp field to not allow nulls if that was the original setting
  migrator.change_fields("user", timestamp=pw.BigIntegerField(null=False))
diff --git a/backend/apps/webui/internal/migrations/008_add_memory.py b/backend/apps/webui/internal/migrations/008_add_memory.py
@@ -1,4 1,4 @@
-"""Peewee migrations -- 009_add_models.py.
 """Peewee migrations -- 002_add_local_sharing.py.
 
 Some examples (model - class or model name)::
 
@@ -35,27 35,19 @@
 
 
 def migrate(migrator: Migrator, database: pw.Database, *, fake=False):
- """Write your migrations here."""
-
  @migrator.create_model
- class Model(pw.Model):
- id = pw.TextField(unique=True)
- user_id = pw.TextField()
- base_model_id = pw.TextField(null=True)
-
- name = pw.TextField()
-
- meta = pw.TextField()
- params = pw.TextField()
-
- created_at = pw.BigIntegerField(null=False)
  class Memory(pw.Model):
  id = pw.CharField(max_length=255, unique=True)
  user_id = pw.CharField(max_length=255)
  content = pw.TextField(null=False)
  updated_at = pw.BigIntegerField(null=False)
  created_at = pw.BigIntegerField(null=False)
 
  class Meta:
- table_name = "model"
  table_name = "memory"
 
 
 def rollback(migrator: Migrator, database: pw.Database, *, fake=False):
  """Write your rollback migrations here."""
 
- migrator.remove_model("model")
  migrator.remove_model("memory")