feat: Surface platform credits in the public API

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 22
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-933dc0de5653397e66fc2618c9aeb6dc1b65bc327fe36346bb59f0bd0c25b2bd.yml
-openapi_spec_hash: fed1d156b2b419f2bee85db21d5870f5
-config_hash: ea21d91999803fd188cdbfdf2c654914
+configured_endpoints: 23
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-439930897f625c088be1b09e394f4053e45060cd2daab7454dfcf6c775cba1fb.yml
+openapi_spec_hash: f0382bea9210d76f906665e8a825a05b
+config_hash: 5a6e285f6e3a958a887b31b972a3f49c

api.md

@@ -94,6 +94,7 @@ Methods:
 - <code title="put /agent/identities/{uid}">client.agent.agent.<a href="./src/oz_agent_sdk/resources/agent/agent_.py">update</a>(uid, \*\*<a href="src/oz_agent_sdk/types/agent/agent_update_params.py">params</a>) -> <a href="./src/oz_agent_sdk/types/agent/agent_response.py">AgentResponse</a></code>
 - <code title="get /agent/identities">client.agent.agent.<a href="./src/oz_agent_sdk/resources/agent/agent_.py">list</a>() -> <a href="./src/oz_agent_sdk/types/agent/list_agent_identities_response.py">ListAgentIdentitiesResponse</a></code>
 - <code title="delete /agent/identities/{uid}">client.agent.agent.<a href="./src/oz_agent_sdk/resources/agent/agent_.py">delete</a>(uid) -> None</code>
+- <code title="get /agent/identities/{uid}">client.agent.agent.<a href="./src/oz_agent_sdk/resources/agent/agent_.py">get</a>(uid) -> <a href="./src/oz_agent_sdk/types/agent/agent_response.py">AgentResponse</a></code>
 
 ## Sessions
 

src/oz_agent_sdk/resources/agent/agent_.py

@@ -222,6 +222,42 @@ def delete(
             cast_to=NoneType,
         )
 
+    def get(
+        self,
+        uid: str,
+        *,
+        # 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentResponse:
+        """Retrieve a single agent by its unique identifier.
+
+        The response includes an
+        `available` flag indicating whether the agent is within the team's plan limit
+        and may be used for runs.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not uid:
+            raise ValueError(f"Expected a non-empty value for `uid` but received {uid!r}")
+        return self._get(
+            path_template("/agent/identities/{uid}", uid=uid),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentResponse,
+        )
+
 
 class AsyncAgentResource(AsyncAPIResource):
     """Operations for running and managing cloud agents"""
@@ -421,6 +457,42 @@ async def delete(
             cast_to=NoneType,
         )
 
+    async def get(
+        self,
+        uid: str,
+        *,
+        # 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AgentResponse:
+        """Retrieve a single agent by its unique identifier.
+
+        The response includes an
+        `available` flag indicating whether the agent is within the team's plan limit
+        and may be used for runs.
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not uid:
+            raise ValueError(f"Expected a non-empty value for `uid` but received {uid!r}")
+        return await self._get(
+            path_template("/agent/identities/{uid}", uid=uid),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AgentResponse,
+        )
+
 
 class AgentResourceWithRawResponse:
     def __init__(self, agent: AgentResource) -> None:
@@ -438,6 +510,9 @@ def __init__(self, agent: AgentResource) -> None:
         self.delete = to_raw_response_wrapper(
             agent.delete,
         )
+        self.get = to_raw_response_wrapper(
+            agent.get,
+        )
 
 
 class AsyncAgentResourceWithRawResponse:
@@ -456,6 +531,9 @@ def __init__(self, agent: AsyncAgentResource) -> None:
         self.delete = async_to_raw_response_wrapper(
             agent.delete,
         )
+        self.get = async_to_raw_response_wrapper(
+            agent.get,
+        )
 
 
 class AgentResourceWithStreamingResponse:
@@ -474,6 +552,9 @@ def __init__(self, agent: AgentResource) -> None:
         self.delete = to_streamed_response_wrapper(
             agent.delete,
         )
+        self.get = to_streamed_response_wrapper(
+            agent.get,
+        )
 
 
 class AsyncAgentResourceWithStreamingResponse:
@@ -492,3 +573,6 @@ def __init__(self, agent: AsyncAgentResource) -> None:
         self.delete = async_to_streamed_response_wrapper(
             agent.delete,
         )
+        self.get = async_to_streamed_response_wrapper(
+            agent.get,
+        )

src/oz_agent_sdk/types/agent/run_item.py

@@ -44,6 +44,9 @@ class RequestUsage(BaseModel):
     inference_cost: Optional[float] = None
     """Cost of LLM inference for the run"""
 
+    platform_cost: Optional[float] = None
+    """Cost of platform usage for the run"""
+
 
 class Schedule(BaseModel):
     """

tests/api_resources/agent/test_agent_.py

@@ -189,6 +189,48 @@ def test_path_params_delete(self, client: OzAPI) -> None:
                 "",
             )
 
+    @pytest.mark.skip(reason="Mock server tests are disabled")
+    @parametrize
+    def test_method_get(self, client: OzAPI) -> None:
+        agent = client.agent.agent.get(
+            "uid",
+        )
+        assert_matches_type(AgentResponse, agent, path=["response"])
+
+    @pytest.mark.skip(reason="Mock server tests are disabled")
+    @parametrize
+    def test_raw_response_get(self, client: OzAPI) -> None:
+        response = client.agent.agent.with_raw_response.get(
+            "uid",
+        )
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        agent = response.parse()
+        assert_matches_type(AgentResponse, agent, path=["response"])
+
+    @pytest.mark.skip(reason="Mock server tests are disabled")
+    @parametrize
+    def test_streaming_response_get(self, client: OzAPI) -> None:
+        with client.agent.agent.with_streaming_response.get(
+            "uid",
+        ) as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            agent = response.parse()
+            assert_matches_type(AgentResponse, agent, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
+    @pytest.mark.skip(reason="Mock server tests are disabled")
+    @parametrize
+    def test_path_params_get(self, client: OzAPI) -> None:
+        with pytest.raises(ValueError, match=r"Expected a non-empty value for `uid` but received ''"):
+            client.agent.agent.with_raw_response.get(
+                "",
+            )
+
 
 class TestAsyncAgent:
     parametrize = pytest.mark.parametrize(
@@ -363,3 +405,45 @@ async def test_path_params_delete(self, async_client: AsyncOzAPI) -> None:
             await async_client.agent.agent.with_raw_response.delete(
                 "",
             )
+
+    @pytest.mark.skip(reason="Mock server tests are disabled")
+    @parametrize
+    async def test_method_get(self, async_client: AsyncOzAPI) -> None:
+        agent = await async_client.agent.agent.get(
+            "uid",
+        )
+        assert_matches_type(AgentResponse, agent, path=["response"])
+
+    @pytest.mark.skip(reason="Mock server tests are disabled")
+    @parametrize
+    async def test_raw_response_get(self, async_client: AsyncOzAPI) -> None:
+        response = await async_client.agent.agent.with_raw_response.get(
+            "uid",
+        )
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        agent = await response.parse()
+        assert_matches_type(AgentResponse, agent, path=["response"])
+
+    @pytest.mark.skip(reason="Mock server tests are disabled")
+    @parametrize
+    async def test_streaming_response_get(self, async_client: AsyncOzAPI) -> None:
+        async with async_client.agent.agent.with_streaming_response.get(
+            "uid",
+        ) as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            agent = await response.parse()
+            assert_matches_type(AgentResponse, agent, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
+    @pytest.mark.skip(reason="Mock server tests are disabled")
+    @parametrize
+    async def test_path_params_get(self, async_client: AsyncOzAPI) -> None:
+        with pytest.raises(ValueError, match=r"Expected a non-empty value for `uid` but received ''"):
+            await async_client.agent.agent.with_raw_response.get(
+                "",
+            )