modelcontextprotocol · dsp-ant · Mar 12, 2025
diff --git a/.gitignore b/.gitignore
@@ -165,4 +165,5 @@ cython_debug/
 #.idea/

 # vscode
-.vscode/
+.vscode/
+.windsurfrules
diff --git a/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py b/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py
@@ -322,8 +322,7 @@ async def process_llm_response(self, llm_response: str) -> str:
                                total = result["total"]
                                percentage = (progress / total) * 100
                                logging.info(
-                                    f"Progress: {progress}/{total} "
-                                    f"({percentage:.1f}%)"
+                                    f"Progress: {progress}/{total} ({percentage:.1f}%)"
                                )

                            return f"Tool execution result: {result}"

diff --git a/pyproject.toml b/pyproject.toml
@@ -43,11 +43,10 @@ mcp = "mcp.cli:app [cli]"
 [tool.uv]
 resolution = "lowest-direct"
 dev-dependencies = [
-    "pyright>=1.1.391",
+    "pyright>=1.1.396",
    "pytest>=8.3.4",
    "ruff>=0.8.5",
    "trio>=0.26.2",
-    "pytest-flakefinder>=1.1.0",
    "pytest-xdist>=3.6.1",
 ]


diff --git a/src/mcp/client/__main__.py b/src/mcp/client/__main__.py
@@ -5,7 +5,11 @@
 from urllib.parse import urlparse

 import anyio
+import httpx
+from pydantic import AnyHttpUrl

+from mcp.client.auth.httpx import McpOAuth
+from mcp.client.auth.oauth import InMemoryOAuthStore, OAuthClient
 from mcp.client.session import ClientSession
 from mcp.client.sse import sse_client
 from mcp.client.stdio import StdioServerParameters, stdio_client
@@ -46,8 +50,15 @@ async def main(command_or_url: str, args: list[str], env: list[tuple[str, str]])

    if urlparse(command_or_url).scheme in ("http", "https"):
        # Use SSE client for HTTP(S) URLs
-        async with sse_client(command_or_url) as streams:
-            await run_session(*streams)
+        oauth_client = OAuthClient(
+            client_name="mcp-client",
+            server_url=AnyHttpUrl(command_or_url),
+            redirect_url=AnyHttpUrl("http://localhost:5999/auth"),
+            provider=InMemoryOAuthStore(),
+        )
+        async with httpx.AsyncClient(auth=McpOAuth(oauth_client)) as http:
+            async with sse_client(http, command_or_url) as streams:
+                await run_session(*streams)
    else:
        # Use stdio client for commands
        server_parameters = StdioServerParameters(

diff --git a/src/mcp/client/auth/__init__.py b/src/mcp/client/auth/__init__.py