OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-31 11:00:49 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

refac

Browse Source 
Co-Authored-By: dulicon <57261058+dulicon@users.noreply.github.com>
This commit is contained in:
Timothy Jaeryang Baek 2025-04-10 20:01:07 -07:00
parent a47a958edd
commit 1921420319
1 changed files with 23 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
backend/open_webui/utils/middleware.py
View File

@ -1660,14 +1660,27 @@ async def process_chat_response(
)
if tool_call_index is not None:
if (
len(response_tool_calls)
<= tool_call_index
):
# Check if the tool call already exists
current_response_tool_call = None
for (
response_tool_call
) in response_tool_calls:
if (
response_tool_call.get("index")
== tool_call_index
):
current_response_tool_call = (
response_tool_call
)
break
if current_response_tool_call is None:
# Add the new tool call
response_tool_calls.append(
delta_tool_call
)
else:
# Update the existing tool call
delta_name = delta_tool_call.get(
"function", {}
).get("name")
@ -1678,16 +1691,14 @@ async def process_chat_response(
)
if delta_name:
response_tool_calls[
tool_call_index
]["function"][
"name"
] += delta_name
current_response_tool_call[
"function"
]["name"] += delta_name
if delta_arguments:
response_tool_calls[
tool_call_index
]["function"][
current_response_tool_call[
"function"
][
"arguments"
] += delta_arguments