fix: Add after_sequence param to axon subscribe (#8497)

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 116
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/runloop-ai%2Frunloop-8ea4486cd6de6410a6648545710d617f175994f8d12753783585029ce22c225c.yml
-openapi_spec_hash: fcb07ce127f4290f1dac51db130365aa
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/runloop-ai%2Frunloop-77cfebef736250545ee47fa63d2210f323f096f0cbff4194c4a460a4d0328fd3.yml
+openapi_spec_hash: 33b5de41f43ca91cd3c745b048e68856
 config_hash: 6649774d90af30c3559d6a242b6cb4b0

api.md

@@ -109,7 +109,7 @@ Methods:
 - <code title="get /v1/axons/{id}">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">retrieve</a>(id) -> <a href="./src/runloop_api_client/types/axon_view.py">AxonView</a></code>
 - <code title="get /v1/axons">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">list</a>(\*\*<a href="src/runloop_api_client/types/axon_list_params.py">params</a>) -> <a href="./src/runloop_api_client/types/axon_view.py">SyncAxonsCursorIDPage[AxonView]</a></code>
 - <code title="post /v1/axons/{id}/publish">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">publish</a>(id, \*\*<a href="src/runloop_api_client/types/axon_publish_params.py">params</a>) -> <a href="./src/runloop_api_client/types/publish_result_view.py">PublishResultView</a></code>
-- <code title="get /v1/axons/{id}/subscribe/sse">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">subscribe_sse</a>(id) -> <a href="./src/runloop_api_client/types/axon_event_view.py">AxonEventView</a></code>
+- <code title="get /v1/axons/{id}/subscribe/sse">client.axons.<a href="./src/runloop_api_client/resources/axons/axons.py">subscribe_sse</a>(id, \*\*<a href="src/runloop_api_client/types/axon_subscribe_sse_params.py">params</a>) -> <a href="./src/runloop_api_client/types/axon_event_view.py">AxonEventView</a></code>
 
 ## Sql
 

src/runloop_api_client/resources/axons/axons.py

@@ -15,7 +15,7 @@
     SqlResourceWithStreamingResponse,
     AsyncSqlResourceWithStreamingResponse,
 )
-from ...types import axon_list_params, axon_create_params, axon_publish_params
+from ...types import axon_list_params, axon_create_params, axon_publish_params, axon_subscribe_sse_params
 from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from ..._utils import path_template, maybe_transform, async_maybe_transform
 from ..._compat import cached_property
@@ -259,6 +259,7 @@ def subscribe_sse(
         self,
         id: str,
         *,
+        after_sequence: int | 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,
@@ -270,6 +271,9 @@ def subscribe_sse(
         [Beta] Subscribe to an axon event stream via server-sent events.
 
         Args:
+          after_sequence: Sequence number after which to start streaming. Events with sequence > this
+              value are returned. If unset, replay from the beginning.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -285,7 +289,13 @@ def subscribe_sse(
         return self._get(
             path_template("/v1/axons/{id}/subscribe/sse", id=id),
             options=make_request_options(
-                extra_headers=merged_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=merged_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {"after_sequence": after_sequence}, axon_subscribe_sse_params.AxonSubscribeSseParams
+                ),
             ),
             cast_to=AxonEventView,
             stream=True,
@@ -516,6 +526,7 @@ async def subscribe_sse(
         self,
         id: str,
         *,
+        after_sequence: int | 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,
@@ -527,6 +538,9 @@ async def subscribe_sse(
         [Beta] Subscribe to an axon event stream via server-sent events.
 
         Args:
+          after_sequence: Sequence number after which to start streaming. Events with sequence > this
+              value are returned. If unset, replay from the beginning.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -542,7 +556,13 @@ async def subscribe_sse(
         return await self._get(
             path_template("/v1/axons/{id}/subscribe/sse", id=id),
             options=make_request_options(
-                extra_headers=merged_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=merged_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {"after_sequence": after_sequence}, axon_subscribe_sse_params.AxonSubscribeSseParams
+                ),
             ),
             cast_to=AxonEventView,
             stream=True,

src/runloop_api_client/types/__init__.py

@@ -82,6 +82,7 @@
 from .mcp_config_update_params import McpConfigUpdateParams as McpConfigUpdateParams
 from .network_policy_list_view import NetworkPolicyListView as NetworkPolicyListView
 from .object_download_url_view import ObjectDownloadURLView as ObjectDownloadURLView
+from .axon_subscribe_sse_params import AxonSubscribeSseParams as AxonSubscribeSseParams
 from .benchmark_job_list_params import BenchmarkJobListParams as BenchmarkJobListParams
 from .benchmark_run_list_params import BenchmarkRunListParams as BenchmarkRunListParams
 from .devbox_send_std_in_result import DevboxSendStdInResult as DevboxSendStdInResult

src/runloop_api_client/types/axon_subscribe_sse_params.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+__all__ = ["AxonSubscribeSseParams"]
+
+
+class AxonSubscribeSseParams(TypedDict, total=False):
+    after_sequence: int
+    """Sequence number after which to start streaming.
+
+    Events with sequence > this value are returned. If unset, replay from the
+    beginning.
+    """

tests/api_resources/test_axons.py

@@ -184,14 +184,22 @@ def test_path_params_publish(self, client: Runloop) -> None:
     @parametrize
     def test_method_subscribe_sse(self, client: Runloop) -> None:
         axon_stream = client.axons.subscribe_sse(
-            "id",
+            id="id",
+        )
+        axon_stream.response.close()
+
+    @parametrize
+    def test_method_subscribe_sse_with_all_params(self, client: Runloop) -> None:
+        axon_stream = client.axons.subscribe_sse(
+            id="id",
+            after_sequence=0,
         )
         axon_stream.response.close()
 
     @parametrize
     def test_raw_response_subscribe_sse(self, client: Runloop) -> None:
         response = client.axons.with_raw_response.subscribe_sse(
-            "id",
+            id="id",
         )
 
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -201,7 +209,7 @@ def test_raw_response_subscribe_sse(self, client: Runloop) -> None:
     @parametrize
     def test_streaming_response_subscribe_sse(self, client: Runloop) -> None:
         with client.axons.with_streaming_response.subscribe_sse(
-            "id",
+            id="id",
         ) as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -215,7 +223,7 @@ def test_streaming_response_subscribe_sse(self, client: Runloop) -> None:
     def test_path_params_subscribe_sse(self, client: Runloop) -> None:
         with pytest.raises(ValueError, match=r"Expected a non-empty value for `id` but received ''"):
             client.axons.with_raw_response.subscribe_sse(
-                "",
+                id="",
             )
 
 
@@ -387,14 +395,22 @@ async def test_path_params_publish(self, async_client: AsyncRunloop) -> None:
     @parametrize
     async def test_method_subscribe_sse(self, async_client: AsyncRunloop) -> None:
         axon_stream = await async_client.axons.subscribe_sse(
-            "id",
+            id="id",
+        )
+        await axon_stream.response.aclose()
+
+    @parametrize
+    async def test_method_subscribe_sse_with_all_params(self, async_client: AsyncRunloop) -> None:
+        axon_stream = await async_client.axons.subscribe_sse(
+            id="id",
+            after_sequence=0,
         )
         await axon_stream.response.aclose()
 
     @parametrize
     async def test_raw_response_subscribe_sse(self, async_client: AsyncRunloop) -> None:
         response = await async_client.axons.with_raw_response.subscribe_sse(
-            "id",
+            id="id",
         )
 
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -404,7 +420,7 @@ async def test_raw_response_subscribe_sse(self, async_client: AsyncRunloop) -> N
     @parametrize
     async def test_streaming_response_subscribe_sse(self, async_client: AsyncRunloop) -> None:
         async with async_client.axons.with_streaming_response.subscribe_sse(
-            "id",
+            id="id",
         ) as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -418,5 +434,5 @@ async def test_streaming_response_subscribe_sse(self, async_client: AsyncRunloop
     async def test_path_params_subscribe_sse(self, async_client: AsyncRunloop) -> None:
         with pytest.raises(ValueError, match=r"Expected a non-empty value for `id` but received ''"):
             await async_client.axons.with_raw_response.subscribe_sse(
-                "",
+                id="",
             )