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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kl-mcp-client
-Version: 2.1.2
+Version: 2.1.3
 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.3}/kl_mcp_client/tools.py

@@ -321,3 +321,104 @@ 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

{kl_mcp_client-2.1.2 → kl_mcp_client-2.1.3}/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.3
 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.3}/pyproject.toml

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