Skip to content

feat: Client updates for Axon SSE methods #778

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
tode-rl wants to merge 11 commits into
base: main
Choose a base branch
from
Open

feat: Client updates for Axon SSE methods #778

Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
ea91605
feat: add CancellationToken support to polling infrastructure
tode-rl Apr 1, 2026
0d0f863
feat: propagate cancellation_token through all resource polling methods
tode-rl Apr 1, 2026
6baf371
feat: complete SDK wrapper cancellation_token propagation
tode-rl Apr 1, 2026
19771bc
feat: add SSE auto-reconnect for Axon subscriptions
tode-rl Apr 1, 2026
a015865
test: add comprehensive unit tests for new features
tode-rl Apr 1, 2026
34d932f
test: add comprehensive list pagination smoke tests (port of TS PR #767)
tode-rl Apr 2, 2026
b596e71
Update tests, pass CI
tode-rl Apr 2, 2026
8ced825
Merge origin/main into feature/ts-pr-765-port
tode-rl Apr 2, 2026
14085b9
fix: resolve merge conflicts and fix type errors in axons.py
tode-rl Apr 2, 2026
498a24a
fix: resolve mypy type errors in axons.py
tode-rl Apr 2, 2026
427cd6c
chore: apply code formatting and update workflow docs
tode-rl Apr 2, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Merge origin/main into feature/ts-pr-765-port 
Resolved conflicts in src/runloop_api_client/resources/axons/axons.py by:
- Keeping auto-reconnection logic from feature branch (ReconnectingStream)
- Adding support for user-provided after_sequence parameter from main
- Initial stream uses user's after_sequence, reconnections use last received sequence
- Both sync and async subscribe_sse methods now properly support both features

Changes from origin/main (v1.15.0):
- Updated version to 1.15.0
- Added axon_subscribe_sse_params type
- Updated benchmark_runs and related tests

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
  • Loading branch information
@tode-rl @claude
tode-rl and claude committed Apr 2, 2026
commit 8ced8257e17fb9ea2af0e878012d2932b91d01ff
32 changes: 23 additions & 9 deletions src/runloop_api_client/resources/axons/axons.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -296,16 +296,22 @@ 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,
stream_cls=Stream[AxonEventView],
)

def create_stream(last_sequence: str | None) -> Stream[AxonEventView]:
# after_sequence is used internally for reconnection only
sequence_int = int(last_sequence) if last_sequence is not None else None
# Use user-provided after_sequence for initial stream, then use last_sequence for reconnections
sequence_to_use = after_sequence if last_sequence is None else int(last_sequence)
return self._get(
path_template("/v1/axons/{id}/subscribe/sse", id=id),
options=make_request_options(
Expand All @@ -314,7 +320,7 @@ def create_stream(last_sequence: str | None) -> Stream[AxonEventView]:
extra_body=extra_body,
timeout=timeout,
query=maybe_transform(
{"after_sequence": sequence_int},
{"after_sequence": sequence_to_use},
axon_subscribe_sse_params.AxonSubscribeSseParams,
),
),
Expand All @@ -341,6 +347,7 @@ def get_sequence(item: AxonEventView) -> str | None:
)



class AsyncAxonsResource(AsyncAPIResource):
@cached_property
def sql(self) -> AsyncSqlResource:
Expand Down Expand Up @@ -599,25 +606,31 @@ 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,
stream_cls=AsyncStream[AxonEventView],
)

async def create_stream(last_sequence: str | None) -> AsyncStream[AxonEventView]:
# after_sequence is used internally for reconnection only
sequence_int = int(last_sequence) if last_sequence is not None else None
# Use user-provided after_sequence for initial stream, then use last_sequence for reconnections
sequence_to_use = after_sequence if last_sequence is None else int(last_sequence)
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,
query=maybe_transform(
{"after_sequence": sequence_int},
query=await async_maybe_transform(
{"after_sequence": sequence_to_use},
axon_subscribe_sse_params.AxonSubscribeSseParams,
),
),
Expand All @@ -644,6 +657,7 @@ def get_sequence(item: AxonEventView) -> str | None:
)



class AxonsResourceWithRawResponse:
def __init__(self, axons: AxonsResource) -> None:
self._axons = axons
Expand Down
Loading
You are viewing a condensed version of this merge commit. You can view the full changes here.