vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2026-04-28 11:40:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: add enriched script version detail endpoint (#5111)

Browse source
This commit is contained in:
pedrohsdb 2026-03-15 12:12:23 -07:00 • committed by GitHub
parent 7427607896
commit f2062b65b3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 123 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

19
skyvern/forge/sdk/db/agent_db.py
View file

@ -6816,12 +6816,24 @@ class AgentDB(BaseAlchemyDB):
async def get_fallback_episodes_count(
self,
organization_id: str,
workflow_permanent_id: str,
workflow_permanent_id: str | None = None,
workflow_run_id: str | None = None,
block_label: str | None = None,
reviewed: bool | None = None,
fallback_type: str | None = None,
script_revision_id: str | None = None,
) -> int:
"""Count fallback episodes matching the given filters.
At least one scoping filter (workflow_permanent_id, workflow_run_id,
or script_revision_id) should be provided. Without any, this returns
the total count for the entire organization which is rarely intended.
"""
if workflow_permanent_id is None and workflow_run_id is None and script_revision_id is None:
LOG.warning(
"get_fallback_episodes_count called without any scoping filter",
organization_id=organization_id,
)
try:
async with self.Session() as session:
query = (
@ -6829,9 +6841,10 @@ class AgentDB(BaseAlchemyDB):
.select_from(ScriptFallbackEpisodeModel)
.filter(
ScriptFallbackEpisodeModel.organization_id == organization_id,
ScriptFallbackEpisodeModel.workflow_permanent_id == workflow_permanent_id,
)
)
if workflow_permanent_id is not None:
query = query.filter(ScriptFallbackEpisodeModel.workflow_permanent_id == workflow_permanent_id)
if workflow_run_id is not None:
query = query.filter(ScriptFallbackEpisodeModel.workflow_run_id == workflow_run_id)
if block_label is not None:
@ -6840,6 +6853,8 @@ class AgentDB(BaseAlchemyDB):
query = query.filter(ScriptFallbackEpisodeModel.reviewed == reviewed)
if fallback_type is not None:
query = query.filter(ScriptFallbackEpisodeModel.fallback_type == fallback_type)
if script_revision_id is not None:
query = query.filter(ScriptFallbackEpisodeModel.script_revision_id == script_revision_id)
result = await session.scalar(query)
return result or 0