UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-03-04 19:38:54 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Fix on ollama to openai conversion - stream can return a single message with content

Browse Source
This commit is contained in:
Simone 2025-02-20 21:25:32 +01:00
parent 472a291c24
commit a8859a8145
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -104,7 +104,7 @@ async def convert_streaming_response_ollama_to_openai(ollama_streaming_response)
data = json.loads(data) data = json.loads(data)
model = data.get("model", "ollama") model = data.get("model", "ollama")
message_content = data.get("message", {}).get("content", "") message_content = data.get("message", {}).get("content", None)
tool_calls = data.get("message", {}).get("tool_calls", None) tool_calls = data.get("message", {}).get("tool_calls", None)
openai_tool_calls = None openai_tool_calls = None
@ -118,7 +118,7 @@ async def convert_streaming_response_ollama_to_openai(ollama_streaming_response)
usage = convert_ollama_usage_to_openai(data) usage = convert_ollama_usage_to_openai(data)
data = openai_chat_chunk_message_template( data = openai_chat_chunk_message_template(
model, message_content if not done else None, openai_tool_calls, usage model, message_content, openai_tool_calls, usage
) )
line = f"data: {json.dumps(data)}\n\n" line = f"data: {json.dumps(data)}\n\n"