hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-10 11:26:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add hosted moderation (#1158)

Browse Source
This commit is contained in:
takatost
2023-09-12 10:26:12 +08:00
committed by GitHub
parent 983834cd52
commit f9082104ed
15 changed files with 240 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/tests/integration_tests/models/moderation/test_openai_moderation.py
View File

@@ -2,7 +2,7 @@ import json
import os
from unittest.mock import patch
from core.model_providers.models.moderation.openai_moderation import OpenAIModeration, DEFAULT_AUDIO_MODEL
from core.model_providers.models.moderation.openai_moderation import OpenAIModeration, DEFAULT_MODEL
from core.model_providers.providers.openai_provider import OpenAIProvider
from models.provider import Provider, ProviderType
@@ -23,7 +23,7 @@ def get_mock_openai_moderation_model():
openai_provider = OpenAIProvider(provider=get_mock_provider(valid_openai_api_key))
return OpenAIModeration(
model_provider=openai_provider,
name=DEFAULT_AUDIO_MODEL
name=DEFAULT_MODEL
)
@@ -36,5 +36,4 @@ def test_run(mock_decrypt):
model = get_mock_openai_moderation_model()
rst = model.run('hello')
assert isinstance(rst, dict)
assert 'id' in rst
assert rst is True