fix: Add streamable MCP server support (#468)

* fix: Add streamable MCP server support(#349)

* “Revert-timeout”

* fix lint and test check

* modify streamable error notify

---------

Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
This commit is contained in:
suntp
2025-07-29 14:04:04 +08:00
committed by GitHub
parent bedf7d4af2
commit e178483971
6 changed files with 27 additions and 6 deletions

View File

@@ -13,8 +13,13 @@ import src.server.mcp_utils as mcp_utils
async def test__get_tools_from_client_session_success(mock_ClientSession):
mock_read = AsyncMock()
mock_write = AsyncMock()
mock_callback = AsyncMock()
mock_context_manager = AsyncMock()
mock_context_manager.__aenter__.return_value = (mock_read, mock_write)
mock_context_manager.__aenter__.return_value = (
mock_read,
mock_write,
mock_callback,
)
mock_context_manager.__aexit__.return_value = None
mock_session = AsyncMock()