sagemaker.core.token_generator.token_generator

sagemaker.core.token_generator.token_generator#

SageMaker Token Generator core signing logic.

Generates short-term bearer tokens for AWS SageMaker API authentication using SigV4 signed pre-signed URLs.

Classes

SageMakerTokenGenerator()

Generate short-lived AWS SageMaker bearer tokens.

class sagemaker.core.token_generator.token_generator.SageMakerTokenGenerator[source]#

Bases: object

Generate short-lived AWS SageMaker bearer tokens.

get_token(credentials: botocore.credentials.Credentials, region: str) str[source]#

Generate a token using provided credentials and region.

Parameters:
  • credentials (Credentials) – AWS credentials to sign the request.

  • region (str) – AWS region.

Returns:

A bearer token string.

Return type:

str

Raises:

ValueError – If inputs are invalid.