feat: expose Cohere and HF embedding support on UI (#236)

2024-09-06 18:18:19 +07:00
parent 4d7f16475f
commit 069f0f3c83
2 changed files with 27 additions and 2 deletions
--- a/libs/ktem/ktem/embeddings/manager.py
+++ b/libs/ktem/ktem/embeddings/manager.py
@@ -55,10 +55,18 @@ class EmbeddingManager:
        from kotaemon.embeddings import (
            AzureOpenAIEmbeddings,
            FastEmbedEmbeddings,
+            LCCohereEmbeddings,
+            LCHuggingFaceEmbeddings,
            OpenAIEmbeddings,
        )

-        self._vendors = [AzureOpenAIEmbeddings, OpenAIEmbeddings, FastEmbedEmbeddings]
+        self._vendors = [
+            AzureOpenAIEmbeddings,
+            OpenAIEmbeddings,
+            FastEmbedEmbeddings,
+            LCCohereEmbeddings,
+            LCHuggingFaceEmbeddings,
+        ]

    def __getitem__(self, key: str) -> BaseEmbeddings:
        """Get model by name"""