kl-mcp-client 2.1.2__tar.gz → 2.1.4__tar.gz

{kl_mcp_client-2.1.2 → kl_mcp_client-2.1.4}/PKG-INFO

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

{kl_mcp_client-2.1.2 → kl_mcp_client-2.1.4}/kl_mcp_client/tools.py

@@ -321,3 +321,241 @@ class MCPTools:
             "getCleanText",
             {"sessionId": sessionId},
         ).get("structuredContent", {})
+
+    @_ensure_client
+    def evaluate_stream(
+        self,
+        sessionId: str,
+        expression: str,
+        chunkSize: int = 100,
+    ) -> Dict[str, Any]:
+        """
+        Evaluate JS expression theo chế độ STREAM.
+        Dùng khi kết quả là array lớn (DOM, list, table, ...)
+
+        Returns (init response):
+        {
+          "stream_id": "...",
+          "total": 1234,
+          "chunk_size": 100
+        }
+        """
+        return self.client.call_tool(
+            "evaluate.stream",
+            {
+                "sessionId": sessionId,
+                "expression": expression,
+                "chunkSize": int(chunkSize),
+            },
+        ).get("structuredContent", {})
+
+    @_ensure_client
+    def stream_pull(
+        self,
+        stream_id: str,
+        offset: int = 0,
+        limit: int = 100,
+    ) -> Dict[str, Any]:
+        """
+        Kéo 1 chunk từ stream đã tạo bởi evaluate.stream
+
+        Returns:
+        {
+          "items": [...],
+          "offset": 0,
+          "has_more": true
+        }
+        """
+        return self.client.call_tool(
+            "stream.pull",
+            {
+                "stream_id": stream_id,
+                "offset": int(offset),
+                "limit": int(limit),
+            },
+        ).get("structuredContent", {})
+
+    @_ensure_client
+    def evaluate_stream_all(
+        self,
+        sessionId: str,
+        expression: str,
+        chunkSize: int = 100,
+        max_items: Optional[int] = None,
+    ):
+        """
+        Helper: evaluate.stream + tự động pull toàn bộ dữ liệu.
+
+        ⚠️ Chỉ dùng khi bạn THỰC SỰ cần full data.
+        """
+        init = self.evaluate_stream(
+            sessionId=sessionId,
+            expression=expression,
+            chunkSize=chunkSize,
+        )
+
+        stream_id = init.get("stream_id")
+        total = init.get("total", 0)
+
+        if not stream_id:
+            return []
+
+        items = []
+        offset = 0
+
+        while True:
+            chunk = self.stream_pull(
+                stream_id=stream_id,
+                offset=offset,
+                limit=chunkSize,
+            )
+
+            part = chunk.get("items", [])
+            items.extend(part)
+
+            if max_items and len(items) >= max_items:
+                return items[:max_items]
+
+            if not chunk.get("has_more"):
+                break
+
+            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", {})

{kl_mcp_client-2.1.2 → kl_mcp_client-2.1.4}/kl_mcp_client.egg-info/PKG-INFO

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

{kl_mcp_client-2.1.2 → kl_mcp_client-2.1.4}/pyproject.toml

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