Python: Forward runtime kwargs to skill resource functions (#4417)

* support code skills

* address pr review comments

* address package and syntax checks

* address pr review comments

* address pr review comment

* address failed check

* rename agentskill and agetnskillprovider

* move agent skills related assets to _skills.py

* address pr review comments

* address review comments

* support kwargs

* address pr review feedback
This commit is contained in:
SergeyMenshykh
2026-03-05 18:01:25 +00:00
committed by GitHub
Unverified
parent 55ddd841b7
commit 8bf4235f4e
4 changed files with 75 additions and 13 deletions
@@ -107,6 +107,15 @@ class SkillResource:
self.content = content
self.function = function
# Precompute whether the function accepts **kwargs to avoid
# repeated inspect.signature() calls on every invocation.
self._accepts_kwargs: bool = False
if function is not None:
sig = inspect.signature(function)
self._accepts_kwargs = any(
p.kind == inspect.Parameter.VAR_KEYWORD for p in sig.parameters.values()
)
class Skill:
"""A skill definition with optional resources.
@@ -511,7 +520,7 @@ class SkillsProvider(BaseContextProvider):
return content
async def _read_skill_resource(self, skill_name: str, resource_name: str) -> str:
async def _read_skill_resource(self, skill_name: str, resource_name: str, **kwargs: Any) -> str:
"""Read a named resource from a skill.
Resolves the resource by case-insensitive name lookup. Static
@@ -521,6 +530,9 @@ class SkillsProvider(BaseContextProvider):
Args:
skill_name: The name of the owning skill.
resource_name: The resource name to look up (case-insensitive).
**kwargs: Runtime keyword arguments forwarded to resource functions
that accept ``**kwargs`` (e.g. arguments passed via
``agent.run(user_id="123")``).
Returns:
The resource content string, or a user-facing error message on
@@ -550,9 +562,11 @@ class SkillsProvider(BaseContextProvider):
if resource.function is not None:
try:
if inspect.iscoroutinefunction(resource.function):
result = await resource.function()
result = (
await resource.function(**kwargs) if resource._accepts_kwargs else await resource.function()
)
else:
result = resource.function()
result = resource.function(**kwargs) if resource._accepts_kwargs else resource.function()
return str(result)
except Exception as exc:
logger.exception("Failed to read resource '%s' from skill '%s'", resource_name, skill_name)
@@ -6,6 +6,7 @@ from __future__ import annotations
import os
from pathlib import Path
from typing import Any
from unittest.mock import AsyncMock
import pytest
@@ -993,6 +994,42 @@ class TestSkillsProviderCodeSkill:
result = await provider._read_skill_resource("prog-skill", "nonexistent")
assert result.startswith("Error:")
async def test_read_callable_resource_sync_with_kwargs(self) -> None:
skill = Skill(name="prog-skill", description="A skill.", content="Body")
@skill.resource
def get_user_config(**kwargs: Any) -> str:
user_id = kwargs.get("user_id", "unknown")
return f"config for {user_id}"
provider = SkillsProvider(skills=[skill])
result = await provider._read_skill_resource("prog-skill", "get_user_config", user_id="user_123")
assert result == "config for user_123"
async def test_read_callable_resource_async_with_kwargs(self) -> None:
skill = Skill(name="prog-skill", description="A skill.", content="Body")
@skill.resource
async def get_user_data(**kwargs: Any) -> str:
token = kwargs.get("auth_token", "none")
return f"data with token={token}"
provider = SkillsProvider(skills=[skill])
result = await provider._read_skill_resource("prog-skill", "get_user_data", auth_token="abc")
assert result == "data with token=abc"
async def test_read_callable_resource_without_kwargs_ignores_extra_args(self) -> None:
"""Resource functions without **kwargs should still work when kwargs are passed."""
skill = Skill(name="prog-skill", description="A skill.", content="Body")
@skill.resource
def static_resource() -> str:
return "static content"
provider = SkillsProvider(skills=[skill])
result = await provider._read_skill_resource("prog-skill", "static_resource", user_id="ignored")
assert result == "static content"
async def test_before_run_injects_code_skills(self) -> None:
skill = Skill(name="prog-skill", description="A code-defined skill.", content="Body")
provider = SkillsProvider(skills=[skill])
@@ -4,12 +4,13 @@ This sample demonstrates how to create **Agent Skills** in Python code, without
## What are Code-Defined Skills?
While file-based skills use `SKILL.md` files discovered on disk, code-defined skills let you define skills entirely in Python using `Skill` and `SkillResource` classes. Two patterns are shown:
While file-based skills use `SKILL.md` files discovered on disk, code-defined skills let you define skills entirely in Python using `Skill` and `SkillResource` classes. Three patterns are shown:
1. **Basic Code Skill** — Create a `Skill` directly with static resources (inline content)
2. **Dynamic Resources** — Attach callable resources via the `@skill.resource` decorator that generate content at invocation time
3. **Dynamic Resources with kwargs** — Attach a callable resource that accepts `**kwargs` to receive runtime arguments passed via `agent.run()`, useful for injecting request-scoped context (user tokens, session data)
Both patterns can be combined with file-based skills in a single `SkillsProvider`.
All patterns can be combined with file-based skills in a single `SkillsProvider`.
## Project Structure
@@ -47,7 +48,7 @@ uv run samples/02-agents/skills/code_skill/code_skill.py
The sample runs two examples:
1. **Code style question** — Uses Pattern 1 (static resources): the agent loads the `code-style` skill and reads the `style-guide` resource to answer naming convention questions
2. **Project info question** — Uses Pattern 2 (dynamic resources): the agent reads dynamically generated `environment` and `team-roster` resources
2. **Project info question** — Uses Patterns 2 & 3 (dynamic resources with kwargs): the agent reads the dynamically generated `team-roster` resource and the `environment` resource which receives `app_version` via runtime kwargs
## Learn More
@@ -4,6 +4,7 @@ import asyncio
import os
import sys
from textwrap import dedent
from typing import Any
from agent_framework import Agent, Skill, SkillResource, SkillsProvider
from agent_framework.azure import AzureOpenAIResponsesClient
@@ -14,7 +15,7 @@ from dotenv import load_dotenv
Code-Defined Agent Skills — Define skills in Python code
This sample demonstrates how to create Agent Skills in code,
without needing SKILL.md files on disk. Two patterns are shown:
without needing SKILL.md files on disk. Three patterns are shown:
Pattern 1: Basic Code Skill
Create a Skill instance directly with static resources (inline content).
@@ -24,6 +25,11 @@ Pattern 2: Dynamic Resources
decorator. Resources can be sync or async functions that generate content at
invocation time.
Pattern 3: Dynamic Resources with kwargs
Attach a callable resource that accepts **kwargs to receive runtime
arguments passed via agent.run(). This is useful for injecting
request-scoped context (user tokens, session data) into skill resources.
Both patterns can be combined with file-based skills in a single SkillsProvider.
"""
@@ -72,12 +78,15 @@ project_info_skill = Skill(
@project_info_skill.resource
def environment() -> str:
def environment(**kwargs: Any) -> str:
"""Get current environment configuration."""
# Access runtime kwargs passed via agent.run(app_version="...")
app_version = kwargs.get("app_version", "unknown")
env = os.environ.get("APP_ENV", "development")
region = os.environ.get("APP_REGION", "us-east-1")
return f"""\
# Environment Configuration
- App Version: {app_version}
- Environment: {env}
- Region: {region}
- Python: {sys.version}
@@ -124,10 +133,11 @@ async def main() -> None:
response = await agent.run("What naming convention should I use for class attributes?")
print(f"Agent: {response}\n")
# Example 2: Project info question (Pattern 2 — dynamic resources)
# Example 2: Project info question (Pattern 2 & 3 — dynamic resources with kwargs)
print("Example 2: Project info question")
print("---------------------------------")
response = await agent.run("What environment are we running in and who is on the team?")
# Pass app_version as a runtime kwarg; it flows to the environment() resource via **kwargs
response = await agent.run("What environment are we running in and who is on the team?", app_version="2.4.1")
print(f"Agent: {response}\n")
"""
@@ -141,9 +151,9 @@ async def main() -> None:
Example 2: Project info question
---------------------------------
Agent: We're running in the development environment in us-east-1.
The team consists of Alice Chen (Tech Lead), Bob Smith (Backend Engineer),
and Carol Davis (Frontend Engineer).
Agent: We're running app version 2.4.1 in the development environment
in us-east-1. The team consists of Alice Chen (Tech Lead), Bob Smith
(Backend Engineer), and Carol Davis (Frontend Engineer).
"""