test_provider.py 591 B

1234567891011121314151617
  1. import os
  2. import pytest
  3. from core.model_runtime.errors.validate import CredentialsValidateFailedError
  4. from core.model_runtime.model_providers.wenxin.wenxin import WenxinProvider
  5. def test_validate_provider_credentials():
  6. provider = WenxinProvider()
  7. with pytest.raises(CredentialsValidateFailedError):
  8. provider.validate_provider_credentials(credentials={"api_key": "hahahaha", "secret_key": "hahahaha"})
  9. provider.validate_provider_credentials(
  10. credentials={"api_key": os.environ.get("WENXIN_API_KEY"), "secret_key": os.environ.get("WENXIN_SECRET_KEY")}
  11. )