Fix ollama embedding response

2024-03-25 16:25:22 +00:00 · 2024-03-25 16:25:22 +00:00 · 2df63cc621
commit 2df63cc621
parent ef69eefcdb
1 changed files with 1 additions and 0 deletions
--- a/litellm/llms/ollama.py
+++ b/litellm/llms/ollama.py
@ -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):