Merge pull request #3 from hatchet-dev/feat--more-typing-and-remove-v2

Feat: fixing all the mypy errors

Author: mrkaye97

conftest.py

@@ -4,7 +4,7 @@
 import time
 from io import BytesIO
 from threading import Thread
-from typing import AsyncGenerator, Callable, cast
+from typing import AsyncGenerator, Callable, Generator, cast
 
 import psutil
 import pytest
@@ -24,7 +24,9 @@ def hatchet() -> Hatchet:
 
 
 @pytest.fixture()
-def worker(request: pytest.FixtureRequest):
+def worker(
+    request: pytest.FixtureRequest,
+) -> Generator[subprocess.Popen[bytes], None, None]:
     example = cast(str, request.param)
 
     command = ["poetry", "run", example]

examples/__init__.py

@@ -6,7 +6,8 @@
 
 client = new_client()
 
-# client.event.push("user:create", {"test": "test"})
 client.event.push(
-    "user:create", {"test": "test"}, options={"additional_metadata": {"hello": "moon"}}
+    "user:create",
+    {"test": "test"},
+    options=PushEventOptions(additional_metadata={"hello": "moon"}),
 )

examples/_deprecated/README.md

@@ -7,7 +7,7 @@
 from dotenv import load_dotenv
 
 from hatchet_sdk import new_client
-from hatchet_sdk.clients.admin import TriggerWorkflowOptions
+from hatchet_sdk.clients.admin import TriggerWorkflowOptions, WorkflowRunDict
 from hatchet_sdk.clients.rest.models.workflow_run import WorkflowRun
 from hatchet_sdk.clients.run_event_listener import StepRunEventType
 
@@ -16,25 +16,22 @@ async def main() -> None:
     load_dotenv()
     hatchet = new_client()
 
-    workflowRuns: list[dict[str, Any]] = []
-
-    # we are going to run the BulkParent workflow 20 which will trigger the Child workflows n times for each n in range(20)
-    for i in range(20):
-        workflowRuns.append(
-            {
-                "workflow_name": "BulkParent",
-                "input": {"n": i},
-                "options": {
-                    "additional_metadata": {
-                        "bulk-trigger": i,
-                        "hello-{i}": "earth-{i}",
-                    },
-                },
-            }
+    workflow_runs = [
+        WorkflowRunDict(
+            workflow_name="BulkParent",
+            input={"n": i},
+            options=TriggerWorkflowOptions(
+                additional_metadata={
+                    "bulk-trigger": i,
+                    "hello-{i}": "earth-{i}",
+                }
+            ),
         )
+        for i in range(20)
+    ]
 
     workflowRunRefs = hatchet.admin.run_workflows(
-        workflowRuns,
+        workflow_runs,
     )
 
     results = await asyncio.gather(

examples/_deprecated/concurrency_limit_rr/event.py

@@ -6,10 +6,9 @@
 
 from dotenv import load_dotenv
 
-from hatchet_sdk import new_client
+from hatchet_sdk import Hatchet, new_client
 from hatchet_sdk.clients.admin import TriggerWorkflowOptions
 from hatchet_sdk.clients.run_event_listener import StepRunEventType
-from hatchet_sdk.v2.hatchet import Hatchet
 
 
 async def main() -> None: