Skip to content

Commit 8dce3ab

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: Add Skill Registry ListSkills and DeleteSkill to SDK
PiperOrigin-RevId: 918571020
1 parent 027acc8 commit 8dce3ab

1 file changed

Lines changed: 27 additions & 51 deletions

File tree

agentplatform/_genai/skills.py

Lines changed: 27 additions & 51 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
import json
2222
import logging
2323
import typing
24-
from typing import Any, Iterator, Optional, Union
24+
from typing import Any, Optional, Union
2525
from urllib.parse import urlencode
2626

2727
from google.genai import _api_module
@@ -509,10 +509,6 @@ def _update(
509509
def _list(
510510
self, *, config: Optional[types.ListSkillsConfigOrDict] = None
511511
) -> types.ListSkillsResponse:
512-
"""
513-
Lists Skills in the Skill Registry.
514-
"""
515-
516512
parameter_model = types._ListSkillsRequestParameters(
517513
config=config,
518514
)
@@ -893,27 +889,6 @@ def update(
893889

894890
return operation
895891

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-
917892
def delete(
918893
self,
919894
*,
@@ -965,6 +940,21 @@ def revisions(self) -> "skill_revisions_module.SkillRevisions":
965940
self._revisions = importlib.import_module(".skill_revisions", __package__)
966941
return self._revisions.SkillRevisions(self._api_client) # type: ignore[no-any-return]
967942

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+
968958

969959
class AsyncSkills(_api_module.BaseModule):
970960
"""Class for managing Skills in the Skill Registry."""
@@ -1263,10 +1253,6 @@ async def _update(
12631253
async def _list(
12641254
self, *, config: Optional[types.ListSkillsConfigOrDict] = None
12651255
) -> types.ListSkillsResponse:
1266-
"""
1267-
Lists Skills in the Skill Registry.
1268-
"""
1269-
12701256
parameter_model = types._ListSkillsRequestParameters(
12711257
config=config,
12721258
)
@@ -1655,27 +1641,6 @@ async def update(
16551641

16561642
return operation
16571643

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-
16791644
async def delete(
16801645
self,
16811646
*,
@@ -1725,3 +1690,14 @@ def revisions(self) -> "skill_revisions_module.AsyncSkillRevisions":
17251690
if self._revisions is None:
17261691
self._revisions = importlib.import_module(".skill_revisions", __package__)
17271692
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

Comments
 (0)