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

2023-09-27 16:29:13 +08:00 · 2023-09-27 16:29:13 +08:00 · d38eac959b
commit d38eac959b
parent 9dbb8acd4b
2 changed files with 7 additions and 1 deletions
--- a/api/core/model_providers/models/llm/wenxin_model.py
+++ b/api/core/model_providers/models/llm/wenxin_model.py
@ -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,