From f08b0dc1900d6107998e739ba3419ac1cca7f8fd Mon Sep 17 00:00:00 2001
From: Chenjiahao <chenjiahao@sensetime.com>
Date: Fri, 11 Apr 2025 11:10:27 +0800
Subject: [PATCH] modify __init__ for sse_client when using thirdparty import

---
 src/mcp/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/mcp/__init__.py b/src/mcp/__init__.py
index 0d3c372ce..6cc5be06c 100644
--- a/src/mcp/__init__.py
+++ b/src/mcp/__init__.py
@@ -1,4 +1,5 @@
 from .client.session import ClientSession
+from .client.sse import sse_client
 from .client.stdio import StdioServerParameters, stdio_client
 from .server.session import ServerSession
 from .server.stdio import stdio_server
@@ -107,6 +108,7 @@
     "Tool",
     "ToolsCapability",
     "UnsubscribeRequest",
+    "sse_client",
     "stdio_client",
     "stdio_server",
     "CompleteRequest",