handle_auth_code_callback_3.py

def handle_auth_code_callback(body, event, auth_code, aad_object_id):
    # ...
    encrypted_token_base64 = base64.b64encode(encrypted_token['CiphertextBlob']).decode('utf-8')

    # Store the access token in DynamoDB, can be used in future transactions for an hour (default expiration)
    dynamodb_client = boto3.client("dynamodb")
    dynamodb_client.put_item(
        TableName=token_table_arn,
        Item={
            "aadObjectId": {"S": aad_object_id},
            "accessToken": {"S": encrypted_token_base64},
            "expiresAt": {"N": str(expires_at)}
        }
    )

    print("🟢 Successfully stored access token in DynamoDB")
    
    # Fetch the conversation body from DynamoDB
    response = dynamodb_client.get_item(
        TableName=conversation_table_arn,
        Key={
            "aadObjectId": {"S": aad_object_id}
        }
    )