clerk-sdk 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

clerk/decorator/models.py

@@ -1,8 +1,19 @@
-from typing import Dict
+from typing import Dict, List, Optional
 from pydantic import BaseModel
 
 
+class File(BaseModel):
+    name: str
+    url: str
+
+
+class Document(BaseModel):
+    id: str
+    message_subject: Optional[str] = None
+    message_content: Optional[str] = None
+    files: List[File] = []
+
+
 class ClerkCodePayload(BaseModel):
-    document_id: str
-    instance_id: str
-    data: Dict
+    document: Document
+    structured_data: Dict

clerk/decorator/task_decorator.py

@@ -1,23 +1,43 @@
-import json
-from typing import Dict
-from prefect import flow
+import pickle
+from typing import Callable
 from functools import wraps
-from prefect.states import Completed
-from pydantic import BaseModel
 from .models import ClerkCodePayload
 
+input_pkl: str = "/app/data/input/input.pkl"
+output_pkl: str = "/app/data/output/output.pkl"
+
 
 def clerk_code():
-    def wrapper(func):
+    def decorator(func: Callable[[ClerkCodePayload], ClerkCodePayload]):
         @wraps(func)
-        @flow(persist_result=False, log_prints=True, result_serializer="json")
-        def wrapped_flow(payload: Dict):
-            payload = ClerkCodePayload(**payload)
-            result = func(payload)
-            if isinstance(result, BaseModel):
-                result = result.model_dump()
-            return Completed(message=json.dumps(result), data=result)
+        def wrapper():
+            # Step 1: Load and parse input
+            try:
+                with open(input_pkl, "rb") as f:
+                    raw_data = pickle.load(f)
+                parsed = (
+                    ClerkCodePayload.model_validate(raw_data)
+                    if not isinstance(raw_data, ClerkCodePayload)
+                    else raw_data
+                )
+            except Exception as e:
+                raise RuntimeError(f"Failed to load and parse input pickle: {e}") from e
+
+            # Step 2: Call the function and validate output
+            try:
+                output = func(parsed)
+                if not isinstance(output, ClerkCodePayload):
+                    raise TypeError("Function must return a ClerkCodePayload instance.")
+            except Exception as e:
+                output = e  # Save exception to output file for later debugging
+
+            # Step 3: Write output or exception to pickle
+            try:
+                with open(output_pkl, "wb") as f:
+                    pickle.dump(output, f)
+            except Exception as e:
+                raise RuntimeError(f"Failed to write output pickle: {e}") from e
 
-        return wrapped_flow
+        return wrapper
 
-    return wrapper
+    return decorator

{clerk_sdk-0.1.0.dist-info → clerk_sdk-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clerk-sdk
-Version: 0.1.0
+Version: 0.1.1
 Summary: Library for interacting with Clerk
 Home-page: https://github.com/F-ONE-Group/clerk_pypi
 Author: F-ONE Group
@@ -11,7 +11,6 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: prefect==3.4.1
 Requires-Dist: pydantic>2.0.0
 Requires-Dist: backoff>2.0.0
 Dynamic: author
@@ -85,7 +84,6 @@ def process_document(payload):
 
 - Python 3.10+
 - Dependencies listed in `requirements.txt`:
-  - `prefect>=3.4.1`
   - `pydantic>2.0.0`
   - `backoff>2.0.0`
 

{clerk_sdk-0.1.0.dist-info → clerk_sdk-0.1.1.dist-info}/RECORD

@@ -1,15 +1,15 @@
 clerk/__init__.py,sha256=LWpbImG7352mUJYC1tRm_zsn5rnt4sFl5ldoq5f0dlo,26
 clerk/client.py,sha256=UEbs5AhnBs6JWstLzT-TPHZ-yxyBwTFDLemv-yh25K4,3386
 clerk/decorator/__init__.py,sha256=4VCGOFNq7pA7iJS410V_R9eWfjxP7mwxwa4thwaUTf8,39
-clerk/decorator/models.py,sha256=K0vFIh0tzklNxVzntllDDSWuzLf5Sn1pnbJ9Ke7dcrs,149
-clerk/decorator/task_decorator.py,sha256=DnsosG6GA3CLktO88bT32Q_3X0YriRUCD7d2xyePmSE,673
+clerk/decorator/models.py,sha256=FKs7oCq50hmvbzYWuscvqlL7boPYzMKv0Vg3tznmwao,361
+clerk/decorator/task_decorator.py,sha256=nLKpPG4kRvNuviO5OLeBzHQppNG2RTHFclGOSARajdM,1536
 clerk/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clerk/models/document.py,sha256=Bbt3YSvGcOrJJjJBYjbDGXpZIaQAYdq_-uIr2urwy5s,525
 clerk/models/document_statuses.py,sha256=ytTQhgACs2m22qz51_7Ti0IxzbVyl-fl7uF7CnDEyLA,279
 clerk/models/file.py,sha256=7n6bV6ukRA-uh7MFxUx_TEMNpVAdNe5O0nKIeneCQhg,459
 clerk/models/response_model.py,sha256=R62daUN1YVOwgnrh_epvFRsQcOwT7R4u97l73egvm-c,232
-clerk_sdk-0.1.0.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
-clerk_sdk-0.1.0.dist-info/METADATA,sha256=Z-raplIGs2ITxdmC2HF9iflHKtVodWZehdKWragyW9U,2685
-clerk_sdk-0.1.0.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-clerk_sdk-0.1.0.dist-info/top_level.txt,sha256=99eQiU6d05_-f41tmSFanfI_SIJeAdh7u9m3LNSfcv4,6
-clerk_sdk-0.1.0.dist-info/RECORD,,
+clerk_sdk-0.1.1.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
+clerk_sdk-0.1.1.dist-info/METADATA,sha256=LYguUv0vCO8dpZOkwO5TSuD6VpQeTzq4DcgbRMIasmI,2634
+clerk_sdk-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+clerk_sdk-0.1.1.dist-info/top_level.txt,sha256=99eQiU6d05_-f41tmSFanfI_SIJeAdh7u9m3LNSfcv4,6
+clerk_sdk-0.1.1.dist-info/RECORD,,

{clerk_sdk-0.1.0.dist-info → clerk_sdk-0.1.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any