0
qwen3-tts-backend/schemas/__init__.py
Normal file
0
qwen3-tts-backend/schemas/__init__.py
Normal file
15
qwen3-tts-backend/schemas/cache.py
Normal file
15
qwen3-tts-backend/schemas/cache.py
Normal file
@@ -0,0 +1,15 @@
|
||||
from datetime import datetime
|
||||
from typing import Optional, Dict, Any
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
|
||||
class CacheEntry(BaseModel):
|
||||
id: int
|
||||
user_id: int
|
||||
ref_audio_hash: str
|
||||
cache_path: str
|
||||
meta_data: Optional[Dict[str, Any]] = None
|
||||
created_at: datetime
|
||||
last_accessed: datetime
|
||||
access_count: int
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
25
qwen3-tts-backend/schemas/job.py
Normal file
25
qwen3-tts-backend/schemas/job.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from datetime import datetime
|
||||
from typing import Optional, Dict, Any, List
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
|
||||
class JobBase(BaseModel):
|
||||
job_type: str
|
||||
|
||||
class JobCreate(JobBase):
|
||||
input_data: Dict[str, Any]
|
||||
|
||||
class Job(JobBase):
|
||||
id: int
|
||||
user_id: int
|
||||
status: str
|
||||
output_path: Optional[str] = None
|
||||
download_url: Optional[str] = None
|
||||
error_message: Optional[str] = None
|
||||
created_at: datetime
|
||||
completed_at: Optional[datetime] = None
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
class JobList(BaseModel):
|
||||
total: int
|
||||
jobs: List[Job]
|
||||
50
qwen3-tts-backend/schemas/tts.py
Normal file
50
qwen3-tts-backend/schemas/tts.py
Normal file
@@ -0,0 +1,50 @@
|
||||
from typing import Optional, List
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
class TTSRequest(BaseModel):
|
||||
text: str = Field(..., min_length=1, max_length=1000)
|
||||
ref_audio: Optional[str] = None
|
||||
ref_text: Optional[str] = None
|
||||
language: str = Field(default="en")
|
||||
speed: float = Field(default=1.0, ge=0.5, le=2.0)
|
||||
|
||||
class TTSResponse(BaseModel):
|
||||
job_id: int
|
||||
status: str
|
||||
audio_url: Optional[str] = None
|
||||
|
||||
|
||||
class CustomVoiceRequest(BaseModel):
|
||||
text: str = Field(..., min_length=1, max_length=1000)
|
||||
language: str = Field(default="Auto")
|
||||
speaker: str
|
||||
instruct: Optional[str] = Field(default="")
|
||||
max_new_tokens: Optional[int] = Field(default=2048, ge=128, le=4096)
|
||||
temperature: Optional[float] = Field(default=0.9, ge=0.1, le=2.0)
|
||||
top_k: Optional[int] = Field(default=50, ge=1, le=100)
|
||||
top_p: Optional[float] = Field(default=1.0, ge=0.0, le=1.0)
|
||||
repetition_penalty: Optional[float] = Field(default=1.05, ge=1.0, le=2.0)
|
||||
|
||||
|
||||
class VoiceDesignRequest(BaseModel):
|
||||
text: str = Field(..., min_length=1, max_length=1000)
|
||||
language: str = Field(default="Auto")
|
||||
instruct: str = Field(..., min_length=1)
|
||||
max_new_tokens: Optional[int] = Field(default=2048, ge=128, le=4096)
|
||||
temperature: Optional[float] = Field(default=0.9, ge=0.1, le=2.0)
|
||||
top_k: Optional[int] = Field(default=50, ge=1, le=100)
|
||||
top_p: Optional[float] = Field(default=1.0, ge=0.0, le=1.0)
|
||||
repetition_penalty: Optional[float] = Field(default=1.05, ge=1.0, le=2.0)
|
||||
|
||||
|
||||
class VoiceCloneRequest(BaseModel):
|
||||
text: str = Field(..., min_length=1, max_length=1000)
|
||||
language: str = Field(default="Auto")
|
||||
ref_text: Optional[str] = Field(default=None, max_length=500)
|
||||
use_cache: bool = Field(default=True)
|
||||
x_vector_only_mode: bool = Field(default=False)
|
||||
max_new_tokens: Optional[int] = Field(default=2048, ge=128, le=4096)
|
||||
temperature: Optional[float] = Field(default=0.9, ge=0.1, le=2.0)
|
||||
top_k: Optional[int] = Field(default=50, ge=1, le=100)
|
||||
top_p: Optional[float] = Field(default=1.0, ge=0.0, le=1.0)
|
||||
repetition_penalty: Optional[float] = Field(default=1.05, ge=1.0, le=2.0)
|
||||
91
qwen3-tts-backend/schemas/user.py
Normal file
91
qwen3-tts-backend/schemas/user.py
Normal file
@@ -0,0 +1,91 @@
|
||||
from datetime import datetime
|
||||
from typing import Optional
|
||||
from pydantic import BaseModel, EmailStr, Field, field_validator, ConfigDict
|
||||
import re
|
||||
|
||||
class UserBase(BaseModel):
|
||||
username: str = Field(..., min_length=3, max_length=50)
|
||||
email: EmailStr
|
||||
|
||||
@field_validator('username')
|
||||
@classmethod
|
||||
def validate_username(cls, v: str) -> str:
|
||||
if not re.match(r'^[a-zA-Z0-9_-]+$', v):
|
||||
raise ValueError('Username must contain only alphanumeric characters, underscores, and dashes')
|
||||
return v
|
||||
|
||||
class UserCreate(UserBase):
|
||||
password: str = Field(..., min_length=8, max_length=128)
|
||||
|
||||
@field_validator('password')
|
||||
@classmethod
|
||||
def validate_password_strength(cls, v: str) -> str:
|
||||
if not re.search(r'[A-Z]', v):
|
||||
raise ValueError('Password must contain at least one uppercase letter')
|
||||
if not re.search(r'[a-z]', v):
|
||||
raise ValueError('Password must contain at least one lowercase letter')
|
||||
if not re.search(r'\d', v):
|
||||
raise ValueError('Password must contain at least one digit')
|
||||
return v
|
||||
|
||||
class User(UserBase):
|
||||
id: int
|
||||
is_active: bool
|
||||
is_superuser: bool
|
||||
created_at: datetime
|
||||
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
class UserCreateByAdmin(UserBase):
|
||||
password: str = Field(..., min_length=8, max_length=128)
|
||||
is_superuser: bool = False
|
||||
|
||||
@field_validator('password')
|
||||
@classmethod
|
||||
def validate_password_strength(cls, v: str) -> str:
|
||||
if not re.search(r'[A-Z]', v):
|
||||
raise ValueError('Password must contain at least one uppercase letter')
|
||||
if not re.search(r'[a-z]', v):
|
||||
raise ValueError('Password must contain at least one lowercase letter')
|
||||
if not re.search(r'\d', v):
|
||||
raise ValueError('Password must contain at least one digit')
|
||||
return v
|
||||
|
||||
class UserUpdate(BaseModel):
|
||||
username: Optional[str] = Field(None, min_length=3, max_length=50)
|
||||
email: Optional[EmailStr] = None
|
||||
password: Optional[str] = Field(None, min_length=8, max_length=128)
|
||||
is_active: Optional[bool] = None
|
||||
is_superuser: Optional[bool] = None
|
||||
|
||||
@field_validator('username')
|
||||
@classmethod
|
||||
def validate_username(cls, v: Optional[str]) -> Optional[str]:
|
||||
if v is not None and not re.match(r'^[a-zA-Z0-9_-]+$', v):
|
||||
raise ValueError('Username must contain only alphanumeric characters, underscores, and dashes')
|
||||
return v
|
||||
|
||||
@field_validator('password')
|
||||
@classmethod
|
||||
def validate_password_strength(cls, v: Optional[str]) -> Optional[str]:
|
||||
if v is not None:
|
||||
if not re.search(r'[A-Z]', v):
|
||||
raise ValueError('Password must contain at least one uppercase letter')
|
||||
if not re.search(r'[a-z]', v):
|
||||
raise ValueError('Password must contain at least one lowercase letter')
|
||||
if not re.search(r'\d', v):
|
||||
raise ValueError('Password must contain at least one digit')
|
||||
return v
|
||||
|
||||
class UserListResponse(BaseModel):
|
||||
users: list[User]
|
||||
total: int
|
||||
skip: int
|
||||
limit: int
|
||||
|
||||
class Token(BaseModel):
|
||||
access_token: str
|
||||
token_type: str
|
||||
|
||||
class TokenData(BaseModel):
|
||||
username: Optional[str] = None
|
||||
Reference in New Issue
Block a user