feat: model capabilities

This commit is contained in:
Timothy J. Baek
2024-05-24 23:34:58 -07:00
parent 89d80b58e1
commit 0715cd2811
8 changed files with 86 additions and 93 deletions

View File

@@ -31,7 +31,6 @@ class ModelParams(BaseModel):
# ModelMeta is a model for the data stored in the meta field of the Model table
# It isn't currently used in the backend, but it's here as a reference
class ModelMeta(BaseModel):
profile_image_url: Optional[str] = "/favicon.png"
@@ -40,10 +39,7 @@ class ModelMeta(BaseModel):
User-facing description of the model.
"""
vision_capable: Optional[bool] = None
"""
A flag indicating if the model is capable of vision and thus image inputs
"""
capabilities: Optional[dict] = None
model_config = ConfigDict(extra="allow")