Introduce a function to create a standard AsyncClient with options (#655)

This commit is contained in:
ihrpr
2025-05-08 20:53:21 +01:00
committed by GitHub
parent 72003d9cc0
commit ed25167fa5
6 changed files with 95 additions and 8 deletions

View File

@@ -10,6 +10,7 @@ from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStre
from httpx_sse import aconnect_sse
import mcp.types as types
from mcp.shared._httpx_utils import create_mcp_http_client
from mcp.shared.message import SessionMessage
logger = logging.getLogger(__name__)
@@ -44,7 +45,7 @@ async def sse_client(
async with anyio.create_task_group() as tg:
try:
logger.info(f"Connecting to SSE endpoint: {remove_request_params(url)}")
async with httpx.AsyncClient(headers=headers) as client:
async with create_mcp_http_client(headers=headers) as client:
async with aconnect_sse(
client,
"GET",