jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
commit 6c832ee328
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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(