refactor: model params
This commit is contained in:
parent
da3c945b56
commit
7e80b951ac
@ -35,8 +35,8 @@ class OpenAIChatModel(MaxKBBaseModel, BaseChatOpenAI):
|
|||||||
streaming = False
|
streaming = False
|
||||||
chat_open_ai = OpenAIChatModel(
|
chat_open_ai = OpenAIChatModel(
|
||||||
model=model_name,
|
model=model_name,
|
||||||
api_base=model_credential.get('api_base'),
|
openai_api_base=model_credential.get('api_base'),
|
||||||
api_key=model_credential.get('api_key'),
|
openai_api_key=model_credential.get('api_key'),
|
||||||
extra_body=optional_params,
|
extra_body=optional_params,
|
||||||
streaming=streaming,
|
streaming=streaming,
|
||||||
custom_get_token_ids=custom_get_token_ids
|
custom_get_token_ids=custom_get_token_ids
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user