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),