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/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"