feat: modelfiles migration support

2023-12-26 23:58:40 -08:00 · 2023-12-26 23:58:40 -08:00 · 21c7f50790
commit 21c7f50790
parent fbb64406d8
5 changed files with 124 additions and 21 deletions
--- a/backend/apps/web/models/modelfiles.py
+++ b/backend/apps/web/models/modelfiles.py
@ -74,11 +74,16 @@ class ModelfilesTable:
                    "timestamp": int(time.time()),
                }
            )
-            result = Modelfile.create(**modelfile.model_dump())
-            if result:
-                return modelfile
-            else:
+
+            try:
+                result = Modelfile.create(**modelfile.model_dump())
+                if result:
+                    return modelfile
+                else:
+                    return None
+            except:
                return None
+
        else:
            return None