ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

convert embedding function name to be more consistence

Browse source
This commit is contained in:
henry 2025-06-04 18:24:27 +02:00
parent cc12e9e1a3
commit 04da2b05f6
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
backend/open_webui/utils/embeddings.py
View file

@ -15,7 +15,7 @@ from open_webui.routers.pipelines import process_pipeline_inlet_filter
from open_webui.utils.payload import convert_embedding_payload_openai_to_ollama from open_webui.utils.payload import convert_embedding_payload_openai_to_ollama
from open_webui.utils.response import convert_response_ollama_to_openai from open_webui.utils.response import convert_embedding_response_ollama_to_openai
logging.basicConfig(stream=sys.stdout, level=GLOBAL_LOG_LEVEL) logging.basicConfig(stream=sys.stdout, level=GLOBAL_LOG_LEVEL)
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -117,7 +117,7 @@ async def generate_embeddings(
form_data=form_obj, form_data=form_obj,
user=user, user=user,
) )
return convert_response_ollama_to_openai(response) return convert_embedding_response_ollama_to_openai(response)
# Default: OpenAI or compatible backend # Default: OpenAI or compatible backend
return await openai_embeddings( return await openai_embeddings(

2
backend/open_webui/utils/response.py
View file

@ -126,7 +126,7 @@ async def convert_streaming_response_ollama_to_openai(ollama_streaming_response)
yield "data: [DONE]\n\n" yield "data: [DONE]\n\n"
def convert_response_ollama_to_openai(response): def convert_embedding_response_ollama_to_openai(response) -> dict:
""" """
Convert the response from Ollama embeddings endpoint to the OpenAI-compatible format. Convert the response from Ollama embeddings endpoint to the OpenAI-compatible format.