vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

join query performance improvement - add organization_id to the workflow_scripts & scripts table join (#4429)

Browse Source
This commit is contained in:
Shuchang Zheng
2026-01-09 20:01:28 -08:00
committed by GitHub
parent dad3590d7a
commit 6cd5d90f55
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
skyvern/forge/sdk/db/agent_db.py
View File

@@ -5806,9 +5806,16 @@ class AgentDB(BaseAlchemyDB):
try: try:
async with self.Session() as session: async with self.Session() as session:
# Build the query: join workflow_scripts with scripts # Build the query: join workflow_scripts with scripts
# Join on both script_id and organization_id to leverage uc_org_script_version index
query = ( query = (
select(ScriptModel) select(ScriptModel)
.join(WorkflowScriptModel, ScriptModel.script_id == WorkflowScriptModel.script_id) .join(
WorkflowScriptModel,
and_(
ScriptModel.organization_id == WorkflowScriptModel.organization_id,
ScriptModel.script_id == WorkflowScriptModel.script_id,
),
)
.where( .where(
WorkflowScriptModel.organization_id == organization_id, WorkflowScriptModel.organization_id == organization_id,
WorkflowScriptModel.workflow_permanent_id == workflow_permanent_id, WorkflowScriptModel.workflow_permanent_id == workflow_permanent_id,