From 7d5f0c8499b91c4d0bc6f535cb4d4d79f0a72d88 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sun, 14 Apr 2024 21:52:50 -0700 Subject: [PATCH] artifact signed url (#190) --- skyvern/forge/sdk/artifact/models.py | 1 + skyvern/forge/sdk/routes/agent_protocol.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/skyvern/forge/sdk/artifact/models.py b/skyvern/forge/sdk/artifact/models.py index 510b06bd..aa00c87f 100644 --- a/skyvern/forge/sdk/artifact/models.py +++ b/skyvern/forge/sdk/artifact/models.py @@ -75,4 +75,5 @@ class Artifact(BaseModel): description="The URI of the artifact.", examples=["/Users/skyvern/hello/world.png"], ) + signed_url: str | None = None organization_id: str | None = None diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index 577df4dd..914b37c9 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -399,6 +399,9 @@ async def get_agent_task_step_artifacts( step_id, organization_id=current_org.organization_id, ) + if SettingsManager.get_settings().ENV != "local": + for artifact in artifacts: + artifact.signed_url = await app.ARTIFACT_MANAGER.get_share_link(artifact) return ORJSONResponse([artifact.model_dump() for artifact in artifacts])