Change Nova Sonic model import and config to follow Strands docs and samples (#1443)
* Refactor Nova Sonic model client configuration Signed-off-by: Gui Ruggiero <guilherme.ruggiero@gmail.com> * Add Gui Ruggiero to the contributors list * Refactor import statement for BidiNovaSonicModel --------- Signed-off-by: Gui Ruggiero <guilherme.ruggiero@gmail.com>
This commit is contained in:
+4
-4
@@ -5,7 +5,7 @@ import traceback
|
||||
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"],
|
||||
}
|
||||
},
|
||||
|
||||
@@ -115,3 +115,4 @@
|
||||
- Anant Murarka (anantmu)
|
||||
- Cristiano Scandura (scandura)
|
||||
- palbiren
|
||||
- Gui Ruggiero (guiruggiero)
|
||||
|
||||
Reference in New Issue
Block a user