From 9afc6878c82ee71332fa09aebecea93dd7829e7f Mon Sep 17 00:00:00 2001 From: Deven Patel Date: Fri, 19 Jan 2024 23:38:09 +0530 Subject: [PATCH] [Update] add test for passing vector dimension in embedder config (#1196) Co-authored-by: Deven Patel --- pyproject.toml | 2 +- tests/test_factory.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 4abe8d80ed..01e9bb5dbc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "embedchain" -version = "0.1.65" +version = "0.1.66" description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data" authors = [ "Taranjeet Singh ", diff --git a/tests/test_factory.py b/tests/test_factory.py index cc2d0b0454..b6d7f62516 100644 --- a/tests/test_factory.py +++ b/tests/test_factory.py @@ -35,7 +35,7 @@ def test_llm_factory_create(self, provider_name, config_data, expected_class): ("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),