Fix: stabilize MCP client imports and unify test environment for CI & Local.

Author: krishnaharshap

.github/workflows/ci.yml

@@ -37,15 +37,25 @@ jobs:
       - name: Install dependencies
         run: |
           python -m venv venv
-          # If mcp is private, uncomment below:
-          # python -m pip install git+https://github.com/your-org/mcp.git
-          ./venv/bin/python -m pip install -r requirements.txt || .\venv\Scripts\python.exe -m pip install -r requirements.txt
-          ./venv/bin/python -m playwright install || .\venv\Scripts\python.exe -m playwright install
+          if [ "${{ runner.os }}" == "Windows" ]; then
+            .\venv\Scripts\python.exe -m pip install -r requirements.txt
+            .\venv\Scripts\python.exe -m playwright install
+          else
+            ./venv/bin/python -m pip install -r requirements.txt
+            ./venv/bin/python -m playwright install
+          fi
+
 
       - name: Run tests
         run: |
-          ./venv/bin/python -m pytest -v --maxfail=1 --disable-warnings --html=report.html --self-contained-html \
-          || .\venv\Scripts\python.exe -m pytest -v --maxfail=1 --disable-warnings --html=report.html --self-contained-html
+          if [ "${{ runner.os }}" == "Windows" ]; then
+            .\venv\Scripts\Activate.ps1
+            pytest -v --maxfail=1 --disable-warnings --html=report.html --self-contained-html
+          else
+            source venv/bin/activate
+            pytest -v --maxfail=1 --disable-warnings --html=report.html --self-contained-html
+          fi
+
 
       - name: Upload HTML Report
         uses: actions/upload-artifact@v4

playwright_mcp/mcp_client.py

@@ -1,25 +1,34 @@
-# playwright_mcp/mcp_client.py
-
-from mcp import client as mcp_client  # import the lowercase module
+from mcp.client.stdio import stdio_client, StdioServerParameters
+from mcp.client.session import ClientSession
 import asyncio
 
 class MCPClient:
     def __init__(self, server_url: str):
         self.server_url = server_url
-        self.client = None
+        self.session = None
 
     async def connect(self):
         print(f"[MCP] Connecting to {self.server_url}...")
-        self.client = mcp_client.Client(self.server_url)  # use Client via mcp_client
-        await self.client.connect()
+
+        # Define the server parameters
+        server_params = StdioServerParameters(
+            command="your_command_here",  # Replace with the actual command to run your MCP server
+            args=["your", "arguments", "here"],  # Replace with any arguments needed
+        )
+
+        # Use stdio_client to connect to the server
+        async with stdio_client(server_params) as (read_stream, write_stream):
+            self.session = ClientSession(read_stream, write_stream)
+            await self.session.connect()
+
         print("[MCP] Connected.")
 
     async def send_context(self, context_data: dict):
-        if not self.client:
-            raise RuntimeError("MCP client not connected.")
-        response = await self.client.query({"context": context_data})
-        return response
+        if self.session is None:
+            raise RuntimeError("MCP session not connected.")
+        result = await self.session.query({"context": context_data})
+        return result
 
     async def close(self):
-        if self.client:
-            await self.client.close()
+        if self.session:
+            await self.session.close()

pytest.ini

@@ -1,6 +1,7 @@
 [pytest]
 # Specify test directories
 testpaths = tests
+pythonpath = .
 
 # Add options for verbose output, HTML report generation
 addopts = -v --maxfail=1 --disable-warnings --html=report.html --self-contained-html

report.html

@@ -1,4 +1,6 @@
-from playwright_mcp.browser import launch_and_get_title
+# tests/test_example.py
+
+from playwright_mcp.browser import launch_and_get_title
 
 def test_example_title():
     title = launch_and_get_title("https://example.com")

tests/test_example.py

@@ -1,7 +1,8 @@
 # tests/test_mcp_integration.py
+
 import asyncio
 from playwright_mcp.mcp_adapter import run_mcp_test_flow
 
 def test_mcp_integration():
-    """Ensure MCP integration and Playwright flow work end-to-end."""
+    """Ensure MCP and Playwright integration flow runs end-to-end."""
     asyncio.run(run_mcp_test_flow())