From 0b8f64e8ab4a66b79da9ea3bdd06c5a9bf332ef9 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Tue, 4 Mar 2025 03:22:16 -0500 Subject: [PATCH] get_total_unique_step_order_count_by_task_ids fix again (#1881) --- skyvern/forge/sdk/db/client.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index 75b3fecd..b0db6f8e 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -3,7 +3,7 @@ from datetime import datetime, timedelta from typing import Any, List, Optional, Sequence import structlog -from sqlalchemy import and_, delete, distinct, func, select, update +from sqlalchemy import and_, delete, distinct, func, select, tuple_, update from sqlalchemy.exc import SQLAlchemyError from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine @@ -356,11 +356,11 @@ class AgentDB: try: async with self.Session() as session: query = ( - select(func.count(distinct(StepModel.task_id, StepModel.order))) + select(func.count(distinct(tuple_(StepModel.task_id, StepModel.order)))) .where(StepModel.task_id.in_(task_ids)) - .filter_by(organization_id=organization_id) + .where(StepModel.organization_id == organization_id) ) - return (await session.scalars(query)).scalar() + return (await session.execute(query)).scalar() except SQLAlchemyError: LOG.error("SQLAlchemyError", exc_info=True) raise