diff --git a/embedchain/config/embedder/google.py b/embedchain/config/embedder/google.py index f42e5b53..7cf5a901 100644 --- 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" diff --git a/embedchain/config/embedder/ollama.py b/embedchain/config/embedder/ollama.py index c0a5f1de..f680328f 100644 --- a/embedchain/config/embedder/ollama.py +++ b/embedchain/config/embedder/ollama.py @@ -10,6 +10,7 @@ class OllamaEmbedderConfig(BaseEmbedderConfig): self, model: Optional[str] = None, base_url: Optional[str] = None, + vector_dimension: Optional[int] = None, ): - super().__init__(model) + super().__init__(model=model, vector_dimension=vector_dimension) self.base_url = base_url or "http://localhost:11434"