start ai-moderation
This commit is contained in:
@@ -0,0 +1,24 @@
|
||||
from fastapi import APIRouter
|
||||
|
||||
from app.models.dto.text_request import TextRequest
|
||||
from app.models.response.moderation_response import ModerationResponse
|
||||
from app.services.text_moderation_service import TextModerationService
|
||||
|
||||
router = APIRouter(
|
||||
prefix="/api/v1/moderation",
|
||||
tags=["Moderation"]
|
||||
)
|
||||
|
||||
service = TextModerationService()
|
||||
|
||||
|
||||
@router.post("/text", response_model=ModerationResponse)
|
||||
def moderate_text(request: TextRequest):
|
||||
|
||||
prediction = service.moderate(request.text)
|
||||
|
||||
return ModerationResponse(
|
||||
approved=prediction.approved,
|
||||
score=prediction.score,
|
||||
reason=prediction.reason
|
||||
)
|
||||
Reference in New Issue
Block a user