From 672d6b4af0b181131b7957c471ad92338eededf5 Mon Sep 17 00:00:00 2001
From: Daniele Tria <36860433+dtria91@users.noreply.github.com>
Date: Wed, 7 Aug 2024 16:15:12 +0200
Subject: [PATCH] feat: add modelUUID as optional value (#163)

---
 api/app/models/model_dto.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/api/app/models/model_dto.py b/api/app/models/model_dto.py
index 6358026e..2229cf8b 100644
--- 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,