Replace SSE test with Pyright/ruff-compliant version

Author: Vinisha-projects

tests/test_sse_client_server.py

@@ -1,31 +1,26 @@
 import asyncio
-from typing import List
+from typing import AsyncGenerator, List
 
 from fastapi import FastAPI
 from starlette.responses import StreamingResponse
 import uvicorn
 from threading import Thread
 import httpx
-
 from mcp.client.sse import aconnect_sse
 
-
 app = FastAPI()
 
-
 @app.get("/sse")
 async def sse_endpoint() -> StreamingResponse:
-    async def event_stream() -> asyncio.AsyncGenerator[str, None]:
+    async def event_stream() -> AsyncGenerator[str, None]:
         for i in range(3):
             yield f"data: Hello {i+1}\n\n"
             await asyncio.sleep(0.1)
     return StreamingResponse(event_stream(), media_type="text/event-stream")
 
-
 def run_mock_server() -> None:
     uvicorn.run(app, host="127.0.0.1", port=8012, log_level="warning")
 
-
 async def test_aconnect_sse_server_response() -> None:
     server_thread = Thread(target=run_mock_server, daemon=True)
     server_thread.start()
@@ -44,10 +39,3 @@ async def test_aconnect_sse_server_response() -> None:
 
     assert messages == ["Hello 1", "Hello 2", "Hello 3"]
     print("\n✅ Test passed! SSE connection via aconnect_sse worked correctly.")
-
-
-
-
-
-
-