Update api response

This commit is contained in:
Aiden Dai committed 2024-06-11 10:53:56 +08:00
1 parent 6ef7641a0d
commit 56786f9e32
2 files changed
+3 -1

No files matched your search

+2 -1
View File
@@ -463,6 +463,7 @@ class BedrockModel(BaseChatModel):
message = ChatResponseMessage( message = ChatResponseMessage(
tool_calls=[ tool_calls=[
ToolCall( ToolCall(
index=0,
type="function", type="function",
id=delta["toolUse"]["toolUseId"], id=delta["toolUse"]["toolUseId"],
function=ResponseFunction( function=ResponseFunction(
@@ -484,7 +485,7 @@ class BedrockModel(BaseChatModel):
message = ChatResponseMessage( message = ChatResponseMessage(
tool_calls=[ tool_calls=[
ToolCall( ToolCall(
type="function", index=0,
function=ResponseFunction( function=ResponseFunction(
arguments=delta["toolUse"]["input"], arguments=delta["toolUse"]["input"],
) )
+1
View File
@@ -22,6 +22,7 @@ class ResponseFunction(BaseModel):
class ToolCall(BaseModel): class ToolCall(BaseModel):
index: int | None = None
id: str | None = None id: str | None = None
type: Literal["function"] = "function" type: Literal["function"] = "function"
function: ResponseFunction function: ResponseFunction