clerk-sdk 0.1.0__py3-none-any.whl → 0.1.2__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,51 @@
-import json
-from typing import Dict
-from prefect import flow
+import pickle
+from typing import Callable, Optional
 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(payload: Optional[ClerkCodePayload] = None) -> ClerkCodePayload:
+            # 1. Load payload from file if not provided
+            if payload is None:
+                try:
+                    with open(input_pkl, "rb") as f:
+                        raw_data = pickle.load(f)
+                    payload = (
+                        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
+
+            # 2. Execute function
+            try:
+                output = func(payload)
+                if not isinstance(output, ClerkCodePayload):
+                    raise TypeError("Function must return a ClerkCodePayload instance.")
+            except Exception as e:
+                output = e
+
+            # 3. Always write to output.pkl
+            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
+
+            # 4. Raise if error or return result
+            if isinstance(output, Exception):
+                raise output
+            return output
 
-        return wrapped_flow
+        return wrapper
 
-    return wrapper
+    return decorator

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clerk-sdk
-Version: 0.1.0
+Version: 0.1.2
 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
@@ -27,14 +26,14 @@ Dynamic: summary
 
 # CLERK
 
-`clerk-sdk` is a Python library designed to simplify interactions with the Clerk API. It provides a robust and user-friendly interface for managing documents, handling API requests, and integrating structured data models into your workflows. With built-in support for Prefect task flows and retry mechanisms, `clerk-sdk` is ideal for developers looking to streamline their integration with Clerk.
+`clerk-sdk` is a Python library designed to simplify interactions with the Clerk API. It provides a robust and user-friendly interface for managing documents, handling API requests, and integrating structured data models into your workflows. `clerk-sdk` is ideal for developers looking to streamline their integration with Clerk.
 
 ## Features
 
 - **Document Management**: Retrieve and manage documents and their associated files.
 - **API Request Handling**: Simplified GET and POST requests with automatic retries and error handling.
 - **Data Models**: Predefined Pydantic models for structured data validation and serialization.
-- **Task Flow Integration**: Prefect-based decorators for creating and managing task flows.
+- **Task Flow Integration**: Decorator for creating and managing task flows.
 - **Extensibility**: Easily extend and customize the library to fit your specific use case.
 
 ## Installation
@@ -70,22 +69,42 @@ for file in files:
     print(file.name)
 ```
 
-### Use the Prefect Task Decorator
+### Use the Task Decorator
+
+#### PROD
 
 ```python
 from clerk.decorator import clerk_code
+from clerk.decorator.models import ClerkCodePayload
 
 @clerk_code()
-def process_document(payload):
-    # Your processing logic here
-    return {"status": "processed"}
+def main(payload: ClerkCodePayload) -> ClerkCodePayload:
+    payload.structured_data["status"] = "ok"
+    return payload
+
+main()
 ```
 
+#### TEST
+
+```python
+from clerk.decorator.models import ClerkCodePayload, Document
+
+def test_main():
+    test_payload = ClerkCodePayload(
+        document=Document(id="doc-123", message_subject="Hello"),
+        structured_data={}
+    )
+
+    result = main(test_payload)  # ✅ Just pass it!
+    assert result.structured_data["status"] == "ok"
+```
+
+
 ## Requirements
 
 - 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.2.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=CW8Fhu-4DanJWB8OaDSZtdeWhDeJQOIq4PofEDf8sqg,1799
 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.2.dist-info/licenses/LICENSE,sha256=GTVQl3vH6ht70wJXKC0yMT8CmXKHxv_YyO_utAgm7EA,1065
+clerk_sdk-0.1.2.dist-info/METADATA,sha256=Xn9ZXuobFoZilR4nKj3VuKTgyFBrnJodRJK_Nt6Ksko,2986
+clerk_sdk-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+clerk_sdk-0.1.2.dist-info/top_level.txt,sha256=99eQiU6d05_-f41tmSFanfI_SIJeAdh7u9m3LNSfcv4,6
+clerk_sdk-0.1.2.dist-info/RECORD,,

{clerk_sdk-0.1.0.dist-info → clerk_sdk-0.1.2.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