OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added support for API tool_calls if stream false

Browse Source
This commit is contained in:
Simone
2025-02-12 09:11:26 +01:00
parent cd2f4142d5
commit 7dc000a6b9
2 changed files with 32 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/open_webui/utils/misc.py
View File

@@ -217,12 +217,19 @@ def openai_chat_chunk_message_template(
def openai_chat_completion_message_template(
model: str, message: Optional[str] = None, usage: Optional[dict] = None
model: str,
message: Optional[str] = None,
tool_calls: Optional[list[dict]] = None,
usage: Optional[dict] = None
) -> dict:
template = openai_chat_message_template(model)
template["object"] = "chat.completion"
if message is not None:
template["choices"][0]["message"] = {"content": message, "role": "assistant"}
if tool_calls:
template["choices"][0]["tool_calls"] = tool_calls
template["choices"][0]["finish_reason"] = "stop"
if usage: