fix: 修改LLMProvider 类型

This commit is contained in:
XIE7654
2025-08-11 10:26:42 +08:00
parent e21a1ac716
commit a88f272c19
7 changed files with 54 additions and 12 deletions

View File

@@ -1,4 +1,5 @@
from pydantic import BaseModel
from llm.enums import LLMProvider
class CreateDrawingTaskRequest(BaseModel):
@@ -6,5 +7,5 @@ class CreateDrawingTaskRequest(BaseModel):
style: str = 'auto'
size: str = '1024*1024'
model: str = 'wanx_v1'
platform: str = 'tongyi'
platform: str = LLMProvider.TONGYI
n: int = 1