phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Fix ollama embedding response

Browse source
This commit is contained in:
onukura 2024-03-25 16:25:22 +00:00
parent ef69eefcdb
commit 2df63cc621
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/llms/ollama.py
View file

@ -396,6 +396,7 @@ async def ollama_aembeddings(
response_json = await response.json() response_json = await response.json()
embeddings = response_json["embedding"] embeddings = response_json["embedding"]
embeddings = [embeddings] # Ollama currently does not support batch embedding
## RESPONSE OBJECT ## RESPONSE OBJECT
output_data = [] output_data = []
for idx, embedding in enumerate(embeddings): for idx, embedding in enumerate(embeddings):