File tree Expand file tree Collapse file tree 3 files changed +20
-8
lines changed Expand file tree Collapse file tree 3 files changed +20
-8
lines changed Original file line number Diff line number Diff line change @@ -403,17 +403,12 @@ def create(
403
403
detail = model_name ,
404
404
** telemetry_kwargs ,
405
405
)
406
- # tracks the shape used for deploying the custom or service models
407
- self .telemetry .record_event_async (
408
- category = f"aqua/{ model_type } /deployment/create" ,
409
- action = "shape" ,
410
- detail = instance_shape ,
411
- )
412
406
# tracks the shape used for deploying the custom or service models by name
413
407
self .telemetry .record_event_async (
414
- category = f"aqua/{ model_type } /{ model_name } / deployment/create" ,
408
+ category = f"aqua/{ model_type } /deployment/create" ,
415
409
action = "shape" ,
416
410
detail = instance_shape ,
411
+ value = model_name ,
417
412
)
418
413
419
414
return AquaDeployment .from_oci_model_deployment (
Original file line number Diff line number Diff line change @@ -683,6 +683,14 @@ def create(
683
683
),
684
684
)
685
685
686
+ # tracks shapes used in evaluation that were created for the given evaluation source
687
+ self .telemetry .record_event_async (
688
+ category = "aqua/evaluation/create" ,
689
+ action = "shape" ,
690
+ detail = create_aqua_evaluation_details .shape_name ,
691
+ value = self ._get_service_model_name (evaluation_source ),
692
+ )
693
+
686
694
# tracks unique evaluation that were created for the given evaluation source
687
695
self .telemetry .record_event_async (
688
696
category = "aqua/evaluation" ,
Original file line number Diff line number Diff line change @@ -462,8 +462,9 @@ def create(
462
462
telemetry_kwargs = (
463
463
{"ocid" : ft_job .id [- 6 :]} if ft_job and len (ft_job .id ) > 6 else {}
464
464
)
465
+ # track shapes that were used for fine-tune creation
465
466
self .telemetry .record_event_async (
466
- category = f"aqua/service/{ source . display_name } / finetune/create/shape/" ,
467
+ category = f"aqua/service/finetune/create/shape/" ,
467
468
action = f"{ create_fine_tuning_details .shape_name } x{ create_fine_tuning_details .replica } " ,
468
469
** telemetry_kwargs ,
469
470
)
@@ -473,6 +474,14 @@ def create(
473
474
category = "aqua/service/finetune" ,
474
475
action = "create" ,
475
476
detail = source .display_name ,
477
+ ** telemetry_kwargs ,
478
+ )
479
+ # track combination of model and shape used for fine-tune creation
480
+ self .telemetry .record_event_async (
481
+ category = "aqua/service/finetune/create" ,
482
+ action = "shape" ,
483
+ detail = f"{ create_fine_tuning_details .shape_name } x{ create_fine_tuning_details .replica } " ,
484
+ value = source .display_name ,
476
485
)
477
486
478
487
return AquaFineTuningSummary (
You can’t perform that action at this time.
0 commit comments