clerk-sdk 0.4.15__py3-none-any.whl → 0.4.16.dev0__py3-none-any.whl

clerk/__init__.py

@@ -1,4 +1,4 @@
 from .client import Clerk
 
 
-__version__ = "0.4.14"
+__version__ = "0.4.16.dev0"

clerk/gui_automation/decorators/gui_automation.py

@@ -2,7 +2,7 @@ import asyncio
 import functools
 import os
 import time
-from typing import Callable, Union
+from typing import Any, Callable, Dict, Sequence, Union
 
 from websockets.asyncio.client import connect, ClientConnection
 from websockets.protocol import State
@@ -24,13 +24,6 @@ global_ws: Union[ClientConnection, None] = None
 clerk_client = RPAClerk()
 wss_uri = "wss://agent-manager.f-one.group/action"
 
-REMOTE_DEVICE_ALLOCATION_TIMEOUT = int(
-    os.getenv("REMOTE_DEVICE_ALLOCATION_TIMEOUT", 60)
-)
-REMOTE_DEVICE_ALLOCATION_MAX_TRIES = int(
-    os.getenv("REMOTE_DEVICE_ALLOCATION_MAX_TRIES", 60)
-)
-
 
 async def connect_to_ws(uri: str) -> ClientConnection:
     # Same knobs as before, just via the new connect()
@@ -54,46 +47,7 @@ async def reconnect_ws():
     global_ws = await connect_to_ws(uri)
 
 
-def _allocate_remote_device(
-    clerk_client: RPAClerk, group_name: str, run_id: str
-) -> RemoteDevice:
-    remote_device = None
-    retries = 0
-
-    while True:
-        try:
-            remote_device = clerk_client.allocate_remote_device(
-                group_name=group_name, run_id=run_id
-            )
-            os.environ["REMOTE_DEVICE_ID"] = remote_device.id
-            os.environ["REMOTE_DEVICE_NAME"] = remote_device.name
-            logger.debug(f"Remote device allocated: {remote_device.name}")
-            return remote_device
-
-        except NoClientsAvailable:
-            logger.warning(
-                f"No clients are available for {group_name} group. Initiating a {REMOTE_DEVICE_ALLOCATION_TIMEOUT} seconds wait. Retry count: {retries}"
-            )
-            if retries >= REMOTE_DEVICE_ALLOCATION_MAX_TRIES:
-                raise ClientAvailabilityTimeout(
-                    f"No clients available for {group_name} group after {REMOTE_DEVICE_ALLOCATION_TIMEOUT * REMOTE_DEVICE_ALLOCATION_MAX_TRIES} seconds"
-                )
-            time.sleep(REMOTE_DEVICE_ALLOCATION_TIMEOUT)
-            retries += 1
-
-
-def _deallocate_target(
-    clerk_client: RPAClerk, remote_device: RemoteDevice, run_id: str
-):
-    clerk_client.deallocate_remote_device(remote_device=remote_device, run_id=run_id)
-    logger.debug("Remote device deallocated")
-    os.environ.pop("REMOTE_DEVICE_ID", None)
-    os.environ.pop("REMOTE_DEVICE_NAME", None)
-
-
-def gui_automation(
-    reserve_client: bool = False,
-):
+def gui_automation():
     """
     Decorator that:
       • Allocates a remote device,
@@ -101,22 +55,22 @@ def gui_automation(
       • Passes control to the wrapped function,
       • Cleans everything up afterwards.
     """
-    group_name: str | None = os.getenv("REMOTE_DEVICE_GROUP")
-    if not group_name:
-        raise ValueError("REMOTE_DEVICE_GROUP environmental variable is required.")
+    remote_device_name: str | None = os.getenv("REMOTE_DEVICE_NAME")
+
+    if not remote_device_name:
+        raise ValueError("REMOTE_DEVICE_NAME environmental variable is required.")
+
+    wss_token = clerk_client.get_wss_token()
 
     def decorator(func: Callable):
         @functools.wraps(func)
-        def wrapper(payload: ClerkCodePayload, *args, **kwargs):
+        def wrapper(
+            payload: ClerkCodePayload, *args: Sequence[Any], **kwargs: Dict[str, Any]
+        ):
             global global_ws
 
-            force_deallocate = False
             os.environ["_document_id"] = payload.document.id
-            os.environ["_run_id"] = payload.run_id
-
-            remote_device = _allocate_remote_device(
-                clerk_client, group_name, payload.run_id
-            )
+            os.environ["_run_id"] = payload.run_id or ""
 
             # Create a dedicated loop for the WebSocket work
             event_loop = asyncio.new_event_loop()
@@ -125,8 +79,8 @@ def gui_automation(
             try:
                 task = event_loop.create_task(
                     connect_to_ws(
-                        f"{wss_uri}/{remote_device.name}/publisher"
-                        f"?token={remote_device.wss_token}"
+                        f"{wss_uri}/{remote_device_name}/publisher"
+                        f"?token={wss_token}"
                     )
                 )
                 global_ws = event_loop.run_until_complete(task)
@@ -138,18 +92,11 @@ def gui_automation(
                     global_ws = None
                     raise WebSocketConnectionFailed()
 
-            except Exception as e:
-                force_deallocate = True
+            except Exception:
                 raise
             finally:
                 os.environ.pop("_run_id", None)
                 os.environ.pop("_document_id", None)
-                if not reserve_client or force_deallocate:
-                    _deallocate_target(clerk_client, remote_device, payload.run_id)
-                else:
-                    logger.warning(
-                        f"The client stayed reserved for the this run id: {payload.run_id}"
-                    )
 
                 if global_ws and global_ws.state is State.OPEN:
                     close_task = event_loop.create_task(close_ws_connection(global_ws))

{clerk_sdk-0.4.15.dist-info → clerk_sdk-0.4.16.dev0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clerk-sdk
-Version: 0.4.15
+Version: 0.4.16.dev0
 Summary: Library for interacting with Clerk
 Home-page: https://github.com/F-ONE-Group/clerk_pypi
 Author: F-ONE Group

{clerk_sdk-0.4.15.dist-info → clerk_sdk-0.4.16.dev0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-clerk/__init__.py,sha256=xGgrQuxBmA5NdRf4kuLF14MHSYwPsVBcmuyxvPhL6TU,51
+clerk/__init__.py,sha256=6KE-7b4AAmpoWge6Ecko_0Spe4J7BC3YRdmmlEmsUFQ,56
 clerk/base.py,sha256=lbFTdpdDfsmYIQUFH93S1aw0-L6GNJwAcubW1tdMFX4,3967
 clerk/client.py,sha256=RdOvC23WK9ZtIXDOYaoSFk9debh3UTmstBXjAswAH6E,4981
 clerk/decorator/__init__.py,sha256=yGGcS17VsZ7cZ-hVGCm3I3vGDJMiJIAqmDGzriIi0DI,65
@@ -17,7 +17,7 @@ clerk/gui_automation/client_actor/client_actor.py,sha256=D2mQAHitF5x-dZzt70ZPw_f
 clerk/gui_automation/client_actor/exception.py,sha256=zdnImHZ88yf52Xq3aMHivEU3aJg-r2c-r8x8XZnI3ic,407
 clerk/gui_automation/client_actor/model.py,sha256=wVpFCi1w2kh4kAV8oNx489vf_SLUQnqhc02rFD5NIJA,6335
 clerk/gui_automation/decorators/__init__.py,sha256=OCgXStEumscgT-RyVy5OKS7ml1w9y-lEnjCVnxuRnQs,43
-clerk/gui_automation/decorators/gui_automation.py,sha256=6BVT74n4JsPJmD_G_b_MNfEvq3WZAukWbQcAOX2aDPM,5770
+clerk/gui_automation/decorators/gui_automation.py,sha256=CwfDmTP9d4BOEsQS4_YGwOoz1gbDwYsA-JVnXz25Cto,3680
 clerk/gui_automation/exceptions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clerk/gui_automation/exceptions/agent_manager.py,sha256=KFWFWQ7X_8Z9XG__SMzb1jKI3yNoVTPJAXzW5O-fSXc,101
 clerk/gui_automation/exceptions/websocket.py,sha256=-MdwSwlf1hbnu55aDgk3L1znkTZ6xJS6po5VpEGD9ck,117
@@ -45,8 +45,8 @@ clerk/models/ui_operator.py,sha256=mKTJUFZgv7PeEt5oys28HVZxHOJsofmRQOcRpqj0dbU,2
 clerk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clerk/utils/logger.py,sha256=NrMIlJfVmRjjRw_N_Jngkl0qqv7btXUbg5wxcRmFEH4,3800
 clerk/utils/save_artifact.py,sha256=94aYkYNVGcSUaSWZmdjiY6Oc-3yCKb2XWCZ56IAXQqk,1158
-clerk_sdk-0.4.15.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
-clerk_sdk-0.4.15.dist-info/METADATA,sha256=5a82iUvHTK2T885B0pUGQMu2px-RzRIUxMKxH8sSd9w,9937
-clerk_sdk-0.4.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-clerk_sdk-0.4.15.dist-info/top_level.txt,sha256=99eQiU6d05_-f41tmSFanfI_SIJeAdh7u9m3LNSfcv4,6
-clerk_sdk-0.4.15.dist-info/RECORD,,
+clerk_sdk-0.4.16.dev0.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
+clerk_sdk-0.4.16.dev0.dist-info/METADATA,sha256=U8QlEn0ANQxSkmowBLby6M8IJL-BKHkXxkQ5BL6lLhc,9942
+clerk_sdk-0.4.16.dev0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+clerk_sdk-0.4.16.dev0.dist-info/top_level.txt,sha256=99eQiU6d05_-f41tmSFanfI_SIJeAdh7u9m3LNSfcv4,6
+clerk_sdk-0.4.16.dev0.dist-info/RECORD,,