[Update] add test for passing vector dimension in embedder config (#1196)

Co-authored-by: Deven Patel <deven298@yahoo.com>
2024-01-19 23:38:09 +05:30
parent 0b5b12575a
commit 9afc6878c8
2 changed files with 2 additions and 2 deletions
--- a/tests/test_factory.py
+++ b/tests/test_factory.py
@@ -35,7 +35,7 @@ class TestFactories:
            ("gpt4all", {}, embedchain.embedder.gpt4all.GPT4AllEmbedder),
            (
                "huggingface",
-                {"model": "sentence-transformers/all-mpnet-base-v2"},
+                {"model": "sentence-transformers/all-mpnet-base-v2", "vector_dimension": 768},
                embedchain.embedder.huggingface.HuggingFaceEmbedder,
            ),
            ("vertexai", {"model": "textembedding-gecko"}, embedchain.embedder.vertexai.VertexAIEmbedder),