agent/agent-framework
1
0
Fork 0
mirror of https://github.com/microsoft/agent-framework.git synced 2026-06-16 21:04:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix pipeline

Browse Source
This commit is contained in:
Tao Chen
2026-03-16 14:05:39 -07:00
Unverified
parent 2fcbd9cec9
commit d3191ffd48
2 changed files with 7 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/core/agent_framework/_mcp.py
+1 -1
View File
@@ -494,7 +494,7 @@ class MCPTool:
self.is_connected: bool = False
self._tools_loaded: bool = False
self._prompts_loaded: bool = False
self._mcp_protocol_version: str | None = None
self._mcp_protocol_version: str | int | None = None
def __str__(self) -> str:
return f"MCPTool(name={self.name}, description={self.description})"
python/packages/core/tests/core/test_mcp.py
+6 -12
View File
@@ -2940,9 +2940,7 @@ def mcp_test_server_class():
# MCP SDK stores _request_id as int; call_tool converts it to str for the span.
self._mock_session._request_id = 42
self._mock_session.call_tool = AsyncMock(
return_value=types.CallToolResult(
content=[types.TextContent(type="text", text="ok")]
)
return_value=types.CallToolResult(content=[types.TextContent(type="text", text="ok")])
)
self._mock_session.get_prompt = AsyncMock(
return_value=types.GetPromptResult(
@@ -2954,9 +2952,7 @@ def mcp_test_server_class():
]
)
)
self._mock_session.list_tools = AsyncMock(
return_value=types.ListToolsResult(tools=[])
)
self._mock_session.list_tools = AsyncMock(return_value=types.ListToolsResult(tools=[]))
async def connect(self):
self.session = self._mock_session
@@ -2998,11 +2994,9 @@ async def test_call_tool_span_sets_error_on_mcp_error(span_exporter, mcp_test_se
server = mcp_test_server_class()
async with server:
server._mock_session.call_tool = AsyncMock(
side_effect=McpError(
types.ErrorData(code=-32600, message="bad request")
)
side_effect=McpError(types.ErrorData(code=-32600, message="bad request"))
)
with pytest.raises(Exception):
with pytest.raises(ToolExecutionException):
await server.call_tool("my_tool")
spans = span_exporter.get_finished_spans()
@@ -3024,7 +3018,7 @@ async def test_call_tool_span_sets_error_on_tool_error_result(span_exporter, mcp
content=[types.TextContent(type="text", text="tool error")],
)
)
with pytest.raises(Exception):
with pytest.raises(ToolExecutionException):
await server.call_tool("my_tool")
spans = span_exporter.get_finished_spans()
@@ -3072,7 +3066,7 @@ async def test_get_prompt_span_sets_error_on_exception(span_exporter, mcp_test_s
server = mcp_test_server_class()
async with server:
server._mock_session.get_prompt = AsyncMock(side_effect=RuntimeError("fail"))
with pytest.raises(Exception):
with pytest.raises(ToolExecutionException):
await server.get_prompt("my_prompt")
spans = span_exporter.get_finished_spans()