From 76bed4267af22b5d7c1073eed29e992cf685cf9a Mon Sep 17 00:00:00 2001 From: Zhedong Cen Date: Mon, 12 Aug 2024 10:16:07 +0800 Subject: [PATCH] update --- rag/llm/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/rag/llm/__init__.py b/rag/llm/__init__.py index eebe8bfa92f..20d01fbbbed 100644 --- 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, }