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

@@ -8,6 +8,7 @@ from api.v1.drawing.vo import CreateDrawingTaskRequest
from db.session import get_db
from deps.auth import get_current_user
from llm.factory import get_adapter
from llm.enums import LLMProvider
from services.drawing_service import get_drawing_page, create_drawing_task, fetch_drawing_task_status
from utils.resp import resp_error, resp_success
@@ -50,7 +51,7 @@ def api_create_image_task(
prompt = req.prompt
model = req.model
api_key = os.getenv("DASHSCOPE_API_KEY")
adapter = get_adapter('tongyi', api_key=api_key, model=model)
adapter = get_adapter(LLMProvider.TONGYI, api_key=api_key, model=model)
try:
rsp = adapter.create_drawing_task(prompt=prompt, n=n, style=style, size=size)
# rsp = {

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