Skip to content

Commit c5c0a3a

Browse files
feat(api): api update
1 parent 7f6d70a commit c5c0a3a

5 files changed

Lines changed: 38 additions & 6 deletions

File tree

.stats.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 64
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp/agentex-sdk-ae2571b5ac5d337ba5ced527cec0ff6e3088296fa67c3c836ed5a06544b25cb8.yml
3-
openapi_spec_hash: 962a2f20444c7823fd3a34f95365146e
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp/agentex-sdk-e925e150285e8390d1faef227d54bbf108901e4d846ac02bb074dd90a9ce90d0.yml
3+
openapi_spec_hash: 85e5220ee09746474843dd55822a8c68
44
config_hash: 138b7c0b394e7393133c8ff16a6d0eb3

src/agentex/resources/tasks.py

Lines changed: 28 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -554,6 +554,7 @@ def update_by_id(
554554
self,
555555
task_id: str,
556556
*,
557+
merge_params: Optional[Dict[str, object]] | Omit = omit,
557558
task_metadata: Optional[Dict[str, object]] | Omit = omit,
558559
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
559560
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -578,7 +579,13 @@ def update_by_id(
578579
raise ValueError(f"Expected a non-empty value for `task_id` but received {task_id!r}")
579580
return self._put(
580581
path_template("/tasks/{task_id}", task_id=task_id),
581-
body=maybe_transform({"task_metadata": task_metadata}, task_update_by_id_params.TaskUpdateByIDParams),
582+
body=maybe_transform(
583+
{
584+
"merge_params": merge_params,
585+
"task_metadata": task_metadata,
586+
},
587+
task_update_by_id_params.TaskUpdateByIDParams,
588+
),
582589
options=make_request_options(
583590
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
584591
),
@@ -589,6 +596,7 @@ def update_by_name(
589596
self,
590597
task_name: str,
591598
*,
599+
merge_params: Optional[Dict[str, object]] | Omit = omit,
592600
task_metadata: Optional[Dict[str, object]] | Omit = omit,
593601
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
594602
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -613,7 +621,13 @@ def update_by_name(
613621
raise ValueError(f"Expected a non-empty value for `task_name` but received {task_name!r}")
614622
return self._put(
615623
path_template("/tasks/name/{task_name}", task_name=task_name),
616-
body=maybe_transform({"task_metadata": task_metadata}, task_update_by_name_params.TaskUpdateByNameParams),
624+
body=maybe_transform(
625+
{
626+
"merge_params": merge_params,
627+
"task_metadata": task_metadata,
628+
},
629+
task_update_by_name_params.TaskUpdateByNameParams,
630+
),
617631
options=make_request_options(
618632
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
619633
),
@@ -1136,6 +1150,7 @@ async def update_by_id(
11361150
self,
11371151
task_id: str,
11381152
*,
1153+
merge_params: Optional[Dict[str, object]] | Omit = omit,
11391154
task_metadata: Optional[Dict[str, object]] | Omit = omit,
11401155
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
11411156
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -1161,7 +1176,11 @@ async def update_by_id(
11611176
return await self._put(
11621177
path_template("/tasks/{task_id}", task_id=task_id),
11631178
body=await async_maybe_transform(
1164-
{"task_metadata": task_metadata}, task_update_by_id_params.TaskUpdateByIDParams
1179+
{
1180+
"merge_params": merge_params,
1181+
"task_metadata": task_metadata,
1182+
},
1183+
task_update_by_id_params.TaskUpdateByIDParams,
11651184
),
11661185
options=make_request_options(
11671186
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -1173,6 +1192,7 @@ async def update_by_name(
11731192
self,
11741193
task_name: str,
11751194
*,
1195+
merge_params: Optional[Dict[str, object]] | Omit = omit,
11761196
task_metadata: Optional[Dict[str, object]] | Omit = omit,
11771197
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
11781198
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -1198,7 +1218,11 @@ async def update_by_name(
11981218
return await self._put(
11991219
path_template("/tasks/name/{task_name}", task_name=task_name),
12001220
body=await async_maybe_transform(
1201-
{"task_metadata": task_metadata}, task_update_by_name_params.TaskUpdateByNameParams
1221+
{
1222+
"merge_params": merge_params,
1223+
"task_metadata": task_metadata,
1224+
},
1225+
task_update_by_name_params.TaskUpdateByNameParams,
12021226
),
12031227
options=make_request_options(
12041228
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout

src/agentex/types/task_update_by_id_params.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,6 @@
99

1010

1111
class TaskUpdateByIDParams(TypedDict, total=False):
12+
merge_params: Optional[Dict[str, object]]
13+
1214
task_metadata: Optional[Dict[str, object]]

src/agentex/types/task_update_by_name_params.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,6 @@
99

1010

1111
class TaskUpdateByNameParams(TypedDict, total=False):
12+
merge_params: Optional[Dict[str, object]]
13+
1214
task_metadata: Optional[Dict[str, object]]

tests/api_resources/test_tasks.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -657,6 +657,7 @@ def test_method_update_by_id(self, client: Agentex) -> None:
657657
def test_method_update_by_id_with_all_params(self, client: Agentex) -> None:
658658
task = client.tasks.update_by_id(
659659
task_id="task_id",
660+
merge_params={"foo": "bar"},
660661
task_metadata={"foo": "bar"},
661662
)
662663
assert_matches_type(Task, task, path=["response"])
@@ -708,6 +709,7 @@ def test_method_update_by_name(self, client: Agentex) -> None:
708709
def test_method_update_by_name_with_all_params(self, client: Agentex) -> None:
709710
task = client.tasks.update_by_name(
710711
task_name="task_name",
712+
merge_params={"foo": "bar"},
711713
task_metadata={"foo": "bar"},
712714
)
713715
assert_matches_type(Task, task, path=["response"])
@@ -1384,6 +1386,7 @@ async def test_method_update_by_id(self, async_client: AsyncAgentex) -> None:
13841386
async def test_method_update_by_id_with_all_params(self, async_client: AsyncAgentex) -> None:
13851387
task = await async_client.tasks.update_by_id(
13861388
task_id="task_id",
1389+
merge_params={"foo": "bar"},
13871390
task_metadata={"foo": "bar"},
13881391
)
13891392
assert_matches_type(Task, task, path=["response"])
@@ -1435,6 +1438,7 @@ async def test_method_update_by_name(self, async_client: AsyncAgentex) -> None:
14351438
async def test_method_update_by_name_with_all_params(self, async_client: AsyncAgentex) -> None:
14361439
task = await async_client.tasks.update_by_name(
14371440
task_name="task_name",
1441+
merge_params={"foo": "bar"},
14381442
task_metadata={"foo": "bar"},
14391443
)
14401444
assert_matches_type(Task, task, path=["response"])

0 commit comments

Comments
 (0)