Support for Cohere Embeddings (#1310)

2024-03-12 12:14:45 -07:00
parent a4d32aec24
commit f77f5b996e
4 changed files with 106 additions and 0 deletions
@@ -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",