Skip to content

Commit 54e543f

Browse files
Pass bedrock runtime client as client parameter.
1 parent 65f5573 commit 54e543f

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

singlestoredb/ai/chat.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,12 @@ def _inject_headers(request: Any, **_ignored: Any) -> None:
147147
)
148148
return ChatBedrockConverse(
149149
model=actual_model,
150+
endpoint_url=info.connection_url, # redirect requests to UMG
151+
region_name='us-east-1', # dummy value; UMG does not use this
152+
aws_access_key_id='placeholder', # dummy value; UMG does not use this
153+
aws_secret_access_key='placeholder', # dummy value; UMG does not use this
150154
disable_streaming=not streaming,
151-
bedrock_runtime_client=client,
155+
client=client,
152156
**kwargs,
153157
)
154158

0 commit comments

Comments
 (0)