fix: Elasticsearch - use correct class attributes (#566)

2023-09-06 00:58:40 +02:00
parent f0844ed923
commit b0d8711b65
2 changed files with 7 additions and 6 deletions
--- a/embedchain/embedder/base_embedder.py
+++ b/embedchain/embedder/base_embedder.py
@@ -24,6 +24,7 @@ class BaseEmbedder:
            self.config = BaseEmbedderConfig()
        else:
            self.config = config
+        self.vector_dimension: int

    def set_embedding_fn(self, embedding_fn: Callable[[list[str]], list[str]]):
        if not hasattr(embedding_fn, "__call__"):