From d5683182bcf250fb626f65e136a88c0f07307e2d Mon Sep 17 00:00:00 2001 From: Abhinav N <114095553+Abhinav-Naikawadi@users.noreply.github.com> Date: Wed, 5 Jul 2023 17:57:13 -0700 Subject: [PATCH] Cohere embeddings support (#432) cohere embedding support Co-authored-by: Abhinav Naikawadi --- src/autolabel/few_shot/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/autolabel/few_shot/__init__.py b/src/autolabel/few_shot/__init__.py index 3dfdd39e..4ce71685 100644 --- a/src/autolabel/few_shot/__init__.py +++ b/src/autolabel/few_shot/__init__.py @@ -4,6 +4,7 @@ from autolabel.configs import AutolabelConfig from autolabel.schema import FewShotAlgorithm, ModelProvider from langchain.embeddings import ( + CohereEmbeddings, HuggingFaceEmbeddings, OpenAIEmbeddings, VertexAIEmbeddings, @@ -30,6 +31,7 @@ ModelProvider.OPENAI: OpenAIEmbeddings, ModelProvider.GOOGLE: VertexAIEmbeddings, ModelProvider.HUGGINGFACE_PIPELINE: HuggingFaceEmbeddings, + ModelProvider.COHERE: CohereEmbeddings, } logger = logging.getLogger(__name__)