Skip to content

Commit

Permalink
fix: model config
Browse files Browse the repository at this point in the history
  • Loading branch information
zhayujie committed May 3, 2024
1 parent 27e507e commit fb72b60
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions bridge/bridge.py
Expand Up @@ -39,20 +39,22 @@ def __init__(self):
if model_type and model_type.startswith("claude-3"):
self.btype["chat"] = const.CLAUDEAPI

if model_type in ["claude"]:
self.btype["chat"] = const.CLAUDEAI

if model_type in ["moonshot-v1-8k", "moonshot-v1-32k", "moonshot-v1-128k"]:
self.btype["chat"] = const.MOONSHOT

if conf().get("use_linkai") and conf().get("linkai_api_key"):
self.btype["chat"] = const.LINKAI
if not conf().get("voice_to_text") or conf().get("voice_to_text") in ["openai"]:
self.btype["voice_to_text"] = const.LINKAI
if not conf().get("text_to_voice") or conf().get("text_to_voice") in ["openai", const.TTS_1, const.TTS_1_HD]:
self.btype["text_to_voice"] = const.LINKAI
if model_type in ["claude"]:
self.btype["chat"] = const.CLAUDEAI

if model_type in ["moonshot-v1-8k", "moonshot-v1-32k", "moonshot-v1-128k"]:
self.btype["chat"] = const.MOONSHOT

self.bots = {}
self.chat_bots = {}

# 模型对应的接口
def get_bot(self, typename):
if self.bots.get(typename) is None:
Expand Down

0 comments on commit fb72b60

Please sign in to comment.