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

chore: format

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-02-13 00:13:33 -08:00
parent 7ff719938a
commit eb568695e7
52 changed files with 57 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -217,10 +217,10 @@ def openai_chat_chunk_message_template(
def openai_chat_completion_message_template(
model: str,
model: str,
message: Optional[str] = None,
tool_calls: Optional[list[dict]] = None,
usage: Optional[dict] = None
usage: Optional[dict] = None,
) -> dict:
template = openai_chat_message_template(model)
template["object"] = "chat.completion"

7
backend/open_webui/utils/response.py
View File

@@ -16,13 +16,14 @@ def convert_ollama_tool_call_to_openai(tool_calls: dict) -> dict:
"function": {
"name": tool_call.get("function", {}).get("name", ""),
"arguments": json.dumps(
tool_call.get('function', {}).get('arguments', {})
tool_call.get("function", {}).get("arguments", {})
),
},
}
openai_tool_calls.append(openai_tool_call)
return openai_tool_calls
def convert_response_ollama_to_openai(ollama_response: dict) -> dict:
model = ollama_response.get("model", "ollama")
message_content = ollama_response.get("message", {}).get("content", "")
@@ -73,7 +74,9 @@ def convert_response_ollama_to_openai(ollama_response: dict) -> dict:
),
}
response = openai_chat_completion_message_template(model, message_content, openai_tool_calls, usage)
response = openai_chat_completion_message_template(
model, message_content, openai_tool_calls, usage
)
return response