scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

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
Show stats 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()
embeddings = response_json["embedding"]
embeddings = [embeddings] # Ollama currently does not support batch embedding
## RESPONSE OBJECT
output_data = []
for idx, embedding in enumerate(embeddings):