feat(api): add schedules to runs

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 12
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta%2Fwarp-api-d3907596f0e1b561fd86deb46529fca2a4802a992a35e2a9c51440451f4e9811.yml
-openapi_spec_hash: 5636d38917d7664a237f4096c1338304
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta%2Fwarp-api-71da0fb8e959241d9767c5a4879871711b501bd0d6cef087fbb7047a0b86791e.yml
+openapi_spec_hash: 8b0828210e5b33f36fc88093dbf066f1
 config_hash: 07820b17df23cbea39cb77fa05292538

src/warp_agent_sdk/resources/agent/runs.py

@@ -92,11 +92,8 @@ def list(
         environment_id: str | Omit = omit,
         limit: int | Omit = omit,
         model_id: str | Omit = omit,
-        schedule_id: str | Omit = omit,
-        skill_spec: str | Omit = omit,
         source: RunSourceType | Omit = omit,
         state: List[RunState] | Omit = omit,
-        updated_after: Union[str, datetime] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -126,17 +123,11 @@ def list(
 
           model_id: Filter by model ID
 
-          schedule_id: Filter runs by the scheduled agent ID that created them
-
-          skill_spec: Filter runs by skill spec (e.g., "owner/repo:path/to/SKILL.md")
-
           source: Filter by run source type
 
           state: Filter by run state. Can be specified multiple times to match any of the given
               states.
 
-          updated_after: Filter runs updated after this timestamp (RFC3339 format)
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -162,11 +153,8 @@ def list(
                         "environment_id": environment_id,
                         "limit": limit,
                         "model_id": model_id,
-                        "schedule_id": schedule_id,
-                        "skill_spec": skill_spec,
                         "source": source,
                         "state": state,
-                        "updated_after": updated_after,
                     },
                     run_list_params.RunListParams,
                 ),
@@ -275,11 +263,8 @@ async def list(
         environment_id: str | Omit = omit,
         limit: int | Omit = omit,
         model_id: str | Omit = omit,
-        schedule_id: str | Omit = omit,
-        skill_spec: str | Omit = omit,
         source: RunSourceType | Omit = omit,
         state: List[RunState] | Omit = omit,
-        updated_after: Union[str, datetime] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -309,17 +294,11 @@ async def list(
 
           model_id: Filter by model ID
 
-          schedule_id: Filter runs by the scheduled agent ID that created them
-
-          skill_spec: Filter runs by skill spec (e.g., "owner/repo:path/to/SKILL.md")
-
           source: Filter by run source type
 
           state: Filter by run state. Can be specified multiple times to match any of the given
               states.
 
-          updated_after: Filter runs updated after this timestamp (RFC3339 format)
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -345,11 +324,8 @@ async def list(
                         "environment_id": environment_id,
                         "limit": limit,
                         "model_id": model_id,
-                        "schedule_id": schedule_id,
-                        "skill_spec": skill_spec,
                         "source": source,
                         "state": state,
-                        "updated_after": updated_after,
                     },
                     run_list_params.RunListParams,
                 ),

src/warp_agent_sdk/types/agent/artifact_item.py

@@ -7,10 +7,10 @@
 from ..._utils import PropertyInfo
 from ..._models import BaseModel
 
-__all__ = ["ArtifactItem", "PlanArtifact", "PlanArtifactData", "PullRequestArtifact", "PullRequestArtifactData"]
+__all__ = ["ArtifactItem", "PlanArtifact", "PlanArtifactPlan", "PullRequestArtifact", "PullRequestArtifactPullRequest"]
 
 
-class PlanArtifactData(BaseModel):
+class PlanArtifactPlan(BaseModel):
     document_uid: str
     """Unique identifier for the plan document"""
 
@@ -22,16 +22,16 @@ class PlanArtifactData(BaseModel):
 
 
 class PlanArtifact(BaseModel):
-    artifact_type: Literal["PLAN"]
+    artifact_type: Literal["plan"]
     """Type of the artifact"""
 
     created_at: datetime
     """Timestamp when the artifact was created (RFC3339)"""
 
-    data: PlanArtifactData
+    plan: PlanArtifactPlan
 
 
-class PullRequestArtifactData(BaseModel):
+class PullRequestArtifactPullRequest(BaseModel):
     branch: str
     """Branch name for the pull request"""
 
@@ -40,13 +40,13 @@ class PullRequestArtifactData(BaseModel):
 
 
 class PullRequestArtifact(BaseModel):
-    artifact_type: Literal["PULL_REQUEST"]
+    artifact_type: Literal["pull_request"]
     """Type of the artifact"""
 
     created_at: datetime
     """Timestamp when the artifact was created (RFC3339)"""
 
-    data: PullRequestArtifactData
+    pull_request: PullRequestArtifactPullRequest
 
 
 ArtifactItem: TypeAlias = Annotated[

src/warp_agent_sdk/types/agent/run_item.py

@@ -10,7 +10,7 @@
 from .run_source_type import RunSourceType
 from ..ambient_agent_config import AmbientAgentConfig
 
-__all__ = ["RunItem", "RequestUsage", "StatusMessage"]
+__all__ = ["RunItem", "RequestUsage", "Schedule", "StatusMessage"]
 
 
 class RequestUsage(BaseModel):
@@ -23,6 +23,21 @@ class RequestUsage(BaseModel):
     """Cost of LLM inference for the run"""
 
 
+class Schedule(BaseModel):
+    """
+    Information about the schedule that triggered this run (only present for scheduled runs)
+    """
+
+    cron_schedule: str
+    """Cron expression at the time the run was created"""
+
+    schedule_id: str
+    """Unique identifier for the schedule"""
+
+    schedule_name: str
+    """Name of the schedule at the time the run was created"""
+
+
 class StatusMessage(BaseModel):
     message: Optional[str] = None
     """Human-readable status message"""
@@ -47,7 +62,6 @@ class RunItem(BaseModel):
     - INPROGRESS: Run is actively being executed
     - SUCCEEDED: Run completed successfully
     - FAILED: Run failed
-    - CANCELLED: Run was cancelled by user
     """
 
     task_id: str
@@ -79,6 +93,12 @@ class RunItem(BaseModel):
     request_usage: Optional[RequestUsage] = None
     """Resource usage information for the run"""
 
+    schedule: Optional[Schedule] = None
+    """
+    Information about the schedule that triggered this run (only present for
+    scheduled runs)
+    """
+
     session_id: Optional[str] = None
     """UUID of the shared session (if available)"""
 

src/warp_agent_sdk/types/agent/run_list_params.py

@@ -29,7 +29,7 @@ class RunListParams(TypedDict, total=False):
     cursor: str
     """Pagination cursor from previous response"""
 
-    environment_id: str
+    environment_id: Annotated[str, PropertyInfo(alias="environmentId")]
     """Filter runs by environment ID"""
 
     limit: int
@@ -38,12 +38,6 @@ class RunListParams(TypedDict, total=False):
     model_id: str
     """Filter by model ID"""
 
-    schedule_id: str
-    """Filter runs by the scheduled agent ID that created them"""
-
-    skill_spec: str
-    """Filter runs by skill spec (e.g., "owner/repo:path/to/SKILL.md")"""
-
     source: RunSourceType
     """Filter by run source type"""
 
@@ -52,6 +46,3 @@ class RunListParams(TypedDict, total=False):
 
     Can be specified multiple times to match any of the given states.
     """
-
-    updated_after: Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]
-    """Filter runs updated after this timestamp (RFC3339 format)"""

src/warp_agent_sdk/types/agent/run_state.py

@@ -4,4 +4,4 @@
 
 __all__ = ["RunState"]
 
-RunState: TypeAlias = Literal["QUEUED", "PENDING", "CLAIMED", "INPROGRESS", "SUCCEEDED", "FAILED", "CANCELLED"]
+RunState: TypeAlias = Literal["QUEUED", "PENDING", "CLAIMED", "INPROGRESS", "SUCCEEDED", "FAILED"]

src/warp_agent_sdk/types/agent_run_response.py

@@ -19,5 +19,4 @@ class AgentRunResponse(BaseModel):
     - INPROGRESS: Run is actively being executed
     - SUCCEEDED: Run completed successfully
     - FAILED: Run failed
-    - CANCELLED: Run was cancelled by user
     """

tests/api_resources/agent/test_runs.py

@@ -75,14 +75,11 @@ def test_method_list_with_all_params(self, client: WarpAPI) -> None:
             created_before=parse_datetime("2019-12-27T18:11:19.117Z"),
             creator="creator",
             cursor="cursor",
-            environment_id="environment_id",
+            environment_id="environmentId",
             limit=1,
             model_id="model_id",
-            schedule_id="schedule_id",
-            skill_spec="skill_spec",
             source="LINEAR",
             state=["QUEUED"],
-            updated_after=parse_datetime("2019-12-27T18:11:19.117Z"),
         )
         assert_matches_type(RunListResponse, run, path=["response"])
 
@@ -213,14 +210,11 @@ async def test_method_list_with_all_params(self, async_client: AsyncWarpAPI) ->
             created_before=parse_datetime("2019-12-27T18:11:19.117Z"),
             creator="creator",
             cursor="cursor",
-            environment_id="environment_id",
+            environment_id="environmentId",
             limit=1,
             model_id="model_id",
-            schedule_id="schedule_id",
-            skill_spec="skill_spec",
             source="LINEAR",
             state=["QUEUED"],
-            updated_after=parse_datetime("2019-12-27T18:11:19.117Z"),
         )
         assert_matches_type(RunListResponse, run, path=["response"])