Update api response

2024-06-11 10:53:56 +08:00
parent 6ef7641a0d
commit 56786f9e32
2 changed files with 3 additions and 1 deletions
@@ -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"],
                            )
@@ -22,6 +22,7 @@ class ResponseFunction(BaseModel):


 class ToolCall(BaseModel):
+    index: int | None = None
    id: str | None = None
    type: Literal["function"] = "function"
    function: ResponseFunction