Commit ae26596d authored by Jun Siang Cheah's avatar Jun Siang Cheah
Browse files

feat: log exceptions on update_all_models

parent 715a4a6c
import json import json
import logging
from typing import Optional from typing import Optional
import peewee as pw import peewee as pw
...@@ -7,6 +8,11 @@ from pydantic import BaseModel ...@@ -7,6 +8,11 @@ from pydantic import BaseModel
from apps.web.internal.db import DB 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 # Models DB Schema
...@@ -16,16 +22,15 @@ from apps.web.internal.db import DB ...@@ -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 # 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 # It isn't currently used in the backend, but it's here as a reference
class ModelParams(BaseModel): class ModelParams(BaseModel):
description: str
""" """
A Pydantic model that represents the parameters of a model. User-facing description of the 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
vision_capable: bool vision_capable: bool
"""
A flag indicating if the model is capable of vision and thus image inputs
"""
class Model(pw.Model): class Model(pw.Model):
...@@ -151,6 +156,7 @@ class ModelsTable: ...@@ -151,6 +156,7 @@ class ModelsTable:
return True return True
except Exception as e: except Exception as e:
log.exception(e)
return False return False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment