Kaynağa Gözat

feat: log exceptions on update_all_models

Jun Siang Cheah 11 ay önce
ebeveyn
işleme
ae26596d0d
1 değiştirilmiş dosya ile 12 ekleme ve 6 silme
  1. 12 6
      backend/apps/web/models/models.py

+ 12 - 6
backend/apps/web/models/models.py

@@ -1,4 +1,5 @@
 import json
+import logging
 from typing import Optional
 
 import peewee as pw
@@ -7,6 +8,11 @@ from pydantic import BaseModel
 
 from apps.web.internal.db import DB
 
+from config import SRC_LOG_LEVELS
+
+log = logging.getLogger(__name__)
+log.setLevel(SRC_LOG_LEVELS["MODELS"])
+
 
 ####################
 # Models DB Schema
@@ -16,16 +22,15 @@ from apps.web.internal.db import DB
 # ModelParams is a model for the data stored in the params field of the Model table
 # It isn't currently used in the backend, but it's here as a reference
 class ModelParams(BaseModel):
+    description: str
     """
-    A Pydantic model that represents the parameters of a model.
-
-    Attributes:
-        description (str): A description of the model.
-        vision_capable (bool): A flag indicating if the model is capable of vision and thus image inputs.
+        User-facing description of the model.
     """
 
-    description: str
     vision_capable: bool
+    """
+        A flag indicating if the model is capable of vision and thus image inputs
+    """
 
 
 class Model(pw.Model):
@@ -151,6 +156,7 @@ class ModelsTable:
 
             return True
         except Exception as e:
+            log.exception(e)
             return False