lightning-sdk 0.1.2__py3-none-any.whl → 0.1.46__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_managed_endpoint.py

@@ -41,34 +41,106 @@ class V1ManagedEndpoint(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'api_key': 'str',
+        'base_url': 'str',
         'description': 'str',
         'id': 'str',
-        'models': 'list[str]',
-        'name': 'str'
+        'models_metadata': 'list[V1ManagedModel]',
+        'name': 'str',
+        'org_id': 'str',
+        'project_id': 'str',
+        'publish_status': 'str',
+        'user_id': 'str'
     }
 
     attribute_map = {
+        'api_key': 'apiKey',
+        'base_url': 'baseUrl',
         'description': 'description',
         'id': 'id',
-        'models': 'models',
-        'name': 'name'
+        'models_metadata': 'modelsMetadata',
+        'name': 'name',
+        'org_id': 'orgId',
+        'project_id': 'projectId',
+        'publish_status': 'publishStatus',
+        'user_id': 'userId'
     }
 
-    def __init__(self, description: 'str' =None, id: 'str' =None, models: 'list[str]' =None, name: 'str' =None):  # noqa: E501
+    def __init__(self, api_key: 'str' =None, base_url: 'str' =None, description: 'str' =None, id: 'str' =None, models_metadata: 'list[V1ManagedModel]' =None, name: 'str' =None, org_id: 'str' =None, project_id: 'str' =None, publish_status: 'str' =None, user_id: 'str' =None):  # noqa: E501
         """V1ManagedEndpoint - a model defined in Swagger"""  # noqa: E501
+        self._api_key = None
+        self._base_url = None
         self._description = None
         self._id = None
-        self._models = None
+        self._models_metadata = None
         self._name = None
+        self._org_id = None
+        self._project_id = None
+        self._publish_status = None
+        self._user_id = None
         self.discriminator = None
+        if api_key is not None:
+            self.api_key = api_key
+        if base_url is not None:
+            self.base_url = base_url
         if description is not None:
             self.description = description
         if id is not None:
             self.id = id
-        if models is not None:
-            self.models = models
+        if models_metadata is not None:
+            self.models_metadata = models_metadata
         if name is not None:
             self.name = name
+        if org_id is not None:
+            self.org_id = org_id
+        if project_id is not None:
+            self.project_id = project_id
+        if publish_status is not None:
+            self.publish_status = publish_status
+        if user_id is not None:
+            self.user_id = user_id
+
+    @property
+    def api_key(self) -> 'str':
+        """Gets the api_key of this V1ManagedEndpoint.  # noqa: E501
+
+
+        :return: The api_key of this V1ManagedEndpoint.  # noqa: E501
+        :rtype: str
+        """
+        return self._api_key
+
+    @api_key.setter
+    def api_key(self, api_key: 'str'):
+        """Sets the api_key of this V1ManagedEndpoint.
+
+
+        :param api_key: The api_key of this V1ManagedEndpoint.  # noqa: E501
+        :type: str
+        """
+
+        self._api_key = api_key
+
+    @property
+    def base_url(self) -> 'str':
+        """Gets the base_url of this V1ManagedEndpoint.  # noqa: E501
+
+
+        :return: The base_url of this V1ManagedEndpoint.  # noqa: E501
+        :rtype: str
+        """
+        return self._base_url
+
+    @base_url.setter
+    def base_url(self, base_url: 'str'):
+        """Sets the base_url of this V1ManagedEndpoint.
+
+
+        :param base_url: The base_url of this V1ManagedEndpoint.  # noqa: E501
+        :type: str
+        """
+
+        self._base_url = base_url
 
     @property
     def description(self) -> 'str':
@@ -113,25 +185,25 @@ class V1ManagedEndpoint(object):
         self._id = id
 
     @property
-    def models(self) -> 'list[str]':
-        """Gets the models of this V1ManagedEndpoint.  # noqa: E501
+    def models_metadata(self) -> 'list[V1ManagedModel]':
+        """Gets the models_metadata of this V1ManagedEndpoint.  # noqa: E501
 
 
-        :return: The models of this V1ManagedEndpoint.  # noqa: E501
-        :rtype: list[str]
+        :return: The models_metadata of this V1ManagedEndpoint.  # noqa: E501
+        :rtype: list[V1ManagedModel]
         """
-        return self._models
+        return self._models_metadata
 
-    @models.setter
-    def models(self, models: 'list[str]'):
-        """Sets the models of this V1ManagedEndpoint.
+    @models_metadata.setter
+    def models_metadata(self, models_metadata: 'list[V1ManagedModel]'):
+        """Sets the models_metadata of this V1ManagedEndpoint.
 
 
-        :param models: The models of this V1ManagedEndpoint.  # noqa: E501
-        :type: list[str]
+        :param models_metadata: The models_metadata of this V1ManagedEndpoint.  # noqa: E501
+        :type: list[V1ManagedModel]
         """
 
-        self._models = models
+        self._models_metadata = models_metadata
 
     @property
     def name(self) -> 'str':
@@ -154,6 +226,90 @@ class V1ManagedEndpoint(object):
 
         self._name = name
 
+    @property
+    def org_id(self) -> 'str':
+        """Gets the org_id of this V1ManagedEndpoint.  # noqa: E501
+
+
+        :return: The org_id of this V1ManagedEndpoint.  # noqa: E501
+        :rtype: str
+        """
+        return self._org_id
+
+    @org_id.setter
+    def org_id(self, org_id: 'str'):
+        """Sets the org_id of this V1ManagedEndpoint.
+
+
+        :param org_id: The org_id of this V1ManagedEndpoint.  # noqa: E501
+        :type: str
+        """
+
+        self._org_id = org_id
+
+    @property
+    def project_id(self) -> 'str':
+        """Gets the project_id of this V1ManagedEndpoint.  # noqa: E501
+
+
+        :return: The project_id of this V1ManagedEndpoint.  # noqa: E501
+        :rtype: str
+        """
+        return self._project_id
+
+    @project_id.setter
+    def project_id(self, project_id: 'str'):
+        """Sets the project_id of this V1ManagedEndpoint.
+
+
+        :param project_id: The project_id of this V1ManagedEndpoint.  # noqa: E501
+        :type: str
+        """
+
+        self._project_id = project_id
+
+    @property
+    def publish_status(self) -> 'str':
+        """Gets the publish_status of this V1ManagedEndpoint.  # noqa: E501
+
+
+        :return: The publish_status of this V1ManagedEndpoint.  # noqa: E501
+        :rtype: str
+        """
+        return self._publish_status
+
+    @publish_status.setter
+    def publish_status(self, publish_status: 'str'):
+        """Sets the publish_status of this V1ManagedEndpoint.
+
+
+        :param publish_status: The publish_status of this V1ManagedEndpoint.  # noqa: E501
+        :type: str
+        """
+
+        self._publish_status = publish_status
+
+    @property
+    def user_id(self) -> 'str':
+        """Gets the user_id of this V1ManagedEndpoint.  # noqa: E501
+
+
+        :return: The user_id of this V1ManagedEndpoint.  # noqa: E501
+        :rtype: str
+        """
+        return self._user_id
+
+    @user_id.setter
+    def user_id(self, user_id: 'str'):
+        """Sets the user_id of this V1ManagedEndpoint.
+
+
+        :param user_id: The user_id of this V1ManagedEndpoint.  # noqa: E501
+        :type: str
+        """
+
+        self._user_id = user_id
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py

@@ -0,0 +1,305 @@
+# 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 V1ManagedModel(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 = {
+        'abilities': 'V1ManagedModelAbilities',
+        'completion_token_price': 'float',
+        'description': 'str',
+        'endpoint_id': 'str',
+        'id': 'str',
+        'name': 'str',
+        'prompt_token_price': 'float',
+        'status': 'V1AssistantModelStatus'
+    }
+
+    attribute_map = {
+        'abilities': 'abilities',
+        'completion_token_price': 'completionTokenPrice',
+        'description': 'description',
+        'endpoint_id': 'endpointId',
+        'id': 'id',
+        'name': 'name',
+        'prompt_token_price': 'promptTokenPrice',
+        'status': 'status'
+    }
+
+    def __init__(self, abilities: 'V1ManagedModelAbilities' =None, completion_token_price: 'float' =None, description: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None, status: 'V1AssistantModelStatus' =None):  # noqa: E501
+        """V1ManagedModel - a model defined in Swagger"""  # noqa: E501
+        self._abilities = None
+        self._completion_token_price = None
+        self._description = None
+        self._endpoint_id = None
+        self._id = None
+        self._name = None
+        self._prompt_token_price = None
+        self._status = 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 description is not None:
+            self.description = description
+        if endpoint_id is not None:
+            self.endpoint_id = endpoint_id
+        if id is not None:
+            self.id = id
+        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
+
+    @property
+    def abilities(self) -> 'V1ManagedModelAbilities':
+        """Gets the abilities of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The abilities of this V1ManagedModel.  # noqa: E501
+        :rtype: V1ManagedModelAbilities
+        """
+        return self._abilities
+
+    @abilities.setter
+    def abilities(self, abilities: 'V1ManagedModelAbilities'):
+        """Sets the abilities of this V1ManagedModel.
+
+
+        :param abilities: The abilities of this V1ManagedModel.  # noqa: E501
+        :type: V1ManagedModelAbilities
+        """
+
+        self._abilities = abilities
+
+    @property
+    def completion_token_price(self) -> 'float':
+        """Gets the completion_token_price of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The completion_token_price of this V1ManagedModel.  # noqa: E501
+        :rtype: float
+        """
+        return self._completion_token_price
+
+    @completion_token_price.setter
+    def completion_token_price(self, completion_token_price: 'float'):
+        """Sets the completion_token_price of this V1ManagedModel.
+
+
+        :param completion_token_price: The completion_token_price of this V1ManagedModel.  # noqa: E501
+        :type: float
+        """
+
+        self._completion_token_price = completion_token_price
+
+    @property
+    def description(self) -> 'str':
+        """Gets the description of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The description of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._description
+
+    @description.setter
+    def description(self, description: 'str'):
+        """Sets the description of this V1ManagedModel.
+
+
+        :param description: The description of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._description = description
+
+    @property
+    def endpoint_id(self) -> 'str':
+        """Gets the endpoint_id of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The endpoint_id of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._endpoint_id
+
+    @endpoint_id.setter
+    def endpoint_id(self, endpoint_id: 'str'):
+        """Sets the endpoint_id of this V1ManagedModel.
+
+
+        :param endpoint_id: The endpoint_id of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._endpoint_id = endpoint_id
+
+    @property
+    def id(self) -> 'str':
+        """Gets the id of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The id of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id: 'str'):
+        """Sets the id of this V1ManagedModel.
+
+
+        :param id: The id of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._id = id
+
+    @property
+    def name(self) -> 'str':
+        """Gets the name of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The name of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name: 'str'):
+        """Sets the name of this V1ManagedModel.
+
+
+        :param name: The name of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._name = name
+
+    @property
+    def prompt_token_price(self) -> 'float':
+        """Gets the prompt_token_price of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The prompt_token_price of this V1ManagedModel.  # noqa: E501
+        :rtype: float
+        """
+        return self._prompt_token_price
+
+    @prompt_token_price.setter
+    def prompt_token_price(self, prompt_token_price: 'float'):
+        """Sets the prompt_token_price of this V1ManagedModel.
+
+
+        :param prompt_token_price: The prompt_token_price of this V1ManagedModel.  # noqa: E501
+        :type: float
+        """
+
+        self._prompt_token_price = prompt_token_price
+
+    @property
+    def status(self) -> 'V1AssistantModelStatus':
+        """Gets the status of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The status of this V1ManagedModel.  # noqa: E501
+        :rtype: V1AssistantModelStatus
+        """
+        return self._status
+
+    @status.setter
+    def status(self, status: 'V1AssistantModelStatus'):
+        """Sets the status of this V1ManagedModel.
+
+
+        :param status: The status of this V1ManagedModel.  # noqa: E501
+        :type: V1AssistantModelStatus
+        """
+
+        self._status = status
+
+    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(V1ManagedModel, 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: 'V1ManagedModel') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ManagedModel):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ManagedModel') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other