diff --git a/mem0/embeddings/huggingface.py b/mem0/embeddings/huggingface.py index 6b187d81..5e56f034 100644 --- a/mem0/embeddings/huggingface.py +++ b/mem0/embeddings/huggingface.py @@ -6,7 +6,7 @@ class HuggingFaceEmbedding(EmbeddingBase): def __init__(self, model_name="multi-qa-MiniLM-L6-cos-v1"): self.model = SentenceTransformer(model_name) - def get_embedding(self, text): + def embed(self, text): """ Get the embedding for the given text using Hugging Face.