Feat: Add Ruff

conftest.py

@@ -1,35 +1,82 @@
 import logging
+import os
 import subprocess
 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
 import pytest_asyncio
 
-from hatchet_sdk import Hatchet
+from hatchet_sdk import ClientConfig, Hatchet
+from hatchet_sdk.loader import ClientTLSConfig
+
+
+@pytest.fixture(scope="session", autouse=True)
+def token() -> str:
+    result = subprocess.run(
+        [
+            "docker",
+            "compose",
+            "run",
+            "--no-deps",
+            "setup-config",
+            "/hatchet/hatchet-admin",
+            "token",
+            "create",
+            "--config",
+            "/hatchet/config",
+            "--tenant-id",
+            "707d0855-80ab-4e1f-a156-f1c4546cbf52",
+        ],
+        capture_output=True,
+        text=True,
+    )
+
+    token = result.stdout.strip()
+
+    os.environ["HATCHET_CLIENT_TOKEN"] = token
+
+    return token
 
 
 @pytest_asyncio.fixture(scope="session")
-async def aiohatchet() -> AsyncGenerator[Hatchet, None]:
-    yield Hatchet(debug=True)
+async def aiohatchet(token: str) -> AsyncGenerator[Hatchet, None]:
+    yield Hatchet(
+        debug=True,
+        config=ClientConfig(
+            token=token,
+            tls_config=ClientTLSConfig(strategy="none"),
+        ),
+    )
 
 
 @pytest.fixture(scope="session")
-def hatchet() -> Hatchet:
-    return Hatchet(debug=True)
+def hatchet(token: str) -> Hatchet:
+    return Hatchet(
+        debug=True,
+        config=ClientConfig(
+            token=token,
+            tls_config=ClientTLSConfig(strategy="none"),
+        ),
+    )
 
 
 @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]
 
     logging.info(f"Starting background worker: {' '.join(command)}")
-    proc = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+    proc = subprocess.Popen(
+        command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=os.environ.copy()
+    )
 
     # Check if the process is still running
     if proc.poll() is not None:

examples/affinity-workers/event.py

@@ -1,13 +1,10 @@
-from dotenv import load_dotenv
-
+from hatchet_sdk.clients.events import PushEventOptions
 from hatchet_sdk.hatchet import Hatchet
 
-load_dotenv()
-
 hatchet = Hatchet(debug=True)
 
 hatchet.event.push(
     "affinity:run",
     {"test": "test"},
-    options={"additional_metadata": {"hello": "moon"}},
+    options=PushEventOptions(additional_metadata={"hello": "moon"}),
 )

examples/affinity-workers/worker.py

@@ -1,22 +1,22 @@
-from dotenv import load_dotenv
+from hatchet_sdk import BaseWorkflow, Context, Hatchet, WorkerLabelComparator
+from hatchet_sdk.labels import DesiredWorkerLabel
 
-from hatchet_sdk import Context, Hatchet, WorkerLabelComparator
+hatchet = Hatchet(debug=True)
 
-load_dotenv()
+wf = hatchet.declare_workflow(on_events=["affinity:run"])
 
-hatchet = Hatchet(debug=True)
 
+class AffinityWorkflow(BaseWorkflow):
+    config = wf.config
 
-@hatchet.workflow(on_events=["affinity:run"])
-class AffinityWorkflow:
     @hatchet.step(
         desired_worker_labels={
-            "model": {"value": "fancy-ai-model-v2", "weight": 10},
-            "memory": {
-                "value": 256,
-                "required": True,
-                "comparator": WorkerLabelComparator.LESS_THAN,
-            },
+            "model": DesiredWorkerLabel(value="fancy-ai-model-v2", weight=10),
+            "memory": DesiredWorkerLabel(
+                value=256,
+                required=True,
+                comparator=WorkerLabelComparator.LESS_THAN,
+            ),
         },
     )
     async def step(self, context: Context) -> dict[str, str | None]:

examples/api/api.py

@@ -1,8 +1,4 @@
-from dotenv import load_dotenv
-
-from hatchet_sdk import Hatchet, WorkflowList
-
-load_dotenv()
+from hatchet_sdk import Hatchet
 
 hatchet = Hatchet(debug=True)
 

examples/api/async_api.py

@@ -1,11 +1,6 @@
 import asyncio
-from typing import cast
 
-from dotenv import load_dotenv
-
-from hatchet_sdk import Hatchet, WorkflowList
-
-load_dotenv()
+from hatchet_sdk import Hatchet
 
 hatchet = Hatchet(debug=True)
 

examples/blocked_async/event.py

@@ -1,12 +1,9 @@
-from dotenv import load_dotenv
-
 from hatchet_sdk import PushEventOptions, new_client
 
-load_dotenv()
-
 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"}),
 )