feat: [Artifacts] public API file artifact downloads

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 14
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta%2Fwarp-api-d57dae004e23149ddae4bd26e3ba05c2378d2c1070df0f303b19e45111c780d5.yml
-openapi_spec_hash: 6a15b456e6520a79981fafed3b0c94b9
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta%2Fwarp-api-a29592b2ba26cba9d89b95969d66506f49c08e140b76ce4aea4189e5c1dccc06.yml
+openapi_spec_hash: 27a5de1f891104d5e47904ad8e4b4bd1
 config_hash: 40327fb76b7cce7b97f23de9b8d48efb

src/oz_agent_sdk/resources/agent/agent.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Iterable
+from typing import Any, Iterable, cast
 from typing_extensions import Literal
 
 import httpx
@@ -163,8 +163,8 @@ def get_artifact(
     ) -> AgentGetArtifactResponse:
         """Retrieve an artifact by its UUID.
 
-        For screenshot artifacts, returns a
-        time-limited signed download URL.
+        For supported downloadable artifacts, returns
+        a time-limited signed download URL.
 
         Args:
           extra_headers: Send extra headers
@@ -177,12 +177,17 @@ def get_artifact(
         """
         if not artifact_uid:
             raise ValueError(f"Expected a non-empty value for `artifact_uid` but received {artifact_uid!r}")
-        return self._get(
-            path_template("/agent/artifacts/{artifact_uid}", artifact_uid=artifact_uid),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+        return cast(
+            AgentGetArtifactResponse,
+            self._get(
+                path_template("/agent/artifacts/{artifact_uid}", artifact_uid=artifact_uid),
+                options=make_request_options(
+                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                ),
+                cast_to=cast(
+                    Any, AgentGetArtifactResponse
+                ),  # Union types cannot be passed in as arguments in the type system
             ),
-            cast_to=AgentGetArtifactResponse,
         )
 
     def run(
@@ -382,8 +387,8 @@ async def get_artifact(
     ) -> AgentGetArtifactResponse:
         """Retrieve an artifact by its UUID.
 
-        For screenshot artifacts, returns a
-        time-limited signed download URL.
+        For supported downloadable artifacts, returns
+        a time-limited signed download URL.
 
         Args:
           extra_headers: Send extra headers
@@ -396,12 +401,17 @@ async def get_artifact(
         """
         if not artifact_uid:
             raise ValueError(f"Expected a non-empty value for `artifact_uid` but received {artifact_uid!r}")
-        return await self._get(
-            path_template("/agent/artifacts/{artifact_uid}", artifact_uid=artifact_uid),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+        return cast(
+            AgentGetArtifactResponse,
+            await self._get(
+                path_template("/agent/artifacts/{artifact_uid}", artifact_uid=artifact_uid),
+                options=make_request_options(
+                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                ),
+                cast_to=cast(
+                    Any, AgentGetArtifactResponse
+                ),  # Union types cannot be passed in as arguments in the type system
             ),
-            cast_to=AgentGetArtifactResponse,
         )
 
     async def run(

src/oz_agent_sdk/resources/agent/runs.py

@@ -87,7 +87,7 @@ def retrieve(
     def list(
         self,
         *,
-        artifact_type: Literal["PLAN", "PULL_REQUEST", "SCREENSHOT"] | Omit = omit,
+        artifact_type: Literal["PLAN", "PULL_REQUEST", "SCREENSHOT", "FILE"] | Omit = omit,
         created_after: Union[str, datetime] | Omit = omit,
         created_before: Union[str, datetime] | Omit = omit,
         creator: str | Omit = omit,
@@ -307,7 +307,7 @@ async def retrieve(
     def list(
         self,
         *,
-        artifact_type: Literal["PLAN", "PULL_REQUEST", "SCREENSHOT"] | Omit = omit,
+        artifact_type: Literal["PLAN", "PULL_REQUEST", "SCREENSHOT", "FILE"] | Omit = omit,
         created_after: Union[str, datetime] | Omit = omit,
         created_before: Union[str, datetime] | Omit = omit,
         creator: str | Omit = omit,

src/oz_agent_sdk/types/agent/artifact_item.py

@@ -15,6 +15,8 @@
     "PullRequestArtifactData",
     "ScreenshotArtifact",
     "ScreenshotArtifactData",
+    "FileArtifact",
+    "FileArtifactData",
 ]
 
 
@@ -78,6 +80,37 @@ class ScreenshotArtifact(BaseModel):
     data: ScreenshotArtifactData
 
 
+class FileArtifactData(BaseModel):
+    artifact_uid: str
+    """Unique identifier for the file artifact"""
+
+    filename: str
+    """Last path component of filepath"""
+
+    filepath: str
+    """Conversation-relative filepath for the uploaded file"""
+
+    mime_type: str
+    """MIME type of the uploaded file"""
+
+    description: Optional[str] = None
+    """Optional description of the file"""
+
+    size_bytes: Optional[int] = None
+    """Size of the uploaded file in bytes"""
+
+
+class FileArtifact(BaseModel):
+    artifact_type: Literal["FILE"]
+    """Type of the artifact"""
+
+    created_at: datetime
+    """Timestamp when the artifact was created (RFC3339)"""
+
+    data: FileArtifactData
+
+
 ArtifactItem: TypeAlias = Annotated[
-    Union[PlanArtifact, PullRequestArtifact, ScreenshotArtifact], PropertyInfo(discriminator="artifact_type")
+    Union[PlanArtifact, PullRequestArtifact, ScreenshotArtifact, FileArtifact],
+    PropertyInfo(discriminator="artifact_type"),
 ]

src/oz_agent_sdk/types/agent/run_list_params.py

@@ -14,7 +14,7 @@
 
 
 class RunListParams(TypedDict, total=False):
-    artifact_type: Literal["PLAN", "PULL_REQUEST", "SCREENSHOT"]
+    artifact_type: Literal["PLAN", "PULL_REQUEST", "SCREENSHOT", "FILE"]
     """Filter runs by artifact type"""
 
     created_after: Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]

src/oz_agent_sdk/types/agent_get_artifact_response.py

@@ -1,14 +1,22 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Optional
+from typing import Union, Optional
 from datetime import datetime
+from typing_extensions import Literal, Annotated, TypeAlias
 
+from .._utils import PropertyInfo
 from .._models import BaseModel
 
-__all__ = ["AgentGetArtifactResponse", "Data"]
+__all__ = [
+    "AgentGetArtifactResponse",
+    "ScreenshotArtifactResponse",
+    "ScreenshotArtifactResponseData",
+    "FileArtifactResponse",
+    "FileArtifactResponseData",
+]
 
 
-class Data(BaseModel):
+class ScreenshotArtifactResponseData(BaseModel):
     """Response data for a screenshot artifact, including a signed download URL."""
 
     content_type: str
@@ -24,17 +32,63 @@ class Data(BaseModel):
     """Optional description of the screenshot"""
 
 
-class AgentGetArtifactResponse(BaseModel):
-    """Response for artifact retrieval. Currently supports screenshot artifacts."""
+class ScreenshotArtifactResponse(BaseModel):
+    """Response for retrieving a screenshot artifact."""
 
-    artifact_type: str
-    """Type of the artifact (e.g., SCREENSHOT)"""
+    artifact_type: Literal["SCREENSHOT"]
+    """Type of the artifact"""
 
     artifact_uid: str
     """Unique identifier (UUID) for the artifact"""
 
     created_at: datetime
     """Timestamp when the artifact was created (RFC3339)"""
 
-    data: Data
+    data: ScreenshotArtifactResponseData
     """Response data for a screenshot artifact, including a signed download URL."""
+
+
+class FileArtifactResponseData(BaseModel):
+    """Response data for a file artifact, including a signed download URL."""
+
+    content_type: str
+    """MIME type of the uploaded file"""
+
+    download_url: str
+    """Time-limited signed URL to download the file"""
+
+    expires_at: datetime
+    """Timestamp when the download URL expires (RFC3339)"""
+
+    filename: str
+    """Last path component of filepath"""
+
+    filepath: str
+    """Conversation-relative filepath for the uploaded file"""
+
+    description: Optional[str] = None
+    """Optional description of the file"""
+
+    size_bytes: Optional[int] = None
+    """Size of the uploaded file in bytes"""
+
+
+class FileArtifactResponse(BaseModel):
+    """Response for retrieving a file artifact."""
+
+    artifact_type: Literal["FILE"]
+    """Type of the artifact"""
+
+    artifact_uid: str
+    """Unique identifier (UUID) for the artifact"""
+
+    created_at: datetime
+    """Timestamp when the artifact was created (RFC3339)"""
+
+    data: FileArtifactResponseData
+    """Response data for a file artifact, including a signed download URL."""
+
+
+AgentGetArtifactResponse: TypeAlias = Annotated[
+    Union[ScreenshotArtifactResponse, FileArtifactResponse], PropertyInfo(discriminator="artifact_type")
+]