mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-04-22 21:59:55 +08:00
16 lines
392 B
Python
16 lines
392 B
Python
import openai
|
|
from models.provider import ProviderName
|
|
|
|
|
|
class Moderation:
|
|
|
|
def __init__(self, provider: str, api_key: str):
|
|
self.provider = provider
|
|
self.api_key = api_key
|
|
|
|
if self.provider == ProviderName.OPENAI.value:
|
|
self.client = openai.Moderation
|
|
|
|
def moderate(self, text):
|
|
return self.client.create(input=text, api_key=self.api_key)
|