|
20 | 20 | from ..._base_client import make_request_options |
21 | 21 | from ...types.agents import ( |
22 | 22 | schedule_list_params, |
| 23 | + schedule_skip_params, |
23 | 24 | schedule_pause_params, |
24 | 25 | schedule_create_params, |
25 | 26 | schedule_resume_params, |
| 27 | + schedule_unskip_params, |
26 | 28 | schedule_update_params, |
27 | 29 | schedule_pause_by_name_params, |
28 | 30 | schedule_resume_by_name_params, |
29 | 31 | schedule_update_by_name_params, |
30 | 32 | ) |
31 | 33 | from ...types.shared.delete_response import DeleteResponse |
32 | 34 | from ...types.agents.schedule_list_response import ScheduleListResponse |
| 35 | +from ...types.agents.schedule_skip_response import ScheduleSkipResponse |
33 | 36 | from ...types.agents.schedule_pause_response import SchedulePauseResponse |
34 | 37 | from ...types.agents.schedule_create_response import ScheduleCreateResponse |
35 | 38 | from ...types.agents.schedule_resume_response import ScheduleResumeResponse |
| 39 | +from ...types.agents.schedule_unskip_response import ScheduleUnskipResponse |
36 | 40 | from ...types.agents.schedule_update_response import ScheduleUpdateResponse |
37 | 41 | from ...types.agents.schedule_trigger_response import ScheduleTriggerResponse |
38 | 42 | from ...types.agents.schedule_retrieve_response import ScheduleRetrieveResponse |
@@ -579,6 +583,48 @@ def retrieve_by_name( |
579 | 583 | cast_to=ScheduleRetrieveByNameResponse, |
580 | 584 | ) |
581 | 585 |
|
| 586 | + def skip( |
| 587 | + self, |
| 588 | + schedule_id: str, |
| 589 | + *, |
| 590 | + agent_id: str, |
| 591 | + scheduled_time: Union[str, datetime], |
| 592 | + # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
| 593 | + # The extra values given here take precedence over values defined on the client or passed to this method. |
| 594 | + extra_headers: Headers | None = None, |
| 595 | + extra_query: Query | None = None, |
| 596 | + extra_body: Body | None = None, |
| 597 | + timeout: float | httpx.Timeout | None | NotGiven = not_given, |
| 598 | + ) -> ScheduleSkipResponse: |
| 599 | + """ |
| 600 | + Skip a recurring fire of the schedule. |
| 601 | +
|
| 602 | + Args: |
| 603 | + scheduled_time: Specific scheduled fire time to skip. |
| 604 | +
|
| 605 | + extra_headers: Send extra headers |
| 606 | +
|
| 607 | + extra_query: Add additional query parameters to the request |
| 608 | +
|
| 609 | + extra_body: Add additional JSON properties to the request |
| 610 | +
|
| 611 | + timeout: Override the client-level default timeout for this request, in seconds |
| 612 | + """ |
| 613 | + if not agent_id: |
| 614 | + raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}") |
| 615 | + if not schedule_id: |
| 616 | + raise ValueError(f"Expected a non-empty value for `schedule_id` but received {schedule_id!r}") |
| 617 | + return self._post( |
| 618 | + path_template( |
| 619 | + "/agents/{agent_id}/schedules/{schedule_id}/skip", agent_id=agent_id, schedule_id=schedule_id |
| 620 | + ), |
| 621 | + body=maybe_transform({"scheduled_time": scheduled_time}, schedule_skip_params.ScheduleSkipParams), |
| 622 | + options=make_request_options( |
| 623 | + extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout |
| 624 | + ), |
| 625 | + cast_to=ScheduleSkipResponse, |
| 626 | + ) |
| 627 | + |
582 | 628 | def trigger( |
583 | 629 | self, |
584 | 630 | schedule_id: str, |
@@ -654,6 +700,48 @@ def trigger_by_name( |
654 | 700 | cast_to=ScheduleTriggerByNameResponse, |
655 | 701 | ) |
656 | 702 |
|
| 703 | + def unskip( |
| 704 | + self, |
| 705 | + schedule_id: str, |
| 706 | + *, |
| 707 | + agent_id: str, |
| 708 | + scheduled_time: Union[str, datetime], |
| 709 | + # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
| 710 | + # The extra values given here take precedence over values defined on the client or passed to this method. |
| 711 | + extra_headers: Headers | None = None, |
| 712 | + extra_query: Query | None = None, |
| 713 | + extra_body: Body | None = None, |
| 714 | + timeout: float | httpx.Timeout | None | NotGiven = not_given, |
| 715 | + ) -> ScheduleUnskipResponse: |
| 716 | + """ |
| 717 | + Remove a skip for a recurring fire of the schedule. |
| 718 | +
|
| 719 | + Args: |
| 720 | + scheduled_time: Specific scheduled fire time to unskip. |
| 721 | +
|
| 722 | + extra_headers: Send extra headers |
| 723 | +
|
| 724 | + extra_query: Add additional query parameters to the request |
| 725 | +
|
| 726 | + extra_body: Add additional JSON properties to the request |
| 727 | +
|
| 728 | + timeout: Override the client-level default timeout for this request, in seconds |
| 729 | + """ |
| 730 | + if not agent_id: |
| 731 | + raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}") |
| 732 | + if not schedule_id: |
| 733 | + raise ValueError(f"Expected a non-empty value for `schedule_id` but received {schedule_id!r}") |
| 734 | + return self._post( |
| 735 | + path_template( |
| 736 | + "/agents/{agent_id}/schedules/{schedule_id}/unskip", agent_id=agent_id, schedule_id=schedule_id |
| 737 | + ), |
| 738 | + body=maybe_transform({"scheduled_time": scheduled_time}, schedule_unskip_params.ScheduleUnskipParams), |
| 739 | + options=make_request_options( |
| 740 | + extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout |
| 741 | + ), |
| 742 | + cast_to=ScheduleUnskipResponse, |
| 743 | + ) |
| 744 | + |
657 | 745 | def update_by_name( |
658 | 746 | self, |
659 | 747 | path_name: str, |
@@ -1274,6 +1362,50 @@ async def retrieve_by_name( |
1274 | 1362 | cast_to=ScheduleRetrieveByNameResponse, |
1275 | 1363 | ) |
1276 | 1364 |
|
| 1365 | + async def skip( |
| 1366 | + self, |
| 1367 | + schedule_id: str, |
| 1368 | + *, |
| 1369 | + agent_id: str, |
| 1370 | + scheduled_time: Union[str, datetime], |
| 1371 | + # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
| 1372 | + # The extra values given here take precedence over values defined on the client or passed to this method. |
| 1373 | + extra_headers: Headers | None = None, |
| 1374 | + extra_query: Query | None = None, |
| 1375 | + extra_body: Body | None = None, |
| 1376 | + timeout: float | httpx.Timeout | None | NotGiven = not_given, |
| 1377 | + ) -> ScheduleSkipResponse: |
| 1378 | + """ |
| 1379 | + Skip a recurring fire of the schedule. |
| 1380 | +
|
| 1381 | + Args: |
| 1382 | + scheduled_time: Specific scheduled fire time to skip. |
| 1383 | +
|
| 1384 | + extra_headers: Send extra headers |
| 1385 | +
|
| 1386 | + extra_query: Add additional query parameters to the request |
| 1387 | +
|
| 1388 | + extra_body: Add additional JSON properties to the request |
| 1389 | +
|
| 1390 | + timeout: Override the client-level default timeout for this request, in seconds |
| 1391 | + """ |
| 1392 | + if not agent_id: |
| 1393 | + raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}") |
| 1394 | + if not schedule_id: |
| 1395 | + raise ValueError(f"Expected a non-empty value for `schedule_id` but received {schedule_id!r}") |
| 1396 | + return await self._post( |
| 1397 | + path_template( |
| 1398 | + "/agents/{agent_id}/schedules/{schedule_id}/skip", agent_id=agent_id, schedule_id=schedule_id |
| 1399 | + ), |
| 1400 | + body=await async_maybe_transform( |
| 1401 | + {"scheduled_time": scheduled_time}, schedule_skip_params.ScheduleSkipParams |
| 1402 | + ), |
| 1403 | + options=make_request_options( |
| 1404 | + extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout |
| 1405 | + ), |
| 1406 | + cast_to=ScheduleSkipResponse, |
| 1407 | + ) |
| 1408 | + |
1277 | 1409 | async def trigger( |
1278 | 1410 | self, |
1279 | 1411 | schedule_id: str, |
@@ -1349,6 +1481,50 @@ async def trigger_by_name( |
1349 | 1481 | cast_to=ScheduleTriggerByNameResponse, |
1350 | 1482 | ) |
1351 | 1483 |
|
| 1484 | + async def unskip( |
| 1485 | + self, |
| 1486 | + schedule_id: str, |
| 1487 | + *, |
| 1488 | + agent_id: str, |
| 1489 | + scheduled_time: Union[str, datetime], |
| 1490 | + # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
| 1491 | + # The extra values given here take precedence over values defined on the client or passed to this method. |
| 1492 | + extra_headers: Headers | None = None, |
| 1493 | + extra_query: Query | None = None, |
| 1494 | + extra_body: Body | None = None, |
| 1495 | + timeout: float | httpx.Timeout | None | NotGiven = not_given, |
| 1496 | + ) -> ScheduleUnskipResponse: |
| 1497 | + """ |
| 1498 | + Remove a skip for a recurring fire of the schedule. |
| 1499 | +
|
| 1500 | + Args: |
| 1501 | + scheduled_time: Specific scheduled fire time to unskip. |
| 1502 | +
|
| 1503 | + extra_headers: Send extra headers |
| 1504 | +
|
| 1505 | + extra_query: Add additional query parameters to the request |
| 1506 | +
|
| 1507 | + extra_body: Add additional JSON properties to the request |
| 1508 | +
|
| 1509 | + timeout: Override the client-level default timeout for this request, in seconds |
| 1510 | + """ |
| 1511 | + if not agent_id: |
| 1512 | + raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}") |
| 1513 | + if not schedule_id: |
| 1514 | + raise ValueError(f"Expected a non-empty value for `schedule_id` but received {schedule_id!r}") |
| 1515 | + return await self._post( |
| 1516 | + path_template( |
| 1517 | + "/agents/{agent_id}/schedules/{schedule_id}/unskip", agent_id=agent_id, schedule_id=schedule_id |
| 1518 | + ), |
| 1519 | + body=await async_maybe_transform( |
| 1520 | + {"scheduled_time": scheduled_time}, schedule_unskip_params.ScheduleUnskipParams |
| 1521 | + ), |
| 1522 | + options=make_request_options( |
| 1523 | + extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout |
| 1524 | + ), |
| 1525 | + cast_to=ScheduleUnskipResponse, |
| 1526 | + ) |
| 1527 | + |
1352 | 1528 | async def update_by_name( |
1353 | 1529 | self, |
1354 | 1530 | path_name: str, |
@@ -1472,12 +1648,18 @@ def __init__(self, schedules: SchedulesResource) -> None: |
1472 | 1648 | self.retrieve_by_name = to_raw_response_wrapper( |
1473 | 1649 | schedules.retrieve_by_name, |
1474 | 1650 | ) |
| 1651 | + self.skip = to_raw_response_wrapper( |
| 1652 | + schedules.skip, |
| 1653 | + ) |
1475 | 1654 | self.trigger = to_raw_response_wrapper( |
1476 | 1655 | schedules.trigger, |
1477 | 1656 | ) |
1478 | 1657 | self.trigger_by_name = to_raw_response_wrapper( |
1479 | 1658 | schedules.trigger_by_name, |
1480 | 1659 | ) |
| 1660 | + self.unskip = to_raw_response_wrapper( |
| 1661 | + schedules.unskip, |
| 1662 | + ) |
1481 | 1663 | self.update_by_name = to_raw_response_wrapper( |
1482 | 1664 | schedules.update_by_name, |
1483 | 1665 | ) |
@@ -1520,12 +1702,18 @@ def __init__(self, schedules: AsyncSchedulesResource) -> None: |
1520 | 1702 | self.retrieve_by_name = async_to_raw_response_wrapper( |
1521 | 1703 | schedules.retrieve_by_name, |
1522 | 1704 | ) |
| 1705 | + self.skip = async_to_raw_response_wrapper( |
| 1706 | + schedules.skip, |
| 1707 | + ) |
1523 | 1708 | self.trigger = async_to_raw_response_wrapper( |
1524 | 1709 | schedules.trigger, |
1525 | 1710 | ) |
1526 | 1711 | self.trigger_by_name = async_to_raw_response_wrapper( |
1527 | 1712 | schedules.trigger_by_name, |
1528 | 1713 | ) |
| 1714 | + self.unskip = async_to_raw_response_wrapper( |
| 1715 | + schedules.unskip, |
| 1716 | + ) |
1529 | 1717 | self.update_by_name = async_to_raw_response_wrapper( |
1530 | 1718 | schedules.update_by_name, |
1531 | 1719 | ) |
@@ -1568,12 +1756,18 @@ def __init__(self, schedules: SchedulesResource) -> None: |
1568 | 1756 | self.retrieve_by_name = to_streamed_response_wrapper( |
1569 | 1757 | schedules.retrieve_by_name, |
1570 | 1758 | ) |
| 1759 | + self.skip = to_streamed_response_wrapper( |
| 1760 | + schedules.skip, |
| 1761 | + ) |
1571 | 1762 | self.trigger = to_streamed_response_wrapper( |
1572 | 1763 | schedules.trigger, |
1573 | 1764 | ) |
1574 | 1765 | self.trigger_by_name = to_streamed_response_wrapper( |
1575 | 1766 | schedules.trigger_by_name, |
1576 | 1767 | ) |
| 1768 | + self.unskip = to_streamed_response_wrapper( |
| 1769 | + schedules.unskip, |
| 1770 | + ) |
1577 | 1771 | self.update_by_name = to_streamed_response_wrapper( |
1578 | 1772 | schedules.update_by_name, |
1579 | 1773 | ) |
@@ -1616,12 +1810,18 @@ def __init__(self, schedules: AsyncSchedulesResource) -> None: |
1616 | 1810 | self.retrieve_by_name = async_to_streamed_response_wrapper( |
1617 | 1811 | schedules.retrieve_by_name, |
1618 | 1812 | ) |
| 1813 | + self.skip = async_to_streamed_response_wrapper( |
| 1814 | + schedules.skip, |
| 1815 | + ) |
1619 | 1816 | self.trigger = async_to_streamed_response_wrapper( |
1620 | 1817 | schedules.trigger, |
1621 | 1818 | ) |
1622 | 1819 | self.trigger_by_name = async_to_streamed_response_wrapper( |
1623 | 1820 | schedules.trigger_by_name, |
1624 | 1821 | ) |
| 1822 | + self.unskip = async_to_streamed_response_wrapper( |
| 1823 | + schedules.unskip, |
| 1824 | + ) |
1625 | 1825 | self.update_by_name = async_to_streamed_response_wrapper( |
1626 | 1826 | schedules.update_by_name, |
1627 | 1827 | ) |
0 commit comments