lightning-sdk 0.2.22__py3-none-any.whl → 0.2.24rc0__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py

@@ -43,25 +43,30 @@ class V1MagicLinkLoginResponse(object):
     swagger_types = {
         'auth_start_url': 'str',
         'sso_enabled': 'bool',
+        'verified': 'bool',
         'waitlisted': 'bool'
     }
 
     attribute_map = {
         'auth_start_url': 'authStartUrl',
         'sso_enabled': 'ssoEnabled',
+        'verified': 'verified',
         'waitlisted': 'waitlisted'
     }
 
-    def __init__(self, auth_start_url: 'str' =None, sso_enabled: 'bool' =None, waitlisted: 'bool' =None):  # noqa: E501
+    def __init__(self, auth_start_url: 'str' =None, sso_enabled: 'bool' =None, verified: 'bool' =None, waitlisted: 'bool' =None):  # noqa: E501
         """V1MagicLinkLoginResponse - a model defined in Swagger"""  # noqa: E501
         self._auth_start_url = None
         self._sso_enabled = None
+        self._verified = None
         self._waitlisted = None
         self.discriminator = None
         if auth_start_url is not None:
             self.auth_start_url = auth_start_url
         if sso_enabled is not None:
             self.sso_enabled = sso_enabled
+        if verified is not None:
+            self.verified = verified
         if waitlisted is not None:
             self.waitlisted = waitlisted
 
@@ -107,6 +112,27 @@ class V1MagicLinkLoginResponse(object):
 
         self._sso_enabled = sso_enabled
 
+    @property
+    def verified(self) -> 'bool':
+        """Gets the verified of this V1MagicLinkLoginResponse.  # noqa: E501
+
+
+        :return: The verified of this V1MagicLinkLoginResponse.  # noqa: E501
+        :rtype: bool
+        """
+        return self._verified
+
+    @verified.setter
+    def verified(self, verified: 'bool'):
+        """Sets the verified of this V1MagicLinkLoginResponse.
+
+
+        :param verified: The verified of this V1MagicLinkLoginResponse.  # noqa: E501
+        :type: bool
+        """
+
+        self._verified = verified
+
     @property
     def waitlisted(self) -> 'bool':
         """Gets the waitlisted of this V1MagicLinkLoginResponse.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py

@@ -46,6 +46,7 @@ class V1MetricsStream(object):
         'cluster_id': 'str',
         'created_at': 'datetime',
         'dark_color': 'str',
+        'experiment_project_name': 'str',
         'id': 'str',
         'job_id': 'str',
         'job_name': 'str',
@@ -77,6 +78,7 @@ class V1MetricsStream(object):
         'cluster_id': 'clusterId',
         'created_at': 'createdAt',
         'dark_color': 'darkColor',
+        'experiment_project_name': 'experimentProjectName',
         'id': 'id',
         'job_id': 'jobId',
         'job_name': 'jobName',
@@ -102,13 +104,14 @@ class V1MetricsStream(object):
         'work_id': 'workId'
     }
 
-    def __init__(self, app_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, dark_color: 'str' =None, id: 'str' =None, job_id: 'str' =None, job_name: 'str' =None, light_color: 'str' =None, name: 'str' =None, persisted: 'bool' =None, phase: 'V1PhaseType' =None, plugin_id: 'str' =None, project_id: 'str' =None, source_id: 'str' =None, stopped_at: 'datetime' =None, store_created_at: 'bool' =None, store_step: 'bool' =None, system_info: 'V1SystemInfo' =None, tags: 'list[V1MetricsTags]' =None, trackers: 'dict(str, V1MetricsTracker)' =None, updated_at: 'datetime' =None, updates_counter: 'int' =None, user_id: 'str' =None, version: 'datetime' =None, version_number: 'int' =None, visibility: 'V1ResourceVisibility' =None, work_id: 'str' =None):  # noqa: E501
+    def __init__(self, app_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, dark_color: 'str' =None, experiment_project_name: 'str' =None, id: 'str' =None, job_id: 'str' =None, job_name: 'str' =None, light_color: 'str' =None, name: 'str' =None, persisted: 'bool' =None, phase: 'V1PhaseType' =None, plugin_id: 'str' =None, project_id: 'str' =None, source_id: 'str' =None, stopped_at: 'datetime' =None, store_created_at: 'bool' =None, store_step: 'bool' =None, system_info: 'V1SystemInfo' =None, tags: 'list[V1MetricsTags]' =None, trackers: 'dict(str, V1MetricsTracker)' =None, updated_at: 'datetime' =None, updates_counter: 'int' =None, user_id: 'str' =None, version: 'datetime' =None, version_number: 'int' =None, visibility: 'V1ResourceVisibility' =None, work_id: 'str' =None):  # noqa: E501
         """V1MetricsStream - a model defined in Swagger"""  # noqa: E501
         self._app_id = None
         self._cloudspace_id = None
         self._cluster_id = None
         self._created_at = None
         self._dark_color = None
+        self._experiment_project_name = None
         self._id = None
         self._job_id = None
         self._job_name = None
@@ -143,6 +146,8 @@ class V1MetricsStream(object):
             self.created_at = created_at
         if dark_color is not None:
             self.dark_color = dark_color
+        if experiment_project_name is not None:
+            self.experiment_project_name = experiment_project_name
         if id is not None:
             self.id = id
         if job_id is not None:
@@ -295,6 +300,27 @@ class V1MetricsStream(object):
 
         self._dark_color = dark_color
 
+    @property
+    def experiment_project_name(self) -> 'str':
+        """Gets the experiment_project_name of this V1MetricsStream.  # noqa: E501
+
+
+        :return: The experiment_project_name of this V1MetricsStream.  # noqa: E501
+        :rtype: str
+        """
+        return self._experiment_project_name
+
+    @experiment_project_name.setter
+    def experiment_project_name(self, experiment_project_name: 'str'):
+        """Sets the experiment_project_name of this V1MetricsStream.
+
+
+        :param experiment_project_name: The experiment_project_name of this V1MetricsStream.  # noqa: E501
+        :type: str
+        """
+
+        self._experiment_project_name = experiment_project_name
+
     @property
     def id(self) -> 'str':
         """Gets the id of this V1MetricsStream.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_organization.py

@@ -58,6 +58,7 @@ class V1Organization(object):
         'created_at': 'datetime',
         'default_machine_image_version': 'str',
         'default_machine_type': 'str',
+        'default_project_id': 'str',
         'description': 'str',
         'disallow_aws_saas': 'bool',
         'disallow_dgx_saas': 'bool',
@@ -108,6 +109,7 @@ class V1Organization(object):
         'created_at': 'createdAt',
         'default_machine_image_version': 'defaultMachineImageVersion',
         'default_machine_type': 'defaultMachineType',
+        'default_project_id': 'defaultProjectId',
         'description': 'description',
         'disallow_aws_saas': 'disallowAwsSaas',
         'disallow_dgx_saas': 'disallowDgxSaas',
@@ -140,7 +142,7 @@ class V1Organization(object):
         'workload_max_run_duration': 'workloadMaxRunDuration'
     }
 
-    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_budgeting: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_guest: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, created_at: 'datetime' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, description: 'str' =None, disallow_aws_saas: 'bool' =None, disallow_dgx_saas: 'bool' =None, disallow_gcp_saas: 'bool' =None, disallow_lambda_saas: 'bool' =None, disallow_lightning_saas: 'bool' =None, disallow_vultr_saas: 'bool' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, full_story_end_date: 'datetime' =None, full_story_start_date: 'datetime' =None, general_teamspace: 'bool' =None, id: 'str' =None, last_storage_overuse_notification_sent_at: 'datetime' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, storage_overuse_bytes: 'str' =None, storage_overuse_deletion_at: 'datetime' =None, storage_overuse_notification_count: 'int' =None, switch_to_default_machine_on_idle: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None, workload_max_run_duration: 'str' =None):  # noqa: E501
+    def __init__(self, alerts_config: 'V1AlertsConfig' =None, allow_budgeting: 'bool' =None, allow_credits_auto_replenish: 'bool' =None, allow_external_project_duplication: 'bool' =None, allow_guest: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, auto_replenish_amount: 'float' =None, auto_replenish_threshold: 'float' =None, auto_switch_machine: 'bool' =None, created_at: 'datetime' =None, default_machine_image_version: 'str' =None, default_machine_type: 'str' =None, default_project_id: 'str' =None, description: 'str' =None, disallow_aws_saas: 'bool' =None, disallow_dgx_saas: 'bool' =None, disallow_gcp_saas: 'bool' =None, disallow_lambda_saas: 'bool' =None, disallow_lightning_saas: 'bool' =None, disallow_vultr_saas: 'bool' =None, display_name: 'str' =None, domain: 'str' =None, email: 'str' =None, featured_gallery: 'bool' =None, full_story_end_date: 'datetime' =None, full_story_start_date: 'datetime' =None, general_teamspace: 'bool' =None, id: 'str' =None, last_storage_overuse_notification_sent_at: 'datetime' =None, location: 'str' =None, name: 'str' =None, preferred_cluster: 'str' =None, preferred_deployment_provider: 'str' =None, preferred_studio_provider: 'str' =None, start_studios_on_spot_instance: 'bool' =None, storage_overuse_bytes: 'str' =None, storage_overuse_deletion_at: 'datetime' =None, storage_overuse_notification_count: 'int' =None, switch_to_default_machine_on_idle: 'bool' =None, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None, workload_max_run_duration: 'str' =None):  # noqa: E501
         """V1Organization - a model defined in Swagger"""  # noqa: E501
         self._alerts_config = None
         self._allow_budgeting = None
@@ -159,6 +161,7 @@ class V1Organization(object):
         self._created_at = None
         self._default_machine_image_version = None
         self._default_machine_type = None
+        self._default_project_id = None
         self._description = None
         self._disallow_aws_saas = None
         self._disallow_dgx_saas = None
@@ -224,6 +227,8 @@ class V1Organization(object):
             self.default_machine_image_version = default_machine_image_version
         if default_machine_type is not None:
             self.default_machine_type = default_machine_type
+        if default_project_id is not None:
+            self.default_project_id = default_project_id
         if description is not None:
             self.description = description
         if disallow_aws_saas is not None:
@@ -644,6 +649,27 @@ class V1Organization(object):
 
         self._default_machine_type = default_machine_type
 
+    @property
+    def default_project_id(self) -> 'str':
+        """Gets the default_project_id of this V1Organization.  # noqa: E501
+
+
+        :return: The default_project_id of this V1Organization.  # noqa: E501
+        :rtype: str
+        """
+        return self._default_project_id
+
+    @default_project_id.setter
+    def default_project_id(self, default_project_id: 'str'):
+        """Sets the default_project_id of this V1Organization.
+
+
+        :param default_project_id: The default_project_id of this V1Organization.  # noqa: E501
+        :type: str
+        """
+
+        self._default_project_id = default_project_id
+
     @property
     def description(self) -> 'str':
         """Gets the description of this V1Organization.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py

@@ -43,6 +43,7 @@ class V1Pipeline(object):
     swagger_types = {
         'cluster_id': 'str',
         'created_at': 'datetime',
+        'created_from_ui': 'bool',
         'display_name': 'str',
         'error': 'str',
         'id': 'str',
@@ -64,6 +65,7 @@ class V1Pipeline(object):
     attribute_map = {
         'cluster_id': 'clusterId',
         'created_at': 'createdAt',
+        'created_from_ui': 'createdFromUi',
         'display_name': 'displayName',
         'error': 'error',
         'id': 'id',
@@ -82,10 +84,11 @@ class V1Pipeline(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, cluster_id: 'str' =None, created_at: 'datetime' =None, display_name: 'str' =None, error: 'str' =None, id: 'str' =None, is_published: 'bool' =None, message: 'str' =None, name: 'str' =None, parameters: 'list[V1PipelineParameter]' =None, parent_pipeline_id: 'str' =None, project_id: 'str' =None, schedule_id: 'str' =None, shared_filesystem: 'V1SharedFilesystem' =None, state: 'V1PipelineState' =None, statuses: 'list[V1PipelineStepStatus]' =None, steps: 'list[V1PipelineStep]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, cluster_id: 'str' =None, created_at: 'datetime' =None, created_from_ui: 'bool' =None, display_name: 'str' =None, error: 'str' =None, id: 'str' =None, is_published: 'bool' =None, message: 'str' =None, name: 'str' =None, parameters: 'list[V1PipelineParameter]' =None, parent_pipeline_id: 'str' =None, project_id: 'str' =None, schedule_id: 'str' =None, shared_filesystem: 'V1SharedFilesystem' =None, state: 'V1PipelineState' =None, statuses: 'list[V1PipelineStepStatus]' =None, steps: 'list[V1PipelineStep]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Pipeline - a model defined in Swagger"""  # noqa: E501
         self._cluster_id = None
         self._created_at = None
+        self._created_from_ui = None
         self._display_name = None
         self._error = None
         self._id = None
@@ -107,6 +110,8 @@ class V1Pipeline(object):
             self.cluster_id = cluster_id
         if created_at is not None:
             self.created_at = created_at
+        if created_from_ui is not None:
+            self.created_from_ui = created_from_ui
         if display_name is not None:
             self.display_name = display_name
         if error is not None:
@@ -182,6 +187,27 @@ class V1Pipeline(object):
 
         self._created_at = created_at
 
+    @property
+    def created_from_ui(self) -> 'bool':
+        """Gets the created_from_ui of this V1Pipeline.  # noqa: E501
+
+
+        :return: The created_from_ui of this V1Pipeline.  # noqa: E501
+        :rtype: bool
+        """
+        return self._created_from_ui
+
+    @created_from_ui.setter
+    def created_from_ui(self, created_from_ui: 'bool'):
+        """Sets the created_from_ui of this V1Pipeline.
+
+
+        :param created_from_ui: The created_from_ui of this V1Pipeline.  # noqa: E501
+        :type: bool
+        """
+
+        self._created_from_ui = created_from_ui
+
     @property
     def display_name(self) -> 'str':
         """Gets the display_name of this V1Pipeline.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py

@@ -41,35 +41,81 @@ class V1SharedFilesystem(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'efs': 'bool',
         'enabled': 'bool',
+        'filestore': 'bool',
+        'gcs_folder': 'bool',
+        'r2_folder': 'bool',
         'resource_id': 'str',
         'resource_name': 'str',
+        's3_folder': 'bool',
         'state': 'str'
     }
 
     attribute_map = {
+        'efs': 'efs',
         'enabled': 'enabled',
+        'filestore': 'filestore',
+        'gcs_folder': 'gcsFolder',
+        'r2_folder': 'r2Folder',
         'resource_id': 'resourceId',
         'resource_name': 'resourceName',
+        's3_folder': 's3Folder',
         'state': 'state'
     }
 
-    def __init__(self, enabled: 'bool' =None, resource_id: 'str' =None, resource_name: 'str' =None, state: 'str' =None):  # noqa: E501
+    def __init__(self, efs: 'bool' =None, enabled: 'bool' =None, filestore: 'bool' =None, gcs_folder: 'bool' =None, r2_folder: 'bool' =None, resource_id: 'str' =None, resource_name: 'str' =None, s3_folder: 'bool' =None, state: 'str' =None):  # noqa: E501
         """V1SharedFilesystem - a model defined in Swagger"""  # noqa: E501
+        self._efs = None
         self._enabled = None
+        self._filestore = None
+        self._gcs_folder = None
+        self._r2_folder = None
         self._resource_id = None
         self._resource_name = None
+        self._s3_folder = None
         self._state = None
         self.discriminator = None
+        if efs is not None:
+            self.efs = efs
         if enabled is not None:
             self.enabled = enabled
+        if filestore is not None:
+            self.filestore = filestore
+        if gcs_folder is not None:
+            self.gcs_folder = gcs_folder
+        if r2_folder is not None:
+            self.r2_folder = r2_folder
         if resource_id is not None:
             self.resource_id = resource_id
         if resource_name is not None:
             self.resource_name = resource_name
+        if s3_folder is not None:
+            self.s3_folder = s3_folder
         if state is not None:
             self.state = state
 
+    @property
+    def efs(self) -> 'bool':
+        """Gets the efs of this V1SharedFilesystem.  # noqa: E501
+
+
+        :return: The efs of this V1SharedFilesystem.  # noqa: E501
+        :rtype: bool
+        """
+        return self._efs
+
+    @efs.setter
+    def efs(self, efs: 'bool'):
+        """Sets the efs of this V1SharedFilesystem.
+
+
+        :param efs: The efs of this V1SharedFilesystem.  # noqa: E501
+        :type: bool
+        """
+
+        self._efs = efs
+
     @property
     def enabled(self) -> 'bool':
         """Gets the enabled of this V1SharedFilesystem.  # noqa: E501
@@ -91,6 +137,69 @@ class V1SharedFilesystem(object):
 
         self._enabled = enabled
 
+    @property
+    def filestore(self) -> 'bool':
+        """Gets the filestore of this V1SharedFilesystem.  # noqa: E501
+
+
+        :return: The filestore of this V1SharedFilesystem.  # noqa: E501
+        :rtype: bool
+        """
+        return self._filestore
+
+    @filestore.setter
+    def filestore(self, filestore: 'bool'):
+        """Sets the filestore of this V1SharedFilesystem.
+
+
+        :param filestore: The filestore of this V1SharedFilesystem.  # noqa: E501
+        :type: bool
+        """
+
+        self._filestore = filestore
+
+    @property
+    def gcs_folder(self) -> 'bool':
+        """Gets the gcs_folder of this V1SharedFilesystem.  # noqa: E501
+
+
+        :return: The gcs_folder of this V1SharedFilesystem.  # noqa: E501
+        :rtype: bool
+        """
+        return self._gcs_folder
+
+    @gcs_folder.setter
+    def gcs_folder(self, gcs_folder: 'bool'):
+        """Sets the gcs_folder of this V1SharedFilesystem.
+
+
+        :param gcs_folder: The gcs_folder of this V1SharedFilesystem.  # noqa: E501
+        :type: bool
+        """
+
+        self._gcs_folder = gcs_folder
+
+    @property
+    def r2_folder(self) -> 'bool':
+        """Gets the r2_folder of this V1SharedFilesystem.  # noqa: E501
+
+
+        :return: The r2_folder of this V1SharedFilesystem.  # noqa: E501
+        :rtype: bool
+        """
+        return self._r2_folder
+
+    @r2_folder.setter
+    def r2_folder(self, r2_folder: 'bool'):
+        """Sets the r2_folder of this V1SharedFilesystem.
+
+
+        :param r2_folder: The r2_folder of this V1SharedFilesystem.  # noqa: E501
+        :type: bool
+        """
+
+        self._r2_folder = r2_folder
+
     @property
     def resource_id(self) -> 'str':
         """Gets the resource_id of this V1SharedFilesystem.  # noqa: E501
@@ -133,6 +242,27 @@ class V1SharedFilesystem(object):
 
         self._resource_name = resource_name
 
+    @property
+    def s3_folder(self) -> 'bool':
+        """Gets the s3_folder of this V1SharedFilesystem.  # noqa: E501
+
+
+        :return: The s3_folder of this V1SharedFilesystem.  # noqa: E501
+        :rtype: bool
+        """
+        return self._s3_folder
+
+    @s3_folder.setter
+    def s3_folder(self, s3_folder: 'bool'):
+        """Sets the s3_folder of this V1SharedFilesystem.
+
+
+        :param s3_folder: The s3_folder of this V1SharedFilesystem.  # noqa: E501
+        :type: bool
+        """
+
+        self._s3_folder = s3_folder
+
     @property
     def state(self) -> 'str':
         """Gets the state of this V1SharedFilesystem.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py

@@ -0,0 +1,175 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1TokenUsage(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'completion_tokens': 'str',
+        'prompt_tokens': 'str',
+        'total_tokens': 'str'
+    }
+
+    attribute_map = {
+        'completion_tokens': 'completionTokens',
+        'prompt_tokens': 'promptTokens',
+        'total_tokens': 'totalTokens'
+    }
+
+    def __init__(self, completion_tokens: 'str' =None, prompt_tokens: 'str' =None, total_tokens: 'str' =None):  # noqa: E501
+        """V1TokenUsage - a model defined in Swagger"""  # noqa: E501
+        self._completion_tokens = None
+        self._prompt_tokens = None
+        self._total_tokens = None
+        self.discriminator = None
+        if completion_tokens is not None:
+            self.completion_tokens = completion_tokens
+        if prompt_tokens is not None:
+            self.prompt_tokens = prompt_tokens
+        if total_tokens is not None:
+            self.total_tokens = total_tokens
+
+    @property
+    def completion_tokens(self) -> 'str':
+        """Gets the completion_tokens of this V1TokenUsage.  # noqa: E501
+
+
+        :return: The completion_tokens of this V1TokenUsage.  # noqa: E501
+        :rtype: str
+        """
+        return self._completion_tokens
+
+    @completion_tokens.setter
+    def completion_tokens(self, completion_tokens: 'str'):
+        """Sets the completion_tokens of this V1TokenUsage.
+
+
+        :param completion_tokens: The completion_tokens of this V1TokenUsage.  # noqa: E501
+        :type: str
+        """
+
+        self._completion_tokens = completion_tokens
+
+    @property
+    def prompt_tokens(self) -> 'str':
+        """Gets the prompt_tokens of this V1TokenUsage.  # noqa: E501
+
+
+        :return: The prompt_tokens of this V1TokenUsage.  # noqa: E501
+        :rtype: str
+        """
+        return self._prompt_tokens
+
+    @prompt_tokens.setter
+    def prompt_tokens(self, prompt_tokens: 'str'):
+        """Sets the prompt_tokens of this V1TokenUsage.
+
+
+        :param prompt_tokens: The prompt_tokens of this V1TokenUsage.  # noqa: E501
+        :type: str
+        """
+
+        self._prompt_tokens = prompt_tokens
+
+    @property
+    def total_tokens(self) -> 'str':
+        """Gets the total_tokens of this V1TokenUsage.  # noqa: E501
+
+
+        :return: The total_tokens of this V1TokenUsage.  # noqa: E501
+        :rtype: str
+        """
+        return self._total_tokens
+
+    @total_tokens.setter
+    def total_tokens(self, total_tokens: 'str'):
+        """Sets the total_tokens of this V1TokenUsage.
+
+
+        :param total_tokens: The total_tokens of this V1TokenUsage.  # noqa: E501
+        :type: str
+        """
+
+        self._total_tokens = total_tokens
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1TokenUsage, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1TokenUsage') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1TokenUsage):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1TokenUsage') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other