update

infiniflow · Aug 12, 2024 · 76bed42 · 76bed42
1 parent f2f326b
commit 76bed42
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/rag/llm/__init__.py b/rag/llm/__init__.py
@@ -38,7 +38,8 @@
     "NVIDIA": NvidiaEmbed,
     "LM-Studio": LmStudioEmbed,
     "OpenAI-API-Compatible": OpenAI_APIEmbed,
-    "cohere": CoHereEmbed
+    "cohere": CoHereEmbed,
+    "Upstage": UpstageEmbed
 }
 
 
@@ -84,7 +85,8 @@
     "LM-Studio": LmStudioChat,
     "OpenAI-API-Compatible": OpenAI_APIChat,
     "cohere": CoHereChat,
-    "LeptonAI": LeptonAIChat
+    "LeptonAI": LeptonAIChat,
+    "Upstage":UpstageChat,
 }