Update api response

2024-06-11 10:53:56 +08:00
parent 6ef7641a0d
commit 56786f9e32
2 changed files with 3 additions and 1 deletions
--- a/src/api/models/bedrock.py
+++ b/src/api/models/bedrock.py
@@ -463,6 +463,7 @@ class BedrockModel(BaseChatModel):
                message = ChatResponseMessage(
                    tool_calls=[
                        ToolCall(
+                            index=0,
                            type="function",
                            id=delta["toolUse"]["toolUseId"],
                            function=ResponseFunction(
@@ -484,7 +485,7 @@ class BedrockModel(BaseChatModel):
                message = ChatResponseMessage(
                    tool_calls=[
                        ToolCall(
-                            type="function",
+                            index=0,
                            function=ResponseFunction(
                                arguments=delta["toolUse"]["input"],
                            )