From 4a52a71956a8d46fcb7294ac71734504bb09bcc2 Mon Sep 17 00:00:00 2001
From: S. Solomon Darnell
Date: Fri, 28 Mar 2025 21:52:21 -0500
Subject: two version of R2R are here

---
 .../hatchet_sdk/context/worker_context.py          | 28 ++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 .venv/lib/python3.12/site-packages/hatchet_sdk/context/worker_context.py

(limited to '.venv/lib/python3.12/site-packages/hatchet_sdk/context/worker_context.py')

diff --git a/.venv/lib/python3.12/site-packages/hatchet_sdk/context/worker_context.py b/.venv/lib/python3.12/site-packages/hatchet_sdk/context/worker_context.py
new file mode 100644
index 00000000..770ae097
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/hatchet_sdk/context/worker_context.py
@@ -0,0 +1,28 @@
+from hatchet_sdk.clients.dispatcher.dispatcher import DispatcherClient
+
+
+class WorkerContext:
+    _worker_id: str | None = None
+    _registered_workflow_names: list[str] = []
+    _labels: dict[str, str | int] = {}
+
+    def __init__(self, labels: dict[str, str | int], client: DispatcherClient):
+        self._labels = labels
+        self.client = client
+
+    def labels(self) -> dict[str, str | int]:
+        return self._labels
+
+    def upsert_labels(self, labels: dict[str, str | int]) -> None:
+        self.client.upsert_worker_labels(self._worker_id, labels)
+        self._labels.update(labels)
+
+    async def async_upsert_labels(self, labels: dict[str, str | int]) -> None:
+        await self.client.async_upsert_worker_labels(self._worker_id, labels)
+        self._labels.update(labels)
+
+    def id(self) -> str | None:
+        return self._worker_id
+
+    # def has_workflow(self, workflow_name: str):
+    #     return workflow_name in self._registered_workflow_names
-- 
cgit v1.2.3