Support for Cohere Embeddings (#1310)

2024-03-13 00:44:45 +05:30
parent a4d32aec24
commit f77f5b996e
4 changed files with 106 additions and 0 deletions
--- a/embedchain/factory.py
+++ b/embedchain/factory.py
@@ -56,6 +56,7 @@ class EmbedderFactory:
        "google": "embedchain.embedder.google.GoogleAIEmbedder",
        "mistralai": "embedchain.embedder.mistralai.MistralAIEmbedder",
        "nvidia": "embedchain.embedder.nvidia.NvidiaEmbedder",
+        "cohere": "embedchain.embedder.cohere.CohereEmbedder",
    }
    provider_to_config_class = {
        "azure_openai": "embedchain.config.embedder.base.BaseEmbedderConfig",