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

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.23"
+__version__ = "0.2.24.rc0"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/base_studio_api.py

@@ -1,10 +1,13 @@
 from typing import Any, List, Optional
 
+from lightning_sdk.lightning_cloud.openapi import (
+    V1CloudSpaceEnvironmentType,
+    V1ListCloudSpaceEnvironmentTemplatesResponse,
+)
 from lightning_sdk.lightning_cloud.openapi.models.update import Update as BaseStudioUpdateBody
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import (
     V1CloudSpaceEnvironmentTemplate,
 )
-from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_type import V1CloudSpaceEnvironmentType
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
 
@@ -21,8 +24,12 @@ class BaseStudioApi:
         except ValueError as e:
             raise ValueError(f"Base studio {base_studio_id} does not exist") from e
 
-    def get_all_base_studios(self, org_id: str) -> List[V1CloudSpaceEnvironmentTemplate]:
+    def get_all_base_studios(self, org_id: str, managed: bool = True) -> V1ListCloudSpaceEnvironmentTemplatesResponse:
         """Retrieve all base studios for a given organization."""
+        if managed:
+            return self._client.cloud_space_environment_template_service_list_managed_cloud_space_environment_templates(
+                org_id=org_id
+            )
         return self._client.cloud_space_environment_template_service_list_cloud_space_environment_templates(
             org_id=org_id
         )

lightning_sdk/api/deployment_api.py

@@ -102,10 +102,10 @@ class ExecHealthCheck(HealthCheck):
     def __init__(
         self,
         command: str,
-        timeout_seconds: int = 60,
-        initial_delay_seconds: int = 5,
-        failure_threshold: int = 12,
-        interval_seconds: int = 5,
+        initial_delay_seconds: int = 0,
+        failure_threshold: int = 600,
+        interval_seconds: int = 1,
+        timeout_seconds: int = 30,
     ) -> None:
         self.command = command
         self.timeout_seconds = timeout_seconds
@@ -131,17 +131,17 @@ class HttpHealthCheck(HealthCheck):
         self,
         path: str,
         port: float,
-        timeout_seconds: int = 60,
-        initial_delay_seconds: int = 5,
-        failure_threshold: int = 12,
-        interval_seconds: int = 5,
+        initial_delay_seconds: int = 0,
+        failure_threshold: int = 600,
+        interval_seconds: int = 1,
+        timeout_seconds: int = 30,
     ) -> None:
         self.path = path
         self.port = port
-        self.timeout_seconds = timeout_seconds
         self.initial_delay_seconds = initial_delay_seconds
         self.failure_threshold = failure_threshold
         self.interval_seconds = interval_seconds
+        self.timeout_seconds = timeout_seconds
 
 
 class AutoScalingMetric:

lightning_sdk/api/llm_api.py

@@ -18,17 +18,11 @@ class LLMApi:
     def __init__(self) -> None:
         self._client = LightningClient(retry=False, max_tries=0)
 
-    def get_public_models(self) -> List[str]:
-        result = self._client.assistants_service_list_assistants(published=True)
-        return result.assistants
-
-    def get_org_models(self, org_id: str) -> List[str]:
-        result = self._client.assistants_service_list_assistants(org_id=org_id)
-        return result.assistants
-
-    def get_user_models(self, user_id: str) -> List[str]:
-        result = self._client.assistants_service_list_assistants(user_id=user_id)
-        return result.assistants
+    def get_assistant(self, model_provider: str, model_name: str, user_name: str, org_name: str) -> str:
+        result = self._client.assistants_service_get_managed_model_assistant(
+            model_provider=model_provider, model_name=model_name, user_name=user_name, org_name=org_name
+        )
+        return result.id
 
     def _parse_stream_line(self, decoded_line: str) -> Optional[V1ConversationResponseChunk]:
         try:

lightning_sdk/api/pipeline_api.py

@@ -1,5 +1,6 @@
-from typing import TYPE_CHECKING, List, Optional
+from typing import TYPE_CHECKING, List, Optional, Union
 
+from lightning_sdk.api.cluster_api import ClusterApi
 from lightning_sdk.lightning_cloud.openapi.models import (
     ProjectIdPipelinesBody,
     ProjectIdSchedulesBody,
@@ -11,6 +12,7 @@ from lightning_sdk.lightning_cloud.openapi.models import (
 )
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
+from lightning_sdk.teamspace import Teamspace
 
 if TYPE_CHECKING:
     from lightning_sdk.pipeline.schedule import Schedule
@@ -20,7 +22,8 @@ class PipelineApi:
     """Internal API client for Pipeline requests (mainly http requests)."""
 
     def __init__(self) -> None:
-        self._client = LightningClient(retry=False, max_tries=0)
+        self._client = LightningClient(max_tries=0, retry=False)
+        self._cluster_api = ClusterApi()
 
     def get_pipeline_by_id(self, project_id: str, pipeline_id_or_name: str) -> Optional[V1Pipeline]:
         if pipeline_id_or_name.startswith("pip_"):
@@ -43,7 +46,7 @@ class PipelineApi:
     def create_pipeline(
         self,
         name: str,
-        project_id: str,
+        teamspace: Teamspace,
         steps: List["V1PipelineStep"],
         shared_filesystem: bool,
         schedules: List["Schedule"],
@@ -52,20 +55,17 @@ class PipelineApi:
         body = ProjectIdPipelinesBody(
             name=name,
             steps=steps,
-            shared_filesystem=V1SharedFilesystem(
-                enabled=shared_filesystem,
-                s3_folder=True,
-            ),
+            shared_filesystem=self._prepare_shared_filesytem(shared_filesystem, steps, teamspace),
             parent_pipeline_id=parent_pipeline_id or "",
         )
 
-        pipeline = self._client.pipelines_service_create_pipeline(body, project_id)
+        pipeline = self._client.pipelines_service_create_pipeline(body, teamspace.id)
 
         # Delete the previous schedules
         if parent_pipeline_id is not None:
-            current_schedules = self._client.schedules_service_list_schedules(project_id).schedules
+            current_schedules = self._client.schedules_service_list_schedules(teamspace.id).schedules
             for schedule in current_schedules:
-                self._client.schedules_service_delete_schedule(project_id, schedule.id)
+                self._client.schedules_service_delete_schedule(teamspace.id, schedule.id)
 
         if len(schedules):
             for schedule in schedules:
@@ -77,7 +77,7 @@ class PipelineApi:
                     resource_type=V1ScheduleResourceType.PIPELINE,
                 )
 
-                self._client.schedules_service_create_schedule(body, project_id)
+                self._client.schedules_service_create_schedule(body, teamspace.id)
 
         return pipeline
 
@@ -88,3 +88,23 @@ class PipelineApi:
 
     def delete(self, project_id: str, pipeline_id: str) -> V1DeletePipelineResponse:
         return self._client.pipelines_service_delete_pipeline(project_id, pipeline_id)
+
+    def _prepare_shared_filesytem(
+        self, shared_filesystem: Union[bool, V1SharedFilesystem], steps: List["V1PipelineStep"], teamspace: Teamspace
+    ) -> V1SharedFilesystem:
+        if not shared_filesystem:
+            return V1SharedFilesystem(enabled=False)
+
+        from lightning_sdk.pipeline.utils import _get_cloud_account
+
+        cluster = self._cluster_api.get_cluster(
+            cluster_id=_get_cloud_account(steps), project_id=teamspace.id, org_id=teamspace.owner.id
+        )
+
+        if cluster.spec.aws_v1:
+            return V1SharedFilesystem(enabled=True, s3_folder=True)
+
+        if cluster.spec.google_cloud_v1:
+            return V1SharedFilesystem(enabled=True, gcs_folder=True)
+
+        raise NotImplementedError("This cluster isn't support yet")

lightning_sdk/api/studio_api.py

@@ -124,6 +124,8 @@ class StudioApi:
         cloud_account: Optional[str] = None,
         source: Optional[V1CloudSpaceSourceType] = None,
         disable_secrets: bool = False,
+        sandbox: bool = False,
+        cloud_space_environment_template_id: Optional[str] = None,
     ) -> V1CloudSpace:
         """Create a Studio with a given name in a given Teamspace on a possibly given cloud_account."""
         body = ProjectIdCloudspacesBody(
@@ -133,6 +135,8 @@ class StudioApi:
             seed_files=[V1CloudSpaceSeedFile(path="main.py", contents="print('Hello, Lightning World!')\n")],
             source=source,
             disable_secrets=disable_secrets,
+            sandbox=sandbox,
+            cloud_space_environment_template_id=cloud_space_environment_template_id,
         )
         studio = self._client.cloud_space_service_create_cloud_space(body, teamspace_id)
 

lightning_sdk/base_studio.py

@@ -1,17 +1,23 @@
+from dataclasses import dataclass
 from typing import List, Optional, Union
 
 from lightning_sdk.api.base_studio_api import BaseStudioApi
 from lightning_sdk.api.user_api import UserApi
 from lightning_sdk.lightning_cloud import login
-from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import (
-    V1CloudSpaceEnvironmentTemplate,
-)
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_type import V1CloudSpaceEnvironmentType
 from lightning_sdk.organization import Organization
 from lightning_sdk.user import User
 from lightning_sdk.utils.resolve import _resolve_org, _resolve_user
 
 
+@dataclass
+class BaseStudioInfo:
+    id: str
+    name: str
+    managed_id: str
+    description: str
+
+
 class BaseStudio:
     def __init__(
         self,
@@ -61,7 +67,7 @@ class BaseStudio:
         environment_type: Optional[V1CloudSpaceEnvironmentType] = None,
         machine_image_version: Optional[str] = None,
         setup_script_text: Optional[str] = None,
-    ) -> V1CloudSpaceEnvironmentTemplate:
+    ) -> None:
         self._base_studio = self._base_studio_api.update_base_studio(
             self._base_studio.id,
             self._org.id,
@@ -74,10 +80,20 @@ class BaseStudio:
             disabled=disabled,
         )
 
-    def list(self) -> List[V1CloudSpaceEnvironmentTemplate]:
+    def list(self, managed: bool = True) -> List[BaseStudioInfo]:
         """List all base studios in the organization.
 
         Returns:
             List[V1CloudSpaceEnvironmentTemplate]: A list of base studio templates.
         """
-        return self._base_studio_api.get_all_base_studios(self._org.id)
+        result = []
+        for template in self._base_studio_api.get_all_base_studios(self._org.id, managed).templates:
+            result.append(
+                BaseStudioInfo(
+                    id=template.id,
+                    name=template.name,
+                    managed_id=template.managed_id,
+                    description=template.description,
+                ),
+            )
+        return result

lightning_sdk/deployment/deployment.py

@@ -197,11 +197,13 @@ class Deployment:
         if commands is not None:
             command = compose_commands(commands)
 
+        autoscaling_metric_name = ("CPU" if machine.is_cpu() else "GPU") if isinstance(machine, Machine) else "CPU"
+
         if autoscale is None:
             autoscale = AutoScaleConfig(
                 min_replicas=0,
                 max_replicas=1,
-                metric="CPU" if machine.is_cpu() else "GPU",
+                metric=autoscaling_metric_name,
                 threshold=90,
             )
 
@@ -380,27 +382,35 @@ class Deployment:
         return None
 
     @property
-    def pending_replicas(self) -> Optional[List[str]]:
+    def pending_replicas(self) -> Optional[int]:
         """The number of pending replicas."""
         if self._deployment:
             self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
-            return self._deployment.status.pending_replicas
+            return int(self._deployment.status.pending_replicas)
+        return None
+
+    @property
+    def running_replicas(self) -> Optional[int]:
+        """The number of failing replicas."""
+        if self._deployment:
+            self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
+            return int(self._deployment.status.ready_replicas)
         return None
 
     @property
-    def failing_replicas(self) -> Optional[List[str]]:
+    def failing_replicas(self) -> Optional[int]:
         """The number of failing replicas."""
         if self._deployment:
             self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
-            return self._deployment.status.failing_replicas
+            return int(self._deployment.status.failing_replicas)
         return None
 
     @property
-    def deleting_replicas(self) -> Optional[List[str]]:
+    def deleting_replicas(self) -> Optional[int]:
         """The number of deleting replicas."""
         if self._deployment:
             self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
-            return self._deployment.status.deleting_replicas
+            return int(self._deployment.status.deleting_replicas)
         return None
 
     @property

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -30,6 +30,7 @@ from lightning_sdk.lightning_cloud.openapi.api.billing_service_api import Billin
 from lightning_sdk.lightning_cloud.openapi.api.blog_posts_service_api import BlogPostsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_environment_template_service_api import CloudSpaceEnvironmentTemplateServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_service_api import CloudSpaceServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.cloudy_service_api import CloudyServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cluster_service_api import ClusterServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.data_connection_service_api import DataConnectionServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.dataset_service_api import DatasetServiceApi
@@ -103,6 +104,8 @@ from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body import C
 from lightning_sdk.lightning_cloud.openapi.models.create import Create
 from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
 from lightning_sdk.lightning_cloud.openapi.models.create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs import CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs
+from lightning_sdk.lightning_cloud.openapi.models.credits_autoreplenish_body import CreditsAutoreplenishBody
+from lightning_sdk.lightning_cloud.openapi.models.credits_autoreplenish_body1 import CreditsAutoreplenishBody1
 from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
 from lightning_sdk.lightning_cloud.openapi.models.dataset_id_versions_body import DatasetIdVersionsBody
 from lightning_sdk.lightning_cloud.openapi.models.dataset_id_visibility_body import DatasetIdVisibilityBody
@@ -175,6 +178,7 @@ from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import
 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.pipelines_id_body1 import PipelinesIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.pipelinetemplates_id_body import PipelinetemplatesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.profiler_captures_body import ProfilerCapturesBody
 from lightning_sdk.lightning_cloud.openapi.models.profiler_enabled_body import ProfilerEnabledBody
@@ -308,10 +312,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_publication_typ
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_seed_file import V1CloudSpaceSeedFile
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import V1CloudSpaceSession
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_source_type import V1CloudSpaceSourceType
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_specialized_view import V1CloudSpaceSpecializedView
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloudy_expert import V1CloudyExpert
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -599,6 +605,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_type import V1JobType
 from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
 from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_knowledge_configuration import V1KnowledgeConfiguration
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1 import V1KubernetesDirectV1
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status import V1KubernetesDirectV1Status
 from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
@@ -634,6 +642,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_tags_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_version_publications_response import V1ListCloudSpaceVersionPublicationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_versions_response import V1ListCloudSpaceVersionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_spaces_response import V1ListCloudSpacesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloudy_experts_response import V1ListCloudyExpertsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_accelerators_response import V1ListClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_availabilities_response import V1ListClusterAvailabilitiesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reservations_response import V1ListClusterCapacityReservationsResponse
@@ -904,6 +913,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated i
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
 from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
+from lightning_sdk.lightning_cloud.openapi.models.v1_token_usage import V1TokenUsage
 from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_cloud_space_response import V1TransferCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse
@@ -929,9 +939,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response im
 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
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_model_visibility_response import V1UpdateModelVisibilityResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_organization_credits_auto_replenish_response import V1UpdateOrganizationCreditsAutoReplenishResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_snowflake_query_response import V1UpdateSnowflakeQueryResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_credits_auto_replenish_response import V1UpdateUserCreditsAutoReplenishResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_request import V1UpdateUserRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_request import V1UpdateUserViewedNewFeaturesRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_response import V1UpdateUserViewedNewFeaturesResponse

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -11,6 +11,7 @@ from lightning_sdk.lightning_cloud.openapi.api.billing_service_api import Billin
 from lightning_sdk.lightning_cloud.openapi.api.blog_posts_service_api import BlogPostsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_environment_template_service_api import CloudSpaceEnvironmentTemplateServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_service_api import CloudSpaceServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.cloudy_service_api import CloudyServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cluster_service_api import ClusterServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.data_connection_service_api import DataConnectionServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.dataset_service_api import DatasetServiceApi

lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py

@@ -952,6 +952,111 @@ class AssistantsServiceApi(object):
             _request_timeout=params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def assistants_service_get_managed_model_assistant(self, model_name: 'str', **kwargs) -> 'V1Assistant':  # noqa: E501
+        """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
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.assistants_service_get_managed_model_assistant(model_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str model_name: (required)
+        :param str user_name:
+        :param str org_name:
+        :param str model_provider:
+        :return: V1Assistant
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        if kwargs.get('async_req'):
+            return self.assistants_service_get_managed_model_assistant_with_http_info(model_name, **kwargs)  # noqa: E501
+        else:
+            (data) = self.assistants_service_get_managed_model_assistant_with_http_info(model_name, **kwargs)  # noqa: E501
+            return data
+
+    def assistants_service_get_managed_model_assistant_with_http_info(self, model_name: 'str', **kwargs) -> 'V1Assistant':  # noqa: E501
+        """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
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.assistants_service_get_managed_model_assistant_with_http_info(model_name, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool
+        :param str model_name: (required)
+        :param str user_name:
+        :param str org_name:
+        :param str model_provider:
+        :return: V1Assistant
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        all_params = ['model_name', 'user_name', 'org_name', 'model_provider']  # noqa: E501
+        all_params.append('async_req')
+        all_params.append('_return_http_data_only')
+        all_params.append('_preload_content')
+        all_params.append('_request_timeout')
+
+        params = locals()
+        for key, val in six.iteritems(params['kwargs']):
+            if key not in all_params:
+                raise TypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method assistants_service_get_managed_model_assistant" % key
+                )
+            params[key] = val
+        del params['kwargs']
+        # verify the required parameter 'model_name' is set
+        if ('model_name' not in params or
+                params['model_name'] is None):
+            raise ValueError("Missing the required parameter `model_name` when calling `assistants_service_get_managed_model_assistant`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'model_name' in params:
+            path_params['modelName'] = params['model_name']  # noqa: E501
+
+        query_params = []
+        if 'user_name' in params:
+            query_params.append(('userName', params['user_name']))  # noqa: E501
+        if 'org_name' in params:
+            query_params.append(('orgName', params['org_name']))  # noqa: E501
+        if 'model_provider' in params:
+            query_params.append(('modelProvider', params['model_provider']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/v1/agents/managed-model/{modelName}', 'GET',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='V1Assistant',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=params.get('async_req'),
+            _return_http_data_only=params.get('_return_http_data_only'),
+            _preload_content=params.get('_preload_content', True),
+            _request_timeout=params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def assistants_service_get_managed_model_by_name(self, project_id: 'str', managed_endpoint_id: 'str', name: 'str', **kwargs) -> 'V1ManagedModel':  # noqa: E501
         """assistants_service_get_managed_model_by_name  # noqa: E501
 
@@ -1169,6 +1274,8 @@ class AssistantsServiceApi(object):
         :param bool published:
         :param str internal_name:
         :param str user_id:
+        :param bool cloudy_compatible:
+        :param bool model_assistants_only:
         :return: V1ListAssistantsResponse
                  If the method is called asynchronously,
                  returns the request thread.
@@ -1195,12 +1302,14 @@ class AssistantsServiceApi(object):
         :param bool published:
         :param str internal_name:
         :param str user_id:
+        :param bool cloudy_compatible:
+        :param bool model_assistants_only:
         :return: V1ListAssistantsResponse
                  If the method is called asynchronously,
                  returns the request thread.
         """
 
-        all_params = ['org_id', 'project_id', 'cloudspace_id', 'published', 'internal_name', 'user_id']  # noqa: E501
+        all_params = ['org_id', 'project_id', 'cloudspace_id', 'published', 'internal_name', 'user_id', 'cloudy_compatible', 'model_assistants_only']  # noqa: E501
         all_params.append('async_req')
         all_params.append('_return_http_data_only')
         all_params.append('_preload_content')
@@ -1233,6 +1342,10 @@ class AssistantsServiceApi(object):
             query_params.append(('internalName', params['internal_name']))  # noqa: E501
         if 'user_id' in params:
             query_params.append(('userId', params['user_id']))  # noqa: E501
+        if 'cloudy_compatible' in params:
+            query_params.append(('cloudyCompatible', params['cloudy_compatible']))  # noqa: E501
+        if 'model_assistants_only' in params:
+            query_params.append(('modelAssistantsOnly', params['model_assistants_only']))  # noqa: E501
 
         header_params = {}