org id not None (#2696)

2025-06-13 10:44:04 -04:00
parent 296d2f903b
commit 687f3ec547
3 changed files with 13 additions and 15 deletions
--- a/skyvern/forge/sdk/routes/agent_protocol.py
+++ b/skyvern/forge/sdk/routes/agent_protocol.py
@@ -1135,13 +1135,10 @@ async def get_artifacts(
        )

    analytics.capture("skyvern-oss-agent-entity-artifacts-get")
-
    params = {
-        "organization_id": current_org.organization_id,
        entity_type_to_param[entity_type]: entity_id,
    }
-
-    artifacts = await app.DATABASE.get_artifacts_by_entity_id(**params)  # type: ignore
+    artifacts = await app.DATABASE.get_artifacts_by_entity_id(organization_id=current_org.organization_id, **params)  # type: ignore

    if settings.ENV != "local" or settings.GENERATE_PRESIGNED_URLS:
        signed_urls = await app.ARTIFACT_MANAGER.get_share_links(artifacts)