yungongzi OnePieceMan commited on
Commit
c8da8f9
·
1 Parent(s): 713589f

Fix VolcEngine BUG (#1165)

Browse files

### 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: 海贼宅 <[email protected]>

Files changed (1) hide show
  1. rag/llm/__init__.py +1 -0
rag/llm/__init__.py CHANGED
@@ -52,6 +52,7 @@ ChatModel = {
52
  "Xinference": XinferenceChat,
53
  "Moonshot": MoonshotChat,
54
  "DeepSeek": DeepSeekChat,
 
55
  "BaiChuan": BaiChuanChat,
56
  "MiniMax": MiniMaxChat,
57
  "Mistral": MistralChat
 
52
  "Xinference": XinferenceChat,
53
  "Moonshot": MoonshotChat,
54
  "DeepSeek": DeepSeekChat,
55
+ "VolcEngine": VolcEngineChat,
56
  "BaiChuan": BaiChuanChat,
57
  "MiniMax": MiniMaxChat,
58
  "Mistral": MistralChat