feat: add modelUUID as optional value (#163)

radicalbit · Aug 7, 2024 · 672d6b4 · 672d6b4
1 parent 28f485a
commit 672d6b4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/api/app/models/model_dto.py b/api/app/models/model_dto.py
@@ -86,6 +86,7 @@ class ModelFeatures(BaseModel):
 
 
 class ModelIn(BaseModel, validate_assignment=True):
+    uuid: Optional[UUID] = None
     name: str
     description: Optional[str] = None
     model_type: ModelType
@@ -179,6 +180,7 @@ def timestamp_must_be_datetime(self) -> Self:
     def to_model(self) -> Model:
         now = datetime.datetime.now(tz=datetime.UTC)
         return Model(
+            uuid=self.uuid,
             name=self.name,
             description=self.description,
             model_type=self.model_type.value,