From 5ed2d7ac2bd05ecc398cfaa100f7f64d5de4a688 Mon Sep 17 00:00:00 2001 From: Paul Shippy Date: Mon, 30 Jun 2025 17:31:31 -0700 Subject: [PATCH] Add session token option for AWS --- src/pipecat/services/aws_nova_sonic/aws.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/pipecat/services/aws_nova_sonic/aws.py b/src/pipecat/services/aws_nova_sonic/aws.py index 77e9575b5..a7832669e 100644 --- a/src/pipecat/services/aws_nova_sonic/aws.py +++ b/src/pipecat/services/aws_nova_sonic/aws.py @@ -194,6 +194,7 @@ class AWSNovaSonicLLMService(LLMService): *, secret_access_key: str, access_key_id: str, + session_token: Optional[str] = None, region: str, model: str = "amazon.nova-sonic-v1:0", voice_id: str = "matthew", # matthew, tiffany, amy @@ -208,6 +209,7 @@ class AWSNovaSonicLLMService(LLMService): Args: secret_access_key: AWS secret access key for authentication. access_key_id: AWS access key ID for authentication. + session_token: AWS session token for authentication. region: AWS region where the service is hosted. model: Model identifier. Defaults to "amazon.nova-sonic-v1:0". voice_id: Voice ID for speech synthesis. Options: matthew, tiffany, amy. @@ -220,6 +222,7 @@ class AWSNovaSonicLLMService(LLMService): super().__init__(**kwargs) self._secret_access_key = secret_access_key self._access_key_id = access_key_id + self._session_token = session_token self._region = region self._model = model self._client: Optional[BedrockRuntimeClient] = None @@ -523,7 +526,9 @@ class AWSNovaSonicLLMService(LLMService): region=self._region, aws_credentials_identity_resolver=StaticCredentialsResolver( credentials=AWSCredentialsIdentity( - access_key_id=self._access_key_id, secret_access_key=self._secret_access_key + access_key_id=self._access_key_id, + secret_access_key=self._secret_access_key, + session_token=self._session_token, ) ), http_auth_scheme_resolver=HTTPAuthSchemeResolver(),