response.py 471 B

1234567891011121314
  1. from open_webui.utils.task import prompt_template
  2. from open_webui.utils.misc import (
  3. openai_chat_completion_message_template,
  4. )
  5. from typing import Callable, Optional
  6. def convert_response_ollama_to_openai(ollama_response: dict) -> dict:
  7. model = ollama_response.get("model", "ollama")
  8. message_content = ollama_response.get("message", {}).get("content", "")
  9. response = openai_chat_completion_message_template(model, message_content)
  10. return response