Python: [BREAKING] changed AIFunction to FunctionTool and @ai_function to @tool (#3413)

* changed AIFunction to FunctionTool and @ai_function to @tool

* test and mypy fixes

* mypy fix

* switch function tool to always_require

* fix noop

* fix github copilot imports

* test fixes

* fix ollama test

* fixes for tests

* fix tests

* reverted change to always_require and extended timeout

* fix test

python/packages/ag-ui/tests/test_ag_ui_client.py

@@ -13,7 +13,7 @@ from agent_framework import (
     ChatResponseUpdate,
     Content,
     Role,
-    ai_function,
+    tool,
 )
 from pytest import MonkeyPatch
 
@@ -231,9 +231,9 @@ class TestAGUIChatClient:
         When server requests a client function, @use_function_invocation decorator
         intercepts and executes it locally. This matches .NET AG-UI implementation.
         """
-        from agent_framework import ai_function
+        from agent_framework import tool
 
-        @ai_function
+        @tool
         def test_tool(param: str) -> str:
             """Test tool."""
             return "result"
@@ -299,7 +299,7 @@ class TestAGUIChatClient:
     async def test_server_tool_calls_not_executed_locally(self, monkeypatch: MonkeyPatch) -> None:
         """Server tools should not trigger local function invocation even when client tools exist."""
 
-        @ai_function
+        @tool
         def client_tool() -> str:
             """Client tool stub."""
             return "client"

python/packages/ag-ui/tests/test_agent_wrapper_comprehensive.py

@@ -680,12 +680,12 @@ async def test_agent_with_use_service_thread_is_true():
 
 async def test_function_approval_mode_executes_tool():
     """Test that function approval with approval_mode='always_require' sends the correct messages."""
-    from agent_framework import ai_function
+    from agent_framework import tool
     from agent_framework.ag_ui import AgentFrameworkAgent
 
     messages_received: list[Any] = []
 
-    @ai_function(
+    @tool(
         name="get_datetime",
         description="Get the current date and time",
         approval_mode="always_require",
@@ -771,12 +771,12 @@ async def test_function_approval_mode_executes_tool():
 
 async def test_function_approval_mode_rejection():
     """Test that function approval rejection creates a rejection response."""
-    from agent_framework import ai_function
+    from agent_framework import tool
     from agent_framework.ag_ui import AgentFrameworkAgent
 
     messages_received: list[Any] = []
 
-    @ai_function(
+    @tool(
         name="delete_all_data",
         description="Delete all user data",
         approval_mode="always_require",

python/packages/ag-ui/tests/test_tooling.py

@@ -2,7 +2,7 @@
 
 from unittest.mock import MagicMock
 
-from agent_framework import ChatAgent, ai_function
+from agent_framework import ChatAgent, tool
 
 from agent_framework_ag_ui._orchestration._tooling import (
     collect_server_tools,
@@ -25,7 +25,7 @@ class MockMCPTool:
         self.is_connected = is_connected
 
 
-@ai_function
+@tool
 def regular_tool() -> str:
     """Regular tool for testing."""
     return "result"
@@ -35,7 +35,7 @@ def _create_chat_agent_with_tool(tool_name: str = "regular_tool") -> ChatAgent:
     """Create a ChatAgent with a mocked chat client and a simple tool.
 
     Note: tool_name parameter is kept for API compatibility but the tool
-    will always be named 'regular_tool' since ai_function uses the function name.
+    will always be named 'regular_tool' since tool uses the function name.
     """
     mock_chat_client = MagicMock()
     return ChatAgent(chat_client=mock_chat_client, tools=[regular_tool])

python/packages/ag-ui/tests/test_utils.py

@@ -252,13 +252,13 @@ def test_make_json_safe_dataclass_with_nested_to_dict_object():
     assert json_str is not None
 
 
-def test_convert_tools_to_agui_format_with_ai_function():
-    """Test converting AIFunction to AG-UI format."""
-    from agent_framework import ai_function
+def test_convert_tools_to_agui_format_with_tool():
+    """Test converting FunctionTool to AG-UI format."""
+    from agent_framework import tool
 
     from agent_framework_ag_ui._utils import convert_tools_to_agui_format
 
-    @ai_function
+    @tool
     def test_func(param: str, count: int = 5) -> str:
         """Test function."""
         return f"{param} {count}"
@@ -318,11 +318,11 @@ def test_convert_tools_to_agui_format_with_none():
 
 def test_convert_tools_to_agui_format_with_single_tool():
     """Test converting single tool (not in list)."""
-    from agent_framework import ai_function
+    from agent_framework import tool
 
     from agent_framework_ag_ui._utils import convert_tools_to_agui_format
 
-    @ai_function
+    @tool
     def single_tool(arg: str) -> str:
         """Single tool."""
         return arg
@@ -336,16 +336,16 @@ def test_convert_tools_to_agui_format_with_single_tool():
 
 def test_convert_tools_to_agui_format_with_multiple_tools():
     """Test converting multiple tools."""
-    from agent_framework import ai_function
+    from agent_framework import tool
 
     from agent_framework_ag_ui._utils import convert_tools_to_agui_format
 
-    @ai_function
+    @tool
     def tool1(x: int) -> int:
         """Tool 1."""
         return x
 
-    @ai_function
+    @tool
     def tool2(y: str) -> str:
         """Tool 2."""
         return y