From 2b93e02d3771ce84e2beb4bef73f8a16ff9a925c Mon Sep 17 00:00:00 2001
From: ahmedlone127 <ahmedlone127@gmail.com>
Date: Thu, 18 Jan 2024 21:47:11 +0500
Subject: [PATCH] adding BGEEmbeddings to resource downloader (#14133)

---
 .../com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main/scala/com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala b/src/main/scala/com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala
index 997884674b676e..7d10c4039d018c 100644
--- a/src/main/scala/com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala
+++ b/src/main/scala/com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala
@@ -681,7 +681,8 @@ object PythonResourceDownloader {
     "InstructorEmbeddings" -> InstructorEmbeddings,
     "E5Embeddings" -> E5Embeddings,
     "MPNetEmbeddings" -> MPNetEmbeddings,
-    "CLIPForZeroShotClassification" -> CLIPForZeroShotClassification)
+    "CLIPForZeroShotClassification" -> CLIPForZeroShotClassification,
+    "BGEEmbeddings" -> BGEEmbeddings)
 
   // List pairs of types such as the one with key type can load a pretrained model from the value type
   val typeMapper: Map[String, String] = Map("ZeroShotNerModel" -> "RoBertaForQuestionAnswering")