kl-mcp-client 2.1.3__tar.gz → 2.1.7__tar.gz

{kl_mcp_client-2.1.3 → kl_mcp_client-2.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kl-mcp-client
-Version: 2.1.3
+Version: 2.1.7
 Summary: MCP Client for Python
 Author-email: Kyle <hngan.it@gmail.com>
 License: MIT

{kl_mcp_client-2.1.3 → kl_mcp_client-2.1.7}/kl_mcp_client/asyncio/tools.py

@@ -319,3 +319,35 @@ class MCPTools:
                 "key": key,
             },
         )
+    
+    # ======================================================
+    # BROWSER RUNTIME (NO SESSION)
+    # ======================================================
+
+    async def create_browser(
+        self,
+        payload: Optional[Dict[str, Any]] = None,
+    ) -> Dict[str, Any]:
+        """
+        Start browser runtime (idempotent).
+        - KHÔNG tạo session
+        - Dùng cho warm-up / admin / scheduler
+
+        payload: ThirdPartyOpenRequest (optional)
+        """
+        res = await self.client.call_tool(
+            "create_browser",
+            payload or {},
+        )
+        return res.get("structuredContent", {})
+
+    async def release_browser(self) -> Dict[str, Any]:
+        """
+        Release / stop browser runtime.
+        - KHÔNG cần payload
+        """
+        res = await self.client.call_tool(
+            "release_browser",
+            {},
+        )
+        return res.get("structuredContent", {})

{kl_mcp_client-2.1.3 → kl_mcp_client-2.1.7}/kl_mcp_client/tools.py

@@ -422,3 +422,166 @@ class MCPTools:
             offset += chunkSize
 
         return items
+
+    @_ensure_client
+    def wait_for_selector(
+        self, sessionId: str, selector: str, timeoutMs: Optional[int] = None
+    ) -> Dict[str, Any]:
+        args = {
+            "sessionId": sessionId,
+            "selector": selector,
+        }
+        if timeoutMs is not None:
+            args["timeout"] = int(timeoutMs)
+
+        return self.client.call_tool("waitForSelector", args).get(
+            "structuredContent", {}
+        )
+
+    @_ensure_client
+    def send_keys(
+        self, sessionId: str, key: str, interval: int = 100
+    ) -> Dict[str, Any]:
+        """
+        Gửi phím: enter, tab, esc, ctrl+c, ctrl+shift+tab, ...
+        """
+        return self.client.call_tool(
+            "sendKeys",
+            {"sessionId": sessionId, "text": key, "interval": interval},
+        ).get("structuredContent", {})
+
+    @_ensure_client
+    def perform(
+        self,
+        sessionId: str,
+        action: str,
+        target: Optional[str] = None,
+        value: Optional[str] = None,
+        x: Optional[float] = None,
+        y: Optional[float] = None,
+        from_point: Optional[dict] = None,
+        to_point: Optional[dict] = None,
+    ) -> Dict[str, Any]:
+        """
+        action:
+        - click (x,y)
+        - move / hover (x,y)
+        - drag (from -> to)
+        - type (target, value)
+        """
+        args = {
+            "sessionId": sessionId,
+            "action": action,
+        }
+
+        if target is not None:
+            args["target"] = target
+        if value is not None:
+            args["value"] = value
+        if x is not None:
+            args["x"] = float(x)
+        if y is not None:
+            args["y"] = float(y)
+        if from_point is not None:
+            args["from"] = from_point
+        if to_point is not None:
+            args["to"] = to_point
+
+        return self.client.call_tool("perform", args).get("structuredContent", {})
+
+    @_ensure_client
+    def drag_and_drop(
+        self,
+        sessionId: str,
+        from_x: float,
+        from_y: float,
+        to_x: float,
+        to_y: float,
+    ):
+        return self.perform(
+            sessionId=sessionId,
+            action="drag",
+            from_point={"x": from_x, "y": from_y},
+            to_point={"x": to_x, "y": to_y},
+        )
+
+    def hover(self, sessionId: str, x: float, y: float):
+        return self.perform(
+            sessionId=sessionId,
+            action="hover",
+            x=x,
+            y=y,
+        )
+
+    @_ensure_client
+    def scroll(
+        self,
+        sessionId: str,
+        *,
+        x: Optional[float] = None,
+        y: Optional[float] = None,
+        selector: Optional[str] = None,
+        position: Optional[str] = None,  # "top" | "bottom"
+    ) -> Dict[str, Any]:
+        """
+        Scroll trang web.
+
+        Cách dùng:
+        - Scroll theo pixel:
+            scroll(sessionId, y=500)
+        - Scroll tới element:
+            scroll(sessionId, selector="#footer")
+        - Scroll top / bottom:
+            scroll(sessionId, position="top")
+            scroll(sessionId, position="bottom")
+        """
+
+        args: Dict[str, Any] = {
+            "sessionId": sessionId,
+        }
+
+        if x is not None:
+            args["x"] = float(x)
+        if y is not None:
+            args["y"] = float(y)
+        if selector is not None:
+            args["selector"] = selector
+        if position is not None:
+            args["position"] = position
+
+        if len(args) == 1:
+            return {
+                "ok": False,
+                "error": "scroll requires x/y or selector or position",
+            }
+
+        return self.client.call_tool(
+            "scroll",
+            args,
+        ).get("structuredContent", {})
+        # ======================================================
+    # BROWSER RUNTIME (NO SESSION)
+    # ======================================================
+    @_ensure_client
+    def create_browser(self, payload: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+        """
+        Start browser runtime (idempotent).
+        Không tạo session, chỉ đảm bảo browser đang RUNNING.
+
+        payload: map cho ThirdPartyOpenRequest
+        """
+        return self.client.call_tool(
+            "create_browser",
+            payload or {},
+        ).get("structuredContent", {})
+
+    @_ensure_client
+    def release_browser(self) -> Dict[str, Any]:
+        """
+        Release / stop browser runtime.
+        Không cần payload.
+        """
+        return self.client.call_tool(
+            "release_browser",
+            {},
+        ).get("structuredContent", {})

{kl_mcp_client-2.1.3 → kl_mcp_client-2.1.7}/kl_mcp_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kl-mcp-client
-Version: 2.1.3
+Version: 2.1.7
 Summary: MCP Client for Python
 Author-email: Kyle <hngan.it@gmail.com>
 License: MIT

{kl_mcp_client-2.1.3 → kl_mcp_client-2.1.7}/kl_mcp_client.egg-info/SOURCES.txt

@@ -11,4 +11,6 @@ kl_mcp_client.egg-info/requires.txt
 kl_mcp_client.egg-info/top_level.txt
 kl_mcp_client/asyncio/__init__.py
 kl_mcp_client/asyncio/client.py
-kl_mcp_client/asyncio/tools.py
+kl_mcp_client/asyncio/tools.py
+tests/test_async_tools.py
+tests/test_tools.py

{kl_mcp_client-2.1.3 → kl_mcp_client-2.1.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "kl-mcp-client"
-version = "2.1.3"
+version = "2.1.7"
 description = "MCP Client for Python"
 dependencies = [ "requests>=2.32.5", "httpx>=0.28.1",]
 readme = "README.md"

kl_mcp_client-2.1.7/tests/test_async_tools.py

@@ -0,0 +1,245 @@
+from typing import Any, Dict
+
+import pytest
+from kl_mcp_client.asyncio import MCPTools
+
+# ======================================================
+# MOCK MCP CLIENT
+# ======================================================
+
+class MockMCPClient:
+    def __init__(self):
+        self.calls = []
+
+    async def create_session(self, cdpUrl: str):
+        self.calls.append(("create_session", cdpUrl))
+        return "session-123"
+
+    async def close_session(self, sessionId: str):
+        self.calls.append(("close_session", sessionId))
+        return True
+
+    def list_local_sessions(self):
+        return ["session-123", "session-456"]
+
+    async def call_tool(self, name: str, payload: Dict[str, Any]):
+        self.calls.append((name, payload))
+
+        # ---- fake responses ----
+        if name == "screenshot":
+            return {
+                "content": [
+                    {
+                        "type": "image",
+                        "mimeType": "image/png",
+                        "data": "BASE64DATA",
+                    }
+                ]
+            }
+
+        if name in ("create_browser", "release_browser"):
+            return {
+                "structuredContent": {
+                    "success": True,
+                }
+            }
+
+        return {
+            "structuredContent": {
+                "ok": True,
+                "tool": name,
+                "payload": payload,
+            }
+        }
+
+
+# ======================================================
+# FIXTURES
+# ======================================================
+
+@pytest.fixture
+def mock_client():
+    return MockMCPClient()
+
+
+@pytest.fixture
+def tools(mock_client):
+    return MCPTools(client=mock_client)
+
+
+# ======================================================
+# SESSION MANAGEMENT
+# ======================================================
+
+@pytest.mark.asyncio
+async def test_create_session(tools):
+    res = await tools.create_session("http://localhost:9222")
+    assert res["sessionId"] == "session-123"
+
+
+@pytest.mark.asyncio
+async def test_close_session(tools):
+    res = await tools.close_session("session-123")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_list_sessions(tools):
+    res = await tools.list_sessions()
+    assert "sessions" in res
+    assert len(res["sessions"]) == 2
+
+
+# ======================================================
+# BROWSER RUNTIME (NO SESSION)
+# ======================================================
+
+@pytest.mark.asyncio
+async def test_create_browser(tools):
+    res = await tools.create_browser({"image": "chrome"})
+    assert res["success"] is True
+
+
+@pytest.mark.asyncio
+async def test_release_browser(tools):
+    res = await tools.release_browser()
+    assert res["success"] is True
+
+
+# ======================================================
+# NAVIGATION & DOM
+# ======================================================
+
+@pytest.mark.asyncio
+async def test_open_page(tools):
+    res = await tools.open_page("session-123", "https://example.com")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_get_html(tools):
+    res = await tools.get_html("session-123")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_screenshot(tools):
+    res = await tools.screenshot("session-123")
+    assert res["type"] == "image"
+    assert res["mimeType"] == "image/png"
+
+
+@pytest.mark.asyncio
+async def test_evaluate(tools):
+    res = await tools.evaluate("session-123", "1+1")
+    assert res["ok"] is True
+
+
+# ======================================================
+# ELEMENT UTILITIES
+# ======================================================
+
+@pytest.mark.asyncio
+async def test_find_element(tools):
+    res = await tools.find_element("session-123", "#main")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_find_all(tools):
+    res = await tools.find_all("session-123", "div")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_get_bounding_box(tools):
+    res = await tools.get_bounding_box("session-123", "#main")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_click_bounding_box(tools):
+    res = await tools.click_bounding_box("session-123", "#btn")
+    assert res["ok"] is True
+
+
+# ======================================================
+# TAB MANAGEMENT
+# ======================================================
+
+@pytest.mark.asyncio
+async def test_new_tab(tools):
+    res = await tools.new_tab("session-123", "https://example.com")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_switch_tab(tools):
+    res = await tools.switch_tab("session-123", "tab-1")
+    assert res["ok"] is True
+
+
+# ======================================================
+# ADVANCED ACTIONS
+# ======================================================
+
+@pytest.mark.asyncio
+async def test_click_to_text(tools):
+    res = await tools.click_to_text("session-123", "Login")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_find_element_xpath(tools):
+    res = await tools.find_element_xpath("session-123", "//div")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_find_element_by_text(tools):
+    res = await tools.find_element_by_text("session-123", "Submit")
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_click_by_node_id(tools):
+    res = await tools.click_by_node_id("session-123", 1001)
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_import_cookies(tools):
+    res = await tools.import_cookies("session-123", {"a": "b"})
+    assert res["ok"] is True
+
+
+# ======================================================
+# PERFORM / MOUSE
+# ======================================================
+
+@pytest.mark.asyncio
+async def test_perform_click_xy(tools):
+    res = await tools.perform_click_xy("session-123", 10, 20)
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_perform_drag(tools):
+    res = await tools.perform_drag("session-123", 0, 0, 100, 100)
+    assert res["ok"] is True
+
+
+@pytest.mark.asyncio
+async def test_perform_hover(tools):
+    res = await tools.perform_hover("session-123", 50, 50)
+    assert res["ok"] is True
+
+
+# ======================================================
+# CLEAN TEXT
+# ======================================================
+
+@pytest.mark.asyncio
+async def test_get_clean_text(tools):
+    res = await tools.get_clean_text("session-123")
+    assert res["ok"] is True

kl_mcp_client-2.1.7/tests/test_tools.py

@@ -0,0 +1,206 @@
+from typing import Any, Dict
+
+import pytest
+from kl_mcp_client import MCPTools  # đổi đúng import path của bạn
+
+# ======================================================
+# MOCK MCP CLIENT (SYNC)
+# ======================================================
+
+class MockMCPClient:
+    def __init__(self):
+        self.calls = []
+
+    def create_session(self, cdpUrl: str):
+        self.calls.append(("create_session", cdpUrl))
+        return "session-123"
+
+    def close_session(self, sessionId: str):
+        self.calls.append(("close_session", sessionId))
+        return True
+
+    def list_local_sessions(self):
+        return ["session-123", "session-456"]
+
+    def call_tool(self, name: str, payload: Dict[str, Any]):
+        self.calls.append((name, payload))
+
+        if name == "screenshot":
+            return {
+                "content": [
+                    {
+                        "type": "image",
+                        "mimeType": "image/png",
+                        "data": "BASE64DATA",
+                    }
+                ]
+            }
+
+        if name in ("create_browser", "release_browser"):
+            return {
+                "structuredContent": {
+                    "success": True,
+                }
+            }
+
+        return {
+            "structuredContent": {
+                "ok": True,
+                "tool": name,
+                "payload": payload,
+            }
+        }
+
+
+# ======================================================
+# FIXTURES
+# ======================================================
+
+@pytest.fixture
+def tools():
+    t = MCPTools()
+    t.client = MockMCPClient()
+    return t
+
+
+# ======================================================
+# SESSION MANAGEMENT
+# ======================================================
+
+def test_connect_mcp(tools):
+    res = tools.connect_mcp("http://localhost:3000")
+    assert res["ok"] is True
+
+
+def test_create_session(tools):
+    res = tools.create_session("http://localhost:9222")
+    assert res["sessionId"] == "session-123"
+
+
+def test_close_session(tools):
+    res = tools.close_session("session-123")
+    assert res["ok"] is True
+
+
+def test_list_sessions(tools):
+    res = tools.list_sessions()
+    assert len(res["sessions"]) == 2
+
+
+# ======================================================
+# BROWSER RUNTIME (NO SESSION)
+# ======================================================
+
+def test_create_browser(tools):
+    res = tools.create_browser({"image": "chrome"})
+    assert res["success"] is True
+
+
+def test_release_browser(tools):
+    res = tools.release_browser()
+    assert res["success"] is True
+
+
+# ======================================================
+# NAVIGATION & DOM
+# ======================================================
+
+def test_open_page(tools):
+    res = tools.open_page("session-123", "https://example.com")
+    assert res["ok"] is True
+
+
+def test_get_html(tools):
+    res = tools.get_html("session-123")
+    assert res["ok"] is True
+
+
+def test_screenshot(tools):
+    res = tools.screenshot("session-123")
+    assert res["type"] == "image"
+
+
+def test_evaluate(tools):
+    res = tools.evaluate("session-123", "1+1")
+    assert res["ok"] is True
+
+
+# ======================================================
+# ELEMENT UTILITIES
+# ======================================================
+
+def test_find_element(tools):
+    res = tools.find_element("session-123", "#main")
+    assert res["ok"] is True
+
+
+def test_find_all(tools):
+    res = tools.find_all("session-123", "div")
+    assert res["ok"] is True
+
+
+def test_get_bounding_box(tools):
+    res = tools.get_bounding_box("session-123", "#box")
+    assert res["ok"] is True
+
+
+def test_click_bounding_box(tools):
+    res = tools.click_bounding_box("session-123", "#btn")
+    assert res["ok"] is True
+
+
+# ======================================================
+# TAB MANAGEMENT
+# ======================================================
+
+def test_new_tab(tools):
+    res = tools.new_tab("session-123", "https://example.com")
+    assert res["ok"] is True
+
+
+def test_switch_tab(tools):
+    res = tools.switch_tab("session-123", "tab-1")
+    assert res["ok"] is True
+
+
+# ======================================================
+# ADVANCED ACTIONS
+# ======================================================
+
+def test_click_to_text(tools):
+    res = tools.click_to_text("session-123", "Login")
+    assert res["ok"] is True
+
+
+def test_find_element_xpath(tools):
+    res = tools.find_element_xpath("session-123", "//div")
+    assert res["ok"] is True
+
+
+def test_find_element_by_text(tools):
+    res = tools.find_element_by_text("session-123", "Submit")
+    assert res["ok"] is True
+
+
+def test_click_by_node_id(tools):
+    res = tools.click_by_node_id("session-123", 10)
+    assert res["ok"] is True
+
+
+def test_import_cookies(tools):
+    res = tools.import_cookies("session-123", {"a": "b"})
+    assert res["ok"] is True
+
+
+# ======================================================
+# SCROLL / PERFORM
+# ======================================================
+
+def test_scroll_pixel(tools):
+    res = tools.scroll("session-123", y=500)
+    assert res["ok"] is True
+
+
+def test_perform_drag(tools):
+    res = tools.drag_and_drop("session-123", 0, 0, 100, 100)
+    assert res["ok"] is True