tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

feat: openai embeddings integration

Browse source
This commit is contained in:
Timothy J. Baek 2024-04-14 19:48:15 -04:00
parent b48e73fa43
commit b1b72441bb
6 changed files with 155 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

4
backend/apps/ollama/main.py
View file

@ -659,7 +659,7 @@ def generate_ollama_embeddings(
url_idx: Optional[int] = None,
):
log.info("generate_ollama_embeddings", form_data)
log.info(f"generate_ollama_embeddings {form_data}")
if url_idx == None:
model = form_data.model
@ -688,7 +688,7 @@ def generate_ollama_embeddings(
data = r.json()
log.info("generate_ollama_embeddings", data)
log.info(f"generate_ollama_embeddings {data}")
if "embedding" in data:
return data["embedding"]