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

fix: remove openllm pypi package because of this package too large (#931)

Browse Source
This commit is contained in:
takatost
2023-08-21 02:12:28 +08:00
committed by GitHub
parent 25264e7852
commit 6c832ee328
5 changed files with 97 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/tests/unit_tests/model_providers/test_openllm_provider.py
View File

@@ -23,8 +23,7 @@ def decrypt_side_effect(tenant_id, encrypted_key):
def test_is_credentials_valid_or_raise_valid(mocker):
mocker.patch('langchain.llms.openllm.OpenLLM._identifying_params', return_value=None)
mocker.patch('langchain.llms.openllm.OpenLLM._call',
mocker.patch('core.third_party.langchain.llms.openllm.OpenLLM._call',
return_value="abc")
MODEL_PROVIDER_CLASS.is_model_credentials_valid_or_raise(
@@ -35,8 +34,6 @@ def test_is_credentials_valid_or_raise_valid(mocker):
def test_is_credentials_valid_or_raise_invalid(mocker):
mocker.patch('langchain.llms.openllm.OpenLLM._identifying_params', return_value=None)
# raise CredentialsValidateFailedError if credential is not in credentials
with pytest.raises(CredentialsValidateFailedError):
MODEL_PROVIDER_CLASS.is_model_credentials_valid_or_raise(