fix: wenxin model name invalid when llm call (#1248)

This commit is contained in:
takatost
2023-09-27 16:29:13 +08:00
committed by GitHub
parent 9dbb8acd4b
commit d38eac959b
2 changed files with 7 additions and 1 deletions

View File

@ -18,6 +18,7 @@ class WenxinModel(BaseLLM):
provider_model_kwargs = self._to_model_kwargs_input(self.model_rules, self.model_kwargs)
# TODO load price_config from configs(db)
return Wenxin(
model=self.name,
streaming=self.streaming,
callbacks=self.callbacks,
**self.credentials,