lightning-sdk 0.1.58__py3-none-any.whl → 0.2.1__py3-none-any.whl

lightning_sdk/deployment/deployment.py

@@ -63,6 +63,7 @@ class Deployment:
         self._request_session = None
 
         self._auth = login.Auth()
+        self._user = None
 
         try:
             self._auth.authenticate()
@@ -73,9 +74,12 @@ class Deployment:
 
         self._name = name
         self._org = _resolve_org(org)
-        self._user = _resolve_user(user)
+        self._user = _resolve_user(self._user or user)
+
         self._teamspace = _resolve_teamspace(
-            teamspace=teamspace, org=self._org, user=self._user if self._org is None else None
+            teamspace=teamspace,
+            org=self._org,
+            user=self._user,
         )
         if self._teamspace is None:
             raise ValueError("You need to pass a teamspace or an org for your deployment.")
@@ -97,13 +101,13 @@ class Deployment:
     def start(
         self,
         machine: Optional[Machine] = None,
-        environment: Optional[str] = None,
+        image: Optional[str] = None,
         autoscale: Optional[AutoScaleConfig] = None,
         ports: Optional[List[float]] = None,
         release_strategy: Optional[ReleaseStrategy] = None,
         entrypoint: Optional[str] = None,
         command: Optional[str] = None,
-        env: Optional[List[Union[Env, Secret]]] = None,
+        env: Union[List[Union[Secret, Env]], Dict[str, str], None] = None,
         spot: Optional[bool] = None,
         replicas: Optional[int] = None,
         health_check: Optional[Union[HttpHealthCheck, ExecHealthCheck]] = None,
@@ -111,6 +115,7 @@ class Deployment:
         cloud_account: Optional[str] = None,
         custom_domain: Optional[str] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        quantity: Optional[int] = None,
     ) -> None:
         """The Lightning AI Deployment.
 
@@ -122,7 +127,7 @@ class Deployment:
             machine: The machine used by the deployment replicas.
             autoscale: The list of the metrics to autoscale on.
             ports: The ports to reach your replica services.
-            environment: The environement used by the deployment. Currentely, only docker images.
+            image: The environement used by the deployment. Currentely, only docker images.
             release_strategy: The release strategy to use when changing core deployment specs.
             entrypoint: The docker container entrypoint.
             command: The docker container command.
@@ -134,6 +139,7 @@ class Deployment:
             cloud_account: The name of the cloud account, the studio should be created on.
                 Doesn't matter when the studio already exists.
             custom_domain: Whether your service would be referenced under a custom doamin.
+            quantity: The number of machines per replica to deploy.
 
         Note:
             Since a teamspace can either be owned by an org or by a user directly,
@@ -161,10 +167,11 @@ class Deployment:
                     command=command,
                     entrypoint=entrypoint,
                     env=env,
-                    environment=environment,
+                    image=image,
                     spot=spot,
                     machine=machine,
                     health_check=health_check,
+                    quantity=quantity,
                 ),
                 strategy=to_strategy(release_strategy),
             )
@@ -178,7 +185,7 @@ class Deployment:
         self,
         # Changing those arguments create a new release
         machine: Optional[Machine] = None,
-        environment: Optional[str] = None,
+        image: Optional[str] = None,
         entrypoint: Optional[str] = None,
         command: Optional[str] = None,
         env: Optional[List[Union[Env, Secret]]] = None,
@@ -195,6 +202,7 @@ class Deployment:
         auth: Optional[Union[BasicAuth, TokenAuth]] = None,
         custom_domain: Optional[str] = None,
         cluster: Optional[str] = None,  # deprecated in favor of cloud_account
+        quantity: Optional[int] = None,
     ) -> None:
         cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
 
@@ -207,7 +215,7 @@ class Deployment:
             max_replicas=max_replicas,
             cloud_account=cloud_account,
             machine=machine,
-            environment=environment,
+            image=image,
             entrypoint=entrypoint,
             command=command,
             ports=ports,
@@ -216,6 +224,7 @@ class Deployment:
             env=env,
             health_check=health_check,
             release_strategy=release_strategy,
+            quantity=quantity,
         )
 
     def stop(self) -> None:
@@ -334,6 +343,22 @@ class Deployment:
             return self._deployment.spec.cluster_id
         return None
 
+    @property
+    def release_id(self) -> Optional[str]:
+        """The release id of the deployment."""
+        if self._deployment:
+            self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
+            return self._deployment.release_id
+        return None
+
+    @property
+    def quantity(self) -> Optional[str]:
+        """The number of machines per replica."""
+        if self._deployment:
+            self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
+            return self._deployment.spec.quantity
+        return None
+
     @property
     def user(self) -> Optional[User]:
         """The teamspace of the deployment."""

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -44,6 +44,7 @@ from lightning_sdk.lightning_cloud.openapi.api.lit_page_service_api import LitPa
 from lightning_sdk.lightning_cloud.openapi.api.lit_registry_service_api import LitRegistryServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.models_store_api import ModelsStoreApi
 from lightning_sdk.lightning_cloud.openapi.api.organizations_service_api import OrganizationsServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.pipelines_service_api import PipelinesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.profiler_service_api import ProfilerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.projects_service_api import ProjectsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.quest_service_api import QuestServiceApi
@@ -131,6 +132,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploa
 from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body1 import IdUploadsBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body import IdVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body1 import IdVisibilityBody1
+from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body2 import IdVisibilityBody2
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
@@ -151,6 +153,7 @@ from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_bod
 from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import OrgIdMembershipsBody
 from lightning_sdk.lightning_cloud.openapi.models.org_id_roles_body import OrgIdRolesBody
 from lightning_sdk.lightning_cloud.openapi.models.orgs_id_body import OrgsIdBody
+from lightning_sdk.lightning_cloud.openapi.models.pipelines_id_body import PipelinesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.profiler_captures_body import ProfilerCapturesBody
 from lightning_sdk.lightning_cloud.openapi.models.profiler_enabled_body import ProfilerEnabledBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_agentmanagedendpoints_body import ProjectIdAgentmanagedendpointsBody
@@ -167,6 +170,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_litregistry_body im
 from lightning_sdk.lightning_cloud.openapi.models.project_id_memberships_body import ProjectIdMembershipsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_models_body import ProjectIdModelsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_multimachinejobs_body import ProjectIdMultimachinejobsBody
+from lightning_sdk.lightning_cloud.openapi.models.project_id_pipelines_body import ProjectIdPipelinesBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_projectclustersbindings_body import ProjectIdProjectclustersbindingsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_serviceexecution_body import ProjectIdServiceexecutionBody
@@ -196,6 +200,7 @@ from lightning_sdk.lightning_cloud.openapi.models.update import Update
 from lightning_sdk.lightning_cloud.openapi.models.upload_id_complete_body import UploadIdCompleteBody
 from lightning_sdk.lightning_cloud.openapi.models.upload_id_parts_body import UploadIdPartsBody
 from lightning_sdk.lightning_cloud.openapi.models.uploads_upload_id_body import UploadsUploadIdBody
+from lightning_sdk.lightning_cloud.openapi.models.uploads_upload_id_body1 import UploadsUploadIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.usagerestrictions_id_body import UsagerestrictionsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.user_id_affiliatelinks_body import UserIdAffiliatelinksBody
 from lightning_sdk.lightning_cloud.openapi.models.user_id_membershiprolebindings_body import UserIdMembershiprolebindingsBody
@@ -289,6 +294,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_complete_onboarding_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload import V1CompleteUpload
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_project_artifact_response import V1CompleteUploadProjectArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_service_execution_artifact_response import V1CompleteUploadServiceExecutionArtifactResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_temporary_artifact_request import V1CompleteUploadTemporaryArtifactRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_completed_part import V1CompletedPart
 from lightning_sdk.lightning_cloud.openapi.models.v1_compute_config import V1ComputeConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_reason import V1ContactAssistantOwnerReason
@@ -308,9 +314,13 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_k
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_keys_response import V1CreateClusterEncryptionKeysResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_request import V1CreateClusterRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_response import V1CreateClusterResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_request import V1CreateDeploymentRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_template_request import V1CreateDeploymentTemplateRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_job_request import V1CreateJobRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_request import V1CreateLitPageRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_response import V1CreateLitPageResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_managed_endpoint_response import V1CreateManagedEndpointResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_machine_job_request import V1CreateMultiMachineJobRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_part_upload_response import V1CreateMultiPartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request import V1CreateOrganizationRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_project_request import V1CreateProjectRequest
@@ -366,6 +376,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_membership_role_binding_response import V1DeleteOrgMembershipRoleBindingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_org_role_response import V1DeleteOrgRoleResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_organization_response import V1DeleteOrganizationResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_pipeline_response import V1DeletePipelineResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_profiler_capture_response import V1DeleteProfilerCaptureResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_artifact_response import V1DeleteProjectArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_cluster_binding_response import V1DeleteProjectClusterBindingResponse
@@ -384,6 +395,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_cache_state impo
 from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_file_info import V1DependencyFileInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deployment
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_api import V1DeploymentAPI
+from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_details import V1DeploymentDetails
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event import V1DeploymentEvent
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event_type import V1DeploymentEventType
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_metrics import V1DeploymentMetrics
@@ -426,6 +438,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_experiment import V1Experim
 from lightning_sdk.lightning_cloud.openapi.models.v1_export_snowflake_query_response import V1ExportSnowflakeQueryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_external_search_user import V1ExternalSearchUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_file_endpoint import V1FileEndpoint
+from lightning_sdk.lightning_cloud.openapi.models.v1_filestore_data_connection import V1FilestoreDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_app import V1FilesystemApp
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_cloud_space import V1FilesystemCloudSpace
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_dataset import V1FilesystemDataset
@@ -496,6 +509,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_invalidate_cloud_space_inst
 from lightning_sdk.lightning_cloud.openapi.models.v1_invite_project_membership_response import V1InviteProjectMembershipResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_job import V1Job
 from lightning_sdk.lightning_cloud.openapi.models.v1_job_action import V1JobAction
+from lightning_sdk.lightning_cloud.openapi.models.v1_job_artifacts_type import V1JobArtifactsType
 from lightning_sdk.lightning_cloud.openapi.models.v1_job_file import V1JobFile
 from lightning_sdk.lightning_cloud.openapi.models.v1_job_health_check_config import V1JobHealthCheckConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_job_log_entry import V1JobLogEntry
@@ -592,6 +606,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_memberships_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_roles_response import V1ListOrgRolesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_organization_cluster_encryption_keys_response import V1ListOrganizationClusterEncryptionKeysResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_organizations_response import V1ListOrganizationsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_pipelines_response import V1ListPipelinesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_profiler_captures_response import V1ListProfilerCapturesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_artifacts_response import V1ListProjectArtifactsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_cluster_accelerators_response import V1ListProjectClusterAcceleratorsResponse
@@ -670,6 +685,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_parameterization_spec impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_path_mapping import V1PathMapping
 from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline import V1Pipeline
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_schedule import V1PipelineSchedule
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step import V1PipelineStep
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_status import V1PipelineStepStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_type import V1PipelineStepType
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugins_list_response import V1PluginsListResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_post_cloud_space_artifact_events_response import V1PostCloudSpaceArtifactEventsResponse
@@ -741,6 +761,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_service_execution import V1
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_health import V1ServiceHealth
 from lightning_sdk.lightning_cloud.openapi.models.v1_setup_confirmed_ssh_public_key_response import V1SetupConfirmedSSHPublicKeyResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_response import V1SetupDataConnectionResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_shared_filesystem import V1SharedFilesystem
 from lightning_sdk.lightning_cloud.openapi.models.v1_should_start_syncing_response import V1ShouldStartSyncingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_signed_url import V1SignedUrl
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_node import V1SlurmNode
@@ -779,6 +800,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_deployment_visibility_response import V1UpdateDeploymentVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_job_visibility_response import V1UpdateJobVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_repository_response import V1UpdateLitRepositoryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visibility_response import V1UpdateMetricsStreamVisibilityResponse
@@ -797,6 +819,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_upload_project_artifact_res
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_service_execution_artifact_parts_response import V1UploadServiceExecutionArtifactPartsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_service_execution_artifact_response import V1UploadServiceExecutionArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_settings_response import V1UploadSettingsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_upload_temporary_artifact_request import V1UploadTemporaryArtifactRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_upstream_cloud_space import V1UpstreamCloudSpace
 from lightning_sdk.lightning_cloud.openapi.models.v1_upstream_job import V1UpstreamJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_upstream_managed import V1UpstreamManaged

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -25,6 +25,7 @@ from lightning_sdk.lightning_cloud.openapi.api.lit_page_service_api import LitPa
 from lightning_sdk.lightning_cloud.openapi.api.lit_registry_service_api import LitRegistryServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.models_store_api import ModelsStoreApi
 from lightning_sdk.lightning_cloud.openapi.api.organizations_service_api import OrganizationsServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.pipelines_service_api import PipelinesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.profiler_service_api import ProfilerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.projects_service_api import ProjectsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.quest_service_api import QuestServiceApi

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

@@ -253,7 +253,7 @@ class AssistantsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def assistants_service_create_assistant_managed_endpoint(self, body: 'ProjectIdAgentmanagedendpointsBody', project_id: 'str', **kwargs) -> 'V1ManagedEndpoint':  # noqa: E501
+    def assistants_service_create_assistant_managed_endpoint(self, body: 'ProjectIdAgentmanagedendpointsBody', project_id: 'str', **kwargs) -> 'V1CreateManagedEndpointResponse':  # noqa: E501
         """assistants_service_create_assistant_managed_endpoint  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -264,7 +264,7 @@ class AssistantsServiceApi(object):
         :param async_req bool
         :param ProjectIdAgentmanagedendpointsBody body: (required)
         :param str project_id: (required)
-        :return: V1ManagedEndpoint
+        :return: V1CreateManagedEndpointResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -275,7 +275,7 @@ class AssistantsServiceApi(object):
             (data) = self.assistants_service_create_assistant_managed_endpoint_with_http_info(body, project_id, **kwargs)  # noqa: E501
             return data
 
-    def assistants_service_create_assistant_managed_endpoint_with_http_info(self, body: 'ProjectIdAgentmanagedendpointsBody', project_id: 'str', **kwargs) -> 'V1ManagedEndpoint':  # noqa: E501
+    def assistants_service_create_assistant_managed_endpoint_with_http_info(self, body: 'ProjectIdAgentmanagedendpointsBody', project_id: 'str', **kwargs) -> 'V1CreateManagedEndpointResponse':  # noqa: E501
         """assistants_service_create_assistant_managed_endpoint  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
@@ -286,7 +286,7 @@ class AssistantsServiceApi(object):
         :param async_req bool
         :param ProjectIdAgentmanagedendpointsBody body: (required)
         :param str project_id: (required)
-        :return: V1ManagedEndpoint
+        :return: V1CreateManagedEndpointResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
@@ -350,7 +350,7 @@ class AssistantsServiceApi(object):
             body=body_params,
             post_params=form_params,
             files=local_var_files,
-            response_type='V1ManagedEndpoint',  # noqa: E501
+            response_type='V1CreateManagedEndpointResponse',  # noqa: E501
             auth_settings=auth_settings,
             async_req=params.get('async_req'),
             _return_http_data_only=params.get('_return_http_data_only'),
@@ -1168,6 +1168,7 @@ class AssistantsServiceApi(object):
         :param str cloudspace_id:
         :param bool published:
         :param str internal_name:
+        :param str user_id:
         :return: V1ListAssistantsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1193,12 +1194,13 @@ class AssistantsServiceApi(object):
         :param str cloudspace_id:
         :param bool published:
         :param str internal_name:
+        :param str user_id:
         :return: V1ListAssistantsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['org_id', 'project_id', 'cloudspace_id', 'published', 'internal_name']  # noqa: E501
+        all_params = ['org_id', 'project_id', 'cloudspace_id', 'published', 'internal_name', 'user_id']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1229,6 +1231,8 @@ class AssistantsServiceApi(object):
             query_params.append(('published', params['published']))  # noqa: E501
         if 'internal_name' in params:
             query_params.append(('internalName', params['internal_name']))  # noqa: E501
+        if 'user_id' in params:
+            query_params.append(('userId', params['user_id']))  # noqa: E501
 
         header_params = {}