Skip to content

Commit aeb21bb

Browse files
committed
generate new unique name if name or description not passed
1 parent 145f530 commit aeb21bb

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

api/schema/aimodel_schema.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -74,11 +74,11 @@ def _update_aimodel_geographies(model: AIModel, geographies: List[str]) -> None:
7474
class CreateAIModelInput:
7575
"""Input for creating a new AI Model."""
7676

77-
name: str
78-
display_name: str
79-
description: str
8077
model_type: AIModelTypeEnum
8178
provider: AIModelProviderEnum
79+
name: Optional[str] = None
80+
display_name: Optional[str] = None
81+
description: Optional[str] = None
8282
version: Optional[str] = None
8383
provider_model_id: Optional[str] = None
8484
supports_streaming: bool = False
@@ -290,6 +290,12 @@ def create_ai_model(
290290
organization = info.context.context.get("organization")
291291
user = info.context.user
292292

293+
# Generate default values if not provided (similar to dataset creation)
294+
timestamp = datetime.datetime.now().strftime("%d %b %Y - %H:%M:%S")
295+
name = input.name or f"untitled-ai-model-{timestamp}"
296+
display_name = input.display_name or f"Untitled AI Model - {timestamp}"
297+
description = input.description or ""
298+
293299
# Prepare supported_languages
294300
supported_languages = input.supported_languages or []
295301

@@ -302,10 +308,10 @@ def create_ai_model(
302308

303309
try:
304310
model = AIModel.objects.create(
305-
name=input.name,
306-
display_name=input.display_name,
311+
name=name,
312+
display_name=display_name,
307313
version=input.version or "",
308-
description=input.description,
314+
description=description,
309315
model_type=input.model_type,
310316
provider=input.provider,
311317
provider_model_id=input.provider_model_id or "",

0 commit comments

Comments
 (0)