Add tests for gpt4all llm (#852)

2023-10-26 20:41:28 -07:00
parent 191ae3ec1e
commit c95d458e52
3 changed files with 62 additions and 1 deletions
--- a/embedchain/factory.py
+++ b/embedchain/factory.py
@@ -50,6 +50,7 @@ class EmbedderFactory:
    provider_to_config_class = {
        "azure_openai": "embedchain.config.embedder.base.BaseEmbedderConfig",
        "openai": "embedchain.config.embedder.base.BaseEmbedderConfig",
+        "gpt4all": "embedchain.config.embedder.base.BaseEmbedderConfig",
    }

    @classmethod