mirror of
https://github.com/open-webui/open-webui
synced 2024-11-07 00:59:52 +00:00
15 lines
471 B
Python
15 lines
471 B
Python
|
from open_webui.utils.task import prompt_template
|
||
|
|
from open_webui.utils.misc import (
|
||
|
|
openai_chat_completion_message_template,
|
||
|
|
)
|
||
|
|
|
||
|
|
from typing import Callable, Optional
|
||
|
|
|
||
|
|
|
||
|
|
def convert_response_ollama_to_openai(ollama_response: dict) -> dict:
|
||
|
|
model = ollama_response.get("model", "ollama")
|
||
|
|
message_content = ollama_response.get("message", {}).get("content", "")
|
||
|
|
|
||
|
|
response = openai_chat_completion_message_template(model, message_content)
|
||
|
|
return response
|