Apppublic

Underground-Digital/Workflow-Engine

sourceHugging Faceupdated 2y agoView on Hugging Face

0likes

rootapitestsintegration_testsmodel_runtimechatglmtest_provider.py

test_provider.py18 linesDownload Raw Back to chatglm

1import os2 3import pytest4 5from core.model_runtime.errors.validate import CredentialsValidateFailedError6from core.model_runtime.model_providers.chatglm.chatglm import ChatGLMProvider7from 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 = ChatGLMProvider()13 14    with pytest.raises(CredentialsValidateFailedError):15        provider.validate_provider_credentials(credentials={"api_base": "hahahaha"})16 17    provider.validate_provider_credentials(credentials={"api_base": os.environ.get("CHATGLM_API_BASE")})18