mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Jacob Alber
94c5d59984
* feat: ModelClient and content types * refactor: Pythonify ChatResponseFormat and ChatRole * feat: Add guardrail interfaces * refactor: Remove CancellationToken * feat: Solidify the Usage APIs * Adds well-known keys for additional_counts, and guidance for how to avoid collisions between providers * Implement sum-aggregation for usage * refactor: Move AITool out of model_client * refactor: Copy editing * fix: CI checks (pyupgrade, ruff, etc.) * ci: Fix pre-commit to use pyright in uv venv The existing pyright precommit hook inside of python-pyright is no longer being maintained by the owner (see https://github.com/RobertCraigie/pyright-python/issues/265) The fix is to define the hook ourselves, relying on `uv run` to drive it. In order for that to work right we need to use the "system" language to break out of the sandbox. * fix: Pyright error fixes * docs: Update models and types design docs * Python: Refinement of content types and model client (#112) * refinement of structure and buildup with ports from semantigen * refined the data and uri contents * refined chat response and updates * moved things and added tests * moved out of src folder * fixed imports and tests * small tweaks * missing build system * upgrade * add mypy * fixed typing for types * fix tests * fixed tool * disable json checks on vscode * remove print --------- Co-authored-by: Eduard van Valkenburg <eavanvalkenburg@users.noreply.github.com> Co-authored-by: eavanvalkenburg <github@vanvalkenburg.eu>
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
from agent_framework import AITool, ai_function
|
|
|
|
|
|
def test_ai_function_decorator():
|
|
"""Test the ai_function decorator."""
|
|
|
|
@ai_function(name="test_tool", description="A test tool")
|
|
def test_tool(x: int, y: int) -> int:
|
|
"""A simple function that adds two numbers."""
|
|
return x + y
|
|
|
|
assert isinstance(test_tool, AITool)
|
|
assert test_tool.name == "test_tool"
|
|
assert test_tool.description == "A test tool"
|
|
assert test_tool.model_json_schema() == {
|
|
"properties": {"x": {"title": "X", "type": "integer"}, "y": {"title": "Y", "type": "integer"}},
|
|
"required": ["x", "y"],
|
|
"title": "test_tool_input",
|
|
"type": "object",
|
|
}
|
|
assert test_tool(1, 2) == 3
|
|
|
|
|
|
async def test_ai_function_decorator_with_async():
|
|
"""Test the ai_function decorator with an async function."""
|
|
|
|
@ai_function(name="async_test_tool", description="An async test tool")
|
|
async def async_test_tool(x: int, y: int) -> int:
|
|
"""An async function that adds two numbers."""
|
|
return x + y
|
|
|
|
assert isinstance(async_test_tool, AITool)
|
|
assert async_test_tool.name == "async_test_tool"
|
|
assert async_test_tool.description == "An async test tool"
|
|
assert async_test_tool.model_json_schema() == {
|
|
"properties": {"x": {"title": "X", "type": "integer"}, "y": {"title": "Y", "type": "integer"}},
|
|
"required": ["x", "y"],
|
|
"title": "async_test_tool_input",
|
|
"type": "object",
|
|
}
|
|
assert (await async_test_tool(1, 2)) == 3
|