Merge pull request #9 from jolovicdev/feat/streamable-http-small

Feat/streamable http small

Author: jolovicdev

docs/tools.md

@@ -133,8 +133,11 @@ belt = Toolbelt()
 
 ## MCP Tool Integration
 
-Connect to MCP (Model Context Protocol) servers and use their tools.
-Use `connect_sse` when pointing at an SSE endpoint.
+Connect to MCP (Model Context Protocol) servers and use their tools. MCP tools are automatically converted to the blackgeorge `Tool` format and can be passed to workers.
+
+### stdio transport
+
+For local MCP servers that run as subprocesses.
 
 ```python
 from blackgeorge.tools import MCPToolProvider
@@ -145,7 +148,42 @@ async with MCPToolProvider() as provider:
     result = await provider.acall_tool("fetch", {"url": "https://example.com"})
 ```
 
-MCP tools are automatically converted to the blackgeorge `Tool` format and can be passed to workers.
+### Streamable HTTP transport
+
+For remote MCP servers over HTTP.
+
+```python
+from blackgeorge.tools import MCPToolProvider
+
+async with MCPToolProvider() as provider:
+    await provider.connect_streamable_http("https://api.example.com/mcp")
+    tools = provider.list_tools()
+    result = await provider.acall_tool("search", {"query": "python"})
+```
+
+For servers requiring authentication, pass a custom `httpx.AsyncClient`:
+
+```python
+import httpx
+from blackgeorge.tools import MCPToolProvider
+
+async with MCPToolProvider() as provider:
+    client = httpx.AsyncClient(headers={"Authorization": "Bearer token"})
+    await provider.connect_streamable_http("https://api.example.com/mcp", http_client=client)
+    tools = provider.list_tools()
+```
+
+### SSE transport
+
+For MCP servers that use Server-Sent Events.
+
+```python
+from blackgeorge.tools import MCPToolProvider
+
+async with MCPToolProvider() as provider:
+    await provider.connect_sse("https://api.example.com/mcp/sse")
+    tools = provider.list_tools()
+```
 
 ## Execution path
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "blackgeorge"
-version = "1.1.1"
+version = "1.1.2"
 description = "Agentic framework with desk/worker/workforce primitives"
 readme = "README.md"
 license = { file = "LICENCE.md" }

src/blackgeorge/tools/mcp.py

@@ -5,6 +5,7 @@
 from mcp.client.session import ClientSession
 from mcp.client.sse import sse_client
 from mcp.client.stdio import StdioServerParameters, stdio_client
+from mcp.client.streamable_http import streamable_http_client
 from pydantic import BaseModel, create_model
 
 from blackgeorge.async_utils import run_coroutine_in_thread, run_coroutine_sync
@@ -88,15 +89,27 @@ async def connect_stdio(
         read_stream, write_stream = await self._context_manager.__aenter__()
         self._session_context = ClientSession(read_stream, write_stream)
         self._session = await self._session_context.__aenter__()
-        await self._session.initialize()
+        await self._session.initialize()  # type: ignore[misc]
         await self._discover_tools()
 
     async def connect_sse(self, url: str) -> None:
         self._context_manager = sse_client(url)
         read_stream, write_stream = await self._context_manager.__aenter__()
         self._session_context = ClientSession(read_stream, write_stream)
         self._session = await self._session_context.__aenter__()
-        await self._session.initialize()
+        await self._session.initialize()  # type: ignore[misc]
+        await self._discover_tools()
+
+    async def connect_streamable_http(
+        self,
+        url: str,
+        http_client: Any | None = None,
+    ) -> None:
+        self._context_manager = streamable_http_client(url, http_client=http_client)
+        read_stream, write_stream, _ = await self._context_manager.__aenter__()
+        self._session_context = ClientSession(read_stream, write_stream)
+        self._session = await self._session_context.__aenter__()
+        await self._session.initialize()  # type: ignore[misc]
         await self._discover_tools()
 
     async def close(self) -> None: