kl-mcp-client 2.0.3__tar.gz → 2.0.4__tar.gz

{kl_mcp_client-2.0.3 → kl_mcp_client-2.0.4}/PKG-INFO

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

{kl_mcp_client-2.0.3 → kl_mcp_client-2.0.4}/kl_mcp_client/tools.py

@@ -126,27 +126,63 @@ class MCPTools:
 
     @_ensure_client
     def upload_file(
-        self, sessionId: str, selector: str, filename: str, base64data: str
+        self,
+        sessionId: str,
+        selector: str,
+        file_path: str,
     ) -> Dict[str, Any]:
         """
-        Upload file vào input[type=file]
+        Upload file (kể cả video lớn) vào input[type=file] theo luồng mới:
+        1. Multipart upload file lên MCP server
+        2. Nhận uploadId
+        3. Gọi MCP tool uploadFile với uploadId
+
         Args:
             sessionId: MCP browser session
             selector: CSS selector, ví dụ 'input[type=file]'
-            filename: tên file trên browser side
-            base64data: dữ liệu base64 (không kèm header)
-        Returns:
-            structured result từ server
+            file_path: đường dẫn file local (video, pdf, doc, ...)
         """
-        return self.client.call_tool(
+
+        if not file_path:
+            return {"ok": False, "error": "file_path is required"}
+
+        # --------------------------------------------------
+        # 1️⃣ Multipart upload file lên MCP server
+        # --------------------------------------------------
+        try:
+            with open(file_path, "rb") as f:
+                resp = self.client.http.post(
+                    "/upload",
+                    files={"file": f},
+                    timeout=300,  # upload file lớn
+                )
+        except Exception as e:
+            return {"ok": False, "error": f"upload http failed: {e}"}
+
+        if resp.status_code != 200:
+            return {
+                "ok": False,
+                "error": f"upload http error {resp.status_code}: {resp.text}",
+            }
+
+        data = resp.json()
+        upload_id = data.get("uploadId")
+        if not upload_id:
+            return {"ok": False, "error": "uploadId not returned from server"}
+
+        # --------------------------------------------------
+        # 2️⃣ Gọi MCP tool uploadFile (PATH MODE)
+        # --------------------------------------------------
+        result = self.client.call_tool(
             "uploadFile",
             {
                 "sessionId": sessionId,
                 "selector": selector,
-                "filename": filename,
-                "data": base64data,
+                "uploadId": upload_id,
             },
-        ).get("structuredContent", {})
+        )
+
+        return result.get("structuredContent", {})
 
     @_ensure_client
     def wait_for_selector(
@@ -229,6 +265,16 @@ class MCPTools:
             {"sessionId": sessionId, "selector": selector, "args": args or {}},
         )
 
+    @_ensure_client
+    def find_element_by_prompt(self, sessionId: str, prompt: str) -> Dict[str, Any]:
+        """
+        Gọi tool findElementByPrompt trên MCP server.
+        Trả về structuredContent gồm: html, nodeId.
+        """
+        return self.client.call_tool(
+            "findElementByPrompt", {"sessionId": sessionId, "prompt": prompt}
+        ).get("structuredContent", {})
+
     # ======================================================
     # AI / CONTENT PARSING
     # ======================================================

{kl_mcp_client-2.0.3 → kl_mcp_client-2.0.4}/kl_mcp_client.egg-info/PKG-INFO

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

{kl_mcp_client-2.0.3 → kl_mcp_client-2.0.4}/pyproject.toml

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