Skip to content

Commit fef4af4

Browse files
Generate modelserving
1 parent 36fab7f commit fef4af4

File tree

1 file changed

+2
-2
lines changed
  • services/modelserving/src/stackit/modelserving/models

1 file changed

+2
-2
lines changed

services/modelserving/src/stackit/modelserving/models/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,8 @@ def displayed_name_validate_regular_expression(cls, value):
7777
@field_validator("type")
7878
def type_validate_enum(cls, value):
7979
"""Validates the enum"""
80-
if value not in set(["chat", "embedding"]):
81-
raise ValueError("must be one of enum values ('chat', 'embedding')")
80+
if value not in set(["chat", "embedding", "audio", "image"]):
81+
raise ValueError("must be one of enum values ('chat', 'embedding', 'audio', 'image')")
8282
return value
8383

8484
model_config = ConfigDict(

0 commit comments

Comments
 (0)