|
21 | 21 | import json |
22 | 22 | import logging |
23 | 23 | import typing |
24 | | -from typing import Any, Iterator, Optional, Union |
| 24 | +from typing import Any, Optional, Union |
25 | 25 | from urllib.parse import urlencode |
26 | 26 |
|
27 | 27 | from google.genai import _api_module |
@@ -509,10 +509,6 @@ def _update( |
509 | 509 | def _list( |
510 | 510 | self, *, config: Optional[types.ListSkillsConfigOrDict] = None |
511 | 511 | ) -> types.ListSkillsResponse: |
512 | | - """ |
513 | | - Lists Skills in the Skill Registry. |
514 | | - """ |
515 | | - |
516 | 512 | parameter_model = types._ListSkillsRequestParameters( |
517 | 513 | config=config, |
518 | 514 | ) |
@@ -893,27 +889,6 @@ def update( |
893 | 889 |
|
894 | 890 | return operation |
895 | 891 |
|
896 | | - def list( |
897 | | - self, |
898 | | - *, |
899 | | - config: Optional[types.ListSkillsConfigOrDict] = None, |
900 | | - ) -> Iterator[types.Skill]: |
901 | | - """Lists Skills in the Skill Registry. |
902 | | -
|
903 | | - Args: |
904 | | - config (ListSkillsConfigOrDict): |
905 | | - Optional. Additional configuration for listing Skills. |
906 | | -
|
907 | | - Returns: |
908 | | - Iterator[Skill]: An iterator (Pager) of Skills. |
909 | | - """ |
910 | | - return Pager( |
911 | | - "skills", |
912 | | - self._list, |
913 | | - self._list(config=config), |
914 | | - config, |
915 | | - ) |
916 | | - |
917 | 892 | def delete( |
918 | 893 | self, |
919 | 894 | *, |
@@ -965,6 +940,21 @@ def revisions(self) -> "skill_revisions_module.SkillRevisions": |
965 | 940 | self._revisions = importlib.import_module(".skill_revisions", __package__) |
966 | 941 | return self._revisions.SkillRevisions(self._api_client) # type: ignore[no-any-return] |
967 | 942 |
|
| 943 | + def list( |
| 944 | + self, *, config: Optional[types.ListSkillsConfigOrDict] = None |
| 945 | + ) -> Pager[types.Skill]: |
| 946 | + """ |
| 947 | + Lists Skills in the Skill Registry. |
| 948 | + """ |
| 949 | + |
| 950 | + list_request = self._list |
| 951 | + return Pager( |
| 952 | + "skills", |
| 953 | + list_request, |
| 954 | + self._list(config=config), |
| 955 | + config, |
| 956 | + ) |
| 957 | + |
968 | 958 |
|
969 | 959 | class AsyncSkills(_api_module.BaseModule): |
970 | 960 | """Class for managing Skills in the Skill Registry.""" |
@@ -1263,10 +1253,6 @@ async def _update( |
1263 | 1253 | async def _list( |
1264 | 1254 | self, *, config: Optional[types.ListSkillsConfigOrDict] = None |
1265 | 1255 | ) -> types.ListSkillsResponse: |
1266 | | - """ |
1267 | | - Lists Skills in the Skill Registry. |
1268 | | - """ |
1269 | | - |
1270 | 1256 | parameter_model = types._ListSkillsRequestParameters( |
1271 | 1257 | config=config, |
1272 | 1258 | ) |
@@ -1655,27 +1641,6 @@ async def update( |
1655 | 1641 |
|
1656 | 1642 | return operation |
1657 | 1643 |
|
1658 | | - async def list( |
1659 | | - self, |
1660 | | - *, |
1661 | | - config: Optional[types.ListSkillsConfigOrDict] = None, |
1662 | | - ) -> AsyncPager[types.Skill]: |
1663 | | - """Lists Skills in the Skill Registry asynchronously. |
1664 | | -
|
1665 | | - Args: |
1666 | | - config (ListSkillsConfigOrDict): |
1667 | | - Optional. Additional configuration for listing Skills. |
1668 | | -
|
1669 | | - Returns: |
1670 | | - AsyncPager[Skill]: An async pager of Skills. |
1671 | | - """ |
1672 | | - return AsyncPager( |
1673 | | - "skills", |
1674 | | - self._list, |
1675 | | - await self._list(config=config), |
1676 | | - config, |
1677 | | - ) |
1678 | | - |
1679 | 1644 | async def delete( |
1680 | 1645 | self, |
1681 | 1646 | *, |
@@ -1725,3 +1690,14 @@ def revisions(self) -> "skill_revisions_module.AsyncSkillRevisions": |
1725 | 1690 | if self._revisions is None: |
1726 | 1691 | self._revisions = importlib.import_module(".skill_revisions", __package__) |
1727 | 1692 | return self._revisions.AsyncSkillRevisions(self._api_client) # type: ignore[no-any-return] |
| 1693 | + |
| 1694 | + async def list( |
| 1695 | + self, *, config: Optional[types.ListSkillsConfigOrDict] = None |
| 1696 | + ) -> AsyncPager[types.Skill]: |
| 1697 | + list_request = self._list |
| 1698 | + return AsyncPager( |
| 1699 | + "skills", |
| 1700 | + list_request, |
| 1701 | + await self._list(config=config), |
| 1702 | + config, |
| 1703 | + ) |
0 commit comments