sagemaker-core 1.0.41__tar.gz → 1.0.42__tar.gz

{sagemaker_core-1.0.41/src/sagemaker_core.egg-info → sagemaker_core-1.0.42}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.41
+Version: 1.0.42
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Repository, https://github.com/aws/sagemaker-core.git

sagemaker_core-1.0.42/VERSION

@@ -0,0 +1 @@
+1.0.42

{sagemaker_core-1.0.41 → sagemaker_core-1.0.42}/src/sagemaker_core/main/code_injection/shape_dag.py

@@ -421,6 +421,18 @@ SHAPE_DAG = {
         "value_shape": "AuthenticationRequestExtraParamsValue",
         "value_type": "string",
     },
+    "AuthorizedUrl": {
+        "members": [
+            {"name": "Url", "shape": "LongS3Uri", "type": "string"},
+            {"name": "LocalPath", "shape": "LocalPath", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "AuthorizedUrlConfigs": {
+        "member_shape": "AuthorizedUrl",
+        "member_type": "structure",
+        "type": "list",
+    },
     "AutoMLAlgorithmConfig": {
         "members": [{"name": "AutoMLAlgorithms", "shape": "AutoMLAlgorithms", "type": "list"}],
         "type": "structure",
@@ -2300,6 +2312,25 @@ SHAPE_DAG = {
         "members": [{"name": "FlowDefinitionArn", "shape": "FlowDefinitionArn", "type": "string"}],
         "type": "structure",
     },
+    "CreateHubContentPresignedUrlsRequest": {
+        "members": [
+            {"name": "HubName", "shape": "HubNameOrArn", "type": "string"},
+            {"name": "HubContentType", "shape": "HubContentType", "type": "string"},
+            {"name": "HubContentName", "shape": "HubContentName", "type": "string"},
+            {"name": "HubContentVersion", "shape": "HubContentVersion", "type": "string"},
+            {"name": "AccessConfig", "shape": "PresignedUrlAccessConfig", "type": "structure"},
+            {"name": "MaxResults", "shape": "MaxResults", "type": "integer"},
+            {"name": "NextToken", "shape": "NextToken", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "CreateHubContentPresignedUrlsResponse": {
+        "members": [
+            {"name": "AuthorizedUrlConfigs", "shape": "AuthorizedUrlConfigs", "type": "list"},
+            {"name": "NextToken", "shape": "NextToken", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "CreateHubContentReferenceRequest": {
         "members": [
             {"name": "HubName", "shape": "HubNameOrArn", "type": "string"},
@@ -12282,6 +12313,13 @@ SHAPE_DAG = {
         "members": [{"name": "PredefinedMetricType", "shape": "String", "type": "string"}],
         "type": "structure",
     },
+    "PresignedUrlAccessConfig": {
+        "members": [
+            {"name": "AcceptEula", "shape": "Boolean", "type": "boolean"},
+            {"name": "ExpectedS3Url", "shape": "S3ModelUri", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "PriorityClass": {
         "members": [
             {"name": "Name", "shape": "ClusterSchedulerPriorityClassName", "type": "string"},
@@ -13861,12 +13899,14 @@ SHAPE_DAG = {
             {"name": "SpaceStorageSettings", "shape": "SpaceStorageSettings", "type": "structure"},
             {"name": "SpaceManagedResources", "shape": "FeatureStatus", "type": "string"},
             {"name": "CustomFileSystems", "shape": "CustomFileSystems", "type": "list"},
+            {"name": "RemoteAccess", "shape": "FeatureStatus", "type": "string"},
         ],
         "type": "structure",
     },
     "SpaceSettingsSummary": {
         "members": [
             {"name": "AppType", "shape": "AppType", "type": "string"},
+            {"name": "RemoteAccess", "shape": "FeatureStatus", "type": "string"},
             {"name": "SpaceStorageSettings", "shape": "SpaceStorageSettings", "type": "structure"},
         ],
         "type": "structure",
@@ -13966,6 +14006,20 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "StartSessionRequest": {
+        "members": [
+            {"name": "ResourceIdentifier", "shape": "ResourceIdentifier", "type": "string"}
+        ],
+        "type": "structure",
+    },
+    "StartSessionResponse": {
+        "members": [
+            {"name": "SessionId", "shape": "SessionId", "type": "string"},
+            {"name": "StreamUrl", "shape": "StreamUrl", "type": "string"},
+            {"name": "TokenValue", "shape": "TokenValue", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "StopAutoMLJobRequest": {
         "members": [{"name": "AutoMLJobName", "shape": "AutoMLJobName", "type": "string"}],
         "type": "structure",

{sagemaker_core-1.0.41 → sagemaker_core-1.0.42}/src/sagemaker_core/main/resources.py

@@ -7402,9 +7402,9 @@ class Domain(Base):
         domain_name: str,
         auth_mode: str,
         default_user_settings: shapes.UserSettings,
-        subnet_ids: List[str],
-        vpc_id: str,
         domain_settings: Optional[shapes.DomainSettings] = Unassigned(),
+        subnet_ids: Optional[List[str]] = Unassigned(),
+        vpc_id: Optional[str] = Unassigned(),
         tags: Optional[List[shapes.Tag]] = Unassigned(),
         app_network_access_type: Optional[str] = Unassigned(),
         home_efs_file_system_kms_key_id: Optional[str] = Unassigned(),
@@ -7422,9 +7422,9 @@ class Domain(Base):
             domain_name: A name for the domain.
             auth_mode: The mode of authentication that members use to access the domain.
             default_user_settings: The default settings to use to create a user profile when UserSettings isn't specified in the call to the CreateUserProfile API.  SecurityGroups is aggregated when specified in both calls. For all other settings in UserSettings, the values specified in CreateUserProfile take precedence over those specified in CreateDomain.
+            domain_settings: A collection of Domain settings.
             subnet_ids: The VPC subnets that the domain uses for communication.
             vpc_id: The ID of the Amazon Virtual Private Cloud (VPC) that the domain uses for communication.
-            domain_settings: A collection of Domain settings.
             tags: Tags to associated with the Domain. Each tag consists of a key and an optional value. Tag keys must be unique per resource. Tags are searchable using the Search API. Tags that you specify for the Domain are also added to all Apps that the Domain launches.
             app_network_access_type: Specifies the VPC used for non-EFS traffic. The default value is PublicInternetOnly.    PublicInternetOnly - Non-EFS traffic is through a VPC managed by Amazon SageMaker AI, which allows direct internet access    VpcOnly - All traffic is through the specified VPC and subnets
             home_efs_file_system_kms_key_id: Use KmsKeyId.
@@ -12765,6 +12765,116 @@ class HubContent(Base):
         )
 
 
+class HubContentPresignedUrls(Base):
+    """
+    Class representing resource HubContentPresignedUrls
+
+    Attributes:
+        hub_name: The name or Amazon Resource Name (ARN) of the hub that contains the content. For public content, use SageMakerPublicHub.
+        hub_content_type: The type of hub content to access. Valid values include Model, Notebook, and ModelReference.
+        hub_content_name: The name of the hub content for which to generate presigned URLs. This identifies the specific model or content within the hub.
+        authorized_url_configs: An array of authorized URL configurations, each containing a presigned URL and its corresponding local file path for proper file organization during download.
+        hub_content_version: The version of the hub content. If not specified, the latest version is used.
+        access_config: Configuration settings for accessing the hub content, including end-user license agreement acceptance for gated models and expected S3 URL validation.
+        max_results: The maximum number of presigned URLs to return in the response. Default value is 100. Large models may contain hundreds of files, requiring pagination to retrieve all URLs.
+        next_token: A token for pagination. If present, indicates that more presigned URLs are available. Use this token in a subsequent request to retrieve additional URLs.
+
+    """
+
+    hub_name: Union[str, object]
+    hub_content_type: str
+    hub_content_name: Union[str, object]
+    authorized_url_configs: List[shapes.AuthorizedUrl]
+    hub_content_version: Optional[str] = Unassigned()
+    access_config: Optional[shapes.PresignedUrlAccessConfig] = Unassigned()
+    max_results: Optional[int] = Unassigned()
+    next_token: Optional[str] = Unassigned()
+
+    def get_name(self) -> str:
+        attributes = vars(self)
+        resource_name = "hub_content_presigned_urls_name"
+        resource_name_split = resource_name.split("_")
+        attribute_name_candidates = []
+
+        l = len(resource_name_split)
+        for i in range(0, l):
+            attribute_name_candidates.append("_".join(resource_name_split[i:l]))
+
+        for attribute, value in attributes.items():
+            if attribute == "name" or attribute in attribute_name_candidates:
+                return value
+        logger.error("Name attribute not found for object hub_content_presigned_urls")
+        return None
+
+    @classmethod
+    @Base.add_validate_call
+    def create(
+        cls,
+        hub_name: Union[str, object],
+        hub_content_type: str,
+        hub_content_name: Union[str, object],
+        hub_content_version: Optional[str] = Unassigned(),
+        access_config: Optional[shapes.PresignedUrlAccessConfig] = Unassigned(),
+        max_results: Optional[int] = Unassigned(),
+        next_token: Optional[str] = Unassigned(),
+    ) -> Optional["HubContentPresignedUrls"]:
+        """
+        Create a HubContentPresignedUrls resource
+
+        Parameters:
+            hub_name: The name or Amazon Resource Name (ARN) of the hub that contains the content. For public content, use SageMakerPublicHub.
+            hub_content_type: The type of hub content to access. Valid values include Model, Notebook, and ModelReference.
+            hub_content_name: The name of the hub content for which to generate presigned URLs. This identifies the specific model or content within the hub.
+            hub_content_version: The version of the hub content. If not specified, the latest version is used.
+            access_config: Configuration settings for accessing the hub content, including end-user license agreement acceptance for gated models and expected S3 URL validation.
+            max_results: The maximum number of presigned URLs to return in the response. Default value is 100. Large models may contain hundreds of files, requiring pagination to retrieve all URLs.
+            next_token:  A token for pagination. Use this token to retrieve the next set of presigned URLs when the response is truncated.
+            session: Boto3 session.
+            region: Region name.
+
+        Returns:
+            The HubContentPresignedUrls resource.
+
+        Raises:
+            botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
+                The error message and error code can be parsed from the exception as follows:
+                ```
+                try:
+                    # AWS service call here
+                except botocore.exceptions.ClientError as e:
+                    error_message = e.response['Error']['Message']
+                    error_code = e.response['Error']['Code']
+                ```
+            ConfigSchemaValidationError: Raised when a configuration file does not adhere to the schema
+            LocalConfigNotFoundError: Raised when a configuration file is not found in local file system
+            S3ConfigNotFoundError: Raised when a configuration file is not found in S3
+        """
+
+        operation_input_args = {
+            "HubName": hub_name,
+            "HubContentType": hub_content_type,
+            "HubContentName": hub_content_name,
+            "HubContentVersion": hub_content_version,
+            "AccessConfig": access_config,
+            "MaxResults": max_results,
+            "NextToken": next_token,
+        }
+        # serialize the input request
+        operation_input_args = serialize(operation_input_args)
+        logger.debug(f"Serialized input request: {operation_input_args}")
+
+        client = Base.get_sagemaker_client(
+            session=session, region_name=region, service_name="sagemaker"
+        )
+
+        logger.debug(f"Calling create_hub_content_presigned_urls API")
+        response = client.create_hub_content_presigned_urls(**operation_input_args)
+        logger.debug(f"Response: {response}")
+
+        transformed_response = transform(response, "CreateHubContentPresignedUrlsResponse")
+        return cls(**operation_input_args, **transformed_response)
+
+
 class HubContentReference(Base):
     """
     Class representing resource HubContentReference
@@ -26953,6 +27063,13 @@ class SagemakerServicecatalogPortfolio(Base):
         return list(response.values())[0]
 
 
+class Session(Base):
+    """
+    Class representing resource Session
+
+    """
+
+
 class Space(Base):
     """
     Class representing resource Space

{sagemaker_core-1.0.41 → sagemaker_core-1.0.42}/src/sagemaker_core/main/shapes.py

@@ -1652,6 +1652,21 @@ class AthenaDatasetDefinition(Base):
     output_compression: Optional[str] = Unassigned()
 
 
+class AuthorizedUrl(Base):
+    """
+    AuthorizedUrl
+      Contains a presigned URL and its associated local file path for downloading hub content artifacts.
+
+    Attributes
+    ----------------------
+    url: The presigned S3 URL that provides temporary, secure access to download the file. URLs expire within 15 minutes for security purposes.
+    local_path: The recommended local file path where the downloaded file should be stored to maintain proper directory structure and file organization.
+    """
+
+    url: Optional[str] = Unassigned()
+    local_path: Optional[str] = Unassigned()
+
+
 class AutoMLAlgorithmConfig(Base):
     """
     AutoMLAlgorithmConfig
@@ -5451,6 +5466,21 @@ class FlowDefinitionOutputConfig(Base):
     kms_key_id: Optional[str] = Unassigned()
 
 
+class PresignedUrlAccessConfig(Base):
+    """
+    PresignedUrlAccessConfig
+      Configuration for accessing hub content through presigned URLs, including license agreement acceptance and URL validation settings.
+
+    Attributes
+    ----------------------
+    accept_eula: Indicates acceptance of the End User License Agreement (EULA) for gated models. Set to true to acknowledge acceptance of the license terms required for accessing gated content.
+    expected_s3_url: The expected S3 URL prefix for validation purposes. This parameter helps ensure consistency between the resolved S3 URIs and the deployment configuration, reducing potential compatibility issues.
+    """
+
+    accept_eula: Optional[bool] = Unassigned()
+    expected_s3_url: Optional[str] = Unassigned()
+
+
 class HubS3StorageConfig(Base):
     """
     HubS3StorageConfig
@@ -7626,6 +7656,7 @@ class SpaceSettings(Base):
     space_storage_settings: The storage settings for a space.
     space_managed_resources: If you enable this option, SageMaker AI creates the following resources on your behalf when you create the space:   The user profile that possesses the space.   The app that the space contains.
     custom_file_systems: A file system, created by you, that you assign to a space for an Amazon SageMaker AI Domain. Permitted users can access this file system in Amazon SageMaker AI Studio.
+    remote_access: A setting that enables or disables remote access for a SageMaker space. When enabled, this allows you to connect to the remote space from your local IDE.
     """
 
     jupyter_server_app_settings: Optional[JupyterServerAppSettings] = Unassigned()
@@ -7636,6 +7667,7 @@ class SpaceSettings(Base):
     space_storage_settings: Optional[SpaceStorageSettings] = Unassigned()
     space_managed_resources: Optional[str] = Unassigned()
     custom_file_systems: Optional[List[CustomFileSystem]] = Unassigned()
+    remote_access: Optional[str] = Unassigned()
 
 
 class OwnershipSettings(Base):
@@ -11572,10 +11604,12 @@ class SpaceSettingsSummary(Base):
     Attributes
     ----------------------
     app_type: The type of app created within the space.
+    remote_access: A setting that enables or disables remote access for a SageMaker space. When enabled, this allows you to connect to the remote space from your local IDE.
     space_storage_settings: The storage settings for a space.
     """
 
     app_type: Optional[str] = Unassigned()
+    remote_access: Optional[str] = Unassigned()
     space_storage_settings: Optional[SpaceStorageSettings] = Unassigned()
 
 

{sagemaker_core-1.0.41 → sagemaker_core-1.0.42/src/sagemaker_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.41
+Version: 1.0.42
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Repository, https://github.com/aws/sagemaker-core.git

sagemaker_core-1.0.41/VERSION

@@ -1 +0,0 @@
-1.0.41