auth: client implementation

Author: dsp-ant

.gitignore

@@ -165,4 +165,5 @@ cython_debug/
 #.idea/
 
 # vscode
-.vscode/
+.vscode/
+.windsurfrules

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}"

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",
 ]
 

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(