Update api response

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

View File

@@ -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"],
)

View File

@@ -22,6 +22,7 @@ class ResponseFunction(BaseModel):
class ToolCall(BaseModel):
index: int | None = None
id: str | None = None
type: Literal["function"] = "function"
function: ResponseFunction