@@ -289,7 +289,8 @@ def chat_model_params(self) -> dict:
289
289
return {}
290
290
291
291
@property
292
- def _standard_chat_model_params (self ) -> dict :
292
+ def standard_chat_model_params (self ) -> dict :
293
+ """:meta private:"""
293
294
return {
294
295
"temperature" : 0 ,
295
296
"max_tokens" : 100 ,
@@ -303,7 +304,7 @@ def model(self) -> BaseChatModel:
303
304
"""Fixture that returns an instance of the chat model. Should not be
304
305
overridden."""
305
306
return self .chat_model_class (
306
- ** {** self ._standard_chat_model_params , ** self .chat_model_params }
307
+ ** {** self .standard_chat_model_params , ** self .chat_model_params }
307
308
)
308
309
309
310
@property
@@ -448,8 +449,9 @@ def init_from_env_params(self) -> Tuple[dict, dict, dict]:
448
449
""" # noqa: E501
449
450
450
451
@property
451
- def _standard_chat_model_params (self ) -> dict :
452
- params = super ()._standard_chat_model_params
452
+ def standard_chat_model_params (self ) -> dict :
453
+ """:meta private:"""
454
+ params = super ().standard_chat_model_params
453
455
params ["api_key" ] = "test"
454
456
return params
455
457
@@ -472,7 +474,7 @@ def test_init(self) -> None:
472
474
2. The model accommodates standard parameters: https://python.langchain.com/docs/concepts/chat_models/#standard-parameters
473
475
""" # noqa: E501
474
476
model = self .chat_model_class (
475
- ** {** self ._standard_chat_model_params , ** self .chat_model_params }
477
+ ** {** self .standard_chat_model_params , ** self .chat_model_params }
476
478
)
477
479
assert model is not None
478
480
@@ -512,7 +514,7 @@ def test_init_streaming(
512
514
"""
513
515
model = self .chat_model_class (
514
516
** {
515
- ** self ._standard_chat_model_params ,
517
+ ** self .standard_chat_model_params ,
516
518
** self .chat_model_params ,
517
519
"streaming" : True ,
518
520
}
0 commit comments