refactor: isolate agent history queries

This commit is contained in:
jxxghp
2026-08-23 14:21:53 +08:00
parent d8b8583d49
commit 8fc9a021b0
11 changed files with 121 additions and 61 deletions
+3 -3
View File
@@ -4,7 +4,7 @@ from sqlalchemy import Index, Integer, String, Text, delete, 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 db_query
from app.db.decorators import legacy_db_query
from app.db.models.agenttask import AgentTask
@@ -249,7 +249,7 @@ class AgentTaskRun(Base):
return True
@classmethod
@db_query
@legacy_db_query
def get_by_run_id(
cls,
db: Session,
@@ -261,7 +261,7 @@ class AgentTaskRun(Base):
).scalars().first()
@classmethod
@db_query
@legacy_db_query
def list_for_task(
cls,
db: Session,