Timothy J. Baek пре 11 месеци
родитељ
комит
363ab562c3

+ 7 - 1
backend/apps/web/internal/migrations/009_add_models.py

@@ -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"
 

+ 10 - 5
backend/apps/web/models/models.py

@@ -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
 
 
 ####################