estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: compatibility issues with the tongyi model. (#1310)

Browse Source
This commit is contained in:
takatost
2023-10-11 18:16:26 +08:00
committed by GitHub
parent b1352ff8b7
commit c536f85b2e
5 changed files with 28 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/model_providers/models/llm/tongyi_model.py
View File

@@ -18,7 +18,6 @@ class TongyiModel(BaseLLM):
def _init_client(self) -> Any:
provider_model_kwargs = self._to_model_kwargs_input(self.model_rules, self.model_kwargs)
del provider_model_kwargs['max_tokens']
return EnhanceTongyi(
model_name=self.name,
max_retries=1,
@@ -58,7 +57,6 @@ class TongyiModel(BaseLLM):
def _set_model_kwargs(self, model_kwargs: ModelKwargs):
provider_model_kwargs = self._to_model_kwargs_input(self.model_rules, model_kwargs)
del provider_model_kwargs['max_tokens']
for k, v in provider_model_kwargs.items():
if hasattr(self.client, k):
setattr(self.client, k, v)