chore: Add experimental tag to SkillToolset

Co-authored-by: Kathy Wu <wukathy@google.com>
PiperOrigin-RevId: 867816959
This commit is contained in:
Kathy Wu
2026-02-09 16:07:35 -08:00
committed by Copybara-Service
parent 19b607684f
commit ec660ed4f0
2 changed files with 9 additions and 0 deletions
@@ -41,6 +41,7 @@ class FeatureName(str, Enum):
PROGRESSIVE_SSE_STREAMING = "PROGRESSIVE_SSE_STREAMING"
PUBSUB_TOOL_CONFIG = "PUBSUB_TOOL_CONFIG"
PUBSUB_TOOLSET = "PUBSUB_TOOLSET"
SKILL_TOOLSET = "SKILL_TOOLSET"
SPANNER_TOOLSET = "SPANNER_TOOLSET"
SPANNER_TOOL_SETTINGS = "SPANNER_TOOL_SETTINGS"
SPANNER_VECTOR_STORE = "SPANNER_VECTOR_STORE"
@@ -123,6 +124,9 @@ _FEATURE_REGISTRY: dict[FeatureName, FeatureConfig] = {
FeatureName.PUBSUB_TOOLSET: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
FeatureName.SKILL_TOOLSET: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
FeatureName.SPANNER_TOOLSET: FeatureConfig(
FeatureStage.EXPERIMENTAL, default_on=True
),
+5
View File
@@ -21,6 +21,8 @@ from typing import Any
from google.genai import types
from ..agents.readonly_context import ReadonlyContext
from ..features import experimental
from ..features import FeatureName
from ..models.llm_request import LlmRequest
from ..skills import models
from ..skills import prompt
@@ -29,6 +31,7 @@ from .base_toolset import BaseToolset
from .tool_context import ToolContext
@experimental(FeatureName.SKILL_TOOLSET)
class LoadSkillTool(BaseTool):
"""Tool to load a skill's instructions."""
@@ -79,6 +82,7 @@ class LoadSkillTool(BaseTool):
}
@experimental(FeatureName.SKILL_TOOLSET)
class LoadSkillResourceTool(BaseTool):
"""Tool to load resources (references or assets) from a skill."""
@@ -167,6 +171,7 @@ class LoadSkillResourceTool(BaseTool):
}
@experimental(FeatureName.SKILL_TOOLSET)
class SkillToolset(BaseToolset):
"""A toolset for managing and interacting with agent skills."""