From ae26596d0d4b5f8209bf559cba6ce659d1f12efc Mon Sep 17 00:00:00 2001
From: Jun Siang Cheah <git@jscheah.me>
Date: Sun, 19 May 2024 22:11:28 +0800
Subject: [PATCH] feat: log exceptions on update_all_models

---
 backend/apps/web/models/models.py | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/backend/apps/web/models/models.py b/backend/apps/web/models/models.py
index 1d7d0fce8..d1cb082e8 100644
--- a/backend/apps/web/models/models.py
+++ b/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):
-    """
-    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.
-    """
-
     description: str
+    """
+        User-facing description of the model.
+    """
+
     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