Apppublic

Underground-Digital/Workflow-Engine

sourceHugging Faceupdated 2y agoView on Hugging Face

0likes

rootapitestsintegration_testsmodel_runtimeanthropictest_provider.py

test_provider.py18 linesDownload Raw Back to anthropic

1import os2 3import pytest4 5from core.model_runtime.errors.validate import CredentialsValidateFailedError6from core.model_runtime.model_providers.anthropic.anthropic import AnthropicProvider7from tests.integration_tests.model_runtime.__mock.anthropic import setup_anthropic_mock8 9 10@pytest.mark.parametrize("setup_anthropic_mock", [["none"]], indirect=True)11def test_validate_provider_credentials(setup_anthropic_mock):12    provider = AnthropicProvider()13 14    with pytest.raises(CredentialsValidateFailedError):15        provider.validate_provider_credentials(credentials={})16 17    provider.validate_provider_credentials(credentials={"anthropic_api_key": os.environ.get("ANTHROPIC_API_KEY")})18