init commit

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
2026-01-26 15:34:31 +08:00
commit 80513a3258
141 changed files with 24966 additions and 0 deletions

View File

View 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)

View 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]

View 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)

View 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