Apppublic

Underground-Digital/Workflow-Engine

sourceHugging Faceupdated 2y agoView on Hugging Face

0likes

rootapitestsintegration_testsmodel_runtimeopenaitest_provider.py

test_provider.py18 linesDownload Raw Back to openai

1import os2 3import pytest4 5from core.model_runtime.errors.validate import CredentialsValidateFailedError6from core.model_runtime.model_providers.openai.openai import OpenAIProvider7from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock8 9 10@pytest.mark.parametrize("setup_openai_mock", [["chat"]], indirect=True)11def test_validate_provider_credentials(setup_openai_mock):12    provider = OpenAIProvider()13 14    with pytest.raises(CredentialsValidateFailedError):15        provider.validate_provider_credentials(credentials={})16 17    provider.validate_provider_credentials(credentials={"openai_api_key": os.environ.get("OPENAI_API_KEY")})18