Skip to content

Commit 6398230

Browse files
committed
Fixes some typos found during testing
1 parent 6d5557c commit 6398230

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

libs/aws/langchain_aws/chat_models/bedrock_converse.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -466,7 +466,6 @@ def _converse_params(
466466
if not toolConfig and tools:
467467
toolChoice = _format_tool_choice(toolChoice) if toolChoice else None
468468
toolConfig = {"tools": _format_tools(tools), "toolChoice": toolChoice}
469-
470469
return _drop_none(
471470
{
472471
"modelId": modelId or self.model_id,
@@ -573,7 +572,7 @@ def _parse_stream_event(event: Dict[str, Any]) -> Optional[BaseMessageChunk]:
573572
tool_call_chunks.append(
574573
ToolCallChunk(
575574
name=block.get("name"),
576-
id=block.get("tool_use_id"),
575+
id=block.get("id"),
577576
args=block.get("input"),
578577
index=event["contentBlockStart"]["contentBlockIndex"],
579578
)
@@ -660,7 +659,7 @@ def _anthropic_to_bedrock(
660659
bedrock_content.append(
661660
{
662661
"toolResult": {
663-
"toolUseId": block["toolUseId"],
662+
"toolUseId": block["tool_use_id"],
664663
"content": _anthropic_to_bedrock(content),
665664
}
666665
}

0 commit comments

Comments
 (0)