@@ -74,11 +74,11 @@ def _update_aimodel_geographies(model: AIModel, geographies: List[str]) -> None:
7474class 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