feat: Add VLMConfig to schema for proper config.yaml VLM support

This commit is contained in:
empty
2026-01-07 09:38:45 +08:00
parent bc077475c6
commit f19804facb
3 changed files with 19 additions and 8 deletions

View File

@@ -26,6 +26,14 @@ class LLMConfig(BaseModel):
model: str = Field(default="", description="LLM Model Name")
class VLMConfig(BaseModel):
"""VLM (Vision Language Model) configuration for character analysis"""
provider: str = Field(default="qwen", description="VLM provider: qwen, glm, openai")
api_key: str = Field(default="", description="VLM API Key")
base_url: str = Field(default="", description="VLM API Base URL (auto-detected if empty)")
model: str = Field(default="", description="VLM Model Name (defaults based on provider)")
class TTSLocalConfig(BaseModel):
"""Local TTS configuration (Edge TTS)"""
voice: str = Field(default="zh-CN-YunjianNeural", description="Edge TTS voice ID")
@@ -92,6 +100,7 @@ class PixelleVideoConfig(BaseModel):
"""Pixelle-Video main configuration"""
project_name: str = Field(default="Pixelle-Video", description="Project name")
llm: LLMConfig = Field(default_factory=LLMConfig)
vlm: VLMConfig = Field(default_factory=VLMConfig)
comfyui: ComfyUIConfig = Field(default_factory=ComfyUIConfig)
template: TemplateConfig = Field(default_factory=TemplateConfig)