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