diff --git a/rag/llm/__init__.py b/rag/llm/__init__.py index d70136a57..18adbfebd 100644 --- a/rag/llm/__init__.py +++ b/rag/llm/__init__.py @@ -52,6 +52,7 @@ ChatModel = { "Xinference": XinferenceChat, "Moonshot": MoonshotChat, "DeepSeek": DeepSeekChat, + "VolcEngine": VolcEngineChat, "BaiChuan": BaiChuanChat, "MiniMax": MiniMaxChat, "Mistral": MistralChat