fix: preserve legacy model query ABI

This commit is contained in:
jxxghp
2026-08-23 01:08:01 +08:00
parent 1c457143f7
commit 5395bbff54
7 changed files with 117 additions and 25 deletions
+29 -11
View File
@@ -4,6 +4,7 @@ from sqlalchemy import Boolean, Index, Integer, String, Text, select, update
from sqlalchemy.orm import Mapped, Session, mapped_column
from app.db.base import Base, execute_dml, get_id_column
from app.db.decorators import run_legacy_sync_query
def _get_for_user_statement(
@@ -86,30 +87,47 @@ class AgentTask(Base):
@classmethod
def get_for_user(
cls,
db: Session,
task_id: int,
db: Session | int | None = None,
task_id: int | None = None,
user_id: Optional[str] = None,
) -> Optional["AgentTask"]:
"""
按任务 ID 和可选用户 ID 查询 Agent 定时任务
按任务 ID 和可选用户 ID 查询,并保留无 Session 的旧插件调用方式
"""
return db.execute(
_get_for_user_statement(cls, task_id=task_id, user_id=user_id)
).scalars().first()
if task_id is None and isinstance(db, int):
task_id, db = db, None
if task_id is None:
raise TypeError("task_id is required")
def query(session: Session) -> Optional["AgentTask"]:
"""在给定会话中读取单个 Agent 任务。"""
return session.execute(
_get_for_user_statement(cls, task_id=task_id, user_id=user_id)
).scalars().first()
if isinstance(db, Session):
return query(db)
return run_legacy_sync_query(query)
@classmethod
def list_for_user(
cls,
db: Session,
db: Session | None = None,
user_id: Optional[str] = None,
enabled: Optional[bool] = None,
) -> list["AgentTask"]:
"""
按用户和启用状态查询 Agent 定时任务
按用户和启用状态查询,并保留无 Session 的旧插件调用方式
"""
return list(db.execute(
_list_for_user_statement(cls, user_id=user_id, enabled=enabled)
).scalars().all())
def query(session: Session) -> list["AgentTask"]:
"""在给定会话中读取 Agent 任务列表。"""
return list(session.execute(
_list_for_user_statement(cls, user_id=user_id, enabled=enabled)
).scalars().all())
if isinstance(db, Session):
return query(db)
return run_legacy_sync_query(query)
@classmethod
def update_task(