Python: Agent Name Sanitization (#1523)

* a2a agent name sanitization

* fix

* small fix

---------

Co-authored-by: Dmytro Struk <13853051+dmytrostruk@users.noreply.github.com>

python/packages/core/tests/core/test_agents.py

@@ -1,7 +1,9 @@
 # Copyright (c) Microsoft. All rights reserved.
 
+import contextlib
 from collections.abc import AsyncIterable, MutableSequence, Sequence
 from typing import Any
+from unittest.mock import AsyncMock, MagicMock
 from uuid import uuid4
 
 from pytest import raises
@@ -23,6 +25,7 @@ from agent_framework import (
     Role,
     TextContent,
 )
+from agent_framework._mcp import MCPTool
 from agent_framework.exceptions import AgentExecutionException
 
 
@@ -506,3 +509,69 @@ async def test_chat_agent_as_tool_with_async_stream_callback(chat_client: ChatCl
     # Result should be concatenation of all streaming updates
     expected_text = "".join(update.text for update in collected_updates)
     assert result == expected_text
+
+
+async def test_chat_agent_as_tool_name_sanitization(chat_client: ChatClientProtocol) -> None:
+    """Test as_tool name sanitization."""
+    test_cases = [
+        ("Invoice & Billing Agent", "Invoice_Billing_Agent"),
+        ("Travel & Logistics Agent", "Travel_Logistics_Agent"),
+        ("Agent@Company.com", "Agent_Company_com"),
+        ("Agent___Multiple___Underscores", "Agent_Multiple_Underscores"),
+        ("123Agent", "_123Agent"),  # Test digit prefix handling
+        ("9to5Helper", "_9to5Helper"),  # Another digit prefix case
+        ("@@@", "agent"),  # Test empty sanitization fallback
+    ]
+
+    for agent_name, expected_tool_name in test_cases:
+        agent = ChatAgent(chat_client=chat_client, name=agent_name, description="Test agent")
+        tool = agent.as_tool()
+        assert tool.name == expected_tool_name, f"Expected {expected_tool_name}, got {tool.name} for input {agent_name}"
+
+
+async def test_chat_agent_as_mcp_server_basic(chat_client: ChatClientProtocol) -> None:
+    """Test basic as_mcp_server functionality."""
+    agent = ChatAgent(chat_client=chat_client, name="TestAgent", description="Test agent for MCP")
+
+    # Create MCP server with default parameters
+    server = agent.as_mcp_server()
+
+    # Verify server is created
+    assert server is not None
+    assert hasattr(server, "name")
+    assert hasattr(server, "version")
+
+
+async def test_chat_agent_run_with_mcp_tools(chat_client: ChatClientProtocol) -> None:
+    """Test run method with MCP tools to cover MCP tool handling code."""
+    agent = ChatAgent(chat_client=chat_client, name="TestAgent", description="Test agent")
+
+    # Create a mock MCP tool
+    mock_mcp_tool = MagicMock(spec=MCPTool)
+    mock_mcp_tool.is_connected = False
+    mock_mcp_tool.functions = [MagicMock()]
+
+    # Mock the async context manager entry
+    mock_mcp_tool.__aenter__ = AsyncMock(return_value=mock_mcp_tool)
+    mock_mcp_tool.__aexit__ = AsyncMock(return_value=None)
+
+    # Test run with MCP tools - this should hit the MCP tool handling code
+    with contextlib.suppress(Exception):
+        # We expect this to fail since we're using mocks, but we want to exercise the code path
+        await agent.run(messages="Test message", tools=[mock_mcp_tool])
+
+
+async def test_chat_agent_with_local_mcp_tools(chat_client: ChatClientProtocol) -> None:
+    """Test agent initialization with local MCP tools."""
+    # Create a mock MCP tool
+    mock_mcp_tool = MagicMock(spec=MCPTool)
+    mock_mcp_tool.is_connected = False
+    mock_mcp_tool.__aenter__ = AsyncMock(return_value=mock_mcp_tool)
+    mock_mcp_tool.__aexit__ = AsyncMock(return_value=None)
+
+    # Test agent with MCP tools in constructor
+    with contextlib.suppress(Exception):
+        agent = ChatAgent(chat_client=chat_client, name="TestAgent", description="Test agent", tools=[mock_mcp_tool])
+        # Test async context manager with MCP tools
+        async with agent:
+            pass

python/packages/core/tests/openai/test_openai_responses_client.py

@@ -439,7 +439,7 @@ async def test_get_streaming_response_with_all_parameters() -> None:
             instructions="Stream response test",
             max_tokens=50,
             parallel_tool_calls=False,
-            model="gpt-4",
+            model_id="gpt-4",
             previous_response_id="stream-prev-123",
             reasoning={"mode": "stream"},
             service_tier="default",