feat(embeddings): add native embed_batch to OllamaEmbedding (#5415)

Co-authored-by: Kartik <kartik.labhshetwar@mem0.ai>
This commit is contained in:
Alok Tripathi
2026-06-17 14:12:53 +05:30
committed by GitHub
co-authored by Kartik
parent f122eb7c65
commit e34f5835bd
2 changed files with 44 additions and 0 deletions
+10
View File
@@ -63,3 +63,13 @@ class OllamaEmbedding(EmbeddingBase):
if not embeddings:
raise ValueError(f"Ollama embed() returned no embeddings for model '{self.config.model}'")
return embeddings[0]
def embed_batch(self, texts, memory_action="add"):
"""Embed multiple texts in a single Ollama API call."""
if not texts:
return []
response = self.client.embed(model=self.config.model, input=texts)
embeddings = response.get("embeddings") or []
if len(embeddings) != len(texts):
raise ValueError(f"Ollama embed() returned {len(embeddings)} embeddings for {len(texts)} texts using model '{self.config.model}'")
return embeddings