Implement get_share_links (#302)

skyvern/forge/sdk/artifact/manager.py

@@ -71,6 +71,9 @@ class ArtifactManager:
     async def get_share_link(self, artifact: Artifact) -> str | None:
         return await app.STORAGE.get_share_link(artifact)
 
+    async def get_share_links(self, artifacts: list[Artifact]) -> list[str] | None:
+        return await app.STORAGE.get_share_links(artifacts)
+
     async def wait_for_upload_aiotasks_for_task(self, task_id: str) -> None:
         try:
             st = time.time()

skyvern/forge/sdk/artifact/storage/base.py

@@ -40,6 +40,10 @@ class BaseStorage(ABC):
     async def get_share_link(self, artifact: Artifact) -> str | None:
         pass
 
+    @abstractmethod
+    async def get_share_links(self, artifacts: list[Artifact]) -> list[str] | None:
+        pass
+
     @abstractmethod
     async def store_artifact_from_path(self, artifact: Artifact, path: str) -> None:
         pass

skyvern/forge/sdk/artifact/storage/local.py

@@ -52,6 +52,9 @@ class LocalStorage(BaseStorage):
     async def get_share_link(self, artifact: Artifact) -> str:
         return artifact.uri
 
+    async def get_share_links(self, artifacts: list[Artifact]) -> list[str]:
+        return [artifact.uri for artifact in artifacts]
+
     @staticmethod
     def _parse_uri_to_path(uri: str) -> str:
         parsed_uri = urlparse(uri)