diff --git a/pyproject.toml b/pyproject.toml index 4abe8d80..01e9bb5d 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 cc2d0b04..b6d7f625 100644 --- 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),