fix(embeddings): guard embed_batch count mismatch in OpenAI and Azure OpenAI (#5966)

mem0/embeddings/azure_openai.py

@@ -69,4 +69,9 @@ class AzureOpenAIEmbedding(EmbeddingBase):
                 model=self.config.model,
             )
             all_embeddings.extend(item.embedding for item in sorted(response.data, key=lambda x: x.index))
+        if len(all_embeddings) != len(texts):
+            raise ValueError(
+                f"Azure OpenAI embed_batch() returned {len(all_embeddings)} embeddings for {len(texts)} texts"
+                f" using model '{self.config.model}'"
+            )
         return all_embeddings

mem0/embeddings/openai.py

@@ -73,4 +73,9 @@ class OpenAIEmbedding(EmbeddingBase):
                 kwargs["dimensions"] = self.config.embedding_dims
             response = self.client.embeddings.create(**kwargs)
             all_embeddings.extend(item.embedding for item in sorted(response.data, key=lambda x: x.index))
+        if len(all_embeddings) != len(texts):
+            raise ValueError(
+                f"OpenAI embed_batch() returned {len(all_embeddings)} embeddings for {len(texts)} texts"
+                f" using model '{self.config.model}'"
+            )
         return all_embeddings