fix AttributeError: 'EmbeddingModel' object has no attribute 'embed_documents'

This commit is contained in:
Adam Wilson
2025-06-24 14:17:55 -06:00
parent f74f8274af
commit d4b3d39cf6

View File

@@ -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}
)
)
@property
def embeddings(self):
return self._embeddings