Skip to content

Commit 2faabbe

Browse files
committed
refactor: bailian
1 parent e22fc95 commit 2faabbe

File tree

1 file changed

+2
-1
lines changed
  • apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/model

1 file changed

+2
-1
lines changed

apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/model/llm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from setting.models_provider.base_model_provider import MaxKBBaseModel
66
from setting.models_provider.impl.base_chat_open_ai import BaseChatOpenAI
77

8+
89
class BaiLianChatModel(MaxKBBaseModel, BaseChatOpenAI):
910
@staticmethod
1011
def is_cache_model():
@@ -13,7 +14,7 @@ def is_cache_model():
1314
@staticmethod
1415
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
1516
optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)
16-
if model_name == 'qwen-omni-turbo':
17+
if 'qwen-omni-turbo' in model_name:
1718
optional_params['streaming'] = True
1819
return BaiLianChatModel(
1920
model=model_name,

0 commit comments

Comments
 (0)