Update vector_store_retriever.py

vector_store_retriever.py

@@ -23,12 +23,12 @@ load_dotenv()
 path_work = "."
 hf_token = os.getenv("HF")
 
-class HuggingFaceInstructEmbeddings(EmbeddingFunction):
+class HuggingFaceInstructEmbeddings(HuggingFaceInstructEmbeddings):
     def __init__(self, model_name: str, model_kwargs: Optional[Dict[str, Any]] = None):
         self.model = AutoModel.from_pretrained(model_name, **(model_kwargs or {}))
         self.tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-    def __call__(self, input: Union[Documents]) -> Embeddings:
+    def __call__(self, input: Union[Documents]) -> HuggingFaceInstructEmbeddings:
         if isinstance(input, Documents):
             texts = [doc.text for doc in input]
             embeddings = self._embed_text(texts)