Apply suggestions from code review

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

Author: dima-aignostics

specifications/SPEC_PLATFORM_SERVICE.md

@@ -559,9 +559,6 @@ class ApplicationRun(_AuthenticatedResource):
         Returns:
             The created AccessGrant.
         """
-        Returns:
-            The created AccessGrant.
-        """
 
     def list_share_grants(
         self,
@@ -583,15 +580,6 @@ class ApplicationRun(_AuthenticatedResource):
         Returns:
             Iterator of AccessGrant objects for this run.
         """
-        """Lists all active access grants on this run.
-
-        Args:
-            subject_type: Optional filter by subject type.
-            subject_id: Optional filter by subject identifier.
-
-        Returns:
-            Iterator of AccessGrant objects for this run.
-        """
 ```
 
 ```python

src/aignostics/application/_cli.py

@@ -1436,7 +1436,7 @@ def run_share_token_list(
         tokens = list(Service().application_run_share_tokens(run_id))
 
         if format == "json":
-            print(json.dumps([t.model_dump() for t in tokens], indent=2, default=str))
+            print(json.dumps([t.model_dump(mode="json") for t in tokens], indent=2))
         else:
             if not tokens:
                 console.print("No active share tokens.")
@@ -1481,7 +1481,7 @@ def run_share_token_create(
     try:
         token = Service().application_run_create_share_token(run_id, expires_at=expires_at_dt)
         if format == "json":
-            print(json.dumps(token.model_dump(), indent=2, default=str))
+            print(json.dumps(token.model_dump(mode="json"), indent=2))
         else:
             expires = token.expires_at.isoformat() if token.expires_at else "never"
             console.print(f"Share token created for run '{run_id}'.")

src/aignostics/application/_service.py

@@ -25,6 +25,7 @@
     ApplicationVersion,
     Client,
     ForbiddenException,
+    ForbiddenException,
     InputArtifact,
     InputItem,
     NotFoundException,
@@ -33,6 +34,7 @@
     RunOutput,
     RunState,
 )
+from aignostics.platform.resources.access import AccessGrant, ShareToken
 from aignostics.platform import Service as PlatformService
 from aignostics.platform.resources.access import AccessGrant, ShareToken
 from aignostics.utils import BaseService, Health, sanitize_path_component
@@ -1384,18 +1386,18 @@ def application_run_share_tokens(
         try:
             client = self._get_platform_client()
             run = client.run(run_id)
-            seen: set[str] = set()
-            for grant in run.list_share_grants(
+
+            tokens = self._get_platform_client().share_tokens.list(run_id=run_id, page_size=page_size)
+            token_grants = set(g.subject_id for g in run.list_share_grants(
                 subject_type=SubjectType.SHARE_TOKEN,
                 page_size=page_size,
                 nocache=True,
-            ):
-                token_id = grant.subject_id
-                if token_id not in seen:
-                    seen.add(token_id)
-                    share_token = ShareToken.for_token_id(token_id)
-                    if not share_token.revoked:
-                        yield share_token
+            ))
+
+            for token in tokens:
+                if token.share_token_id in token_grants:
+                    yield token
+
         except NotFoundException as e:
             message = f"Application run with ID '{run_id}' not found: {e}"
             logger.warning(message)

src/aignostics/platform/resources/runs.py

@@ -721,6 +721,7 @@ def fetch_grant_page(cached_run_id: str, **kwargs: object) -> list[GrantReadResp
                 lambda **kw: fetch_grant_page(
                     run_id,
                     nocache=nocache,
+                    _cache_run_id=self.run_id,
                     subject_type=subject_type,
                     subject_id=subject_id,
                     relation=relation,