From c68767acdde0bbe521fad7f202f01130d7aa4ce3 Mon Sep 17 00:00:00 2001
From: yungongzi <yun_gong_zi@yeah.net>
Date: Fri, 14 Jun 2024 19:49:28 +0800
Subject: [PATCH] Fix VolcEngine BUG (#1165)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

### What problem does this PR solve?
- Fix a bug for VolcEngine
- After testing, the current VolcEngine configuration also supports the
Doubao series
_Briefly describe what this PR aims to solve. Include background context
that will help reviewers understand the purpose of the PR._

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Co-authored-by: 海贼宅 <stu_xyx@163.com>
---
 rag/llm/__init__.py | 1 +
 1 file changed, 1 insertion(+)

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