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..addf7af 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-f17568757e003a96eb9843fb7f173f5df491bc3aa6b39a5189e9c33602912121.yml +openapi_spec_hash: a391afecf639f9c4993f37e5bbd06369 +config_hash: 236823a4936c76818117c16aa5c188df diff --git a/CHANGELOG.md b/CHANGELOG.md index 4431ffa..8a3bbbe 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,22 @@ # Changelog +## 0.13.0 (2026-05-09) + +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 ([af81ef3](https://github.com/warpdotdev/oz-sdk-python/commit/af81ef3a2d48f378209cf69a3074997cb02b1b6c)) +* **api:** api update ([99b2d31](https://github.com/warpdotdev/oz-sdk-python/commit/99b2d31ba1a4c2c3d79fc11c5eb8d611c55613b8)) +* **memory:** agent identity memory store attachments — API layer ([94b5348](https://github.com/warpdotdev/oz-sdk-python/commit/94b5348152c6b0bfb03b0d3887366c4a65e397fb)) +* **memory:** wire memory stores into run pipeline and add listing endpoint ([6bb74c2](https://github.com/warpdotdev/oz-sdk-python/commit/6bb74c2b695cd268fe8466fc6099f082370ba54e)) +* Retrieve memories in third party harnesses ([7689e12](https://github.com/warpdotdev/oz-sdk-python/commit/7689e121d6f22efad3d81828721f8ed900b9cd28)) + + +### Bug Fixes + +* **client:** add missing f-string prefix in file type error message ([17a8e5b](https://github.com/warpdotdev/oz-sdk-python/commit/17a8e5bf17b882a440a067fee1569caf679f8b55)) + ## 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/_files.py b/src/oz_agent_sdk/_files.py index 0fdce17..76da9e0 100644 --- a/src/oz_agent_sdk/_files.py +++ b/src/oz_agent_sdk/_files.py @@ -99,7 +99,7 @@ async def async_to_httpx_files(files: RequestFiles | None) -> HttpxRequestFiles elif is_sequence_t(files): files = [(key, await _async_transform_file(file)) for key, file in files] else: - raise TypeError("Unexpected file type input {type(files)}, expected mapping or sequence") + raise TypeError(f"Unexpected file type input {type(files)}, expected mapping or sequence") return files 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..5466a06 100644 --- a/src/oz_agent_sdk/resources/agent/agent_.py +++ b/src/oz_agent_sdk/resources/agent/agent_.py @@ -50,7 +50,9 @@ def create( self, *, name: str, + base_model: Optional[str] | Omit = omit, description: Optional[str] | Omit = omit, + memory_stores: Iterable[agent_create_params.MemoryStore] | Omit = omit, secrets: Iterable[agent_create_params.Secret] | Omit = omit, skills: SequenceNotStr[str] | Omit = omit, # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. @@ -68,8 +70,14 @@ 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 + memory_stores: Optional list of memory stores to attach to the agent. Each store must be + team-owned by the same team as the agent. Duplicate UIDs within a single request + are rejected. + secrets: Optional list of secrets associated with the agent. Duplicate names within a single request are rejected. Each entry is unioned into the run-time secret scope when the agent executes. @@ -94,7 +102,9 @@ def create( body=maybe_transform( { "name": name, + "base_model": base_model, "description": description, + "memory_stores": memory_stores, "secrets": secrets, "skills": skills, }, @@ -110,7 +120,9 @@ def update( self, uid: str, *, + base_model: Optional[str] | Omit = omit, description: Optional[str] | Omit = omit, + memory_stores: Optional[Iterable[agent_update_params.MemoryStore]] | Omit = omit, name: str | Omit = omit, secrets: Optional[Iterable[agent_update_params.Secret]] | Omit = omit, skills: Optional[SequenceNotStr[str]] | Omit = omit, @@ -124,11 +136,17 @@ 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. + memory_stores: Replacement list of memory stores. Omit to leave unchanged, pass an empty array + to clear, or pass a non-empty array to replace. + name: The new name for the agent secrets: Replacement list of secrets. Omit to leave unchanged, pass an empty array to @@ -151,7 +169,9 @@ def update( path_template("/agent/identities/{uid}", uid=uid), body=maybe_transform( { + "base_model": base_model, "description": description, + "memory_stores": memory_stores, "name": name, "secrets": secrets, "skills": skills, @@ -222,6 +242,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,7 +305,9 @@ async def create( self, *, name: str, + base_model: Optional[str] | Omit = omit, description: Optional[str] | Omit = omit, + memory_stores: Iterable[agent_create_params.MemoryStore] | Omit = omit, secrets: Iterable[agent_create_params.Secret] | Omit = omit, skills: SequenceNotStr[str] | Omit = omit, # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. @@ -267,8 +325,14 @@ 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 + memory_stores: Optional list of memory stores to attach to the agent. Each store must be + team-owned by the same team as the agent. Duplicate UIDs within a single request + are rejected. + secrets: Optional list of secrets associated with the agent. Duplicate names within a single request are rejected. Each entry is unioned into the run-time secret scope when the agent executes. @@ -293,7 +357,9 @@ async def create( body=await async_maybe_transform( { "name": name, + "base_model": base_model, "description": description, + "memory_stores": memory_stores, "secrets": secrets, "skills": skills, }, @@ -309,7 +375,9 @@ async def update( self, uid: str, *, + base_model: Optional[str] | Omit = omit, description: Optional[str] | Omit = omit, + memory_stores: Optional[Iterable[agent_update_params.MemoryStore]] | Omit = omit, name: str | Omit = omit, secrets: Optional[Iterable[agent_update_params.Secret]] | Omit = omit, skills: Optional[SequenceNotStr[str]] | Omit = omit, @@ -323,11 +391,17 @@ async def update( """Update an existing agent. Args: - description: Replacement description. + base_model: Replacement base model. - Omit or pass `null` to leave unchanged, or use an empty + Omit or pass `null` to leave unchanged, or pass an empty + string to clear. + + description: Replacement description. Omit or pass `null` to leave unchanged, or use an empty value to clear. + memory_stores: Replacement list of memory stores. Omit to leave unchanged, pass an empty array + to clear, or pass a non-empty array to replace. + name: The new name for the agent secrets: Replacement list of secrets. Omit to leave unchanged, pass an empty array to @@ -350,7 +424,9 @@ async def update( path_template("/agent/identities/{uid}", uid=uid), body=await async_maybe_transform( { + "base_model": base_model, "description": description, + "memory_stores": memory_stores, "name": name, "secrets": secrets, "skills": skills, @@ -421,6 +497,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 +550,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 +571,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 +592,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 +613,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..3ab1178 100644 --- a/src/oz_agent_sdk/types/agent/agent_create_params.py +++ b/src/oz_agent_sdk/types/agent/agent_create_params.py @@ -3,20 +3,30 @@ from __future__ import annotations from typing import Iterable, Optional -from typing_extensions import Required, TypedDict +from typing_extensions import Literal, Required, TypedDict from ..._types import SequenceNotStr -__all__ = ["AgentCreateParams", "Secret"] +__all__ = ["AgentCreateParams", "MemoryStore", "Secret"] 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""" + memory_stores: Iterable[MemoryStore] + """ + Optional list of memory stores to attach to the agent. Each store must be + team-owned by the same team as the agent. Duplicate UIDs within a single request + are rejected. + """ + secrets: Iterable[Secret] """ Optional list of secrets associated with the agent. Duplicate names within a @@ -34,6 +44,19 @@ class AgentCreateParams(TypedDict, total=False): """ +class MemoryStore(TypedDict, total=False): + """Reference to a memory store to attach to an agent.""" + + access: Required[Literal["read_write", "read_only"]] + """Access level for the store.""" + + instructions: Required[str] + """Instructions for how the agent should use this memory store. Must not be empty.""" + + uid: Required[str] + """UID of the memory store.""" + + class Secret(TypedDict, total=False): """Reference to a managed secret by name.""" diff --git a/src/oz_agent_sdk/types/agent/agent_response.py b/src/oz_agent_sdk/types/agent/agent_response.py index 8b46c85..2189a46 100644 --- a/src/oz_agent_sdk/types/agent/agent_response.py +++ b/src/oz_agent_sdk/types/agent/agent_response.py @@ -2,10 +2,24 @@ from typing import List, Optional from datetime import datetime +from typing_extensions import Literal from ..._models import BaseModel -__all__ = ["AgentResponse", "Secret"] +__all__ = ["AgentResponse", "MemoryStore", "Secret"] + + +class MemoryStore(BaseModel): + """Reference to a memory store to attach to an agent.""" + + access: Literal["read_write", "read_only"] + """Access level for the store.""" + + instructions: str + """Instructions for how the agent should use this memory store. Must not be empty.""" + + uid: str + """UID of the memory store.""" class Secret(BaseModel): @@ -22,6 +36,12 @@ class AgentResponse(BaseModel): created_at: datetime """When the agent was created (RFC3339)""" + memory_stores: List[MemoryStore] + """ + Memory stores attached to this agent. Always present; empty when no stores are + attached. + """ + name: str """Name of the agent""" @@ -37,5 +57,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..6a4ad86 100644 --- a/src/oz_agent_sdk/types/agent/agent_update_params.py +++ b/src/oz_agent_sdk/types/agent/agent_update_params.py @@ -3,20 +3,33 @@ from __future__ import annotations from typing import Iterable, Optional -from typing_extensions import Required, TypedDict +from typing_extensions import Literal, Required, TypedDict from ..._types import SequenceNotStr -__all__ = ["AgentUpdateParams", "Secret"] +__all__ = ["AgentUpdateParams", "MemoryStore", "Secret"] 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. Omit or pass `null` to leave unchanged, or use an empty value to clear. """ + memory_stores: Optional[Iterable[MemoryStore]] + """Replacement list of memory stores. + + Omit to leave unchanged, pass an empty array to clear, or pass a non-empty array + to replace. + """ + name: str """The new name for the agent""" @@ -35,6 +48,19 @@ class AgentUpdateParams(TypedDict, total=False): """ +class MemoryStore(TypedDict, total=False): + """Reference to a memory store to attach to an agent.""" + + access: Required[Literal["read_write", "read_only"]] + """Access level for the store.""" + + instructions: Required[str] + """Instructions for how the agent should use this memory store. Must not be empty.""" + + uid: Required[str] + """UID of the memory store.""" + + class Secret(TypedDict, total=False): """Reference to a managed secret by name.""" 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/ambient_agent_config.py b/src/oz_agent_sdk/types/ambient_agent_config.py index 148931d..b522e51 100644 --- a/src/oz_agent_sdk/types/ambient_agent_config.py +++ b/src/oz_agent_sdk/types/ambient_agent_config.py @@ -1,6 +1,6 @@ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details. -from typing import Dict, Optional +from typing import Dict, List, Optional from typing_extensions import Literal from pydantic import Field as FieldInfo @@ -8,7 +8,7 @@ from .._models import BaseModel from .mcp_server_config import McpServerConfig -__all__ = ["AmbientAgentConfig", "Harness", "HarnessAuthSecrets", "SessionSharing"] +__all__ = ["AmbientAgentConfig", "Harness", "HarnessAuthSecrets", "MemoryStore", "SessionSharing"] class Harness(BaseModel): @@ -41,6 +41,19 @@ class HarnessAuthSecrets(BaseModel): """ +class MemoryStore(BaseModel): + """Reference to a memory store to attach to an agent.""" + + access: Literal["read_write", "read_only"] + """Access level for the store.""" + + instructions: str + """Instructions for how the agent should use this memory store. Must not be empty.""" + + uid: str + """UID of the memory store.""" + + class SessionSharing(BaseModel): """ Configures sharing behavior for the run's shared session. @@ -100,6 +113,9 @@ class AmbientAgentConfig(BaseModel): mcp_servers: Optional[Dict[str, McpServerConfig]] = None """Map of MCP server configurations by name""" + memory_stores: Optional[List[MemoryStore]] = None + """Memory stores to attach to this run.""" + api_model_id: Optional[str] = FieldInfo(alias="model_id", default=None) """LLM model to use (uses team default if not specified)""" diff --git a/src/oz_agent_sdk/types/ambient_agent_config_param.py b/src/oz_agent_sdk/types/ambient_agent_config_param.py index f4c6ee4..3a4b7d0 100644 --- a/src/oz_agent_sdk/types/ambient_agent_config_param.py +++ b/src/oz_agent_sdk/types/ambient_agent_config_param.py @@ -2,12 +2,12 @@ from __future__ import annotations -from typing import Dict -from typing_extensions import Literal, TypedDict +from typing import Dict, Iterable +from typing_extensions import Literal, Required, TypedDict from .mcp_server_config_param import McpServerConfigParam -__all__ = ["AmbientAgentConfigParam", "Harness", "HarnessAuthSecrets", "SessionSharing"] +__all__ = ["AmbientAgentConfigParam", "Harness", "HarnessAuthSecrets", "MemoryStore", "SessionSharing"] class Harness(TypedDict, total=False): @@ -40,6 +40,19 @@ class HarnessAuthSecrets(TypedDict, total=False): """ +class MemoryStore(TypedDict, total=False): + """Reference to a memory store to attach to an agent.""" + + access: Required[Literal["read_write", "read_only"]] + """Access level for the store.""" + + instructions: Required[str] + """Instructions for how the agent should use this memory store. Must not be empty.""" + + uid: Required[str] + """UID of the memory store.""" + + class SessionSharing(TypedDict, total=False): """ Configures sharing behavior for the run's shared session. @@ -99,6 +112,9 @@ class AmbientAgentConfigParam(TypedDict, total=False): mcp_servers: Dict[str, McpServerConfigParam] """Map of MCP server configurations by name""" + memory_stores: Iterable[MemoryStore] + """Memory stores to attach to this run.""" + model_id: str """LLM model to use (uses team default if not specified)""" diff --git a/tests/api_resources/agent/test_agent_.py b/tests/api_resources/agent/test_agent_.py index a23b05e..11507d9 100644 --- a/tests/api_resources/agent/test_agent_.py +++ b/tests/api_resources/agent/test_agent_.py @@ -33,7 +33,15 @@ 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", + memory_stores=[ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], secrets=[{"name": "name"}], skills=["string"], ) @@ -78,7 +86,15 @@ 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", + memory_stores=[ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], name="name", secrets=[{"name": "name"}], skills=["string"], @@ -189,6 +205,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,7 +266,15 @@ 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", + memory_stores=[ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], secrets=[{"name": "name"}], skills=["string"], ) @@ -253,7 +319,15 @@ 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", + memory_stores=[ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], name="name", secrets=[{"name": "name"}], skills=["string"], @@ -363,3 +437,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..4a33c97 100644 --- a/tests/api_resources/agent/test_schedules.py +++ b/tests/api_resources/agent/test_schedules.py @@ -53,6 +53,13 @@ def test_method_create_with_all_params(self, client: OzAPI) -> None: "warp_id": "warp_id", } }, + "memory_stores": [ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], "model_id": "model_id", "name": "name", "session_sharing": {"public_access": "VIEWER"}, @@ -173,6 +180,13 @@ def test_method_update_with_all_params(self, client: OzAPI) -> None: "warp_id": "warp_id", } }, + "memory_stores": [ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], "model_id": "model_id", "name": "name", "session_sharing": {"public_access": "VIEWER"}, @@ -420,6 +434,13 @@ async def test_method_create_with_all_params(self, async_client: AsyncOzAPI) -> "warp_id": "warp_id", } }, + "memory_stores": [ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], "model_id": "model_id", "name": "name", "session_sharing": {"public_access": "VIEWER"}, @@ -540,6 +561,13 @@ async def test_method_update_with_all_params(self, async_client: AsyncOzAPI) -> "warp_id": "warp_id", } }, + "memory_stores": [ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], "model_id": "model_id", "name": "name", "session_sharing": {"public_access": "VIEWER"}, diff --git a/tests/api_resources/test_agent.py b/tests/api_resources/test_agent.py index a10a54e..d0d2742 100644 --- a/tests/api_resources/test_agent.py +++ b/tests/api_resources/test_agent.py @@ -174,6 +174,13 @@ def test_method_run_with_all_params(self, client: OzAPI) -> None: "warp_id": "warp_id", } }, + "memory_stores": [ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], "model_id": "model_id", "name": "name", "session_sharing": {"public_access": "VIEWER"}, @@ -371,6 +378,13 @@ async def test_method_run_with_all_params(self, async_client: AsyncOzAPI) -> Non "warp_id": "warp_id", } }, + "memory_stores": [ + { + "access": "read_write", + "instructions": "instructions", + "uid": "uid", + } + ], "model_id": "model_id", "name": "name", "session_sharing": {"public_access": "VIEWER"},