refactor: bailian

This commit is contained in:
wxg0103 2025-03-21 11:02:43 +08:00
parent e22fc95ee9
commit 2faabbe392

View File

@ -5,6 +5,7 @@ from typing import Dict
from setting.models_provider.base_model_provider import MaxKBBaseModel
from setting.models_provider.impl.base_chat_open_ai import BaseChatOpenAI
class BaiLianChatModel(MaxKBBaseModel, BaseChatOpenAI):
@staticmethod
def is_cache_model():
@ -13,7 +14,7 @@ class BaiLianChatModel(MaxKBBaseModel, BaseChatOpenAI):
@staticmethod
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)
if model_name == 'qwen-omni-turbo':
if 'qwen-omni-turbo' in model_name:
optional_params['streaming'] = True
return BaiLianChatModel(
model=model_name,