prefect 3.2.8.dev2__tar.gz → 3.2.8.dev4__tar.gz

{prefect-3.2.8.dev2 → prefect-3.2.8.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect
-Version: 3.2.8.dev2
+Version: 3.2.8.dev4
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io

{prefect-3.2.8.dev2 → prefect-3.2.8.dev4}/pyproject.toml

@@ -87,6 +87,7 @@ Changelog = "https://github.com/PrefectHQ/prefect/releases"
 Documentation = "https://docs.prefect.io"
 Source = "https://github.com/PrefectHQ/prefect"
 Tracker = "https://github.com/PrefectHQ/prefect/issues"
+
 [project.optional-dependencies]
 otel = [
     "opentelemetry-distro>=0.48b0,<1.0.0",

prefect-3.2.8.dev4/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.2.8.dev4"
+__build_date__ = "2025-02-27 08:08:45.254471+00:00"
+__git_commit__ = "48500836ca9bc8ec63d5b850b74d9d1570c6461a"
+__dirty__ = False

{prefect-3.2.8.dev2 → prefect-3.2.8.dev4}/src/prefect/_experimental/bundles.py

@@ -7,6 +7,7 @@ import multiprocessing
 import multiprocessing.context
 import os
 import subprocess
+import sys
 from typing import Any, TypedDict
 
 import cloudpickle
@@ -19,6 +20,7 @@ from prefect.flow_engine import run_flow
 from prefect.flows import Flow
 from prefect.settings.context import get_current_settings
 from prefect.settings.models.root import Settings
+from prefect.utilities.slugify import slugify
 
 
 class SerializedBundle(TypedDict):
@@ -147,6 +149,8 @@ def execute_bundle_in_subprocess(
     if dependencies := bundle.get("dependencies"):
         subprocess.check_call(
             [uv.find_uv_bin(), "pip", "install", *dependencies.split("\n")],
+            # Copy the current environment to ensure we install into the correct venv
+            env=os.environ,
         )
 
     process = ctx.Process(
@@ -161,3 +165,52 @@ def execute_bundle_in_subprocess(
     process.start()
 
     return process
+
+
+def convert_step_to_command(step: dict[str, Any], key: str) -> list[str]:
+    """
+    Converts a bundle upload or execution step to a command.
+
+    Args:
+        step: The step to convert.
+        key: The key to use for the remote file when downloading or uploading.
+
+    Returns:
+        A list of strings representing the command to run the step.
+    """
+    # Start with uv run
+    command = ["uv", "run"]
+
+    step_keys = list(step.keys())
+
+    if len(step_keys) != 1:
+        raise ValueError("Expected exactly one function in step")
+
+    function_fqn = step_keys[0]
+    function_args = step[function_fqn]
+
+    # Add the `--with` argument to handle dependencies for running the step
+    requires: list[str] | str = function_args.get("requires", [])
+    if isinstance(requires, str):
+        requires = [requires]
+    if requires:
+        command.extend(["--with", ",".join(requires)])
+
+    # Add the `--python` argument to handle the Python version for running the step
+    python_version = sys.version_info
+    command.extend(["--python", f"{python_version.major}.{python_version.minor}"])
+
+    # Add the `-m` argument to defined the function to run
+    command.extend(["-m", function_fqn])
+
+    # Add any arguments with values defined in the step
+    for arg_name, arg_value in function_args.items():
+        if arg_name == "requires":
+            continue
+
+        command.extend([f"--{slugify(arg_name)}", arg_value])
+
+    # Add the `--key` argument to specify the remote file name
+    command.extend(["--key", key])
+
+    return command

{prefect-3.2.8.dev2 → prefect-3.2.8.dev4}/src/prefect/cli/cloud/__init__.py

@@ -197,7 +197,6 @@ def prompt_select_from_list(
     Returns:
         str: the selected option
     """
-
     current_idx = 0
     selected_option = None
 
@@ -243,12 +242,16 @@ def prompt_select_from_list(
                 exit_with_error("")
             elif key == readchar.key.ENTER or key == readchar.key.CR:
                 selected_option = options[current_idx]
-                if isinstance(selected_option, tuple):
-                    selected_option = selected_option[0]
+                # Break out of the loop immediately after setting selected_option
+                break
 
             live.update(build_table(), refresh=True)
 
-        return selected_option
+    # Convert tuple to its first element if needed
+    if isinstance(selected_option, tuple):
+        selected_option = selected_option[0]
+
+    return selected_option
 
 
 async def login_with_browser() -> str:
@@ -364,11 +367,13 @@ async def _prompt_for_account_and_workspace(
         None,
         "[bold]Go back to account selection[/bold]",
     )
+
     result = prompt_select_from_list(
         app.console,
         "Which workspace would you like to use?",
         options=workspace_options + [go_back_option],
     )
+
     if not result:
         return None, True
     else:
@@ -688,10 +693,22 @@ async def set(
             if not workspaces:
                 exit_with_error("No workspaces found in the selected account.")
 
+            # Store the original list of workspaces
+            original_workspaces = workspaces.copy()
+
             go_back = True
             workspace = None
+            loop_count = 0
+
             while go_back:
+                loop_count += 1
+
+                # If we're going back, use the original list of workspaces
+                if loop_count > 1:
+                    workspaces = original_workspaces.copy()
+
                 workspace, go_back = await _prompt_for_account_and_workspace(workspaces)
+
             if workspace is None:
                 exit_with_error("No workspace selected.")
 

{prefect-3.2.8.dev2 → prefect-3.2.8.dev4}/src/prefect/workers/base.py

@@ -566,8 +566,6 @@ class BaseWorker(abc.ABC, Generic[C, V, R]):
         healthcheck_thread = None
         try:
             async with self as worker:
-                # wait for an initial heartbeat to configure the worker
-                await worker.sync_with_backend()
                 # schedule the scheduled flow run polling loop
                 async with anyio.create_task_group() as loops_task_group:
                     loops_task_group.start_soon(
@@ -655,6 +653,8 @@ class BaseWorker(abc.ABC, Generic[C, V, R]):
         await self._exit_stack.enter_async_context(self._client)
         await self._exit_stack.enter_async_context(self._runs_task_group)
 
+        await self.sync_with_backend()
+
         self.is_setup = True
 
     async def teardown(self, *exc_info: Any) -> None:

prefect-3.2.8.dev2/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.2.8.dev2"
-__build_date__ = "2025-02-25 08:09:51.795333+00:00"
-__git_commit__ = "9f97ceee40727350e347b83db7c2dbf5dfbc0189"
-__dirty__ = False