kl-mcp-client 2.0.2__py3-none-any.whl → 2.0.4__py3-none-any.whl

kl_mcp_client/asyncio/client.py

@@ -23,7 +23,7 @@ class MCPClient:
         headers: Optional[Dict[str, str]] = None,
         timeout: int = 30,
         retries: int = 1,
-        proxies: Optional[Dict[str, str]] = None,
+        proxies=None,
     ):
         """
         base_url: full MCP HTTP endpoint e.g. http://localhost:3000/mcp
@@ -47,7 +47,7 @@ class MCPClient:
         self._client = httpx.AsyncClient(
             headers=self.headers,
             timeout=httpx.Timeout(self.timeout),
-            proxies=proxies,
+            proxy=proxies,
         )
 
         # local session cache

kl_mcp_client/client.py

@@ -22,7 +22,7 @@ class MCPClient:
         headers: Optional[Dict[str, str]] = None,
         timeout: int = 30,
         retries: int = 1,
-        proxies: Optional[Dict[str, str]] = None,
+        proxies=None,
     ):
         """
         base_url: full MCP HTTP endpoint e.g. http://localhost:3000/mcp
@@ -49,7 +49,7 @@ class MCPClient:
             base_url=self.base_url,
             headers=self.headers,
             timeout=httpx.Timeout(self.timeout),
-            proxies=proxies,
+            proxy=proxies,
         )
 
     def close(self):

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.2.dist-info → kl_mcp_client-2.0.4.dist-info}/METADATA

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

kl_mcp_client-2.0.4.dist-info/RECORD

@@ -0,0 +1,11 @@
+kl_mcp_client/__init__.py,sha256=pdJdBcEH5HaRuHSe2B7VUyRgH5ad3u1dDc4euZMDRMY,106
+kl_mcp_client/__version__.py,sha256=CCYQUt19bi0kfo0Q8q8snnwgw_c9F3s6FFxTMsX1yDE,22
+kl_mcp_client/client.py,sha256=ujhRp0zC5GJKnBYziIPMdKZqrhv8K2t6tBAuwYw7q_k,4513
+kl_mcp_client/tools.py,sha256=yJ8eTZrIPCyAwLm462e1eZOz6yaPk39J4gd13BzYXbw,10974
+kl_mcp_client/asyncio/__init__.py,sha256=pdJdBcEH5HaRuHSe2B7VUyRgH5ad3u1dDc4euZMDRMY,106
+kl_mcp_client/asyncio/client.py,sha256=bgooGIzdU-BKAo0f2wHZKkuIHlKrj5GJ8vxAuwU6Xn4,4409
+kl_mcp_client/asyncio/tools.py,sha256=tCeHUc4ao99Hu4ck0BmvDo95vmTtG1ChvwHamaIx_zE,7655
+kl_mcp_client-2.0.4.dist-info/METADATA,sha256=95PHfR0WCTh4FcvCc6F9a1HLkGYAaZpdJd9N1lGPGWI,4443
+kl_mcp_client-2.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+kl_mcp_client-2.0.4.dist-info/top_level.txt,sha256=wd_HFFyGjiKavwACuj8Ny0svtVyNsrxCSVU48EkoQ7c,14
+kl_mcp_client-2.0.4.dist-info/RECORD,,

kl_mcp_client-2.0.2.dist-info/RECORD

@@ -1,11 +0,0 @@
-kl_mcp_client/__init__.py,sha256=pdJdBcEH5HaRuHSe2B7VUyRgH5ad3u1dDc4euZMDRMY,106
-kl_mcp_client/__version__.py,sha256=CCYQUt19bi0kfo0Q8q8snnwgw_c9F3s6FFxTMsX1yDE,22
-kl_mcp_client/client.py,sha256=3NN18vvrxV32hkKFGuWqfeZLZ303jl5m8atpe14U69Q,4543
-kl_mcp_client/tools.py,sha256=kcxg4Cw1sAf8QmPCF6DW7pET55hZs64ztQxBDoe27go,9350
-kl_mcp_client/asyncio/__init__.py,sha256=pdJdBcEH5HaRuHSe2B7VUyRgH5ad3u1dDc4euZMDRMY,106
-kl_mcp_client/asyncio/client.py,sha256=giOz_h8x_RoO5wA1ebiWBy1eU7owU_YOxXAFV2Z_MCk,4439
-kl_mcp_client/asyncio/tools.py,sha256=tCeHUc4ao99Hu4ck0BmvDo95vmTtG1ChvwHamaIx_zE,7655
-kl_mcp_client-2.0.2.dist-info/METADATA,sha256=Kkl2XxG4d2ftuKO1_1BR7C84bJNSZQN6NnMBBvQx8yo,4443
-kl_mcp_client-2.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kl_mcp_client-2.0.2.dist-info/top_level.txt,sha256=wd_HFFyGjiKavwACuj8Ny0svtVyNsrxCSVU48EkoQ7c,14
-kl_mcp_client-2.0.2.dist-info/RECORD,,