1import boto3
2
3
4def get_aws_session(*, role_arn: str) -> boto3.Session:
5    sts_client = boto3.client("sts")
6    assumed_role_object = sts_client.assume_role(
7        RoleArn=role_arn, RoleSessionName="AssumeRoleSession1"
8    )
9    credentials = assumed_role_object["Credentials"]
10
11    return boto3.Session(
12        aws_access_key_id=credentials["AccessKeyId"],
13        aws_secret_access_key=credentials["SecretAccessKey"],
14        aws_session_token=credentials["SessionToken"],
15    )