refac: migration

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

View File

@ -40,11 +40,17 @@ def migrate(migrator: Migrator, database: pw.Database, *, fake=False):
@migrator.create_model
class Model(pw.Model):
id = pw.TextField(unique=True)
meta = pw.TextField()
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"

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.
"""
meta = JSONField()
"""
Holds a JSON encoded blob of metadata, see `ModelMeta`.
"""
user_id = pw.TextField()
base_model_id = pw.TextField(null=True)
"""
@ -66,16 +63,24 @@ class Model(pw.Model):
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:
database = DB
class ModelModel(BaseModel):
id: str
meta: ModelMeta
base_model_id: Optional[str] = None
name: str
params: ModelParams
meta: ModelMeta
####################