Files
dify-plugin-sdks/python/dify_plugin/entities/model/moderation.py
2024-12-03 15:19:19 +08:00

22 lines
435 B
Python

from pydantic import BaseModel, ConfigDict
from dify_plugin.entities.model import BaseModelConfig, ModelType
class ModerationModelConfig(BaseModelConfig):
"""
Model class for moderation model config.
"""
model_type: ModelType = ModelType.MODERATION
model_config = ConfigDict(protected_namespaces=())
class ModerationResult(BaseModel):
"""
Model class for moderation result.
"""
result: bool