labelbox-recursion-sdk 0.0.26__tar.gz → 0.0.28__tar.gz

{labelbox_recursion_sdk-0.0.26 → labelbox_recursion_sdk-0.0.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: labelbox-recursion-sdk
-Version: 0.0.26
+Version: 0.0.28
 Summary: Python SDK for the Recursion RL Platform API
 Requires-Python: >=3.11
 Requires-Dist: attrs>=22.2

{labelbox_recursion_sdk-0.0.26 → labelbox_recursion_sdk-0.0.28}/pyproject.toml

@@ -7,7 +7,7 @@ name = "labelbox-recursion-sdk"
 # highest `python-sdk-v*` git tag at publish time, stamped in the runner only.
 # The Python SDK releases on its own tag line, independent of `sdk-v*` (TypeScript)
 # and `cli-v*`.
-version = "0.0.26"
+version = "0.0.28"
 description = "Python SDK for the Recursion RL Platform API"
 readme = "README.md"
 # 3.11, not 3.10: the generated `client.py` imports `typing.Self`, which does not

{labelbox_recursion_sdk-0.0.26 → labelbox_recursion_sdk-0.0.28}/src/recursion_sdk/api/managed_agents/list_sessions.py

@@ -28,6 +28,7 @@ def _get_kwargs(
     metadata_key: list[str] | Unset = UNSET,
     root_only: bool | Unset = UNSET,
     limit: int | Unset = 100,
+    page_token: str | Unset = UNSET,
 
 ) -> dict[str, Any]:
     
@@ -66,6 +67,8 @@ def _get_kwargs(
 
     params["limit"] = limit
 
+    params["page_token"] = page_token
+
 
     params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
 
@@ -166,6 +169,7 @@ def sync_detailed(
     metadata_key: list[str] | Unset = UNSET,
     root_only: bool | Unset = UNSET,
     limit: int | Unset = 100,
+    page_token: str | Unset = UNSET,
 
 ) -> Response[ApiErrorBody | ApiErrorResponse | ManagedAgentsSessionListResponse]:
     """ List sessions
@@ -195,7 +199,12 @@ def sync_detailed(
             metadata key with any value. Repeatable; every listed key must be present.
         root_only (bool | Unset): Return only root sessions. Use this for model-cost list rows so
             children cannot consume the bounded page before their tree root.
-        limit (int | Unset):  Default: 100.
+        limit (int | Unset): Page size. When more rows match, the response carries
+            next_page_token; pass it back as page_token with the same filters and limit to read the
+            next page. Default: 100.
+        page_token (str | Unset): Signed continuation token from a previous page's
+            next_page_token. Bound to the organization and to the exact filters and limit it was
+            issued for; valid for one hour.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -217,6 +226,7 @@ metadata=metadata,
 metadata_key=metadata_key,
 root_only=root_only,
 limit=limit,
+page_token=page_token,
 
     )
 
@@ -239,6 +249,7 @@ def sync(
     metadata_key: list[str] | Unset = UNSET,
     root_only: bool | Unset = UNSET,
     limit: int | Unset = 100,
+    page_token: str | Unset = UNSET,
 
 ) -> ApiErrorBody | ApiErrorResponse | ManagedAgentsSessionListResponse | None:
     """ List sessions
@@ -268,7 +279,12 @@ def sync(
             metadata key with any value. Repeatable; every listed key must be present.
         root_only (bool | Unset): Return only root sessions. Use this for model-cost list rows so
             children cannot consume the bounded page before their tree root.
-        limit (int | Unset):  Default: 100.
+        limit (int | Unset): Page size. When more rows match, the response carries
+            next_page_token; pass it back as page_token with the same filters and limit to read the
+            next page. Default: 100.
+        page_token (str | Unset): Signed continuation token from a previous page's
+            next_page_token. Bound to the organization and to the exact filters and limit it was
+            issued for; valid for one hour.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -291,6 +307,7 @@ metadata=metadata,
 metadata_key=metadata_key,
 root_only=root_only,
 limit=limit,
+page_token=page_token,
 
     ).parsed
 
@@ -307,6 +324,7 @@ async def asyncio_detailed(
     metadata_key: list[str] | Unset = UNSET,
     root_only: bool | Unset = UNSET,
     limit: int | Unset = 100,
+    page_token: str | Unset = UNSET,
 
 ) -> Response[ApiErrorBody | ApiErrorResponse | ManagedAgentsSessionListResponse]:
     """ List sessions
@@ -336,7 +354,12 @@ async def asyncio_detailed(
             metadata key with any value. Repeatable; every listed key must be present.
         root_only (bool | Unset): Return only root sessions. Use this for model-cost list rows so
             children cannot consume the bounded page before their tree root.
-        limit (int | Unset):  Default: 100.
+        limit (int | Unset): Page size. When more rows match, the response carries
+            next_page_token; pass it back as page_token with the same filters and limit to read the
+            next page. Default: 100.
+        page_token (str | Unset): Signed continuation token from a previous page's
+            next_page_token. Bound to the organization and to the exact filters and limit it was
+            issued for; valid for one hour.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -358,6 +381,7 @@ metadata=metadata,
 metadata_key=metadata_key,
 root_only=root_only,
 limit=limit,
+page_token=page_token,
 
     )
 
@@ -380,6 +404,7 @@ async def asyncio(
     metadata_key: list[str] | Unset = UNSET,
     root_only: bool | Unset = UNSET,
     limit: int | Unset = 100,
+    page_token: str | Unset = UNSET,
 
 ) -> ApiErrorBody | ApiErrorResponse | ManagedAgentsSessionListResponse | None:
     """ List sessions
@@ -409,7 +434,12 @@ async def asyncio(
             metadata key with any value. Repeatable; every listed key must be present.
         root_only (bool | Unset): Return only root sessions. Use this for model-cost list rows so
             children cannot consume the bounded page before their tree root.
-        limit (int | Unset):  Default: 100.
+        limit (int | Unset): Page size. When more rows match, the response carries
+            next_page_token; pass it back as page_token with the same filters and limit to read the
+            next page. Default: 100.
+        page_token (str | Unset): Signed continuation token from a previous page's
+            next_page_token. Bound to the organization and to the exact filters and limit it was
+            issued for; valid for one hour.
 
     Raises:
         errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -432,5 +462,6 @@ metadata=metadata,
 metadata_key=metadata_key,
 root_only=root_only,
 limit=limit,
+page_token=page_token,
 
     )).parsed

labelbox_recursion_sdk-0.0.28/src/recursion_sdk/api/problem_runs/get_usage.py

@@ -0,0 +1,230 @@
+from http import HTTPStatus
+from typing import Any, cast
+from urllib.parse import quote
+
+import httpx
+
+from ...client import AuthenticatedClient, Client
+from ...types import Response, UNSET
+from ... import errors
+
+from ...models.api_error_body import ApiErrorBody
+from ...models.api_error_response import ApiErrorResponse
+from ...models.problem_run_usage_response_dto import ProblemRunUsageResponseDto
+from typing import cast
+from uuid import UUID
+
+
+
+def _get_kwargs(
+    problem_run_id: UUID,
+
+) -> dict[str, Any]:
+    
+
+    
+
+    
+
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": "/v1/problem-runs/{problem_run_id}/usage".format(problem_run_id=quote(str(problem_run_id), safe=""),),
+    }
+
+
+    return _kwargs
+
+
+
+def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto | None:
+    if response.status_code == 200:
+        response_200 = ProblemRunUsageResponseDto.from_dict(response.json())
+
+
+
+        return response_200
+
+    if response.status_code == 400:
+        response_400 = ApiErrorResponse.from_dict(response.json())
+
+
+
+        return response_400
+
+    if response.status_code == 401:
+        response_401 = ApiErrorBody.from_dict(response.json())
+
+
+
+        return response_401
+
+    if response.status_code == 403:
+        response_403 = cast(Any, None)
+        return response_403
+
+    if response.status_code == 404:
+        response_404 = cast(Any, None)
+        return response_404
+
+    if response.status_code == 429:
+        response_429 = ApiErrorResponse.from_dict(response.json())
+
+
+
+        return response_429
+
+    if response.status_code == 500:
+        response_500 = ApiErrorBody.from_dict(response.json())
+
+
+
+        return response_500
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Response[Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    problem_run_id: UUID,
+    *,
+    client: AuthenticatedClient | Client,
+
+) -> Response[Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto]:
+    """ Get cost and resource usage for a problem run
+
+     One row per agent-service run attributed to the problem run (the solver, then each grader), each
+    with its cost and, when the runner reported it, what the container consumed against what it was
+    granted. Rows appear as the underlying runs finish.
+
+    Args:
+        problem_run_id (UUID): Stable problem-run identifier (UUID). One solver attempt at one
+            problem version.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto]
+     """
+
+
+    kwargs = _get_kwargs(
+        problem_run_id=problem_run_id,
+
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+def sync(
+    problem_run_id: UUID,
+    *,
+    client: AuthenticatedClient | Client,
+
+) -> Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto | None:
+    """ Get cost and resource usage for a problem run
+
+     One row per agent-service run attributed to the problem run (the solver, then each grader), each
+    with its cost and, when the runner reported it, what the container consumed against what it was
+    granted. Rows appear as the underlying runs finish.
+
+    Args:
+        problem_run_id (UUID): Stable problem-run identifier (UUID). One solver attempt at one
+            problem version.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto
+     """
+
+
+    return sync_detailed(
+        problem_run_id=problem_run_id,
+client=client,
+
+    ).parsed
+
+async def asyncio_detailed(
+    problem_run_id: UUID,
+    *,
+    client: AuthenticatedClient | Client,
+
+) -> Response[Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto]:
+    """ Get cost and resource usage for a problem run
+
+     One row per agent-service run attributed to the problem run (the solver, then each grader), each
+    with its cost and, when the runner reported it, what the container consumed against what it was
+    granted. Rows appear as the underlying runs finish.
+
+    Args:
+        problem_run_id (UUID): Stable problem-run identifier (UUID). One solver attempt at one
+            problem version.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto]
+     """
+
+
+    kwargs = _get_kwargs(
+        problem_run_id=problem_run_id,
+
+    )
+
+    response = await client.get_async_httpx_client().request(
+        **kwargs
+    )
+
+    return _build_response(client=client, response=response)
+
+async def asyncio(
+    problem_run_id: UUID,
+    *,
+    client: AuthenticatedClient | Client,
+
+) -> Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto | None:
+    """ Get cost and resource usage for a problem run
+
+     One row per agent-service run attributed to the problem run (the solver, then each grader), each
+    with its cost and, when the runner reported it, what the container consumed against what it was
+    granted. Rows appear as the underlying runs finish.
+
+    Args:
+        problem_run_id (UUID): Stable problem-run identifier (UUID). One solver attempt at one
+            problem version.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Any | ApiErrorBody | ApiErrorResponse | ProblemRunUsageResponseDto
+     """
+
+
+    return (await asyncio_detailed(
+        problem_run_id=problem_run_id,
+client=client,
+
+    )).parsed

{labelbox_recursion_sdk-0.0.26 → labelbox_recursion_sdk-0.0.28}/src/recursion_sdk/facade.py

@@ -1115,6 +1115,10 @@ class _ProblemRuns:
         from .api.problem_runs import get_transcript as _endpoint
         return _unwrap(await _endpoint.asyncio_detailed(*args, client=self._client, **kwargs))
 
+    async def get_usage(self, *args: Any, **kwargs: Any) -> Any:
+        from .api.problem_runs import get_usage as _endpoint
+        return _unwrap(await _endpoint.asyncio_detailed(*args, client=self._client, **kwargs))
+
     async def get_workspace_file_download_url(self, *args: Any, **kwargs: Any) -> Any:
         from .api.problem_runs import get_workspace_file_download_url as _endpoint
         return _unwrap(await _endpoint.asyncio_detailed(*args, client=self._client, **kwargs))

{labelbox_recursion_sdk-0.0.26 → labelbox_recursion_sdk-0.0.28}/src/recursion_sdk/models/__init__.py

@@ -1119,6 +1119,13 @@ from .problem_run_response_list_envelope_dto_schema_0_rubric_type import Problem
 from .problem_run_response_list_envelope_dto_schema_0_weighted_sum import ProblemRunResponseListEnvelopeDtoSchema0WeightedSum
 from .problem_run_response_list_envelope_dto_schema_0_weighted_sum_children_item import ProblemRunResponseListEnvelopeDtoSchema0WeightedSumChildrenItem
 from .problem_run_response_list_envelope_dto_schema_0_weighted_sum_type import ProblemRunResponseListEnvelopeDtoSchema0WeightedSumType
+from .problem_run_usage_response_dto import ProblemRunUsageResponseDto
+from .problem_run_usage_response_dto_items_item import ProblemRunUsageResponseDtoItemsItem
+from .problem_run_usage_response_dto_items_item_resource_usage_type_0 import ProblemRunUsageResponseDtoItemsItemResourceUsageType0
+from .problem_run_usage_response_dto_items_item_resource_usage_type_0_cpu import ProblemRunUsageResponseDtoItemsItemResourceUsageType0Cpu
+from .problem_run_usage_response_dto_items_item_resource_usage_type_0_disk_type_0 import ProblemRunUsageResponseDtoItemsItemResourceUsageType0DiskType0
+from .problem_run_usage_response_dto_items_item_resource_usage_type_0_gpu_type_0 import ProblemRunUsageResponseDtoItemsItemResourceUsageType0GpuType0
+from .problem_run_usage_response_dto_items_item_resource_usage_type_0_memory import ProblemRunUsageResponseDtoItemsItemResourceUsageType0Memory
 from .problem_secret_dto import ProblemSecretDto
 from .problem_secret_list_dto_item import ProblemSecretListDtoItem
 from .problem_summaries_response_dto import ProblemSummariesResponseDto
@@ -2976,6 +2983,13 @@ __all__ = (
     "ProblemRunResponseListEnvelopeDtoSchema0WeightedSum",
     "ProblemRunResponseListEnvelopeDtoSchema0WeightedSumChildrenItem",
     "ProblemRunResponseListEnvelopeDtoSchema0WeightedSumType",
+    "ProblemRunUsageResponseDto",
+    "ProblemRunUsageResponseDtoItemsItem",
+    "ProblemRunUsageResponseDtoItemsItemResourceUsageType0",
+    "ProblemRunUsageResponseDtoItemsItemResourceUsageType0Cpu",
+    "ProblemRunUsageResponseDtoItemsItemResourceUsageType0DiskType0",
+    "ProblemRunUsageResponseDtoItemsItemResourceUsageType0GpuType0",
+    "ProblemRunUsageResponseDtoItemsItemResourceUsageType0Memory",
     "ProblemSecretDto",
     "ProblemSecretListDtoItem",
     "ProblemSummariesResponseDto",

{labelbox_recursion_sdk-0.0.26 → labelbox_recursion_sdk-0.0.28}/src/recursion_sdk/models/managed_agents_session_list_response.py

@@ -24,20 +24,22 @@ T = TypeVar("T", bound="ManagedAgentsSessionListResponse")
 
 @_attrs_define
 class ManagedAgentsSessionListResponse:
-    """ Response body of GET /v1/sessions. This list is limit-truncated rather than cursored: it carries no continuation
-    token, so a caller that hits the limit must narrow the filters (status, kind, agent_id, external_source_*) rather
-    than page onward. Token and cost totals are not included; fetch them for a batch of ids via GET /v1/sessions/usage.
+    """ Response body of GET /v1/sessions. Keyset-paginated over (updated_at, id): a page is a stable window at the moment
+    it is read, but a session updated between two page reads moves to the front of the list, so a live list may show it
+    twice or skip it once. Each row is a summary without the start-time snapshots; read one session for those. Token and
+    cost totals are not included; fetch them for a batch of ids via GET /v1/sessions/usage and GET /v1/sessions/costs.
 
         Example:
-            {'sessions': [{'access_policy': 'admin_only', 'agent_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d',
-                'agent_snapshot': {'key': 'example'}, 'agent_version_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'config':
-                {'key': 'example'}, 'created_at': '2026-02-18T09:30:00Z', 'credential_refs': [{'credential_id':
-                '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'vault_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d'}],
-                'credential_refs_configured': True, 'environment_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'execution_state':
-                'provisioning', 'external_source_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'external_source_type': 'example',
-                'failure': {'at': '2026-02-18T09:30:00Z', 'category': 'transient', 'code': 'example', 'message': 'example',
-                'phase': 'example', 'retryable': True}, 'forked_at_event_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'kind':
-                'api_call', 'last_activity_at': '2026-02-18T09:30:00Z', 'metadata': {'key': 'example'}, 'model_ref_id':
+            {'next_page_token': 'example', 'sessions': [{'access_policy': 'admin_only', 'agent_id':
+                '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'agent_snapshot': {'key': 'example'}, 'agent_version_id':
+                '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'config': {'key': 'example'}, 'created_at': '2026-02-18T09:30:00Z',
+                'credential_refs': [{'credential_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'vault_id':
+                '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d'}], 'credential_refs_configured': True, 'environment_id':
+                '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'execution_state': 'provisioning', 'external_source_id':
+                '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'external_source_type': 'example', 'failure': {'at':
+                '2026-02-18T09:30:00Z', 'category': 'transient', 'code': 'example', 'message': 'example', 'phase': 'example',
+                'retryable': True}, 'forked_at_event_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'kind': 'api_call',
+                'last_activity_at': '2026-02-18T09:30:00Z', 'metadata': {'key': 'example'}, 'model_ref_id':
                 '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'model_snapshot': {'key': 'example'}, 'organization_id':
                 '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'parent_session_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d',
                 'project_id': '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'project_source': 'example', 'root_session_id':
@@ -48,11 +50,16 @@ class ManagedAgentsSessionListResponse:
                 '9f8b1c2d-3e4f-5a6b-7c8d-9e0f1a2b3c4d', 'vault_ids': ['example'], 'wake_at': '2026-02-18T09:30:00Z'}]}
 
         Attributes:
-            sessions (list[ManagedAgentsSession] | None | Unset): Sessions matching the query filters, newest first, capped
-                by limit (default 100). May be null or an empty array; both mean nothing matched. Rows the caller's access
-                policy does not permit are dropped silently rather than reported.
+            next_page_token (str | Unset): Present when more sessions match. Pass it as page_token with the same filters and
+                limit to read the next page. Absent on the last page.
+            sessions (list[ManagedAgentsSession] | None | Unset): Sessions matching the query filters, most recently updated
+                first, at most limit (default 100) per page. May be null or an empty array; both mean nothing matched. Rows the
+                caller's access policy does not permit are dropped silently rather than reported. Rows are summaries:
+                agent_snapshot and model_snapshot are omitted and config is an empty object; GET /v1/sessions/{session_id}
+                returns them.
      """
 
+    next_page_token: str | Unset = UNSET
     sessions: list[ManagedAgentsSession] | None | Unset = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -62,6 +69,8 @@ class ManagedAgentsSessionListResponse:
 
     def to_dict(self) -> dict[str, Any]:
         from ..models.managed_agents_session import ManagedAgentsSession # noqa: PLC0415
+        next_page_token = self.next_page_token
+
         sessions: list[dict[str, Any]] | None | Unset
         if isinstance(self.sessions, Unset):
             sessions = UNSET
@@ -80,6 +89,8 @@ class ManagedAgentsSessionListResponse:
         field_dict.update(self.additional_properties)
         field_dict.update({
         })
+        if next_page_token is not UNSET:
+            field_dict["next_page_token"] = next_page_token
         if sessions is not UNSET:
             field_dict["sessions"] = sessions
 
@@ -91,6 +102,8 @@ class ManagedAgentsSessionListResponse:
     def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
         from ..models.managed_agents_session import ManagedAgentsSession # noqa: PLC0415
         d = dict(src_dict)
+        next_page_token = d.pop("next_page_token", UNSET)
+
         def _parse_sessions(data: object) -> list[ManagedAgentsSession] | None | Unset:
             if data is None:
                 return data
@@ -117,6 +130,7 @@ class ManagedAgentsSessionListResponse:
 
 
         managed_agents_session_list_response = cls(
+            next_page_token=next_page_token,
             sessions=sessions,
         )
 

labelbox_recursion_sdk-0.0.28/src/recursion_sdk/models/problem_run_usage_response_dto.py

@@ -0,0 +1,122 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar, BinaryIO, TextIO, TYPE_CHECKING, Generator
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+from typing import cast
+
+if TYPE_CHECKING:
+  from ..models.problem_run_usage_response_dto_items_item import ProblemRunUsageResponseDtoItemsItem
+
+
+
+
+
+T = TypeVar("T", bound="ProblemRunUsageResponseDto")
+
+
+
+@_attrs_define
+class ProblemRunUsageResponseDto:
+    """ Cost and consumed-vs-requested resources for every agent-service run behind a problem run.
+
+        Example:
+            {'items': [{'runType': 'solver', 'modelName': 'claude-sonnet-5', 'agentCostUsd': 0.3121, 'computeCostUsd':
+                0.0184, 'totalCostUsd': 0.3305, 'inputTokens': 18432, 'outputTokens': 7910, 'cacheReadInputTokens': 102400,
+                'cacheCreationInputTokens': 4096, 'resourceUsage': {'scope': 'cgroup', 'sampleCount': 47, 'sampledSeconds':
+                231.4, 'cpu': {'avgMillicores': 410, 'peakMillicores': 1220, 'capacityMillicores': 2000, 'coreSeconds': 94.9},
+                'memory': {'avgWorkingSetBytes': 1181116006, 'peakWorkingSetBytes': 1717986918, 'capacityBytes': 4294967296},
+                'gpu': None, 'disk': {'peakUsedBytes': 3435973836, 'capacityBytes': 52512901529}}}, {'runType':
+                'agentic_grading', 'modelName': 'claude-sonnet-5', 'agentCostUsd': 0.0212, 'computeCostUsd': None,
+                'totalCostUsd': 0.0212, 'inputTokens': 6210, 'outputTokens': 880, 'cacheReadInputTokens': 0,
+                'cacheCreationInputTokens': 0, 'resourceUsage': None}], 'totalCostUsd': 0.3517}
+
+        Attributes:
+            items (list[ProblemRunUsageResponseDtoItemsItem]): One row per agent-service run attributed to this problem run:
+                the solver first, then graders.
+            total_cost_usd (float | None): Sum of every row that reported a cost. Null when no row has cost data yet.
+     """
+
+    items: list[ProblemRunUsageResponseDtoItemsItem]
+    total_cost_usd: float | None
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+
+
+
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.problem_run_usage_response_dto_items_item import ProblemRunUsageResponseDtoItemsItem # noqa: PLC0415
+        items = []
+        for items_item_data in self.items:
+            items_item = items_item_data.to_dict()
+            items.append(items_item)
+
+
+
+        total_cost_usd: float | None
+        total_cost_usd = self.total_cost_usd
+
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({
+            "items": items,
+            "totalCostUsd": total_cost_usd,
+        })
+
+        return field_dict
+
+
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.problem_run_usage_response_dto_items_item import ProblemRunUsageResponseDtoItemsItem # noqa: PLC0415
+        d = dict(src_dict)
+        items = []
+        _items = d.pop("items")
+        for items_item_data in (_items):
+            items_item = ProblemRunUsageResponseDtoItemsItem.from_dict(items_item_data)
+
+
+
+            items.append(items_item)
+
+
+        def _parse_total_cost_usd(data: object) -> float | None:
+            if data is None:
+                return data
+            return cast(float | None, data)
+
+        total_cost_usd = _parse_total_cost_usd(d.pop("totalCostUsd"))
+
+
+        problem_run_usage_response_dto = cls(
+            items=items,
+            total_cost_usd=total_cost_usd,
+        )
+
+
+        problem_run_usage_response_dto.additional_properties = d
+        return problem_run_usage_response_dto
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties