feat: add example

Author: mrkaye97

examples/v2/trigger.py

@@ -0,0 +1,11 @@
+from examples.v2.workflows import ExampleWorkflowInput, example_workflow
+
+
+def main() -> None:
+    example_workflow.run(
+        input=ExampleWorkflowInput(message="Hello, world!"),
+    )
+
+
+if __name__ == "__main__":
+    main()

examples/v2/worker.py

@@ -0,0 +1,25 @@
+from examples.v2.workflows import example_workflow, hatchet
+from hatchet_sdk import Context
+from hatchet_sdk.v2 import BaseWorkflowImpl
+
+
+class ExampleV2Workflow(BaseWorkflowImpl):
+    config = example_workflow.config
+
+    @hatchet.step(timeout="11s", retries=3)
+    def step1(self, context: Context) -> None:
+        input = example_workflow.workflow_input(context)
+
+        print(input.message)
+
+        return None
+
+
+def main() -> None:
+    worker = hatchet.worker("test-worker", max_runs=1)
+    worker.register_workflow(ExampleV2Workflow())
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/v2/workflows.py

@@ -0,0 +1,20 @@
+from dotenv import load_dotenv
+from pydantic import BaseModel
+
+from hatchet_sdk.v2 import Hatchet
+
+load_dotenv()
+
+hatchet = Hatchet(debug=True)
+
+
+class ExampleWorkflowInput(BaseModel):
+    message: str
+
+
+example_workflow = hatchet.declare_workflow(
+    name="example-workflow",
+    on_events=["example-event"],
+    timeout="10m",
+    input_validator=ExampleWorkflowInput,
+)

hatchet_sdk/clients/admin.py

@@ -91,6 +91,8 @@ def _prepare_workflow_request(
             payload_data = json.dumps(input)
             _options = options.model_dump()
 
+            _options.pop("namespace")
+
             try:
                 _options["additional_metadata"] = json.dumps(
                     options.additional_metadata

hatchet_sdk/v2/workflows.py

@@ -178,6 +178,9 @@ def run(self, input: TWorkflowInput | None = None) -> Any:
             workflow_name=self.config.name, input=input.model_dump() if input else {}
         )
 
+    def workflow_input(self, ctx: Context) -> TWorkflowInput:
+        return cast(TWorkflowInput, ctx.workflow_input())
+
 
 class BaseWorkflowImpl:
     """
@@ -186,9 +189,10 @@ class BaseWorkflowImpl:
     Configuration is passed to the workflow implementation via the `config` attribute.
     """
 
-    declaration: WorkflowDeclaration = WorkflowDeclaration(
-        config=WorkflowConfig(), hatchet=None
-    )
+    config: WorkflowConfig = WorkflowConfig()
+
+    def __init__(self) -> None:
+        self.config.name = self.config.name or str(self.__class__.__name__)
 
     def get_service_name(self, namespace: str) -> str:
         return f"{namespace}{self.config.name.lower()}"
@@ -219,10 +223,6 @@ def steps(self) -> list[Step[Any]]:
     def create_action_name(self, namespace: str, step: Step[Any]) -> str:
         return self.get_service_name(namespace) + ":" + step.name
 
-    def __init__(self) -> None:
-        self.config = self.declaration.config
-        self.config.name = self.config.name or str(self.__class__.__name__)
-
     def get_name(self, namespace: str) -> str:
         return namespace + self.config.name