the key for worker metadata is just "metadata"

PrefectHQ · Nov 9, 2024 · ae3e2a9 · ae3e2a9
1 parent 6c45d95
commit ae3e2a9
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/prefect/client/orchestration.py b/src/prefect/client/orchestration.py
@@ -2613,10 +2613,12 @@ async def send_worker_heartbeat(
             "heartbeat_interval_seconds": heartbeat_interval_seconds,
         }
         if worker_metadata:
-            params["worker_metadata"] = worker_metadata.model_dump(mode="json")
+            params["metadata"] = worker_metadata.model_dump(mode="json")
         if get_worker_id:
             params["return_id"] = get_worker_id
 
+        print(f"params: {params}")
+
         resp = await self._client.post(
             f"/work_pools/{work_pool_name}/workers/heartbeat",
             json=params,

diff --git a/tests/client/test_prefect_client.py b/tests/client/test_prefect_client.py
@@ -2737,7 +2737,7 @@ async def test_worker_heartbeat_sends_metadata_if_passed(
             assert mock_post.call_args[1]["json"] == {
                 "name": "test-worker",
                 "heartbeat_interval_seconds": 10,
-                "worker_metadata": {
+                "metadata": {
                     "integrations": [{"name": "prefect-aws", "version": "1.0.0"}]
                 },
             }

diff --git a/tests/workers/test_base_worker.py b/tests/workers/test_base_worker.py
@@ -1914,7 +1914,7 @@ async def test_worker_heartbeat_sends_integrations(
                     json={
                         "name": worker.name,
                         "heartbeat_interval_seconds": worker.heartbeat_interval_seconds,
-                        "worker_metadata": {
+                        "metadata": {
                             "integrations": [
                                 {"name": "prefect-aws", "version": "1.0.0"}
                             ]
@@ -1972,7 +1972,7 @@ async def _worker_metadata(self) -> WorkerMetadata:
                     json={
                         "name": worker.name,
                         "heartbeat_interval_seconds": worker.heartbeat_interval_seconds,
-                        "worker_metadata": {
+                        "metadata": {
                             "integrations": [
                                 {"name": "prefect-aws", "version": "1.0.0"}
                             ],