Underground-Digital/Workflow-Engine

1import os2 3import pytest4 5from core.model_runtime.errors.validate import CredentialsValidateFailedError6from core.model_runtime.model_providers.openai.moderation.moderation import OpenAIModerationModel7from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock8 9 10@pytest.mark.parametrize("setup_openai_mock", [["moderation"]], indirect=True)11def test_validate_credentials(setup_openai_mock):12    model = OpenAIModerationModel()13 14    with pytest.raises(CredentialsValidateFailedError):15        model.validate_credentials(model="text-moderation-stable", credentials={"openai_api_key": "invalid_key"})16 17    model.validate_credentials(18        model="text-moderation-stable", credentials={"openai_api_key": os.environ.get("OPENAI_API_KEY")}19    )20 21 22@pytest.mark.parametrize("setup_openai_mock", [["moderation"]], indirect=True)23def test_invoke_model(setup_openai_mock):24    model = OpenAIModerationModel()25 26    result = model.invoke(27        model="text-moderation-stable",28        credentials={"openai_api_key": os.environ.get("OPENAI_API_KEY")},29        text="hello",30        user="abc-123",31    )32 33    assert isinstance(result, bool)34    assert result is False35 36    result = model.invoke(37        model="text-moderation-stable",38        credentials={"openai_api_key": os.environ.get("OPENAI_API_KEY")},39        text="i will kill you",40        user="abc-123",41    )42 43    assert isinstance(result, bool)44    assert result is True45