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

@@ -7,6 +7,7 @@ from http import HTTPStatus
from sqlalchemy import desc
from llm.factory import get_adapter
from llm.enums import LLMProvider
from models.ai import Drawing
from sqlalchemy.orm import Session
@@ -40,7 +41,7 @@ def fetch_drawing_task_status(db: Session, drawing_id: int):
return None, "任务不存在"
if drawing.status in ("PENDING", 'RUNNING'):
api_key = os.getenv("DASHSCOPE_API_KEY")
adapter = get_adapter('tongyi', api_key=api_key, model='')
adapter = get_adapter(LLMProvider.TONGYI, api_key=api_key, model='')
rsp = adapter.fetch_drawing_task_status(drawing.task_id)
if rsp['status_code'] == HTTPStatus.OK:
# 可根据 status.output.task_status 更新数据库