From d4b3d39cf62d01b12b3c22236f91c8dd5c29fd40 Mon Sep 17 00:00:00 2001 From: Adam Wilson Date: Tue, 24 Jun 2025 14:17:55 -0600 Subject: [PATCH] fix AttributeError: 'EmbeddingModel' object has no attribute 'embed_documents' --- src/text_generation/adapters/llm/embedding_model.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/src/text_generation/adapters/llm/embedding_model.py b/src/text_generation/adapters/llm/embedding_model.py index aff8f85e1..bdfecb69e 100644 --- a/src/text_generation/adapters/llm/embedding_model.py +++ b/src/text_generation/adapters/llm/embedding_model.py @@ -3,11 +3,13 @@ from src.text_generation.adapters.llm.abstract_embedding_model import AbstractEm class EmbeddingModel(AbstractEmbeddingModel): - - @property - def embeddings(self): - return HuggingFaceEmbeddings( + def __init__(self): + self._embeddings = HuggingFaceEmbeddings( model_name='sentence-transformers/all-MiniLM-L6-v2', model_kwargs={'device': 'cpu'}, encode_kwargs={'normalize_embeddings': True} - ) \ No newline at end of file + ) + + @property + def embeddings(self): + return self._embeddings \ No newline at end of file