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

fix: integration

Browse source
This commit is contained in:
Timothy J. Baek 2024-04-14 18:47:45 -04:00
parent 9cdb5bf9fe
commit 36ce157907
3 changed files with 28 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -658,6 +658,9 @@ def generate_ollama_embeddings(
form_data: GenerateEmbeddingsForm,
url_idx: Optional[int] = None,
):
log.info("generate_ollama_embeddings", form_data)
if url_idx == None:
model = form_data.model
@ -685,6 +688,8 @@ def generate_ollama_embeddings(
data = r.json()
log.info("generate_ollama_embeddings", data)
if "embedding" in data:
return data["embedding"]
else: