feat: Share runs via Python SDK by dima-aignostics · Pull Request #667 · aignostics/python-sdk

dima-aignostics · 2026-06-04T11:10:38Z

No description provided.

Copilot

Pull request overview

This PR aims to add “run sharing” capabilities to the Python SDK by introducing access-control resource wrappers (grants + share tokens), wiring them into the platform client, and exposing sharing workflows through the application service and CLI.

Changes:

Added new access-control resource layer (AccessGrant, ShareToken, ShareTokens) and new Run helpers (list_share_grants, grant_access).
Added new application service + CLI commands under run share ... to list/grant/revoke access and manage share tokens.
Regenerated codegen artifacts/docs for updated API parameters (e.g., grants relation filter, additional run filters, run organization_id field).

Reviewed changes

Copilot reviewed 8 out of 61 changed files in this pull request and generated 18 comments.

Show a summary per file

File	Description
`src/aignostics/platform/resources/access.py`	New access-control resource models and share-token listing/creation logic.
`src/aignostics/platform/resources/runs.py`	Adds run-level grant listing/creation helpers with caching/retry integration.
`src/aignostics/platform/_client.py`	Exposes a `share_tokens` resource on `Client`.
`src/aignostics/application/_service.py`	Adds service methods for run sharing and token management.
`src/aignostics/application/_cli.py`	Adds `run share ...` CLI command group.
`tests/aignostics/platform/resources/access_test.py`	Unit tests for access resources.
`tests/aignostics/platform/resources/run_sharing_test.py`	Unit tests for run-sharing workflows (currently mismatched with implementation).
`codegen/in/openapi.json` + `codegen/out/**`	Codegen updates reflecting API spec changes.

codecov · 2026-06-04T11:35:18Z

Codecov Report

❌ Patch coverage is 89.16084% with 31 lines in your changes missing coverage. Please review.
✅ All tests successful. No failed tests found.

Files with missing lines	Patch %	Lines
src/aignostics/application/_service.py	71.79%	22 Missing ⚠️
src/aignostics/platform/resources/access.py	88.33%	7 Missing ⚠️
src/aignostics/application/_cli.py	98.48%	1 Missing and 1 partial ⚠️

Files with missing lines	Coverage Δ
src/aignostics/platform/_client.py	`93.18% <100.00%> (+0.15%)`	⬆️
src/aignostics/platform/resources/runs.py	`85.41% <100.00%> (+0.74%)`	⬆️
src/aignostics/application/_cli.py	`77.79% <98.48%> (+2.24%)`	⬆️
src/aignostics/platform/resources/access.py	`88.33% <88.33%> (ø)`
src/aignostics/application/_service.py	`62.12% <71.79%> (-2.88%)`	⬇️

... and 6 files with indirect coverage changes

Copilot

Pull request overview

Copilot reviewed 8 out of 61 changed files in this pull request and generated 20 comments.

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

Wraps the create_grant API call in a Retrying block with exponential backoff and jitter, consistent with list_share_grants and other methods. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

… grants - Add --expires-at option to `application run share token create` CLI command; accepts ISO 8601 string, defaults to UTC if no timezone given - Add expires_at parameter to Service.application_run_create_share_token() - Promote organization_id from --option to positional [ORGANIZATION_ID] argument in both `share organization grant` and `share organization revoke` commands - Add organization_id parameter to application_run_unshare_with_organization() so revoke filters grants by org (defaults to authenticated user's own org) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Copilot

Pull request overview

Copilot reviewed 8 out of 61 changed files in this pull request and generated 9 comments.

Copilot

Pull request overview

Copilot reviewed 9 out of 63 changed files in this pull request and generated 7 comments.

Copilot

Pull request overview

Copilot reviewed 10 out of 64 changed files in this pull request and generated 4 comments.

- Add 34 integration tests covering all run share CLI subcommands (status, org list/grant/revoke, token list/create/revoke) including success, not-found, and error paths with both text and JSON output - Add 15 unit tests for the 6 new service sharing methods - Fix ShareTokens.list() cache key isolation bug: run_id was captured from closure scope and not included in the @cached_operation key, causing queries for different run IDs to share cache entries - Fix --expires-at Z suffix parsing: remove unnecessary .replace("Z", "+00:00") workaround since Python ≥ 3.11 handles the Z UTC designator natively in datetime.fromisoformat() Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Copilot

Pull request overview

Copilot reviewed 12 out of 66 changed files in this pull request and generated 4 comments.

- Fix extra space before colon in share-with-organization error message - Fix docstring for application_run_revoke_share_token: NotFoundException is raised when the share token is not found, not the run - Fix run_share_token_revoke CLI handler to print the exception message instead of a hardcoded "Run ... not found" string, so users see the correct context (token ID not found vs run ID not found) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Copilot

Pull request overview

Copilot reviewed 12 out of 66 changed files in this pull request and generated 4 comments.

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

Copilot

Pull request overview

Copilot reviewed 12 out of 66 changed files in this pull request and generated 1 comment.

application_run_revoke_share_token now finds the SHARE_TOKEN grant on the specific run and revokes that grant, leaving the token itself intact for any other runs it may be shared with. Raises NotFoundException when no grant for the token exists on the run. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

GrantReadResponse carries these three fields but AccessGrant was silently dropping them. Adding them lets callers (e.g. ShareToken. list_share_grants()) inspect which resource a grant applies to and who created it, without having to go back to the raw codegen model. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

sonarqubecloud · 2026-06-08T09:28:42Z

Quality Gate passed

Issues
0 New issues
0 Accepted issues

Measures
0 Security Hotspots
0.0% Coverage on New Code
0.0% Duplication on New Code

See analysis details on SonarQube Cloud

Copilot

Pull request overview

Copilot reviewed 12 out of 66 changed files in this pull request and generated 3 comments.

+    def list_share_grants(self, *, page_size: int = 100) -> Iterator[AccessGrant]:
+        """List all active grants where this token is the subject.
+
+        Each returned ``AccessGrant`` represents a resource this token has been
+        granted access to.  Call ``grant.revoke()`` to remove access to a
+        specific resource without invalidating the token itself.
+
+        Args:
+            page_size: Number of grants to fetch per page (max 100).
+
+        Returns:
+            Iterator of ``AccessGrant`` objects for this token.
+
+        Raises:
+            Exception: If the API request fails.
+
+        Example::
+
+            token = client.share_tokens.create()
+            for grant in token.list_share_grants():
+                print(grant.grant_id, grant.relation)
+        """
+
+        def fetch_page(**kwargs: object) -> list[GrantReadResponse]:
+            return cast(


+    def list(self, *, run_id: str | None = None, nocache: bool = False, page_size: int = 100) -> Iterator[ShareToken]:
+        """List all share tokens for the authenticated user.
+
+        Results are cached for ``run_cache_ttl`` seconds and retried on
+        transient network or server errors.
+
+        Args:
+            run_id: Optional run ID to filter tokens by the run they are associated with.
+                Defaults to ``None`` (no filter).
+            nocache: If ``True``, bypass the local cache and fetch fresh data
+                from the API.  The fetched result is still written to the cache.
+                Defaults to ``False``.
+            page_size: Number of tokens to fetch per page (max 100).
+                Defaults to 100.
+
+        Returns:
+            Iterator of ``ShareToken`` objects.
+
+        Raises:
+            Exception: If the API request fails after all retries.
+
+        Example::
+
+            for token in client.share_tokens.list():
+                print(token.share_token_id, token.revoked)
+
+            # Force a fresh fetch after creating a new token
+            for token in client.share_tokens.list(nocache=True):
+                print(token.share_token_id)
+        """
+
+        @cached_operation(ttl=settings().run_cache_ttl, token_provider=self._api.token_provider)
+        def list_data_with_retry(cached_run_id: str | None, **kwargs: object) -> builtins.list[ShareToken]:
+            return Retrying(


      "OAuth2AuthorizationCodeBearer": {
        "type": "oauth2",
        "flows": {
          "authorizationCode": {
            "scopes": {},
-            "authorizationUrl": "https://aignostics-platform-staging.eu.auth0.com/authorize",
-            "tokenUrl": "https://aignostics-platform-staging.eu.auth0.com/oauth/token"
+            "authorizationUrl": "https://dev-8ouohmmrbuh2h4vu.eu.auth0.com/authorize",
+            "tokenUrl": "https://dev-8ouohmmrbuh2h4vu.eu.auth0.com/oauth/token"
          }


Copilot AI review requested due to automatic review settings June 4, 2026 11:10

dima-aignostics requested a review from a team as a code owner June 4, 2026 11:10

Copilot started reviewing on behalf of dima-aignostics June 4, 2026 11:10 View session

Copilot AI reviewed Jun 4, 2026

View reviewed changes

Copilot AI review requested due to automatic review settings June 4, 2026 14:41

Copilot started reviewing on behalf of dima-aignostics June 4, 2026 14:41 View session

Copilot AI reviewed Jun 4, 2026

View reviewed changes

Copilot AI review requested due to automatic review settings June 4, 2026 18:45

Copilot started reviewing on behalf of dima-aignostics June 4, 2026 18:45 View session

dima-aignostics and others added 6 commits June 4, 2026 20:47

feat: Share runs via Python SDK

60098fa

Update docs

57bd384

Potential fix for pull request finding

4292122

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

feat: add retry logic to grant_access

7f6ee1a

Wraps the create_grant API call in a Retrying block with exponential backoff and jitter, consistent with list_share_grants and other methods. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Update CLI methods

f668663

Copilot AI reviewed Jun 4, 2026

View reviewed changes

dima-aignostics force-pushed the feat/PYSDK-132-add-run-sharing-to-python-sdk branch from 3bbec8d to 1131c91 Compare June 4, 2026 18:53

Copilot AI review requested due to automatic review settings June 4, 2026 18:55

dima-aignostics force-pushed the feat/PYSDK-132-add-run-sharing-to-python-sdk branch from 1131c91 to fa0f3b9 Compare June 4, 2026 18:55

Copilot started reviewing on behalf of dima-aignostics June 4, 2026 18:55 View session

Copilot AI reviewed Jun 4, 2026

View reviewed changes

Update package

c4e64be

dima-aignostics force-pushed the feat/PYSDK-132-add-run-sharing-to-python-sdk branch from fa0f3b9 to c4e64be Compare June 4, 2026 19:26

Update expected test results for e2e

77167fb

Copilot AI review requested due to automatic review settings June 5, 2026 13:41

Copilot started reviewing on behalf of dima-aignostics June 5, 2026 13:41 View session

Copilot AI reviewed Jun 5, 2026

View reviewed changes

Comment thread src/aignostics/platform/resources/access.py

Comment thread src/aignostics/application/_cli.py

Comment thread src/aignostics/application/_service.py

Comment thread src/aignostics/application/_cli.py Outdated

Copilot AI review requested due to automatic review settings June 8, 2026 08:01

Copilot started reviewing on behalf of dima-aignostics June 8, 2026 08:01 View session

Update pip package

ece8b91

dima-aignostics force-pushed the feat/PYSDK-132-add-run-sharing-to-python-sdk branch from ffa17dc to ece8b91 Compare June 8, 2026 08:06

Copilot AI reviewed Jun 8, 2026

View reviewed changes

Comment thread src/aignostics/application/_cli.py

Comment thread src/aignostics/application/_cli.py Outdated

Comment thread src/aignostics/application/_service.py

Comment thread src/aignostics/application/_service.py Outdated

Copilot AI review requested due to automatic review settings June 8, 2026 08:34

Copilot started reviewing on behalf of dima-aignostics June 8, 2026 08:34 View session

Copilot AI reviewed Jun 8, 2026

View reviewed changes

Comment thread src/aignostics/application/_cli.py

Comment thread src/aignostics/platform/resources/access.py

Comment thread src/aignostics/platform/resources/access.py

Comment thread src/aignostics/application/_service.py Outdated

Potential fix for pull request finding

cc3c58f

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

Copilot AI review requested due to automatic review settings June 8, 2026 09:14

Copilot started reviewing on behalf of dima-aignostics June 8, 2026 09:14 View session

Copilot AI reviewed Jun 8, 2026

View reviewed changes

Comment thread src/aignostics/application/_cli.py

dima-aignostics and others added 3 commits June 8, 2026 11:23

Potential fix for pull request finding

cb21185

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

Copilot AI review requested due to automatic review settings June 8, 2026 09:26

Copilot started reviewing on behalf of dima-aignostics June 8, 2026 09:27 View session

Copilot AI reviewed Jun 8, 2026

View reviewed changes

Conversation

dima-aignostics commented Jun 4, 2026

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

codecov Bot commented Jun 4, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Codecov Report

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

codecov Bot commented Jun 4, 2026 •

edited

Loading