diff --git a/01-tutorials/01-AgentCore-runtime/06-bi-directional-streaming/02-strands-ws/websocket/agent.py b/01-tutorials/01-AgentCore-runtime/06-bi-directional-streaming/02-strands-ws/websocket/agent.py index 8646cddc8..7b25e7ca6 100644 --- a/01-tutorials/01-AgentCore-runtime/06-bi-directional-streaming/02-strands-ws/websocket/agent.py +++ b/01-tutorials/01-AgentCore-runtime/06-bi-directional-streaming/02-strands-ws/websocket/agent.py @@ -5,7 +5,7 @@ from fastapi import WebSocket, WebSocketDisconnect from strands.experimental.bidi.agent import BidiAgent -from strands.experimental.bidi.models.nova_sonic import BidiNovaSonicModel +from strands.experimental.bidi.models import BidiNovaSonicModel logger = logging.getLogger(__name__) @@ -189,12 +189,12 @@ def _create_model(config: dict, effective_gateway_arns: list, api_key: str = Non # Nova Sonic if model_id.startswith("amazon.nova"): return BidiNovaSonicModel( - region=config.get("region", "us-east-1"), + client_config={"region": config.get("region", "us-east-1")}, model_id=model_id, provider_config={ "audio": { - "input_sample_rate": config["input_sample_rate"], - "output_sample_rate": config["output_sample_rate"], + "input_rate": config["input_sample_rate"], + "output_rate": config["output_sample_rate"], "voice": config["voice"], } }, diff --git a/CONTRIBUTORS.md b/CONTRIBUTORS.md index a4cfaddae..1460dc751 100644 --- a/CONTRIBUTORS.md +++ b/CONTRIBUTORS.md @@ -115,3 +115,4 @@ - Anant Murarka (anantmu) - Cristiano Scandura (scandura) - palbiren +- Gui Ruggiero (guiruggiero)