diff --git a/.release-please-manifest.json b/.release-please-manifest.json index a713055..d52d2b9 100644 --- a/.release-please-manifest.json +++ b/.release-please-manifest.json @@ -1,3 +1,3 @@ { - ".": "0.12.0" + ".": "0.13.0" } \ No newline at end of file diff --git a/.stats.yml b/.stats.yml index 2cb0b3b..d5a9669 100644 --- a/.stats.yml +++ b/.stats.yml @@ -1,4 +1,4 @@ -configured_endpoints: 22 -openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-1fecc5f5d6ee664d804b81bd1aa6eec4d3f170ffa788d214fead4f7e95ab9d4e.yml -openapi_spec_hash: 82990b03bd5a93e45bfc79db56ae7fc0 -config_hash: f52e7636f248f25c4ea0b086e7326816 +configured_endpoints: 23 +openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-e752e75a35d88b84870729ef94b0c32783172983420cbff1b204ca14375553f7.yml +openapi_spec_hash: 34787afc1e1c84a643431a0f0eb352ae +config_hash: 5a6e285f6e3a958a887b31b972a3f49c diff --git a/CHANGELOG.md b/CHANGELOG.md index 4431ffa..93ab1f8 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,13 @@ # Changelog +## 0.13.0 (2026-05-07) + +Full Changelog: [v0.12.0...v0.13.0](https://github.com/warpdotdev/oz-sdk-python/compare/v0.12.0...v0.13.0) + +### Features + +* **api:** api update ([99b2d31](https://github.com/warpdotdev/oz-sdk-python/commit/99b2d31ba1a4c2c3d79fc11c5eb8d611c55613b8)) + ## 0.12.0 (2026-05-07) Full Changelog: [v0.11.0...v0.12.0](https://github.com/warpdotdev/oz-sdk-python/compare/v0.11.0...v0.12.0) diff --git a/api.md b/api.md index bd33474..e82443c 100644 --- a/api.md +++ b/api.md @@ -94,6 +94,7 @@ Methods: - client.agent.agent.update(uid, \*\*params) -> AgentResponse - client.agent.agent.list() -> ListAgentIdentitiesResponse - client.agent.agent.delete(uid) -> None +- client.agent.agent.get(uid) -> AgentResponse ## Sessions diff --git a/pyproject.toml b/pyproject.toml index 367b68a..533a8e4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "oz-agent-sdk" -version = "0.12.0" +version = "0.13.0" description = "The official Python library for the oz-api API" dynamic = ["readme"] license = "Apache-2.0" diff --git a/src/oz_agent_sdk/_version.py b/src/oz_agent_sdk/_version.py index b96fedf..1762ab1 100644 --- a/src/oz_agent_sdk/_version.py +++ b/src/oz_agent_sdk/_version.py @@ -1,4 +1,4 @@ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details. __title__ = "oz_agent_sdk" -__version__ = "0.12.0" # x-release-please-version +__version__ = "0.13.0" # x-release-please-version diff --git a/src/oz_agent_sdk/resources/agent/agent_.py b/src/oz_agent_sdk/resources/agent/agent_.py index ab7835e..0bfc72b 100644 --- a/src/oz_agent_sdk/resources/agent/agent_.py +++ b/src/oz_agent_sdk/resources/agent/agent_.py @@ -50,6 +50,7 @@ def create( self, *, name: str, + base_model: Optional[str] | Omit = omit, description: Optional[str] | Omit = omit, secrets: Iterable[agent_create_params.Secret] | Omit = omit, skills: SequenceNotStr[str] | Omit = omit, @@ -68,6 +69,8 @@ def create( Args: name: A name for the agent + base_model: Optional base model for runs executed by this agent. + description: Optional description of the agent secrets: Optional list of secrets associated with the agent. Duplicate names within a @@ -94,6 +97,7 @@ def create( body=maybe_transform( { "name": name, + "base_model": base_model, "description": description, "secrets": secrets, "skills": skills, @@ -110,6 +114,7 @@ def update( self, uid: str, *, + base_model: Optional[str] | Omit = omit, description: Optional[str] | Omit = omit, name: str | Omit = omit, secrets: Optional[Iterable[agent_update_params.Secret]] | Omit = omit, @@ -124,9 +129,12 @@ def update( """Update an existing agent. Args: - description: Replacement description. + base_model: Replacement base model. + + Omit or pass `null` to leave unchanged, or pass an empty + string to clear. - Omit or pass `null` to leave unchanged, or use an empty + description: Replacement description. Omit or pass `null` to leave unchanged, or use an empty value to clear. name: The new name for the agent @@ -151,6 +159,7 @@ def update( path_template("/agent/identities/{uid}", uid=uid), body=maybe_transform( { + "base_model": base_model, "description": description, "name": name, "secrets": secrets, @@ -222,6 +231,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""" @@ -249,6 +294,7 @@ async def create( self, *, name: str, + base_model: Optional[str] | Omit = omit, description: Optional[str] | Omit = omit, secrets: Iterable[agent_create_params.Secret] | Omit = omit, skills: SequenceNotStr[str] | Omit = omit, @@ -267,6 +313,8 @@ async def create( Args: name: A name for the agent + base_model: Optional base model for runs executed by this agent. + description: Optional description of the agent secrets: Optional list of secrets associated with the agent. Duplicate names within a @@ -293,6 +341,7 @@ async def create( body=await async_maybe_transform( { "name": name, + "base_model": base_model, "description": description, "secrets": secrets, "skills": skills, @@ -309,6 +358,7 @@ async def update( self, uid: str, *, + base_model: Optional[str] | Omit = omit, description: Optional[str] | Omit = omit, name: str | Omit = omit, secrets: Optional[Iterable[agent_update_params.Secret]] | Omit = omit, @@ -323,9 +373,12 @@ async def update( """Update an existing agent. Args: - description: Replacement description. + base_model: Replacement base model. + + Omit or pass `null` to leave unchanged, or pass an empty + string to clear. - Omit or pass `null` to leave unchanged, or use an empty + description: Replacement description. Omit or pass `null` to leave unchanged, or use an empty value to clear. name: The new name for the agent @@ -350,6 +403,7 @@ async def update( path_template("/agent/identities/{uid}", uid=uid), body=await async_maybe_transform( { + "base_model": base_model, "description": description, "name": name, "secrets": secrets, @@ -421,6 +475,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 +528,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 +549,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 +570,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 +591,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, + ) diff --git a/src/oz_agent_sdk/types/agent/agent_create_params.py b/src/oz_agent_sdk/types/agent/agent_create_params.py index 5f31e19..73d8734 100644 --- a/src/oz_agent_sdk/types/agent/agent_create_params.py +++ b/src/oz_agent_sdk/types/agent/agent_create_params.py @@ -14,6 +14,9 @@ class AgentCreateParams(TypedDict, total=False): name: Required[str] """A name for the agent""" + base_model: Optional[str] + """Optional base model for runs executed by this agent.""" + description: Optional[str] """Optional description of the agent""" diff --git a/src/oz_agent_sdk/types/agent/agent_response.py b/src/oz_agent_sdk/types/agent/agent_response.py index 8b46c85..1370bda 100644 --- a/src/oz_agent_sdk/types/agent/agent_response.py +++ b/src/oz_agent_sdk/types/agent/agent_response.py @@ -37,5 +37,15 @@ class AgentResponse(BaseModel): uid: str """Unique identifier for the agent""" + base_model: Optional[str] = None + """Base model for runs executed by this agent. + + The precedence order for model resolution is: + + 1. The model specified on the run itself + 2. The agent's base model + 3. The team's default model + """ + description: Optional[str] = None """Optional description of the agent""" diff --git a/src/oz_agent_sdk/types/agent/agent_update_params.py b/src/oz_agent_sdk/types/agent/agent_update_params.py index d3cc68a..271122d 100644 --- a/src/oz_agent_sdk/types/agent/agent_update_params.py +++ b/src/oz_agent_sdk/types/agent/agent_update_params.py @@ -11,6 +11,12 @@ class AgentUpdateParams(TypedDict, total=False): + base_model: Optional[str] + """Replacement base model. + + Omit or pass `null` to leave unchanged, or pass an empty string to clear. + """ + description: Optional[str] """Replacement description. diff --git a/src/oz_agent_sdk/types/agent/run_item.py b/src/oz_agent_sdk/types/agent/run_item.py index 5b802d8..4293ce2 100644 --- a/src/oz_agent_sdk/types/agent/run_item.py +++ b/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): """ diff --git a/src/oz_agent_sdk/types/agent/scheduled_agent_item.py b/src/oz_agent_sdk/types/agent/scheduled_agent_item.py index d307d25..d66ddfa 100644 --- a/src/oz_agent_sdk/types/agent/scheduled_agent_item.py +++ b/src/oz_agent_sdk/types/agent/scheduled_agent_item.py @@ -41,9 +41,6 @@ class ScheduledAgentItem(BaseModel): agent_config: Optional[AmbientAgentConfig] = None """Configuration for a cloud agent run""" - agent_uid: Optional[str] = None - """UID of the agent that this schedule runs as""" - created_by: Optional[UserProfile] = None environment: Optional[CloudEnvironmentConfig] = None diff --git a/tests/api_resources/agent/test_agent_.py b/tests/api_resources/agent/test_agent_.py index a23b05e..5d0632e 100644 --- a/tests/api_resources/agent/test_agent_.py +++ b/tests/api_resources/agent/test_agent_.py @@ -33,6 +33,7 @@ def test_method_create(self, client: OzAPI) -> None: def test_method_create_with_all_params(self, client: OzAPI) -> None: agent = client.agent.agent.create( name="name", + base_model="base_model", description="description", secrets=[{"name": "name"}], skills=["string"], @@ -78,6 +79,7 @@ def test_method_update(self, client: OzAPI) -> None: def test_method_update_with_all_params(self, client: OzAPI) -> None: agent = client.agent.agent.update( uid="uid", + base_model="base_model", description="description", name="name", secrets=[{"name": "name"}], @@ -189,6 +191,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( @@ -208,6 +252,7 @@ async def test_method_create(self, async_client: AsyncOzAPI) -> None: async def test_method_create_with_all_params(self, async_client: AsyncOzAPI) -> None: agent = await async_client.agent.agent.create( name="name", + base_model="base_model", description="description", secrets=[{"name": "name"}], skills=["string"], @@ -253,6 +298,7 @@ async def test_method_update(self, async_client: AsyncOzAPI) -> None: async def test_method_update_with_all_params(self, async_client: AsyncOzAPI) -> None: agent = await async_client.agent.agent.update( uid="uid", + base_model="base_model", description="description", name="name", secrets=[{"name": "name"}], @@ -363,3 +409,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( + "", + ) diff --git a/tests/api_resources/agent/test_schedules.py b/tests/api_resources/agent/test_schedules.py index 0bf7d09..0beb0e2 100644 --- a/tests/api_resources/agent/test_schedules.py +++ b/tests/api_resources/agent/test_schedules.py @@ -59,7 +59,7 @@ def test_method_create_with_all_params(self, client: OzAPI) -> None: "skill_spec": "skill_spec", "worker_host": "worker_host", }, - agent_uid="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e", + agent_uid="agent_uid", enabled=True, mode="normal", prompt="Review open pull requests and provide feedback", @@ -179,7 +179,7 @@ def test_method_update_with_all_params(self, client: OzAPI) -> None: "skill_spec": "skill_spec", "worker_host": "worker_host", }, - agent_uid="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e", + agent_uid="agent_uid", mode="normal", prompt="prompt", ) @@ -426,7 +426,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncOzAPI) -> "skill_spec": "skill_spec", "worker_host": "worker_host", }, - agent_uid="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e", + agent_uid="agent_uid", enabled=True, mode="normal", prompt="Review open pull requests and provide feedback", @@ -546,7 +546,7 @@ async def test_method_update_with_all_params(self, async_client: AsyncOzAPI) -> "skill_spec": "skill_spec", "worker_host": "worker_host", }, - agent_uid="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e", + agent_uid="agent_uid", mode="normal", prompt="prompt", )