zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: bailian

Browse Source
This commit is contained in:
wxg0103 2025-03-21 11:02:43 +08:00
parent e22fc95ee9
commit 2faabbe392
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/model/llm.py
View File

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