diff --git a/api/core/model_providers/models/llm/zhipuai_model.py b/api/core/model_providers/models/llm/zhipuai_model.py index 7f32c1dc70..f94eb4e729 100644 --- a/api/core/model_providers/models/llm/zhipuai_model.py +++ b/api/core/model_providers/models/llm/zhipuai_model.py @@ -16,6 +16,7 @@ class ZhipuAIModel(BaseLLM): def _init_client(self) -> Any: provider_model_kwargs = self._to_model_kwargs_input(self.model_rules, self.model_kwargs) return ZhipuAIChatLLM( + model=self.name, streaming=self.streaming, callbacks=self.callbacks, **self.credentials,