Skip to content

Commit 34bf74a

Browse files
committed
Fixes some typos found during testing
1 parent 78d8ff9 commit 34bf74a

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

libs/aws/langchain_aws/chat_models/bedrock_converse.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,6 @@ def _converse_params(
258258
if not toolConfig and tools:
259259
toolChoice = _format_tool_choice(toolChoice) if toolChoice else None
260260
toolConfig = {"tools": _format_tools(tools), "toolChoice": toolChoice}
261-
262261
return _drop_none(
263262
{
264263
"modelId": modelId or self.model_id,
@@ -307,7 +306,7 @@ def _messages_to_bedrock(
307306

308307
# TODO: Add ToolMessage.status support.
309308
curr["content"].append(
310-
{"toolResult": {"content": content, "toolUseID": msg.tool_call_id}}
309+
{"toolResult": {"content": content, "toolUseId": msg.tool_call_id}}
311310
)
312311
bedrock_messages.append(curr)
313312
else:
@@ -347,7 +346,7 @@ def _parse_stream_event(event: Dict[str, Any]) -> Optional[BaseMessageChunk]:
347346
tool_call_chunks.append(
348347
ToolCallChunk(
349348
name=block.get("name"),
350-
id=block.get("tool_use_id"),
349+
id=block.get("id"),
351350
args=block.get("input"),
352351
index=event["contentBlockStart"]["contentBlockIndex"],
353352
)
@@ -421,7 +420,7 @@ def _anthropic_to_bedrock(
421420
bedrock_content.append(
422421
{
423422
"toolResult": {
424-
"toolUseID": block["tool_use_id"],
423+
"toolUseId": block["tool_use_id"],
425424
"content": _anthropic_to_bedrock(content),
426425
}
427426
}

0 commit comments

Comments
 (0)