mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 18:35:29 -04:00
-LAN-
23 lines
696 B
Python
23 lines
696 B
Python
from collections.abc import Generator
|
|
|
|
from dify_plugin.entities.model.moderation import ModerationModelConfig
|
|
from dify_plugin.entities.tool import ToolInvokeMessage
|
|
from dify_plugin.interfaces.tool import Tool
|
|
|
|
|
|
class Moderation(Tool):
|
|
def _invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None, None]:
|
|
response = self.session.model.moderation.invoke(
|
|
model_config=ModerationModelConfig(
|
|
provider="openai",
|
|
model="text-moderation-stable",
|
|
),
|
|
text=tool_parameters["text"],
|
|
)
|
|
|
|
yield self.create_json_message(
|
|
{
|
|
"data": response,
|
|
}
|
|
)
|