clerk-sdk 0.3.7__tar.gz → 0.4.0__tar.gz

{clerk_sdk-0.3.7/clerk_sdk.egg-info → clerk_sdk-0.4.0}/PKG-INFO

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

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/base.py

@@ -2,7 +2,7 @@ import os
 
 import requests
 import backoff
-from typing import Dict, Optional, Self
+from typing import Any, Dict, List, Optional, Self, Tuple
 
 
 from pydantic import BaseModel, model_validator, Field
@@ -10,7 +10,7 @@ from pydantic import BaseModel, model_validator, Field
 from .models.response_model import StandardResponse
 
 
-def giveup_handler(e):
+def giveup_handler(e: Any):
     return (
         isinstance(e, requests.exceptions.HTTPError)
         and e.response is not None
@@ -42,16 +42,15 @@ class BaseClerk(BaseModel):
         (requests.exceptions.RequestException,),
         max_tries=3,
         jitter=None,
-        # on_backoff=backoff_handler,
         giveup=giveup_handler,
     )
     def get_request(
         self,
         endpoint: str,
         headers: Dict[str, str] = {},
-        json: Dict = {},
-        params: Dict = {},
-    ) -> StandardResponse:
+        json: Dict[str, Any] = {},
+        params: Dict[str, Any] = {},
+    ) -> StandardResponse[Dict[str, Any]]:
 
         merged_headers = {**self.headers, **headers}
         url = f"{self.base_url}{endpoint}"
@@ -70,18 +69,17 @@ class BaseClerk(BaseModel):
         (requests.exceptions.RequestException,),
         max_tries=3,
         jitter=None,
-        # on_backoff=backoff_handler,
         giveup=giveup_handler,
     )
     def post_request(
         self,
         endpoint: str,
         headers: Dict[str, str] = {},
-        json: Dict = {},
-        params: Dict = {},
-        data: Dict = None,
-        files: Dict = {},
-    ) -> StandardResponse:
+        json: Dict[str, Any] = {},
+        params: Dict[str, Any] = {},
+        data: Dict[str, Any] | None = None,
+        files: List[Tuple[str, Tuple[str, bytes, str | None]]] | None = None,
+    ) -> StandardResponse[Dict[str, Any]]:
 
         merged_headers = {**self.headers, **headers}
         url = f"{self.base_url}{endpoint}"
@@ -96,7 +94,43 @@ class BaseClerk(BaseModel):
             json=json,
             params=params,
             data=data,
-            files=files,
+            files=files,  # type: ignore
+        )
+        response.raise_for_status()
+
+        return StandardResponse(**response.json())
+
+    @backoff.on_exception(
+        backoff.expo,
+        (requests.exceptions.RequestException,),
+        max_tries=3,
+        jitter=None,
+        giveup=giveup_handler,
+    )
+    def put_request(
+        self,
+        endpoint: str,
+        headers: Dict[str, str] = {},
+        json: Dict[str, Any] = {},
+        params: Dict[str, Any] = {},
+        data: Dict[str, Any] | None = None,
+        files: List[Tuple[str, Tuple[str, bytes, str | None]]] | None = None,
+    ) -> StandardResponse[Dict[str, Any]]:
+
+        merged_headers = {**self.headers, **headers}
+        url = f"{self.base_url}{endpoint}"
+        if self.root_endpoint:
+            url = f"{self.base_url}{self.root_endpoint}{endpoint}"
+
+        # logger.info(f"POST {url} | body={json} | params={params}")
+
+        response = requests.put(
+            url,
+            headers=merged_headers,
+            json=json,
+            params=params,
+            data=data,
+            files=files,  # type: ignore
         )
         response.raise_for_status()
 

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/client.py

@@ -1,11 +1,28 @@
-from typing import List, Literal
-from xml.dom.minidom import Document
+from typing import Any, Dict, List, Literal
 
 from clerk.base import BaseClerk
+from clerk.models.document import Document, UploadDocumentRequest
 from .models.file import ParsedFile, UploadFile
 
 
 class Clerk(BaseClerk):
+
+    def upload_document(self, request: UploadDocumentRequest) -> Document:
+        endpoint = "/document"
+        res = self.post_request(
+            endpoint=endpoint, data=request.data, files=request.files_
+        )
+        return Document(**res.data[0])
+
+    def update_document_structured_data(
+        self, document_id: str, updated_structured_data: Dict[str, Any]
+    ) -> Document:
+        endpoint = f"/document/{document_id}"
+        payload = dict(structured_data=updated_structured_data)
+        res = self.put_request(endpoint, json=payload)
+
+        return Document(**res.data[0])
+
     def get_document(self, document_id: str) -> Document:
         endpoint = f"/document/{document_id}"
         res = self.get_request(endpoint=endpoint)

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/decorator/__init__.py

@@ -1 +1,3 @@
 from .task_decorator import clerk_code
+
+__all__ = ["clerk_code"]

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/decorator/models.py

@@ -1,4 +1,4 @@
-from typing import Dict, List, Optional
+from typing import Any, Dict, List, Optional
 from pydantic import BaseModel
 
 
@@ -16,5 +16,5 @@ class Document(BaseModel):
 
 class ClerkCodePayload(BaseModel):
     document: Document
-    structured_data: Dict
+    structured_data: Dict[str, Any]
     run_id: Optional[str] = None

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/decorator/task_decorator.py

@@ -16,14 +16,14 @@ def clerk_code():
         def wrapper(payload: Optional[ClerkCodePayload] = None) -> ClerkCodePayload:
             # 1. Load payload from file if not provided
             use_pickle = False
-            output = None
+            output: ClerkCodePayload | BaseException | None = None
             error_occurred = False
             error_info = None
             if payload is None:
                 use_pickle = True
-                # Write a placeholder output file in case of early failure                
+                # Write a placeholder output file in case of early failure
                 with open(output_pkl, "wb") as f:
-                    pickle.dump({"error": "Early failure"}, f)                
+                    pickle.dump({"error": "Early failure"}, f)
                 try:
                     with open(input_pkl, "rb") as f:
                         raw_data = pickle.load(f)
@@ -40,11 +40,13 @@ def clerk_code():
                         traceback=traceback.format_exc(),
                     )
 
+            assert payload is not None
+
             # 2. Execute function
             if not error_occurred:
                 try:
                     output = func(payload)
-                    if not isinstance(output, ClerkCodePayload):
+                    if not isinstance(output, ClerkCodePayload):  # type: ignore
                         raise TypeError(
                             "Function must return a ClerkCodePayload instance."
                         )
@@ -58,7 +60,7 @@ def clerk_code():
 
             # 3. write to output.pkl
             try:
-                if use_pickle:
+                if use_pickle and output is not None:
                     with open(output_pkl, "wb") as f:
                         if error_occurred:
                             pickle.dump(error_info, f)
@@ -66,6 +68,7 @@ def clerk_code():
                             pickle.dump(output, f)
                         else:
                             pickle.dump(output.model_dump(mode="json"), f)
+
             except Exception as e:
                 # If writing output.pkl fails, try to write a minimal error
                 try:
@@ -85,6 +88,9 @@ def clerk_code():
             if isinstance(output, Exception):
                 raise output
 
+            if output is None:
+                raise RuntimeError("output object cannot be None")
+
             return output
 
         return wrapper

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/exceptions/exceptions.py

@@ -1,10 +1,10 @@
-from typing import Optional
+from typing import Any, Optional
 
 
 class AppBaseException(Exception):
     def __init__(
         self,
-        *args,
+        *args: Any,
         type_: Optional[str] = None,
         message: Optional[str] = None,
         traceback: Optional[str] = None,
@@ -23,17 +23,17 @@ class AppBaseException(Exception):
         self.traceback = traceback
 
     # (Optional) make pickling round-trip the extra fields explicitly
-    def __reduce__(self):
+    def __reduce__(self):  # type: ignore
         # Reconstruct with message-only (what Exception expects) and restore extras via state
         return (
             self.__class__,
             (self.message,),
             {"type": self.type, "traceback": self.traceback},
-        )
+        )  # type: ignore
 
-    def __setstate__(self, state):
-        for k, v in state.items():
-            setattr(self, k, v)
+    def __setstate__(self, state):  # type: ignore
+        for k, v in state.items():  # type: ignore
+            setattr(self, k, v)  # type: ignore
 
 
 class ApplicationException(AppBaseException):

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/action_model/model.py

@@ -1,7 +1,7 @@
 import base64
 import os
 from typing import List, Literal, Optional, Union
-from pydantic import BaseModel, Field, validator
+from pydantic import BaseModel
 
 CoordsType = Union[List[float], List[int]]
 

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/client.py

@@ -1,4 +1,4 @@
-from typing import Dict, List, Optional
+from typing import Any, Dict, List, Optional, Type
 
 from pydantic import BaseModel
 from clerk.base import BaseClerk
@@ -16,7 +16,7 @@ from clerk.models.ui_operator import UiOperatorTask
 
 class RPAClerk(BaseClerk):
 
-    root_endpoint: str = "/gui_automation"
+    root_endpoint = "/gui_automation"
 
     def allocate_remote_device(self, group_name: str, run_id: str):
         endpoint = "/remote_device/allocate"
@@ -24,7 +24,7 @@ class RPAClerk(BaseClerk):
             endpoint=endpoint, json={"group_name": group_name, "run_id": run_id}
         )
 
-        if res.data[0] is None:
+        if res.data[0] is None:  # type: ignore
             raise NoClientsAvailable()
 
         return RemoteDevice(**res.data[0])
@@ -40,14 +40,14 @@ class RPAClerk(BaseClerk):
             json={"id": remote_device.id, "name": remote_device.name, "run_id": run_id},
         )
 
-    def get_coordinates(self, payload: Dict) -> Coords:
+    def get_coordinates(self, payload: Dict[str, Any]) -> Coords:
         endpoint = "/action_model/get_coordinates"
         res = self.post_request(endpoint=endpoint, json=payload)
-        if res.data[0] is None:
+        if res.data[0] is None:  # type: ignore
             raise RuntimeError("No coordinates found in the response.")
         return Coords(**res.data[0])
 
-    def create_ui_operator_task(self, payload: Dict) -> UiOperatorTask:
+    def create_ui_operator_task(self, payload: Dict[str, Any]) -> UiOperatorTask:
         endpoint = "/ui_operator"
         res = self.post_request(endpoint=endpoint, json=payload)
         return UiOperatorTask(**res.data[0])
@@ -59,7 +59,7 @@ class RPAClerk(BaseClerk):
 
 
 class GUIVisionClerk(BaseClerk):
-    root_endpoint: str = "/gui_automation/vision"
+    root_endpoint = "/gui_automation/vision"
 
     def find_target(self, screen_b64: str, use_ocr: bool, target_prompt: str):
         endpoint = "/find_target"
@@ -89,8 +89,12 @@ class GUIVisionClerk(BaseClerk):
         return BaseState(**res.data[0])
 
     def answer(
-        self, screen_b64: str, use_ocr: bool, question: str, output_model: BaseModel
-    ) -> Dict:
+        self,
+        screen_b64: str,
+        use_ocr: bool,
+        question: str,
+        output_model: Type[BaseModel],
+    ) -> BaseModel:
         endpoint = "/answer"
         res = self.post_request(
             endpoint=endpoint,
@@ -102,7 +106,7 @@ class GUIVisionClerk(BaseClerk):
             },
         )
 
-        return output_model(**res.data[0])
+        return output_model.model_validate(**res.data[0])
 
     def classify_state(
         self, screen_b64: str, use_ocr: bool, possible_states: List[Dict[str, str]]
@@ -136,12 +140,12 @@ class GUIVisionClerk(BaseClerk):
 
 
 class CourseCorrectorClerk(BaseClerk):
-    root_endpoint: str = "/gui_automation/course_corrector"
+    root_endpoint = "/gui_automation/course_correction"
 
     def get_corrective_actions(
         self,
         screen_b64: str,
-        use_ocr: str,
+        use_ocr: bool,
         goal: str,
         custom_instructions: Optional[str] = None,
     ) -> ActionString:

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/client_actor/client_actor.py

@@ -21,7 +21,7 @@ from .model import PerformActionResponse, ActionStates
 from .exception import PerformActionException, GetScreenError
 
 
-async def _perform_action_ws(payload: Dict) -> PerformActionResponse:
+async def _perform_action_ws(payload: Dict[str, Any]) -> PerformActionResponse:
     """Perform an action over a WebSocket connection.
 
     Args:
@@ -65,7 +65,7 @@ async def _get_screen_async() -> str:
         This function sends a request to perform a screenshot action over a WebSocket connection
         and returns the base64 encoded image of the screen captured.
     """
-    payload = {
+    payload: Dict[str, Any] = {
         "proc_inst_id": os.getenv("_run_id"),
         "client_name": os.getenv("REMOTE_DEVICE_NAME"),
         "headless": True,
@@ -131,7 +131,7 @@ async def _perform_action_async(
     Raises:
         PerformActionException: If the action fails with an error message.
     """
-    req_payload: Dict = {
+    req_payload: Dict[str, Any] = {
         "proc_inst_id": os.getenv("_run_id"),
         "client_name": os.getenv("REMOTE_DEVICE_NAME"),
         "headless": (

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/client_actor/model.py

@@ -1,20 +1,22 @@
-from typing import Any, List, Literal, Optional, Union
+from typing import Any, List, Literal, Optional, TypeAlias, Union
 from pydantic import BaseModel, Field
 from enum import Enum
 
+from clerk.gui_automation.ui_actions.base import ActionTypes
 
-ActionTypes = Literal[
-    "left_click",
-    "right_click",
-    "middle_click",
-    "double_click",
-    "send_keys",
-    "press_keys",
-    "hot_keys",
-    "paste_text",
-    "get_text",
-    "scroll",
-]
+
+# ActionTypes = Literal[
+#     "left_click",
+#     "right_click",
+#     "middle_click",
+#     "double_click",
+#     "send_keys",
+#     "press_keys",
+#     "hot_keys",
+#     "paste_text",
+#     "get_text",
+#     "scroll",
+# ]
 
 
 class ActionStates(Enum):
@@ -77,6 +79,7 @@ class WindowExecutePayload(BaseModel):
         "close_window",
         "activate_window",
     ]
+
     window_name: str
     timeout: int = Field(default=10)
 

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/decorators/gui_automation.py

@@ -79,7 +79,7 @@ def gui_automation(
       • Passes control to the wrapped function,
       • Cleans everything up afterwards.
     """
-    group_name: str = os.getenv("REMOTE_DEVICE_GROUP")
+    group_name: str | None = os.getenv("REMOTE_DEVICE_GROUP")
     if not group_name:
         raise ValueError("REMOTE_DEVICE_GROUP environmental variable is required.")
 

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/ui_actions/actions.py

@@ -45,7 +45,7 @@ class File(BaseModel):
 
     @field_validator("content", mode="before")
     @classmethod
-    def convert_to_bytes(cls, v) -> bytes:
+    def convert_to_bytes(cls, v: str | bytes) -> bytes:
         if isinstance(v, str):
             from base64 import b64decode
 
@@ -75,7 +75,7 @@ class LeftClick(BaseAction):
         LeftClick(target="Suche").above("Kalender").do()
     """
 
-    action_type: Literal["left_click"] = "left_click"
+    action_type = "left_click"
 
     @backoff.on_exception(
         backoff.expo,
@@ -100,7 +100,7 @@ class LeftClick(BaseAction):
 
     @property
     def actionable_string(self):
-        return f"LeftClick(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchor}', relation='{self.relation}').do()"
+        return f"LeftClick(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchors}').do()"
 
 
 class RightClick(BaseAction):
@@ -118,7 +118,7 @@ class RightClick(BaseAction):
         RightClick(target="Suche").above("Kalender").do()
     """
 
-    action_type: Literal["right_click"] = "right_click"
+    action_type = "right_click"
 
     @backoff.on_exception(
         backoff.expo,
@@ -143,7 +143,7 @@ class RightClick(BaseAction):
 
     @property
     def actionable_string(self):
-        return f"RightClick(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchor}', relation='{self.relation}').do()"
+        return f"RightClick(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchors}').do()"
 
 
 class MiddleClickAction(BaseAction):
@@ -161,7 +161,7 @@ class MiddleClickAction(BaseAction):
         MiddleClickAction(target="Suche").above("Kalender").do()
     """
 
-    action_type: Literal["middle_click"] = "middle_click"
+    action_type = "middle_click"
 
     @backoff.on_exception(
         backoff.expo,
@@ -186,7 +186,7 @@ class MiddleClickAction(BaseAction):
 
     @property
     def actionable_string(self):
-        return f"MiddleClickAction(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchor}', relation='{self.relation}').do()"
+        return f"MiddleClickAction(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchors}').do()"
 
 
 class DoubleClick(BaseAction):
@@ -204,7 +204,7 @@ class DoubleClick(BaseAction):
         DoubleClick(target="Suche").above("Kalender").do()
     """
 
-    action_type: Literal["double_click"] = "double_click"
+    action_type = "double_click"
 
     @backoff.on_exception(
         backoff.expo,
@@ -229,7 +229,7 @@ class DoubleClick(BaseAction):
 
     @property
     def actionable_string(self):
-        return f"DoubleClick(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchor}', relation='{self.relation}').do()"
+        return f"DoubleClick(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchors}').do()"
 
 
 class Scroll(BaseAction):
@@ -249,7 +249,7 @@ class Scroll(BaseAction):
         DoubleClick(target="Suche").above("Kalender").do()
     """
 
-    action_type: Literal["scroll"] = "scroll"
+    action_type = "scroll"
     clicks: int
     click_coords: List[int] = Field(default=[])
 
@@ -333,7 +333,7 @@ class SendKeys(BaseAction):
 
     @property
     def actionable_string(self):
-        return f"SendKeys(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchor}', relation='{self.relation}', keys='{self.keys}').do()"
+        return f"SendKeys(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchors}', keys='{self.keys}').do()"
 
 
 class PressKeys(BaseAction):
@@ -353,7 +353,7 @@ class PressKeys(BaseAction):
         PressKeys(keys='ctrl+shift+esc').do()
     """
 
-    action_type: ActionTypes = "press_keys"
+    action_type = "press_keys"
     keys: str
 
     def do(self):
@@ -366,7 +366,7 @@ class PressKeys(BaseAction):
 
     @property
     def actionable_string(self):
-        return f"PressKeys(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchor}', relation='{self.relation}', keys='{self.keys}').do()"
+        return f"PressKeys(action_type='{self.action_type}', target='{self.target}', anchor='{self.anchors}', keys='{self.keys}').do()"
 
 
 class WaitFor(BaseAction):
@@ -385,7 +385,7 @@ class WaitFor(BaseAction):
         WaitFor("element").do(timeout=60)
     """
 
-    action_type: Literal["wait_for"] = "wait_for"
+    action_type = "wait_for"
     retry_timeout: float = 0.5
     is_awaited: bool = True
 
@@ -450,7 +450,7 @@ class OpenApplication(BaseAction):
         OpenApplication(app_path="/path/to/application.exe", app_window_name="Application Window").do()
     """
 
-    action_type: Literal["open_app"] = "open_app"
+    action_type = "open_app"
     app_path: str = Field(description="Absolute path of the application")
     app_window_name: str = Field(
         description="Name of the application window once open. Wildcard logic enabled."
@@ -479,7 +479,7 @@ class ForceCloseApplication(BaseAction):
             Executes the action to force close the application by creating and performing an ApplicationExecutePayload with the specified process name.
     """
 
-    action_type: Literal["force_close_app"] = "force_close_app"
+    action_type = "force_close_app"
     process_name: str = Field(
         description="Process name from task manager. Example: process.exe"
     )
@@ -509,7 +509,7 @@ class SaveFiles(BaseAction):
         SaveFiles(save_location="/path/to/", files=["/path/to/file_1", "/path/to/file_2"]).do()
     """
 
-    action_type: ActionTypes = "save_files"
+    action_type = "save_files"
     save_location: str
     files: Union[List[str], List[FileDetails]]
 
@@ -580,7 +580,7 @@ class GetFile(BaseAction):
         GetFile(file_location="/path/to/file_1").do()
     """
 
-    action_type: Literal["get_file"] = "get_file"
+    action_type = "get_file"
     file_location: str
 
     def do(self) -> File:
@@ -605,7 +605,7 @@ class MaximizeWindow(BaseAction):
         MaximizeWindow(window_name="MyWindow").do()
     """
 
-    action_type: Literal["maximize_window"] = "maximize_window"
+    action_type = "maximize_window"
     window_name: str
 
     def do(self, timeout: int = 10):
@@ -630,7 +630,7 @@ class MinimizeWindow(BaseAction):
         MinimizeWindow(window_name="MyWindow").do()
     """
 
-    action_type: Literal["minimize_window"] = "minimize_window"
+    action_type = "minimize_window"
     window_name: str
 
     def do(self, timeout: int = 10):
@@ -655,7 +655,7 @@ class CloseWindow(BaseAction):
         CloseWindow(window_name="MyWindow").do()
     """
 
-    action_type: Literal["close_window"] = "close_window"
+    action_type = "close_window"
     window_name: str
 
     def do(self, timeout: int = 10):
@@ -680,7 +680,7 @@ class ActivateWindow(BaseAction):
         ActivateWindow(window_name="MyWindow").do()
     """
 
-    action_type: Literal["activate_window"] = "activate_window"
+    action_type = "activate_window"
     window_name: str
 
     def do(self, timeout: int = 10):
@@ -709,7 +709,7 @@ class GetText(BaseAction):
 
     """
 
-    action_type: Literal["get_text"] = "get_text"
+    action_type = "get_text"
 
     @backoff.on_exception(
         backoff.expo,
@@ -752,7 +752,7 @@ class PasteText(BaseAction):
 
     """
 
-    action_type: Literal["paste_text"] = "paste_text"
+    action_type = "paste_text"
     keys: Union[str, List[str]]
     followed_by: Optional[str] = Field(default=None)
 

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/ui_actions/base.py

@@ -1,4 +1,4 @@
-from typing import Literal, Self, Union, List, Optional
+from typing import Literal, Self, TypeAlias, Union, List, Optional
 from pydantic import BaseModel, Field, model_validator
 from ..client_actor import get_screen
 from ..exceptions.modality.exc import TargetModalityError
@@ -24,12 +24,13 @@ def to_full_img_path(img: Union[str, ImageB64]) -> str:
     return os.path.join(TARGET_IMAGES_PATH, img)
 
 
-ActionTypes = Literal[
+ActionTypes: TypeAlias = Literal[
     "left_click",
     "right_click",
     "middle_click",
     "double_click",
     "send_keys",
+    "hot_keys",
     "press_keys",
     "wait_for",
     "open_app",
@@ -130,6 +131,8 @@ class BaseAction(BaseModel):
         return [xcenter, ycenter]
 
     def _prepare_payload(self):
+        if not self.target:
+            raise ValueError("target cannot be None.")
         payload: Screenshot = Screenshot(
             screen_b64=ImageB64(value=get_screen()),
             target=self.target,

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/ui_state_inspector/gui_vision.py

@@ -115,7 +115,7 @@ class Vision(BaseModel):
             The current state of the GUI (BaseState or a subclass of BaseState)
         """
         screen_b64 = get_screen()
-        state = self.verify_state(
+        state = self.clerk_client.verify_state(
             screen_b64,
             self.use_ocr,
             possible_states,
@@ -160,7 +160,7 @@ class Vision(BaseModel):
             screen_b64, self.use_ocr, possible_states
         )
         # if output_model is provided, return the model, otherwise return the id and description of the default model
-        if output_model is not None:
+        if output_model is not None:  # type: ignore
             return state
         assert isinstance(state, BaseState)
         return state.id, state.description

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/ui_state_inspector/models.py

@@ -1,5 +1,5 @@
-from pydantic import BaseModel, field_validator, model_validator
-from typing import List, Dict, Optional
+from pydantic import BaseModel, field_validator
+from typing import Any, List, Dict, Optional
 
 
 class BaseState(BaseModel):
@@ -21,7 +21,7 @@ class BaseState(BaseModel):
 
     id: str
     description: str
-    screenshots: List = []
+    screenshots: List[Dict[str, str]] = []
 
     def add_screenshot(self, bucket_name: str, file_name: str):
         self.screenshots.append({"bucket_name": bucket_name, "file_name": file_name})
@@ -137,7 +137,7 @@ class TargetWithAnchor(BaseModel):
 
     @field_validator("target", mode="before")
     @classmethod
-    def retain_one_word(cls, v):
+    def retain_one_word(cls, v: str):
         return v.split(" ")[-1]
 
 
@@ -174,7 +174,7 @@ class ActionString(BaseModel):
 
     @field_validator("action_string", mode="before")
     @classmethod
-    def ensure_format(cls, v):
+    def ensure_format(cls, v: Any):
         if not isinstance(v, str):
             raise ValueError("Action string must be a string")
         if not v.startswith("LeftClick") and not v.startswith("NoAction"):

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/gui_automation/ui_state_machine/models.py

@@ -1,5 +1,5 @@
-from typing import Optional
-from pydantic import BaseModel, field_validator, model_validator
+from typing import Any, Optional
+from pydantic import BaseModel, field_validator
 
 
 class ActionString(BaseModel):
@@ -24,7 +24,7 @@ class ActionString(BaseModel):
 
     @field_validator("action_string", mode="before")
     @classmethod
-    def ensure_format(cls, v):
+    def ensure_format(cls, v: Any):
         if not isinstance(v, str):
             raise ValueError("Action string must be a string")
         if not v.endswith(".do()") and not v.startswith("NoAction"):
@@ -32,7 +32,7 @@ class ActionString(BaseModel):
         return v
 
     @field_validator("interrupt_process", mode="before")
-    def convert_to_bool(cls, v):
+    def convert_to_bool(cls, v: str | bool | None):
         if v is None:
             return False
         elif isinstance(v, str):

clerk_sdk-0.4.0/clerk/models/document.py

@@ -0,0 +1,81 @@
+from datetime import datetime
+import mimetypes
+import os
+from typing import Any, Dict, List, Optional
+from pydantic import BaseModel
+
+from clerk.models.document_statuses import DocumentStatuses
+from clerk.models.file import ParsedFile
+
+
+class Document(BaseModel):
+    id: str
+    project_id: str
+    title: str
+    upload_date: datetime
+    requestor: Optional[str] = None
+    message_subject: Optional[str] = None
+    message_content: Optional[str] = None
+    message_html: Optional[str] = None
+    structured_data: Optional[Dict[str, Any]] = None
+    status: DocumentStatuses
+    created_at: datetime
+    updated_at: datetime
+
+
+class UploadDocumentRequest(BaseModel):
+    project_id: str
+    message_subject: Optional[str] = None
+    message_content: Optional[str] = None
+    files: List[str | ParsedFile] = []
+
+    def _define_files(self):
+        formatted_files: List[
+            tuple[
+                str,
+                tuple[
+                    str,
+                    bytes,
+                    str | None,
+                ],
+            ]
+        ] = []
+
+        for file in self.files:
+            if isinstance(file, str):
+                if os.path.exists(file):
+                    tmp = (
+                        "files",
+                        (
+                            os.path.basename(file).replace(" ", "_"),
+                            open(file, "rb").read(),
+                            mimetypes.guess_type(file)[0],
+                        ),
+                    )
+
+                else:
+                    raise FileExistsError(file)
+            else:
+                tmp = (
+                    "files",
+                    (
+                        file.name,
+                        file.decoded_content,
+                        file.mimetype,
+                    ),
+                )
+            formatted_files.append(tmp)
+
+        return formatted_files
+
+    @property
+    def data(self) -> Dict[str, Any]:
+        return dict(
+            project_id=self.project_id,
+            message_subject=self.message_subject,
+            mesasge_content=self.message_content,
+        )
+
+    @property
+    def files_(self):
+        return self._define_files()

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/clerk/models/file.py

@@ -18,8 +18,10 @@ class ParsedFile(BaseModel):
 
 class UploadFile(BaseModel):
     name: str
-    mimetype: str
+    mimetype: str | None
     content: bytes
 
-    def to_multipart_format(self, key: str = "files") -> Tuple:
+    def to_multipart_format(
+        self, key: str = "files"
+    ) -> Tuple[str, Tuple[str, bytes, str | None]]:
         return (key, (self.name, self.content, self.mimetype))

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0/clerk_sdk.egg-info}/PKG-INFO

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

{clerk_sdk-0.3.7 → clerk_sdk-0.4.0}/setup.py

@@ -13,7 +13,7 @@ gui_requirements = get_requirements("./clerk/gui_automation")
 
 setup(
     name="clerk-sdk",
-    version="0.3.7",
+    version="0.4.0",
     description="Library for interacting with Clerk",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",

clerk_sdk-0.3.7/clerk/models/document.py

@@ -1,20 +0,0 @@
-import datetime
-from typing import Dict, Optional
-from pydantic import BaseModel
-
-from clerk.models.document_statuses import DocumentStatuses
-
-
-class Document(BaseModel):
-    id: str
-    project_id: str
-    title: str
-    upload_date: datetime
-    requestor: Optional[str] = None
-    message_subject: Optional[str] = None
-    message_content: Optional[str] = None
-    message_html: Optional[str] = None
-    structured_data: Optional[Dict] = None
-    status: DocumentStatuses
-    created_at: datetime
-    updated_at: datetime