File size: 631 Bytes
7c7af61 |
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 26 27 |
from pydantic import BaseModel
from typing import List, Dict, Optional
from enum import Enum
class ChatRequest(BaseModel):
message: str
temperature: float = 0.7
max_new_tokens: int = 512
history: Optional[List[Dict]] = None
format: Optional[str] = "clean"
class VoiceInputRequest(BaseModel):
audio_format: str = "wav"
language: str = "en-US"
class VoiceOutputRequest(BaseModel):
text: str
language: str = "en"
slow: bool = False
return_format: str = "mp3"
class RiskLevel(str, Enum):
NONE = "none"
LOW = "low"
MODERATE = "moderate"
HIGH = "high"
SEVERE = "severe" |