Apppublic

Underground-Digital/Workflow-Engine

sourceHugging Faceupdated 2y agoView on Hugging Face

0likes

rootapitestsintegration_testsmodel_runtimegoogletest_provider.py

test_provider.py18 linesDownload Raw Back to google

1import os2 3import pytest4 5from core.model_runtime.errors.validate import CredentialsValidateFailedError6from core.model_runtime.model_providers.google.google import GoogleProvider7from tests.integration_tests.model_runtime.__mock.google import setup_google_mock8 9 10@pytest.mark.parametrize("setup_google_mock", [["none"]], indirect=True)11def test_validate_provider_credentials(setup_google_mock):12    provider = GoogleProvider()13 14    with pytest.raises(CredentialsValidateFailedError):15        provider.validate_provider_credentials(credentials={})16 17    provider.validate_provider_credentials(credentials={"google_api_key": os.environ.get("GOOGLE_API_KEY")})18