lightning-sdk 0.2.3__py3-none-any.whl → 0.2.5__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py

@@ -43,43 +43,57 @@ class V1ManagedModel(object):
     swagger_types = {
         'abilities': 'V1ManagedModelAbilities',
         'completion_token_price': 'float',
+        'context_length': 'str',
         'deployment_details': 'V1DeploymentDetails',
         'description': 'str',
         'endpoint_id': 'str',
         'id': 'str',
+        'max_completion_tokens': 'str',
         'name': 'str',
         'prompt_token_price': 'float',
-        'status': 'V1AssistantModelStatus'
+        'status': 'V1AssistantModelStatus',
+        'temperature': 'float',
+        'top_k': 'str'
     }
 
     attribute_map = {
         'abilities': 'abilities',
         'completion_token_price': 'completionTokenPrice',
+        'context_length': 'contextLength',
         'deployment_details': 'deploymentDetails',
         'description': 'description',
         'endpoint_id': 'endpointId',
         'id': 'id',
+        'max_completion_tokens': 'maxCompletionTokens',
         'name': 'name',
         'prompt_token_price': 'promptTokenPrice',
-        'status': 'status'
+        'status': 'status',
+        'temperature': 'temperature',
+        'top_k': 'topK'
     }
 
-    def __init__(self, abilities: 'V1ManagedModelAbilities' =None, completion_token_price: 'float' =None, deployment_details: 'V1DeploymentDetails' =None, description: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None, status: 'V1AssistantModelStatus' =None):  # noqa: E501
+    def __init__(self, abilities: 'V1ManagedModelAbilities' =None, completion_token_price: 'float' =None, context_length: 'str' =None, deployment_details: 'V1DeploymentDetails' =None, description: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, max_completion_tokens: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None, status: 'V1AssistantModelStatus' =None, temperature: 'float' =None, top_k: 'str' =None):  # noqa: E501
         """V1ManagedModel - a model defined in Swagger"""  # noqa: E501
         self._abilities = None
         self._completion_token_price = None
+        self._context_length = None
         self._deployment_details = None
         self._description = None
         self._endpoint_id = None
         self._id = None
+        self._max_completion_tokens = None
         self._name = None
         self._prompt_token_price = None
         self._status = None
+        self._temperature = None
+        self._top_k = None
         self.discriminator = None
         if abilities is not None:
             self.abilities = abilities
         if completion_token_price is not None:
             self.completion_token_price = completion_token_price
+        if context_length is not None:
+            self.context_length = context_length
         if deployment_details is not None:
             self.deployment_details = deployment_details
         if description is not None:
@@ -88,12 +102,18 @@ class V1ManagedModel(object):
             self.endpoint_id = endpoint_id
         if id is not None:
             self.id = id
+        if max_completion_tokens is not None:
+            self.max_completion_tokens = max_completion_tokens
         if name is not None:
             self.name = name
         if prompt_token_price is not None:
             self.prompt_token_price = prompt_token_price
         if status is not None:
             self.status = status
+        if temperature is not None:
+            self.temperature = temperature
+        if top_k is not None:
+            self.top_k = top_k
 
     @property
     def abilities(self) -> 'V1ManagedModelAbilities':
@@ -137,6 +157,27 @@ class V1ManagedModel(object):
 
         self._completion_token_price = completion_token_price
 
+    @property
+    def context_length(self) -> 'str':
+        """Gets the context_length of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The context_length of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._context_length
+
+    @context_length.setter
+    def context_length(self, context_length: 'str'):
+        """Sets the context_length of this V1ManagedModel.
+
+
+        :param context_length: The context_length of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._context_length = context_length
+
     @property
     def deployment_details(self) -> 'V1DeploymentDetails':
         """Gets the deployment_details of this V1ManagedModel.  # noqa: E501
@@ -221,6 +262,27 @@ class V1ManagedModel(object):
 
         self._id = id
 
+    @property
+    def max_completion_tokens(self) -> 'str':
+        """Gets the max_completion_tokens of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The max_completion_tokens of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._max_completion_tokens
+
+    @max_completion_tokens.setter
+    def max_completion_tokens(self, max_completion_tokens: 'str'):
+        """Sets the max_completion_tokens of this V1ManagedModel.
+
+
+        :param max_completion_tokens: The max_completion_tokens of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._max_completion_tokens = max_completion_tokens
+
     @property
     def name(self) -> 'str':
         """Gets the name of this V1ManagedModel.  # noqa: E501
@@ -284,6 +346,48 @@ class V1ManagedModel(object):
 
         self._status = status
 
+    @property
+    def temperature(self) -> 'float':
+        """Gets the temperature of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The temperature of this V1ManagedModel.  # noqa: E501
+        :rtype: float
+        """
+        return self._temperature
+
+    @temperature.setter
+    def temperature(self, temperature: 'float'):
+        """Sets the temperature of this V1ManagedModel.
+
+
+        :param temperature: The temperature of this V1ManagedModel.  # noqa: E501
+        :type: float
+        """
+
+        self._temperature = temperature
+
+    @property
+    def top_k(self) -> 'str':
+        """Gets the top_k of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The top_k of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._top_k
+
+    @top_k.setter
+    def top_k(self, top_k: 'str'):
+        """Sets the top_k of this V1ManagedModel.
+
+
+        :param top_k: The top_k of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._top_k = top_k
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_organization.py

@@ -45,6 +45,7 @@ class V1Organization(object):
         'allow_gcp_saas': 'bool',
         'allow_guest': 'bool',
         'allow_lambda_saas': 'bool',
+        'allow_marketplace': 'bool',
         'allow_member_invitations': 'bool',
         'allow_member_teamspace_creation': 'bool',
         'allow_vultr_saas': 'bool',
@@ -52,6 +53,7 @@ class V1Organization(object):
         'auto_join_domain_validations': 'dict(str, V1AutoJoinDomainValidation)',
         'auto_join_domains': 'list[str]',
         'created_at': 'datetime',
+        'default_machine_image_version': 'str',
         'description': 'str',
         'display_name': 'str',
         'domain': 'str',
@@ -76,6 +78,7 @@ class V1Organization(object):
         'allow_gcp_saas': 'allowGcpSaas',
         'allow_guest': 'allowGuest',
         'allow_lambda_saas': 'allowLambdaSaas',
+        'allow_marketplace': 'allowMarketplace',
         'allow_member_invitations': 'allowMemberInvitations',
         'allow_member_teamspace_creation': 'allowMemberTeamspaceCreation',
         'allow_vultr_saas': 'allowVultrSaas',
@@ -83,6 +86,7 @@ class V1Organization(object):
         'auto_join_domain_validations': 'autoJoinDomainValidations',
         'auto_join_domains': 'autoJoinDomains',
         'created_at': 'createdAt',
+        'default_machine_image_version': 'defaultMachineImageVersion',
         'description': 'description',
         'display_name': 'displayName',
         'domain': 'domain',
@@ -102,12 +106,13 @@ class V1Organization(object):
         'updated_at': 'updatedAt'
     }
 
-    def __init__(self, allow_aws_saas: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, created_at: 'datetime' =None, description: 'str' =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, id: 'str' =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, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None):  # noqa: E501
+    def __init__(self, allow_aws_saas: 'bool' =None, allow_gcp_saas: 'bool' =None, allow_guest: 'bool' =None, allow_lambda_saas: 'bool' =None, allow_marketplace: 'bool' =None, allow_member_invitations: 'bool' =None, allow_member_teamspace_creation: 'bool' =None, allow_vultr_saas: 'bool' =None, auto_invite_by_domain: 'bool' =None, auto_join_domain_validations: 'dict(str, V1AutoJoinDomainValidation)' =None, auto_join_domains: 'list[str]' =None, created_at: 'datetime' =None, default_machine_image_version: 'str' =None, description: 'str' =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, id: 'str' =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, teamspace_default_credits: 'float' =None, twitter_username: 'str' =None, updated_at: 'datetime' =None):  # noqa: E501
         """V1Organization - a model defined in Swagger"""  # noqa: E501
         self._allow_aws_saas = None
         self._allow_gcp_saas = None
         self._allow_guest = None
         self._allow_lambda_saas = None
+        self._allow_marketplace = None
         self._allow_member_invitations = None
         self._allow_member_teamspace_creation = None
         self._allow_vultr_saas = None
@@ -115,6 +120,7 @@ class V1Organization(object):
         self._auto_join_domain_validations = None
         self._auto_join_domains = None
         self._created_at = None
+        self._default_machine_image_version = None
         self._description = None
         self._display_name = None
         self._domain = None
@@ -141,6 +147,8 @@ class V1Organization(object):
             self.allow_guest = allow_guest
         if allow_lambda_saas is not None:
             self.allow_lambda_saas = allow_lambda_saas
+        if allow_marketplace is not None:
+            self.allow_marketplace = allow_marketplace
         if allow_member_invitations is not None:
             self.allow_member_invitations = allow_member_invitations
         if allow_member_teamspace_creation is not None:
@@ -155,6 +163,8 @@ class V1Organization(object):
             self.auto_join_domains = auto_join_domains
         if created_at is not None:
             self.created_at = created_at
+        if default_machine_image_version is not None:
+            self.default_machine_image_version = default_machine_image_version
         if description is not None:
             self.description = description
         if display_name is not None:
@@ -274,6 +284,27 @@ class V1Organization(object):
 
         self._allow_lambda_saas = allow_lambda_saas
 
+    @property
+    def allow_marketplace(self) -> 'bool':
+        """Gets the allow_marketplace of this V1Organization.  # noqa: E501
+
+
+        :return: The allow_marketplace of this V1Organization.  # noqa: E501
+        :rtype: bool
+        """
+        return self._allow_marketplace
+
+    @allow_marketplace.setter
+    def allow_marketplace(self, allow_marketplace: 'bool'):
+        """Sets the allow_marketplace of this V1Organization.
+
+
+        :param allow_marketplace: The allow_marketplace of this V1Organization.  # noqa: E501
+        :type: bool
+        """
+
+        self._allow_marketplace = allow_marketplace
+
     @property
     def allow_member_invitations(self) -> 'bool':
         """Gets the allow_member_invitations of this V1Organization.  # noqa: E501
@@ -421,6 +452,27 @@ class V1Organization(object):
 
         self._created_at = created_at
 
+    @property
+    def default_machine_image_version(self) -> 'str':
+        """Gets the default_machine_image_version of this V1Organization.  # noqa: E501
+
+
+        :return: The default_machine_image_version of this V1Organization.  # noqa: E501
+        :rtype: str
+        """
+        return self._default_machine_image_version
+
+    @default_machine_image_version.setter
+    def default_machine_image_version(self, default_machine_image_version: 'str'):
+        """Sets the default_machine_image_version of this V1Organization.
+
+
+        :param default_machine_image_version: The default_machine_image_version of this V1Organization.  # noqa: E501
+        :type: str
+        """
+
+        self._default_machine_image_version = default_machine_image_version
+
     @property
     def description(self) -> 'str':
         """Gets the description of this V1Organization.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py

@@ -46,11 +46,13 @@ class V1Pipeline(object):
         'display_name': 'str',
         'error': 'str',
         'id': 'str',
+        'is_published': 'bool',
         'message': 'str',
         'name': 'str',
+        'parameters': 'list[V1PipelineParameter]',
         'parent_pipeline_id': 'str',
         'project_id': 'str',
-        'schedules': 'list[V1PipelineSchedule]',
+        'schedule_id': 'str',
         'shared_filesystem': 'V1SharedFilesystem',
         'state': 'str',
         'statuses': 'list[V1PipelineStepStatus]',
@@ -65,11 +67,13 @@ class V1Pipeline(object):
         'display_name': 'displayName',
         'error': 'error',
         'id': 'id',
+        'is_published': 'isPublished',
         'message': 'message',
         'name': 'name',
+        'parameters': 'parameters',
         'parent_pipeline_id': 'parentPipelineId',
         'project_id': 'projectId',
-        'schedules': 'schedules',
+        'schedule_id': 'scheduleId',
         'shared_filesystem': 'sharedFilesystem',
         'state': 'state',
         'statuses': 'statuses',
@@ -78,18 +82,20 @@ 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, message: 'str' =None, name: 'str' =None, parent_pipeline_id: 'str' =None, project_id: 'str' =None, schedules: 'list[V1PipelineSchedule]' =None, shared_filesystem: 'V1SharedFilesystem' =None, state: 'str' =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, 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: 'str' =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._display_name = None
         self._error = None
         self._id = None
+        self._is_published = None
         self._message = None
         self._name = None
+        self._parameters = None
         self._parent_pipeline_id = None
         self._project_id = None
-        self._schedules = None
+        self._schedule_id = None
         self._shared_filesystem = None
         self._state = None
         self._statuses = None
@@ -107,16 +113,20 @@ class V1Pipeline(object):
             self.error = error
         if id is not None:
             self.id = id
+        if is_published is not None:
+            self.is_published = is_published
         if message is not None:
             self.message = message
         if name is not None:
             self.name = name
+        if parameters is not None:
+            self.parameters = parameters
         if parent_pipeline_id is not None:
             self.parent_pipeline_id = parent_pipeline_id
         if project_id is not None:
             self.project_id = project_id
-        if schedules is not None:
-            self.schedules = schedules
+        if schedule_id is not None:
+            self.schedule_id = schedule_id
         if shared_filesystem is not None:
             self.shared_filesystem = shared_filesystem
         if state is not None:
@@ -235,6 +245,27 @@ class V1Pipeline(object):
 
         self._id = id
 
+    @property
+    def is_published(self) -> 'bool':
+        """Gets the is_published of this V1Pipeline.  # noqa: E501
+
+
+        :return: The is_published of this V1Pipeline.  # noqa: E501
+        :rtype: bool
+        """
+        return self._is_published
+
+    @is_published.setter
+    def is_published(self, is_published: 'bool'):
+        """Sets the is_published of this V1Pipeline.
+
+
+        :param is_published: The is_published of this V1Pipeline.  # noqa: E501
+        :type: bool
+        """
+
+        self._is_published = is_published
+
     @property
     def message(self) -> 'str':
         """Gets the message of this V1Pipeline.  # noqa: E501
@@ -277,6 +308,27 @@ class V1Pipeline(object):
 
         self._name = name
 
+    @property
+    def parameters(self) -> 'list[V1PipelineParameter]':
+        """Gets the parameters of this V1Pipeline.  # noqa: E501
+
+
+        :return: The parameters of this V1Pipeline.  # noqa: E501
+        :rtype: list[V1PipelineParameter]
+        """
+        return self._parameters
+
+    @parameters.setter
+    def parameters(self, parameters: 'list[V1PipelineParameter]'):
+        """Sets the parameters of this V1Pipeline.
+
+
+        :param parameters: The parameters of this V1Pipeline.  # noqa: E501
+        :type: list[V1PipelineParameter]
+        """
+
+        self._parameters = parameters
+
     @property
     def parent_pipeline_id(self) -> 'str':
         """Gets the parent_pipeline_id of this V1Pipeline.  # noqa: E501
@@ -320,25 +372,25 @@ class V1Pipeline(object):
         self._project_id = project_id
 
     @property
-    def schedules(self) -> 'list[V1PipelineSchedule]':
-        """Gets the schedules of this V1Pipeline.  # noqa: E501
+    def schedule_id(self) -> 'str':
+        """Gets the schedule_id of this V1Pipeline.  # noqa: E501
 
 
-        :return: The schedules of this V1Pipeline.  # noqa: E501
-        :rtype: list[V1PipelineSchedule]
+        :return: The schedule_id of this V1Pipeline.  # noqa: E501
+        :rtype: str
         """
-        return self._schedules
+        return self._schedule_id
 
-    @schedules.setter
-    def schedules(self, schedules: 'list[V1PipelineSchedule]'):
-        """Sets the schedules of this V1Pipeline.
+    @schedule_id.setter
+    def schedule_id(self, schedule_id: 'str'):
+        """Sets the schedule_id of this V1Pipeline.
 
 
-        :param schedules: The schedules of this V1Pipeline.  # noqa: E501
-        :type: list[V1PipelineSchedule]
+        :param schedule_id: The schedule_id of this V1Pipeline.  # noqa: E501
+        :type: str
         """
 
-        self._schedules = schedules
+        self._schedule_id = schedule_id
 
     @property
     def shared_filesystem(self) -> 'V1SharedFilesystem':