Skip to content

Commit 71d7204

Browse files
feat: Fix the harness type in the openAPI spec
1 parent 5514980 commit 71d7204

File tree

5 files changed

+36
-16
lines changed

5 files changed

+36
-16
lines changed

.stats.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 14
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta%2Fwarp-api-753a2c218019502904eeefce9318e46c152ddf6fe4556e232a3a5fc6931f9b80.yml
3-
openapi_spec_hash: 8d9a460280c5943425e79300f79296c9
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta%2Fwarp-api-e281c64ee44d0db13ba9ce62d3a61e8b17fcd1fad9bffb5c3b0e3dfb2b5e66e1.yml
3+
openapi_spec_hash: 6bd667f546147c5fac3bffe34522ba2f
44
config_hash: 1888db8b2f33dc16874aea51a90e78f7

src/oz_agent_sdk/types/ambient_agent_config.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,17 @@
77
from .._models import BaseModel
88
from .mcp_server_config import McpServerConfig
99

10-
__all__ = ["AmbientAgentConfig"]
10+
__all__ = ["AmbientAgentConfig", "Harness"]
11+
12+
13+
class Harness(BaseModel):
14+
"""
15+
Specifies which execution harness to use for the agent run.
16+
Default (nil/empty) uses Warp's built-in Oz harness.
17+
"""
18+
19+
type: Optional[str] = None
20+
"""The harness type identifier (e.g. "claude")."""
1121

1222

1323
class AmbientAgentConfig(BaseModel):
@@ -25,10 +35,10 @@ class AmbientAgentConfig(BaseModel):
2535
environment_id: Optional[str] = None
2636
"""UID of the environment to run the agent in"""
2737

28-
harness: Optional[str] = None
38+
harness: Optional[Harness] = None
2939
"""
30-
Agent harness to use for the agent run. Default (empty) uses Warp's built-in Oz
31-
harness.
40+
Specifies which execution harness to use for the agent run. Default (nil/empty)
41+
uses Warp's built-in Oz harness.
3242
"""
3343

3444
idle_timeout_minutes: Optional[int] = None

src/oz_agent_sdk/types/ambient_agent_config_param.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,17 @@
77

88
from .mcp_server_config_param import McpServerConfigParam
99

10-
__all__ = ["AmbientAgentConfigParam"]
10+
__all__ = ["AmbientAgentConfigParam", "Harness"]
11+
12+
13+
class Harness(TypedDict, total=False):
14+
"""
15+
Specifies which execution harness to use for the agent run.
16+
Default (nil/empty) uses Warp's built-in Oz harness.
17+
"""
18+
19+
type: str
20+
"""The harness type identifier (e.g. "claude")."""
1121

1222

1323
class AmbientAgentConfigParam(TypedDict, total=False):
@@ -25,10 +35,10 @@ class AmbientAgentConfigParam(TypedDict, total=False):
2535
environment_id: str
2636
"""UID of the environment to run the agent in"""
2737

28-
harness: str
38+
harness: Harness
2939
"""
30-
Agent harness to use for the agent run. Default (empty) uses Warp's built-in Oz
31-
harness.
40+
Specifies which execution harness to use for the agent run. Default (nil/empty)
41+
uses Warp's built-in Oz harness.
3242
"""
3343

3444
idle_timeout_minutes: int

tests/api_resources/agent/test_schedules.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def test_method_create_with_all_params(self, client: OzAPI) -> None:
4040
"base_prompt": "base_prompt",
4141
"computer_use_enabled": True,
4242
"environment_id": "environment_id",
43-
"harness": "harness",
43+
"harness": {"type": "type"},
4444
"idle_timeout_minutes": 1,
4545
"mcp_servers": {
4646
"foo": {
@@ -156,7 +156,7 @@ def test_method_update_with_all_params(self, client: OzAPI) -> None:
156156
"base_prompt": "base_prompt",
157157
"computer_use_enabled": True,
158158
"environment_id": "environment_id",
159-
"harness": "harness",
159+
"harness": {"type": "type"},
160160
"idle_timeout_minutes": 1,
161161
"mcp_servers": {
162162
"foo": {
@@ -399,7 +399,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncOzAPI) ->
399399
"base_prompt": "base_prompt",
400400
"computer_use_enabled": True,
401401
"environment_id": "environment_id",
402-
"harness": "harness",
402+
"harness": {"type": "type"},
403403
"idle_timeout_minutes": 1,
404404
"mcp_servers": {
405405
"foo": {
@@ -515,7 +515,7 @@ async def test_method_update_with_all_params(self, async_client: AsyncOzAPI) ->
515515
"base_prompt": "base_prompt",
516516
"computer_use_enabled": True,
517517
"environment_id": "environment_id",
518-
"harness": "harness",
518+
"harness": {"type": "type"},
519519
"idle_timeout_minutes": 1,
520520
"mcp_servers": {
521521
"foo": {

tests/api_resources/test_agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def test_method_run_with_all_params(self, client: OzAPI) -> None:
123123
"base_prompt": "base_prompt",
124124
"computer_use_enabled": True,
125125
"environment_id": "environment_id",
126-
"harness": "harness",
126+
"harness": {"type": "type"},
127127
"idle_timeout_minutes": 1,
128128
"mcp_servers": {
129129
"foo": {
@@ -280,7 +280,7 @@ async def test_method_run_with_all_params(self, async_client: AsyncOzAPI) -> Non
280280
"base_prompt": "base_prompt",
281281
"computer_use_enabled": True,
282282
"environment_id": "environment_id",
283-
"harness": "harness",
283+
"harness": {"type": "type"},
284284
"idle_timeout_minutes": 1,
285285
"mcp_servers": {
286286
"foo": {

0 commit comments

Comments
 (0)