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

make get signed_url requests async in get_agent_task_step_artifacts api (#192)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-04-15 12:27:58 -07:00
committed by GitHub
parent 6b7b9ab414
commit bd485daaf4
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
skyvern/forge/sdk/routes/agent_protocol.py
View File

@@ -1,3 +1,4 @@
import asyncio
from typing import Annotated, Any
import structlog
@@ -400,8 +401,9 @@ async def get_agent_task_step_artifacts(
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)
signed_urls = await asyncio.gather(*[app.ARTIFACT_MANAGER.get_share_link(artifact) for artifact in artifacts])
for i, artifact in enumerate(artifacts):
artifact.signed_url = signed_urls[i]
return ORJSONResponse([artifact.model_dump() for artifact in artifacts])