clerk-sdk 0.3.3__tar.gz → 0.3.5__tar.gz

{clerk_sdk-0.3.3/clerk_sdk.egg-info → clerk_sdk-0.3.5}/PKG-INFO

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

{clerk_sdk-0.3.3 → clerk_sdk-0.3.5}/clerk/base.py

@@ -79,6 +79,8 @@ class BaseClerk(BaseModel):
         headers: Dict[str, str] = {},
         json: Dict = {},
         params: Dict = {},
+        data: Dict = None,
+        files: Dict = {},
     ) -> StandardResponse:
 
         merged_headers = {**self.headers, **headers}
@@ -88,7 +90,14 @@ class BaseClerk(BaseModel):
 
         # logger.info(f"POST {url} | body={json} | params={params}")
 
-        response = requests.post(url, headers=merged_headers, json=json, params=params)
+        response = requests.post(
+            url,
+            headers=merged_headers,
+            json=json,
+            params=params,
+            data=data,
+            files=files,
+        )
         response.raise_for_status()
 
         return StandardResponse(**response.json())

{clerk_sdk-0.3.3 → clerk_sdk-0.3.5}/clerk/client.py

@@ -1,8 +1,8 @@
-from typing import List
+from typing import List, Literal
 from xml.dom.minidom import Document
 
 from clerk.base import BaseClerk
-from .models.file import ParsedFile
+from .models.file import ParsedFile, UploadFile
 
 
 class Clerk(BaseClerk):
@@ -15,3 +15,15 @@ class Clerk(BaseClerk):
         endpoint = f"/document/{document_id}/files"
         res = self.get_request(endpoint=endpoint)
         return [ParsedFile(**d) for d in res.data]
+
+    def add_files_to_document(
+        self,
+        document_id: str,
+        type: Literal["input", "output"],
+        files: List[UploadFile],
+    ):
+        endpoint = f"/document/{document_id}/files/upload"
+        params = {"type": type}
+        files_data = []
+        files_data.append(f.to_multipart_format() for f in files)
+        self.post_request(endpoint, params=params)

{clerk_sdk-0.3.3 → clerk_sdk-0.3.5}/clerk/decorator/task_decorator.py

@@ -21,6 +21,9 @@ def clerk_code():
             error_info = None
             if payload is None:
                 use_pickle = True
+                # Write a placeholder output file in case of early failure                
+                with open(output_pkl, "wb") as f:
+                    pickle.dump({"error": "Early failure"}, f)                
                 try:
                     with open(input_pkl, "rb") as f:
                         raw_data = pickle.load(f)

{clerk_sdk-0.3.3 → clerk_sdk-0.3.5}/clerk/models/file.py

@@ -1,5 +1,5 @@
 import base64
-from typing import Optional
+from typing import Optional, Tuple
 from pydantic import BaseModel, Field
 
 
@@ -14,3 +14,12 @@ class ParsedFile(BaseModel):
             return base64.b64decode(self.content.encode("utf-8"))
         except Exception as e:
             raise ValueError(f"Invalid base64 content: {e}")
+
+
+class UploadFile(BaseModel):
+    name: str
+    mimetype: str
+    content: bytes
+
+    def to_multipart_format(self, key: str = "files") -> Tuple:
+        return (key, (self.name, self.content, self.mimetype))

{clerk_sdk-0.3.3 → clerk_sdk-0.3.5/clerk_sdk.egg-info}/PKG-INFO

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

{clerk_sdk-0.3.3 → clerk_sdk-0.3.5}/setup.py

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