Add vector dimension to Ollama embedder (#1435)

2024-06-21 21:26:46 +05:30
parent 50512a5f03
commit b43a116b3c
2 changed files with 4 additions and 2 deletions
--- a/embedchain/config/embedder/google.py
+++ b/embedchain/config/embedder/google.py
@@ -10,9 +10,10 @@ class GoogleAIEmbedderConfig(BaseEmbedderConfig):
        self,
        model: Optional[str] = None,
        deployment_name: Optional[str] = None,
+        vector_dimension: Optional[int] = None,
        task_type: Optional[str] = None,
        title: Optional[str] = None,
    ):
-        super().__init__(model, deployment_name)
+        super().__init__(model, deployment_name, vector_dimension)
        self.task_type = task_type or "retrieval_document"
        self.title = title or "Embeddings for Embedchain"