Apppublic

Underground-Digital/Workflow-Engine

sourceHugging Faceupdated 2y agoView on Hugging Face

0likes

rootapitestsintegration_testsmodel_runtimezhinaotest_provider.py

test_provider.py16 linesDownload Raw Back to zhinao

1import os2 3import pytest4 5from core.model_runtime.errors.validate import CredentialsValidateFailedError6from core.model_runtime.model_providers.zhinao.zhinao import ZhinaoProvider7 8 9def test_validate_provider_credentials():10    provider = ZhinaoProvider()11 12    with pytest.raises(CredentialsValidateFailedError):13        provider.validate_provider_credentials(credentials={})14 15    provider.validate_provider_credentials(credentials={"api_key": os.environ.get("ZHINAO_API_KEY")})16