Skip to content

Commit 65f9613

Browse files
feat(api): api update
1 parent 0b1c06d commit 65f9613

File tree

4 files changed

+20
-2
lines changed

4 files changed

+20
-2
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-a29592b2ba26cba9d89b95969d66506f49c08e140b76ce4aea4189e5c1dccc06.yml
3-
openapi_spec_hash: 27a5de1f891104d5e47904ad8e4b4bd1
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta%2Fwarp-api-8f9c749573846b07a55a3131b66456f0a592838c6bfc986ab30948df66cd6f11.yml
3+
openapi_spec_hash: 59f1ac98ad6cf13b12c59196bcecffd7
44
config_hash: 60052b2c1c0862014416821aba875574

src/oz_agent_sdk/resources/agent/agent.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,6 +193,7 @@ def get_artifact(
193193
def run(
194194
self,
195195
*,
196+
agent_identity_uid: str | Omit = omit,
196197
attachments: Iterable[agent_run_params.Attachment] | Omit = omit,
197198
config: AmbientAgentConfigParam | Omit = omit,
198199
conversation_id: str | Omit = omit,
@@ -215,6 +216,9 @@ def run(
215216
runs. Behavior is identical to POST /agent/run.
216217
217218
Args:
219+
agent_identity_uid: Optional agent identity UID to use as the execution principal for the run. This
220+
is only valid for runs that are team owned.
221+
218222
attachments: Optional file attachments to include with the prompt (max 5). Attachments are
219223
uploaded to cloud storage and made available to the agent.
220224
@@ -256,6 +260,7 @@ def run(
256260
"/agent/runs",
257261
body=maybe_transform(
258262
{
263+
"agent_identity_uid": agent_identity_uid,
259264
"attachments": attachments,
260265
"config": config,
261266
"conversation_id": conversation_id,
@@ -417,6 +422,7 @@ async def get_artifact(
417422
async def run(
418423
self,
419424
*,
425+
agent_identity_uid: str | Omit = omit,
420426
attachments: Iterable[agent_run_params.Attachment] | Omit = omit,
421427
config: AmbientAgentConfigParam | Omit = omit,
422428
conversation_id: str | Omit = omit,
@@ -439,6 +445,9 @@ async def run(
439445
runs. Behavior is identical to POST /agent/run.
440446
441447
Args:
448+
agent_identity_uid: Optional agent identity UID to use as the execution principal for the run. This
449+
is only valid for runs that are team owned.
450+
442451
attachments: Optional file attachments to include with the prompt (max 5). Attachments are
443452
uploaded to cloud storage and made available to the agent.
444453
@@ -480,6 +489,7 @@ async def run(
480489
"/agent/runs",
481490
body=await async_maybe_transform(
482491
{
492+
"agent_identity_uid": agent_identity_uid,
483493
"attachments": attachments,
484494
"config": config,
485495
"conversation_id": conversation_id,

src/oz_agent_sdk/types/agent_run_params.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,12 @@
1414

1515

1616
class AgentRunParams(TypedDict, total=False):
17+
agent_identity_uid: str
18+
"""
19+
Optional agent identity UID to use as the execution principal for the run. This
20+
is only valid for runs that are team owned.
21+
"""
22+
1723
attachments: Iterable[Attachment]
1824
"""
1925
Optional file attachments to include with the prompt (max 5). Attachments are

tests/api_resources/test_agent.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,7 @@ def test_method_run(self, client: OzAPI) -> None:
112112
@parametrize
113113
def test_method_run_with_all_params(self, client: OzAPI) -> None:
114114
agent = client.agent.run(
115+
agent_identity_uid="agent_identity_uid",
115116
attachments=[
116117
{
117118
"data": "U3RhaW5sZXNzIHJvY2tz",
@@ -272,6 +273,7 @@ async def test_method_run(self, async_client: AsyncOzAPI) -> None:
272273
@parametrize
273274
async def test_method_run_with_all_params(self, async_client: AsyncOzAPI) -> None:
274275
agent = await async_client.agent.run(
276+
agent_identity_uid="agent_identity_uid",
275277
attachments=[
276278
{
277279
"data": "U3RhaW5sZXNzIHJvY2tz",

0 commit comments

Comments
 (0)