lightning-sdk 2025.8.28__py3-none-any.whl → 2025.9.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (45) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/teamspace_api.py +2 -0
  3. lightning_sdk/lightning_cloud/openapi/__init__.py +16 -0
  4. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +388 -0
  5. lightning_sdk/lightning_cloud/openapi/api/auth_service_api.py +97 -0
  6. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +202 -0
  7. lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +101 -0
  8. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +4 -4
  9. lightning_sdk/lightning_cloud/openapi/models/__init__.py +16 -0
  10. lightning_sdk/lightning_cloud/openapi/models/cluster_id_metrics_body.py +27 -1
  11. lightning_sdk/lightning_cloud/openapi/models/models_model_id_body.py +109 -31
  12. lightning_sdk/lightning_cloud/openapi/models/models_model_id_body1.py +149 -0
  13. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  14. lightning_sdk/lightning_cloud/openapi/models/v1_aggregated_pod_metrics.py +799 -0
  15. lightning_sdk/lightning_cloud/openapi/models/v1_cancel_running_cloud_space_instance_transfer_response.py +97 -0
  16. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_transfer_metadata.py +27 -1
  17. lightning_sdk/lightning_cloud/openapi/models/v1_cloudflare_v1.py +3 -29
  18. lightning_sdk/lightning_cloud/openapi/models/v1_create_model_metrics_response.py +97 -0
  19. lightning_sdk/lightning_cloud/openapi/models/v1_daily_model_metrics.py +149 -0
  20. lightning_sdk/lightning_cloud/openapi/models/v1_filestore_data_connection.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_required_balance_status_response.py +149 -0
  22. lightning_sdk/lightning_cloud/openapi/models/v1_get_latest_model_metrics_response.py +123 -0
  23. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_metrics_response.py +123 -0
  24. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_total_usage_metrics_response.py +149 -0
  25. lightning_sdk/lightning_cloud/openapi/models/v1_get_temp_bucket_credentials_response.py +201 -0
  26. lightning_sdk/lightning_cloud/openapi/models/v1_group_node_metrics.py +1215 -0
  27. lightning_sdk/lightning_cloud/openapi/models/v1_guest_login_request.py +177 -0
  28. lightning_sdk/lightning_cloud/openapi/models/v1_guest_login_response.py +149 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_guest_user.py +201 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +29 -3
  31. lightning_sdk/lightning_cloud/openapi/models/v1_list_aggregated_pod_metrics_response.py +6 -6
  32. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_model_metrics.py +175 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_nebius_direct_v1.py +29 -3
  35. lightning_sdk/lightning_cloud/openapi/models/v1_node_metrics.py +81 -3
  36. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_required_balance_reason.py +107 -0
  38. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +53 -27
  39. lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py +29 -3
  40. {lightning_sdk-2025.8.28.dist-info → lightning_sdk-2025.9.4.dist-info}/METADATA +1 -1
  41. {lightning_sdk-2025.8.28.dist-info → lightning_sdk-2025.9.4.dist-info}/RECORD +45 -29
  42. {lightning_sdk-2025.8.28.dist-info → lightning_sdk-2025.9.4.dist-info}/LICENSE +0 -0
  43. {lightning_sdk-2025.8.28.dist-info → lightning_sdk-2025.9.4.dist-info}/WHEEL +0 -0
  44. {lightning_sdk-2025.8.28.dist-info → lightning_sdk-2025.9.4.dist-info}/entry_points.txt +0 -0
  45. {lightning_sdk-2025.8.28.dist-info → lightning_sdk-2025.9.4.dist-info}/top_level.txt +0 -0
lightning_sdk/__init__.py CHANGED
@@ -32,6 +32,6 @@ __all__ = [
32
32
  "User",
33
33
  ]
34
34
 
35
- __version__ = "2025.08.28"
35
+ __version__ = "2025.09.04"
36
36
  _check_version_and_prompt_upgrade(__version__)
37
37
  _set_tqdm_envvars_noninteractive()
@@ -390,6 +390,8 @@ class TeamspaceApi:
390
390
  params=query_params,
391
391
  stream=True,
392
392
  )
393
+ if r.status_code == 404:
394
+ raise FileNotFoundError(f"File {path} not found")
393
395
  total_length = int(r.headers.get("content-length"))
394
396
 
395
397
  if progress_bar:
@@ -173,6 +173,7 @@ from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import
173
173
  from lightning_sdk.lightning_cloud.openapi.models.model_id_visibility_body import ModelIdVisibilityBody
174
174
  from lightning_sdk.lightning_cloud.openapi.models.models_id_body import ModelsIdBody
175
175
  from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
176
+ from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body1 import ModelsModelIdBody1
176
177
  from lightning_sdk.lightning_cloud.openapi.models.multimachinejobs_id_body import MultimachinejobsIdBody
177
178
  from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
178
179
  from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import OrgIdMembershipsBody
@@ -250,6 +251,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job import V1AgentJob
250
251
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_job_artifact import V1AgentJobArtifact
251
252
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_multipart_url import V1AgentUploadMultipartUrl
252
253
  from lightning_sdk.lightning_cloud.openapi.models.v1_agent_upload_part_response import V1AgentUploadPartResponse
254
+ from lightning_sdk.lightning_cloud.openapi.models.v1_aggregated_pod_metrics import V1AggregatedPodMetrics
253
255
  from lightning_sdk.lightning_cloud.openapi.models.v1_ai_pod_v1 import V1AiPodV1
254
256
  from lightning_sdk.lightning_cloud.openapi.models.v1_alert_method import V1AlertMethod
255
257
  from lightning_sdk.lightning_cloud.openapi.models.v1_alerts_config import V1AlertsConfig
@@ -280,6 +282,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_body import V1Body
280
282
  from lightning_sdk.lightning_cloud.openapi.models.v1_build_spec import V1BuildSpec
281
283
  from lightning_sdk.lightning_cloud.openapi.models.v1_cpu_system_metrics import V1CPUSystemMetrics
282
284
  from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_cloud_space_instance_switch_response import V1CancelCloudSpaceInstanceSwitchResponse
285
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_running_cloud_space_instance_transfer_response import V1CancelRunningCloudSpaceInstanceTransferResponse
283
286
  from lightning_sdk.lightning_cloud.openapi.models.v1_cancellation_metadata import V1CancellationMetadata
284
287
  from lightning_sdk.lightning_cloud.openapi.models.v1_capacity_block_offering import V1CapacityBlockOffering
285
288
  from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_request import V1CheckClusterNameAvailabilityRequest
@@ -382,6 +385,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_dataset_multi_pa
382
385
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_request import V1CreateLitPageRequest
383
386
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_response import V1CreateLitPageResponse
384
387
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_managed_endpoint_response import V1CreateManagedEndpointResponse
388
+ from lightning_sdk.lightning_cloud.openapi.models.v1_create_model_metrics_response import V1CreateModelMetricsResponse
385
389
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_machine_job_request import V1CreateMultiMachineJobRequest
386
390
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_part_upload_response import V1CreateMultiPartUploadResponse
387
391
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request import V1CreateOrganizationRequest
@@ -395,6 +399,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_snowflake_connection
395
399
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_request import V1CreateSubscriptionCheckoutSessionRequest
396
400
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_subscription_checkout_session_response import V1CreateSubscriptionCheckoutSessionResponse
397
401
  from lightning_sdk.lightning_cloud.openapi.models.v1_create_user_secret_request import V1CreateUserSecretRequest
402
+ from lightning_sdk.lightning_cloud.openapi.models.v1_daily_model_metrics import V1DailyModelMetrics
398
403
  from lightning_sdk.lightning_cloud.openapi.models.v1_daily_usage import V1DailyUsage
399
404
  from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection import V1DataConnection
400
405
  from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_artifact import V1DataConnectionArtifact
@@ -548,6 +553,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_
548
553
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_open_ports_response import V1GetCloudSpaceInstanceOpenPortsResponse
549
554
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
550
555
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_system_metrics_aggregate_response import V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
556
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_required_balance_status_response import V1GetCloudSpaceRequiredBalanceStatusResponse
551
557
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
552
558
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
553
559
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
@@ -558,6 +564,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_tele
558
564
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_folder_index_response import V1GetFolderIndexResponse
559
565
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_stats_response import V1GetJobStatsResponse
560
566
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_system_metrics_response import V1GetJobSystemMetricsResponse
567
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_latest_model_metrics_response import V1GetLatestModelMetricsResponse
561
568
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightning_run_source_code_download_url_response import V1GetLightningRunSourceCodeDownloadUrlResponse
562
569
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_open_ports_response import V1GetLightningappInstanceOpenPortsResponse
563
570
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_system_metrics_response import V1GetLightningappInstanceSystemMetricsResponse
@@ -570,6 +577,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_upload_urls_
570
577
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
571
578
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_response import V1GetModelFilesResponse
572
579
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_url_response import V1GetModelFilesUrlResponse
580
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_metrics_response import V1GetModelMetricsResponse
581
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_total_usage_metrics_response import V1GetModelTotalUsageMetricsResponse
573
582
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_organization_storage_metadata_response import V1GetOrganizationStorageMetadataResponse
574
583
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_response import V1GetProjectArtifactResponse
575
584
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
@@ -577,6 +586,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_compute_usage_r
577
586
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_storage_metadata_response import V1GetProjectStorageMetadataResponse
578
587
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_settings_response import V1GetSettingsResponse
579
588
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_snowflake_query_response import V1GetSnowflakeQueryResponse
589
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_temp_bucket_credentials_response import V1GetTempBucketCredentialsResponse
580
590
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_usage_details_response import V1GetUsageDetailsResponse
581
591
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_balance_response import V1GetUserBalanceResponse
582
592
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_notification_preferences_response import V1GetUserNotificationPreferencesResponse
@@ -586,6 +596,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_volume_response import
586
596
  from lightning_sdk.lightning_cloud.openapi.models.v1_git_credentials import V1GitCredentials
587
597
  from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
588
598
  from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status
599
+ from lightning_sdk.lightning_cloud.openapi.models.v1_group_node_metrics import V1GroupNodeMetrics
600
+ from lightning_sdk.lightning_cloud.openapi.models.v1_guest_login_request import V1GuestLoginRequest
601
+ from lightning_sdk.lightning_cloud.openapi.models.v1_guest_login_response import V1GuestLoginResponse
602
+ from lightning_sdk.lightning_cloud.openapi.models.v1_guest_user import V1GuestUser
589
603
  from lightning_sdk.lightning_cloud.openapi.models.v1_header import V1Header
590
604
  from lightning_sdk.lightning_cloud.openapi.models.v1_health_check_exec import V1HealthCheckExec
591
605
  from lightning_sdk.lightning_cloud.openapi.models.v1_health_check_http_get import V1HealthCheckHttpGet
@@ -779,6 +793,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tags import V1Metri
779
793
  from lightning_sdk.lightning_cloud.openapi.models.v1_metrics_tracker import V1MetricsTracker
780
794
  from lightning_sdk.lightning_cloud.openapi.models.v1_model import V1Model
781
795
  from lightning_sdk.lightning_cloud.openapi.models.v1_model_file import V1ModelFile
796
+ from lightning_sdk.lightning_cloud.openapi.models.v1_model_metrics import V1ModelMetrics
782
797
  from lightning_sdk.lightning_cloud.openapi.models.v1_model_version_archive import V1ModelVersionArchive
783
798
  from lightning_sdk.lightning_cloud.openapi.models.v1_modify_filesystem_volume_response import V1ModifyFilesystemVolumeResponse
784
799
  from lightning_sdk.lightning_cloud.openapi.models.v1_mount_target import V1MountTarget
@@ -873,6 +888,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_request_cloud_space_access_
873
888
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
874
889
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
875
890
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
891
+ from lightning_sdk.lightning_cloud.openapi.models.v1_required_balance_reason import V1RequiredBalanceReason
876
892
  from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_details import V1ReservationDetails
877
893
  from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
878
894
  from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
@@ -358,6 +358,111 @@ class AssistantsServiceApi(object):
358
358
  _request_timeout=params.get('_request_timeout'),
359
359
  collection_formats=collection_formats)
360
360
 
361
+ def assistants_service_create_model_metrics(self, body: 'ModelsModelIdBody', model_id: 'str', **kwargs) -> 'V1CreateModelMetricsResponse': # noqa: E501
362
+ """assistants_service_create_model_metrics # noqa: E501
363
+
364
+ This method makes a synchronous HTTP request by default. To make an
365
+ asynchronous HTTP request, please pass async_req=True
366
+ >>> thread = api.assistants_service_create_model_metrics(body, model_id, async_req=True)
367
+ >>> result = thread.get()
368
+
369
+ :param async_req bool
370
+ :param ModelsModelIdBody body: (required)
371
+ :param str model_id: (required)
372
+ :return: V1CreateModelMetricsResponse
373
+ If the method is called asynchronously,
374
+ returns the request thread.
375
+ """
376
+ kwargs['_return_http_data_only'] = True
377
+ if kwargs.get('async_req'):
378
+ return self.assistants_service_create_model_metrics_with_http_info(body, model_id, **kwargs) # noqa: E501
379
+ else:
380
+ (data) = self.assistants_service_create_model_metrics_with_http_info(body, model_id, **kwargs) # noqa: E501
381
+ return data
382
+
383
+ def assistants_service_create_model_metrics_with_http_info(self, body: 'ModelsModelIdBody', model_id: 'str', **kwargs) -> 'V1CreateModelMetricsResponse': # noqa: E501
384
+ """assistants_service_create_model_metrics # noqa: E501
385
+
386
+ This method makes a synchronous HTTP request by default. To make an
387
+ asynchronous HTTP request, please pass async_req=True
388
+ >>> thread = api.assistants_service_create_model_metrics_with_http_info(body, model_id, async_req=True)
389
+ >>> result = thread.get()
390
+
391
+ :param async_req bool
392
+ :param ModelsModelIdBody body: (required)
393
+ :param str model_id: (required)
394
+ :return: V1CreateModelMetricsResponse
395
+ If the method is called asynchronously,
396
+ returns the request thread.
397
+ """
398
+
399
+ all_params = ['body', 'model_id'] # noqa: E501
400
+ all_params.append('async_req')
401
+ all_params.append('_return_http_data_only')
402
+ all_params.append('_preload_content')
403
+ all_params.append('_request_timeout')
404
+
405
+ params = locals()
406
+ for key, val in six.iteritems(params['kwargs']):
407
+ if key not in all_params:
408
+ raise TypeError(
409
+ "Got an unexpected keyword argument '%s'"
410
+ " to method assistants_service_create_model_metrics" % key
411
+ )
412
+ params[key] = val
413
+ del params['kwargs']
414
+ # verify the required parameter 'body' is set
415
+ if ('body' not in params or
416
+ params['body'] is None):
417
+ raise ValueError("Missing the required parameter `body` when calling `assistants_service_create_model_metrics`") # noqa: E501
418
+ # verify the required parameter 'model_id' is set
419
+ if ('model_id' not in params or
420
+ params['model_id'] is None):
421
+ raise ValueError("Missing the required parameter `model_id` when calling `assistants_service_create_model_metrics`") # noqa: E501
422
+
423
+ collection_formats = {}
424
+
425
+ path_params = {}
426
+ if 'model_id' in params:
427
+ path_params['modelId'] = params['model_id'] # noqa: E501
428
+
429
+ query_params = []
430
+
431
+ header_params = {}
432
+
433
+ form_params = []
434
+ local_var_files = {}
435
+
436
+ body_params = None
437
+ if 'body' in params:
438
+ body_params = params['body']
439
+ # HTTP header `Accept`
440
+ header_params['Accept'] = self.api_client.select_header_accept(
441
+ ['application/json']) # noqa: E501
442
+
443
+ # HTTP header `Content-Type`
444
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
445
+ ['application/json']) # noqa: E501
446
+
447
+ # Authentication setting
448
+ auth_settings = [] # noqa: E501
449
+
450
+ return self.api_client.call_api(
451
+ '/v1/agents/metrics/models/{modelId}', 'POST',
452
+ path_params,
453
+ query_params,
454
+ header_params,
455
+ body=body_params,
456
+ post_params=form_params,
457
+ files=local_var_files,
458
+ response_type='V1CreateModelMetricsResponse', # noqa: E501
459
+ auth_settings=auth_settings,
460
+ async_req=params.get('async_req'),
461
+ _return_http_data_only=params.get('_return_http_data_only'),
462
+ _preload_content=params.get('_preload_content', True),
463
+ _request_timeout=params.get('_request_timeout'),
464
+ collection_formats=collection_formats)
465
+
361
466
  def assistants_service_delete_assistant(self, project_id: 'str', id: 'str', **kwargs) -> 'V1DeleteAssistantResponse': # noqa: E501
362
467
  """assistants_service_delete_assistant # noqa: E501
363
468
 
@@ -952,6 +1057,95 @@ class AssistantsServiceApi(object):
952
1057
  _request_timeout=params.get('_request_timeout'),
953
1058
  collection_formats=collection_formats)
954
1059
 
1060
+ def assistants_service_get_latest_model_metrics(self, **kwargs) -> 'V1GetLatestModelMetricsResponse': # noqa: E501
1061
+ """assistants_service_get_latest_model_metrics # noqa: E501
1062
+
1063
+ This method makes a synchronous HTTP request by default. To make an
1064
+ asynchronous HTTP request, please pass async_req=True
1065
+ >>> thread = api.assistants_service_get_latest_model_metrics(async_req=True)
1066
+ >>> result = thread.get()
1067
+
1068
+ :param async_req bool
1069
+ :param str model_id:
1070
+ :return: V1GetLatestModelMetricsResponse
1071
+ If the method is called asynchronously,
1072
+ returns the request thread.
1073
+ """
1074
+ kwargs['_return_http_data_only'] = True
1075
+ if kwargs.get('async_req'):
1076
+ return self.assistants_service_get_latest_model_metrics_with_http_info(**kwargs) # noqa: E501
1077
+ else:
1078
+ (data) = self.assistants_service_get_latest_model_metrics_with_http_info(**kwargs) # noqa: E501
1079
+ return data
1080
+
1081
+ def assistants_service_get_latest_model_metrics_with_http_info(self, **kwargs) -> 'V1GetLatestModelMetricsResponse': # noqa: E501
1082
+ """assistants_service_get_latest_model_metrics # noqa: E501
1083
+
1084
+ This method makes a synchronous HTTP request by default. To make an
1085
+ asynchronous HTTP request, please pass async_req=True
1086
+ >>> thread = api.assistants_service_get_latest_model_metrics_with_http_info(async_req=True)
1087
+ >>> result = thread.get()
1088
+
1089
+ :param async_req bool
1090
+ :param str model_id:
1091
+ :return: V1GetLatestModelMetricsResponse
1092
+ If the method is called asynchronously,
1093
+ returns the request thread.
1094
+ """
1095
+
1096
+ all_params = ['model_id'] # noqa: E501
1097
+ all_params.append('async_req')
1098
+ all_params.append('_return_http_data_only')
1099
+ all_params.append('_preload_content')
1100
+ all_params.append('_request_timeout')
1101
+
1102
+ params = locals()
1103
+ for key, val in six.iteritems(params['kwargs']):
1104
+ if key not in all_params:
1105
+ raise TypeError(
1106
+ "Got an unexpected keyword argument '%s'"
1107
+ " to method assistants_service_get_latest_model_metrics" % key
1108
+ )
1109
+ params[key] = val
1110
+ del params['kwargs']
1111
+
1112
+ collection_formats = {}
1113
+
1114
+ path_params = {}
1115
+
1116
+ query_params = []
1117
+ if 'model_id' in params:
1118
+ query_params.append(('modelId', params['model_id'])) # noqa: E501
1119
+
1120
+ header_params = {}
1121
+
1122
+ form_params = []
1123
+ local_var_files = {}
1124
+
1125
+ body_params = None
1126
+ # HTTP header `Accept`
1127
+ header_params['Accept'] = self.api_client.select_header_accept(
1128
+ ['application/json']) # noqa: E501
1129
+
1130
+ # Authentication setting
1131
+ auth_settings = [] # noqa: E501
1132
+
1133
+ return self.api_client.call_api(
1134
+ '/v1/agents/latest-metrics', 'GET',
1135
+ path_params,
1136
+ query_params,
1137
+ header_params,
1138
+ body=body_params,
1139
+ post_params=form_params,
1140
+ files=local_var_files,
1141
+ response_type='V1GetLatestModelMetricsResponse', # noqa: E501
1142
+ auth_settings=auth_settings,
1143
+ async_req=params.get('async_req'),
1144
+ _return_http_data_only=params.get('_return_http_data_only'),
1145
+ _preload_content=params.get('_preload_content', True),
1146
+ _request_timeout=params.get('_request_timeout'),
1147
+ collection_formats=collection_formats)
1148
+
955
1149
  def assistants_service_get_managed_model_assistant(self, model_name: 'str', **kwargs) -> 'V1Assistant': # noqa: E501
956
1150
  """Each managed model has a dedicated assistant for direct interaction. By using user_name, org_name, or model_provider as query parameters, this endpoint retrieves that specific assistant only—excluding any other assistants that may use the same model. # noqa: E501
957
1151
 
@@ -1275,6 +1469,200 @@ class AssistantsServiceApi(object):
1275
1469
  _request_timeout=params.get('_request_timeout'),
1276
1470
  collection_formats=collection_formats)
1277
1471
 
1472
+ def assistants_service_get_model_metrics(self, model_id: 'str', **kwargs) -> 'V1GetModelMetricsResponse': # noqa: E501
1473
+ """assistants_service_get_model_metrics # noqa: E501
1474
+
1475
+ This method makes a synchronous HTTP request by default. To make an
1476
+ asynchronous HTTP request, please pass async_req=True
1477
+ >>> thread = api.assistants_service_get_model_metrics(model_id, async_req=True)
1478
+ >>> result = thread.get()
1479
+
1480
+ :param async_req bool
1481
+ :param str model_id: (required)
1482
+ :param datetime _from:
1483
+ :param datetime to:
1484
+ :return: V1GetModelMetricsResponse
1485
+ If the method is called asynchronously,
1486
+ returns the request thread.
1487
+ """
1488
+ kwargs['_return_http_data_only'] = True
1489
+ if kwargs.get('async_req'):
1490
+ return self.assistants_service_get_model_metrics_with_http_info(model_id, **kwargs) # noqa: E501
1491
+ else:
1492
+ (data) = self.assistants_service_get_model_metrics_with_http_info(model_id, **kwargs) # noqa: E501
1493
+ return data
1494
+
1495
+ def assistants_service_get_model_metrics_with_http_info(self, model_id: 'str', **kwargs) -> 'V1GetModelMetricsResponse': # noqa: E501
1496
+ """assistants_service_get_model_metrics # noqa: E501
1497
+
1498
+ This method makes a synchronous HTTP request by default. To make an
1499
+ asynchronous HTTP request, please pass async_req=True
1500
+ >>> thread = api.assistants_service_get_model_metrics_with_http_info(model_id, async_req=True)
1501
+ >>> result = thread.get()
1502
+
1503
+ :param async_req bool
1504
+ :param str model_id: (required)
1505
+ :param datetime _from:
1506
+ :param datetime to:
1507
+ :return: V1GetModelMetricsResponse
1508
+ If the method is called asynchronously,
1509
+ returns the request thread.
1510
+ """
1511
+
1512
+ all_params = ['model_id', '_from', 'to'] # noqa: E501
1513
+ all_params.append('async_req')
1514
+ all_params.append('_return_http_data_only')
1515
+ all_params.append('_preload_content')
1516
+ all_params.append('_request_timeout')
1517
+
1518
+ params = locals()
1519
+ for key, val in six.iteritems(params['kwargs']):
1520
+ if key not in all_params:
1521
+ raise TypeError(
1522
+ "Got an unexpected keyword argument '%s'"
1523
+ " to method assistants_service_get_model_metrics" % key
1524
+ )
1525
+ params[key] = val
1526
+ del params['kwargs']
1527
+ # verify the required parameter 'model_id' is set
1528
+ if ('model_id' not in params or
1529
+ params['model_id'] is None):
1530
+ raise ValueError("Missing the required parameter `model_id` when calling `assistants_service_get_model_metrics`") # noqa: E501
1531
+
1532
+ collection_formats = {}
1533
+
1534
+ path_params = {}
1535
+ if 'model_id' in params:
1536
+ path_params['modelId'] = params['model_id'] # noqa: E501
1537
+
1538
+ query_params = []
1539
+ if '_from' in params:
1540
+ query_params.append(('from', params['_from'])) # noqa: E501
1541
+ if 'to' in params:
1542
+ query_params.append(('to', params['to'])) # noqa: E501
1543
+
1544
+ header_params = {}
1545
+
1546
+ form_params = []
1547
+ local_var_files = {}
1548
+
1549
+ body_params = None
1550
+ # HTTP header `Accept`
1551
+ header_params['Accept'] = self.api_client.select_header_accept(
1552
+ ['application/json']) # noqa: E501
1553
+
1554
+ # Authentication setting
1555
+ auth_settings = [] # noqa: E501
1556
+
1557
+ return self.api_client.call_api(
1558
+ '/v1/agents/metrics/models/{modelId}', 'GET',
1559
+ path_params,
1560
+ query_params,
1561
+ header_params,
1562
+ body=body_params,
1563
+ post_params=form_params,
1564
+ files=local_var_files,
1565
+ response_type='V1GetModelMetricsResponse', # noqa: E501
1566
+ auth_settings=auth_settings,
1567
+ async_req=params.get('async_req'),
1568
+ _return_http_data_only=params.get('_return_http_data_only'),
1569
+ _preload_content=params.get('_preload_content', True),
1570
+ _request_timeout=params.get('_request_timeout'),
1571
+ collection_formats=collection_formats)
1572
+
1573
+ def assistants_service_get_model_total_usage_metrics(self, model_id: 'str', **kwargs) -> 'V1GetModelTotalUsageMetricsResponse': # noqa: E501
1574
+ """model metrics # noqa: E501
1575
+
1576
+ This method makes a synchronous HTTP request by default. To make an
1577
+ asynchronous HTTP request, please pass async_req=True
1578
+ >>> thread = api.assistants_service_get_model_total_usage_metrics(model_id, async_req=True)
1579
+ >>> result = thread.get()
1580
+
1581
+ :param async_req bool
1582
+ :param str model_id: (required)
1583
+ :return: V1GetModelTotalUsageMetricsResponse
1584
+ If the method is called asynchronously,
1585
+ returns the request thread.
1586
+ """
1587
+ kwargs['_return_http_data_only'] = True
1588
+ if kwargs.get('async_req'):
1589
+ return self.assistants_service_get_model_total_usage_metrics_with_http_info(model_id, **kwargs) # noqa: E501
1590
+ else:
1591
+ (data) = self.assistants_service_get_model_total_usage_metrics_with_http_info(model_id, **kwargs) # noqa: E501
1592
+ return data
1593
+
1594
+ def assistants_service_get_model_total_usage_metrics_with_http_info(self, model_id: 'str', **kwargs) -> 'V1GetModelTotalUsageMetricsResponse': # noqa: E501
1595
+ """model metrics # noqa: E501
1596
+
1597
+ This method makes a synchronous HTTP request by default. To make an
1598
+ asynchronous HTTP request, please pass async_req=True
1599
+ >>> thread = api.assistants_service_get_model_total_usage_metrics_with_http_info(model_id, async_req=True)
1600
+ >>> result = thread.get()
1601
+
1602
+ :param async_req bool
1603
+ :param str model_id: (required)
1604
+ :return: V1GetModelTotalUsageMetricsResponse
1605
+ If the method is called asynchronously,
1606
+ returns the request thread.
1607
+ """
1608
+
1609
+ all_params = ['model_id'] # noqa: E501
1610
+ all_params.append('async_req')
1611
+ all_params.append('_return_http_data_only')
1612
+ all_params.append('_preload_content')
1613
+ all_params.append('_request_timeout')
1614
+
1615
+ params = locals()
1616
+ for key, val in six.iteritems(params['kwargs']):
1617
+ if key not in all_params:
1618
+ raise TypeError(
1619
+ "Got an unexpected keyword argument '%s'"
1620
+ " to method assistants_service_get_model_total_usage_metrics" % key
1621
+ )
1622
+ params[key] = val
1623
+ del params['kwargs']
1624
+ # verify the required parameter 'model_id' is set
1625
+ if ('model_id' not in params or
1626
+ params['model_id'] is None):
1627
+ raise ValueError("Missing the required parameter `model_id` when calling `assistants_service_get_model_total_usage_metrics`") # noqa: E501
1628
+
1629
+ collection_formats = {}
1630
+
1631
+ path_params = {}
1632
+ if 'model_id' in params:
1633
+ path_params['modelId'] = params['model_id'] # noqa: E501
1634
+
1635
+ query_params = []
1636
+
1637
+ header_params = {}
1638
+
1639
+ form_params = []
1640
+ local_var_files = {}
1641
+
1642
+ body_params = None
1643
+ # HTTP header `Accept`
1644
+ header_params['Accept'] = self.api_client.select_header_accept(
1645
+ ['application/json']) # noqa: E501
1646
+
1647
+ # Authentication setting
1648
+ auth_settings = [] # noqa: E501
1649
+
1650
+ return self.api_client.call_api(
1651
+ '/v1/agents/{modelId}/total-usage', 'GET',
1652
+ path_params,
1653
+ query_params,
1654
+ header_params,
1655
+ body=body_params,
1656
+ post_params=form_params,
1657
+ files=local_var_files,
1658
+ response_type='V1GetModelTotalUsageMetricsResponse', # noqa: E501
1659
+ auth_settings=auth_settings,
1660
+ async_req=params.get('async_req'),
1661
+ _return_http_data_only=params.get('_return_http_data_only'),
1662
+ _preload_content=params.get('_preload_content', True),
1663
+ _request_timeout=params.get('_request_timeout'),
1664
+ collection_formats=collection_formats)
1665
+
1278
1666
  def assistants_service_get_published_managed_endpoint_by_model_id(self, id: 'str', **kwargs) -> 'V1ManagedEndpoint': # noqa: E501
1279
1667
  """GetPublishedManagedEndpointModel returns a managed endpoint with a single specific managed endpoint model included in modelsMetadata # noqa: E501
1280
1668
 
@@ -213,6 +213,103 @@ class AuthServiceApi(object):
213
213
  _request_timeout=params.get('_request_timeout'),
214
214
  collection_formats=collection_formats)
215
215
 
216
+ def auth_service_guest_login(self, body: 'V1GuestLoginRequest', **kwargs) -> 'V1GuestLoginResponse': # noqa: E501
217
+ """auth_service_guest_login # noqa: E501
218
+
219
+ This method makes a synchronous HTTP request by default. To make an
220
+ asynchronous HTTP request, please pass async_req=True
221
+ >>> thread = api.auth_service_guest_login(body, async_req=True)
222
+ >>> result = thread.get()
223
+
224
+ :param async_req bool
225
+ :param V1GuestLoginRequest body: (required)
226
+ :return: V1GuestLoginResponse
227
+ If the method is called asynchronously,
228
+ returns the request thread.
229
+ """
230
+ kwargs['_return_http_data_only'] = True
231
+ if kwargs.get('async_req'):
232
+ return self.auth_service_guest_login_with_http_info(body, **kwargs) # noqa: E501
233
+ else:
234
+ (data) = self.auth_service_guest_login_with_http_info(body, **kwargs) # noqa: E501
235
+ return data
236
+
237
+ def auth_service_guest_login_with_http_info(self, body: 'V1GuestLoginRequest', **kwargs) -> 'V1GuestLoginResponse': # noqa: E501
238
+ """auth_service_guest_login # noqa: E501
239
+
240
+ This method makes a synchronous HTTP request by default. To make an
241
+ asynchronous HTTP request, please pass async_req=True
242
+ >>> thread = api.auth_service_guest_login_with_http_info(body, async_req=True)
243
+ >>> result = thread.get()
244
+
245
+ :param async_req bool
246
+ :param V1GuestLoginRequest body: (required)
247
+ :return: V1GuestLoginResponse
248
+ If the method is called asynchronously,
249
+ returns the request thread.
250
+ """
251
+
252
+ all_params = ['body'] # noqa: E501
253
+ all_params.append('async_req')
254
+ all_params.append('_return_http_data_only')
255
+ all_params.append('_preload_content')
256
+ all_params.append('_request_timeout')
257
+
258
+ params = locals()
259
+ for key, val in six.iteritems(params['kwargs']):
260
+ if key not in all_params:
261
+ raise TypeError(
262
+ "Got an unexpected keyword argument '%s'"
263
+ " to method auth_service_guest_login" % key
264
+ )
265
+ params[key] = val
266
+ del params['kwargs']
267
+ # verify the required parameter 'body' is set
268
+ if ('body' not in params or
269
+ params['body'] is None):
270
+ raise ValueError("Missing the required parameter `body` when calling `auth_service_guest_login`") # noqa: E501
271
+
272
+ collection_formats = {}
273
+
274
+ path_params = {}
275
+
276
+ query_params = []
277
+
278
+ header_params = {}
279
+
280
+ form_params = []
281
+ local_var_files = {}
282
+
283
+ body_params = None
284
+ if 'body' in params:
285
+ body_params = params['body']
286
+ # HTTP header `Accept`
287
+ header_params['Accept'] = self.api_client.select_header_accept(
288
+ ['application/json']) # noqa: E501
289
+
290
+ # HTTP header `Content-Type`
291
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
292
+ ['application/json']) # noqa: E501
293
+
294
+ # Authentication setting
295
+ auth_settings = [] # noqa: E501
296
+
297
+ return self.api_client.call_api(
298
+ '/v1/auth/guest-login', 'POST',
299
+ path_params,
300
+ query_params,
301
+ header_params,
302
+ body=body_params,
303
+ post_params=form_params,
304
+ files=local_var_files,
305
+ response_type='V1GuestLoginResponse', # noqa: E501
306
+ auth_settings=auth_settings,
307
+ async_req=params.get('async_req'),
308
+ _return_http_data_only=params.get('_return_http_data_only'),
309
+ _preload_content=params.get('_preload_content', True),
310
+ _request_timeout=params.get('_request_timeout'),
311
+ collection_formats=collection_formats)
312
+
216
313
  def auth_service_login(self, body: 'V1LoginRequest', **kwargs) -> 'V1LoginResponse': # noqa: E501
217
314
  """auth_service_login # noqa: E501
218
315