UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-21 12:29:29 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

refac: migration

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-23 22:58:26 -07:00
parent 7d159ca723
commit 363ab562c3
2 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/apps/web/internal/migrations/009_add_models.py
View File

@ -40,11 +40,17 @@ def migrate(migrator: Migrator, database: pw.Database, *, fake=False):
@migrator.create_model @migrator.create_model
class Model(pw.Model): class Model(pw.Model):
id = pw.TextField(unique=True) id = pw.TextField(unique=True)
meta = pw.TextField() user_id = pw.TextField()
base_model_id = pw.TextField(null=True) base_model_id = pw.TextField(null=True)
name = pw.TextField() name = pw.TextField()
meta = pw.TextField()
params = pw.TextField() params = pw.TextField()
created_at = (pw.BigIntegerField(null=False),)
updated_at = (pw.BigIntegerField(null=False),)
class Meta: class Meta:
table_name = "model" table_name = "model"

15
backend/apps/web/models/models.py
View File

@ -45,10 +45,7 @@ class Model(pw.Model):
The model's id as used in the API. If set to an existing model, it will override the model. The model's id as used in the API. If set to an existing model, it will override the model.
""" """
meta = JSONField() user_id = pw.TextField()
"""
Holds a JSON encoded blob of metadata, see `ModelMeta`.
"""
base_model_id = pw.TextField(null=True) base_model_id = pw.TextField(null=True)
""" """
@ -66,16 +63,24 @@ class Model(pw.Model):
Holds a JSON encoded blob of parameters, see `ModelParams`. Holds a JSON encoded blob of parameters, see `ModelParams`.
""" """
meta = JSONField()
"""
Holds a JSON encoded blob of metadata, see `ModelMeta`.
"""
updated_at: int # timestamp in epoch
created_at: int # timestamp in epoch
class Meta: class Meta:
database = DB database = DB
class ModelModel(BaseModel): class ModelModel(BaseModel):
id: str id: str
meta: ModelMeta
base_model_id: Optional[str] = None base_model_id: Optional[str] = None
name: str name: str
params: ModelParams params: ModelParams
meta: ModelMeta
#################### ####################