feat: Add VLMConfig to schema for proper config.yaml VLM support
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user