AVB/bedrock-access-gateway
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update api response

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

3
src/api/models/bedrock.py
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
src/api/schema.py
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