File size: 422 Bytes
acb544e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from typing import Dict

from pydantic import BaseModel


class AnalyzePromptRequest(BaseModel):
    prompt: str


class AnalyzePromptResponse(BaseModel):
    sanitized_prompt: str
    is_valid: bool
    scanners: Dict[str, float]


class AnalyzeOutputRequest(BaseModel):
    prompt: str
    output: str


class AnalyzeOutputResponse(BaseModel):
    sanitized_output: str
    is_valid: bool
    scanners: Dict[str, float]