From 918823d805a2773ae0f9bdce88748699f0e4bbfc Mon Sep 17 00:00:00 2001 From: Mathew Shen Date: Fri, 2 Aug 2024 04:01:19 +0800 Subject: [PATCH] fix: method name should be same with abstrct base class (#1592) --- mem0/embeddings/huggingface.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mem0/embeddings/huggingface.py b/mem0/embeddings/huggingface.py index 6b187d81e..5e56f0349 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.