Asyncify the flow run engine (#16095)

PrefectHQ · Nov 22, 2024 · d64c0d7 · d64c0d7
1 parent ede23b1
commit d64c0d7
Show file tree

Hide file tree

Showing 4 changed files with 173 additions and 118 deletions.
diff --git a/src/prefect/client/orchestration.py b/src/prefect/client/orchestration.py
@@ -2209,6 +2209,13 @@ async def read_flow_run_states(
             response.json()
         )
 
+    async def set_flow_run_name(self, flow_run_id: UUID, name: str):
+        flow_run_data = FlowRunUpdate(name=name)
+        return await self._client.patch(
+            f"/flow_runs/{flow_run_id}",
+            json=flow_run_data.model_dump(mode="json", exclude_unset=True),
+        )
+
     async def set_task_run_name(self, task_run_id: UUID, name: str):
         task_run_data = TaskRunUpdate(name=name)
         return await self._client.patch(
@@ -4019,7 +4026,7 @@ def set_flow_run_state(
         return OrchestrationResult.model_validate(response.json())
 
     def set_flow_run_name(self, flow_run_id: UUID, name: str):
-        flow_run_data = TaskRunUpdate(name=name)
+        flow_run_data = FlowRunUpdate(name=name)
         return self._client.patch(
             f"/flow_runs/{flow_run_id}",
             json=flow_run_data.model_dump(mode="json", exclude_unset=True),