prefect 3.5.1.dev2__tar.gz → 3.5.1.dev4__tar.gz

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

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

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

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.5.1.dev4"
+__build_date__ = "2025-11-07 08:09:27.106698+00:00"
+__git_commit__ = "3cc4d69a6c0e0b4374a15fc512642411d6cf5323"
+__dirty__ = False

prefect-3.5.1.dev4/src/prefect/_internal/lazy.py

@@ -0,0 +1,47 @@
+"""Utilities for lazy-loading module-level objects."""
+
+from typing import Callable, Generic, TypeVar
+
+K = TypeVar("K")
+V = TypeVar("V")
+
+
+class LazyDict(Generic[K, V]):
+    """
+    A dictionary-like object that defers loading its contents until first access.
+
+    Useful for module-level registries that import heavy dependencies. The loader
+    function is called once on first access, and the result is cached.
+
+    Example:
+        >>> def load_plugins() -> dict[str, type]:
+        ...     from heavy_module import PluginA, PluginB
+        ...     return {"a": PluginA, "b": PluginB}
+        >>>
+        >>> plugins: LazyDict[str, type] = LazyDict(load_plugins)
+        >>> # Nothing loaded yet
+        >>> plugins.get("a")  # Triggers load, returns PluginA
+        >>> plugins.get("b")  # Uses cached result, returns PluginB
+    """
+
+    def __init__(self, loader_func: Callable[[], dict[K, V]]) -> None:
+        self._loader = loader_func
+        self._cache: dict[K, V] | None = None
+
+    def _ensure_loaded(self) -> dict[K, V]:
+        """Load the dictionary contents if not already loaded."""
+        if self._cache is None:
+            self._cache = self._loader()
+        return self._cache
+
+    def get(self, key: K, default: V | None = None) -> V | None:
+        """Get an item from the lazy dict, loading if necessary."""
+        return self._ensure_loaded().get(key, default)
+
+    def __contains__(self, key: K) -> bool:
+        """Check if key is in the lazy dict, loading if necessary."""
+        return key in self._ensure_loaded()
+
+    def __getitem__(self, key: K) -> V:
+        """Get an item from the lazy dict, loading if necessary."""
+        return self._ensure_loaded()[key]

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/blocks/redis.py

@@ -5,19 +5,20 @@ from pathlib import Path
 from typing import AsyncGenerator, Optional
 
 try:
-    import redis.asyncio as redis
+    import redis
+    import redis.asyncio as aioredis
 except ImportError:
     raise ImportError(
         "`redis-py` must be installed to use the `RedisStorageContainer` block. "
-        "You can install it with `pip install redis>=5.0.1"
+        "You can install it with `pip install redis>=5.0.1`"
     )
 
 from pydantic import Field, HttpUrl
 from pydantic.types import SecretStr
 from typing_extensions import Self
 
+from prefect._internal.compatibility.async_dispatch import async_dispatch
 from prefect.filesystems import WritableFileSystem
-from prefect.utilities.asyncutils import sync_compatible
 
 
 class RedisStorageContainer(WritableFileSystem):
@@ -73,35 +74,72 @@ class RedisStorageContainer(WritableFileSystem):
                 "Initialization error: 'username' is provided, but 'password' is missing. Both are required."
             )
 
-    @sync_compatible
-    async def read_path(self, path: Path | str):
+    async def aread_path(self, path: Path | str) -> Optional[bytes]:
         """Read the redis content at `path`
 
         Args:
             path: Redis key to read from
 
         Returns:
-            Contents at key as bytes
+            Contents at key as bytes, or None if key does not exist
         """
         async with self._client() as client:
             return await client.get(str(path))
 
-    @sync_compatible
-    async def write_path(self, path: Path | str, content: bytes):
+    @async_dispatch(aread_path)
+    def read_path(self, path: Path | str) -> Optional[bytes]:
+        """Read the redis content at `path`
+
+        Args:
+            path: Redis key to read from
+
+        Returns:
+            Contents at key as bytes, or None if key does not exist
+        """
+        if self.connection_string:
+            client = redis.Redis.from_url(self.connection_string.get_secret_value())
+        else:
+            assert self.host
+            client = redis.Redis(
+                host=self.host,
+                port=self.port,
+                username=self.username.get_secret_value() if self.username else None,
+                password=self.password.get_secret_value() if self.password else None,
+                db=self.db,
+            )
+
+        try:
+            return client.get(str(path))
+        finally:
+            client.close()
+
+    async def awrite_path(self, path: Path | str, content: bytes) -> bool:
         """Write `content` to the redis at `path`
 
         Args:
             path: Redis key to write to
             content: Binary object to write
+
+        Returns:
+            True if the key was set successfully
         """
 
         async with self._client() as client:
             return await client.set(str(path), content)
 
-    @asynccontextmanager
-    async def _client(self) -> AsyncGenerator[redis.Redis, None]:
+    @async_dispatch(awrite_path)
+    def write_path(self, path: Path | str, content: bytes) -> bool:
+        """Write `content` to the redis at `path`
+
+        Args:
+            path: Redis key to write to
+            content: Binary object to write
+
+        Returns:
+            True if the key was set successfully
+        """
         if self.connection_string:
-            client = redis.Redis.from_url(self.connection_string.get_secret_value())  # pyright: ignore[reportUnknownMemberType] incomplete typing for redis-py
+            client = redis.Redis.from_url(self.connection_string.get_secret_value())
         else:
             assert self.host
             client = redis.Redis(
@@ -112,6 +150,25 @@ class RedisStorageContainer(WritableFileSystem):
                 db=self.db,
             )
 
+        try:
+            return client.set(str(path), content)
+        finally:
+            client.close()
+
+    @asynccontextmanager
+    async def _client(self) -> AsyncGenerator[aioredis.Redis, None]:
+        if self.connection_string:
+            client = aioredis.Redis.from_url(self.connection_string.get_secret_value())  # pyright: ignore[reportUnknownMemberType] incomplete typing for redis-py
+        else:
+            assert self.host
+            client = aioredis.Redis(
+                host=self.host,
+                port=self.port,
+                username=self.username.get_secret_value() if self.username else None,
+                password=self.password.get_secret_value() if self.password else None,
+                db=self.db,
+            )
+
         try:
             yield client
         finally:

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/_types.py

@@ -6,7 +6,7 @@ import asyncio
 import functools
 import sys
 from datetime import datetime
-from typing import Any, Callable, List, Optional
+from typing import TYPE_CHECKING, Any, Callable, List, Optional
 
 import typer
 from rich.console import Console
@@ -14,11 +14,14 @@ from rich.theme import Theme
 
 from prefect._internal.compatibility.deprecated import generate_deprecation_message
 from prefect.cli._utilities import with_cli_exception_handling
-from prefect.settings import PREFECT_CLI_COLORS, Setting
+from prefect.settings import get_current_settings
 from prefect.utilities.asyncutils import is_async_fn
 
+if TYPE_CHECKING:
+    from prefect.settings.legacy import Setting
 
-def SettingsOption(setting: Setting, *args: Any, **kwargs: Any) -> Any:
+
+def SettingsOption(setting: "Setting", *args: Any, **kwargs: Any) -> Any:
     """Custom `typer.Option` factory to load the default value from settings"""
 
     return typer.Option(
@@ -34,7 +37,7 @@ def SettingsOption(setting: Setting, *args: Any, **kwargs: Any) -> Any:
     )
 
 
-def SettingsArgument(setting: Setting, *args: Any, **kwargs: Any) -> Any:
+def SettingsArgument(setting: "Setting", *args: Any, **kwargs: Any) -> Any:
     """Custom `typer.Argument` factory to load the default value from settings"""
 
     # See comments in `SettingsOption`
@@ -91,7 +94,7 @@ class PrefectTyper(typer.Typer):
         self.console = Console(
             highlight=False,
             theme=Theme({"prompt.choices": "bold blue"}),
-            color_system="auto" if PREFECT_CLI_COLORS else None,
+            color_system="auto" if get_current_settings().cli.colors else None,
         )
 
     def add_typer(
@@ -201,7 +204,7 @@ class PrefectTyper(typer.Typer):
     def setup_console(self, soft_wrap: bool, prompt: bool) -> None:
         self.console = Console(
             highlight=False,
-            color_system="auto" if PREFECT_CLI_COLORS else None,
+            color_system="auto" if get_current_settings().cli.colors else None,
             theme=Theme({"prompt.choices": "bold blue"}),
             soft_wrap=not soft_wrap,
             force_interactive=prompt,

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/_utilities.py

@@ -12,7 +12,7 @@ import typer
 from click.exceptions import ClickException
 
 from prefect.exceptions import MissingProfileError
-from prefect.settings import PREFECT_TEST_MODE
+from prefect.settings import get_current_settings
 
 
 def exit_with_error(message: str | Exception, code: int = 1, **kwargs: Any) -> NoReturn:
@@ -47,7 +47,7 @@ def with_cli_exception_handling(fn: Callable[..., Any]) -> Callable[..., Any]:
         except MissingProfileError as exc:
             exit_with_error(exc)
         except Exception:
-            if PREFECT_TEST_MODE.value():
+            if get_current_settings().testing.test_mode:
                 raise  # Reraise exceptions during test mode
             traceback.print_exc()
             exit_with_error("An exception occurred.")

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/block.py

@@ -28,7 +28,7 @@ from prefect.exceptions import (
     ScriptError,
     exception_traceback,
 )
-from prefect.settings import PREFECT_UI_URL
+from prefect.settings import get_current_settings
 from prefect.utilities.asyncutils import run_sync_in_worker_thread
 from prefect.utilities.importtools import load_script_as_module
 
@@ -244,7 +244,7 @@ async def register(
         "go to the Blocks page in the Prefect UI.\n"
     )
 
-    if ui_url := PREFECT_UI_URL:
+    if ui_url := get_current_settings().ui_url:
         if determine_server_type() == ServerType.CLOUD:
             block_catalog_url = f"{ui_url}/settings/blocks/catalog"
         else:
@@ -349,17 +349,16 @@ async def block_create(
             app.console.print(f"Available block types: {', '.join(slugs)}")
             raise typer.Exit(1)
 
-        if not PREFECT_UI_URL:
+        ui_url = get_current_settings().ui_url
+        if not ui_url:
             exit_with_error(
                 "Prefect must be configured to use a hosted Prefect server or "
                 "Prefect Cloud to display the Prefect UI"
             )
         if determine_server_type() == ServerType.CLOUD:
-            block_link = f"{PREFECT_UI_URL.value()}/settings/blocks/catalog/{block_type.slug}/create"
+            block_link = f"{ui_url}/settings/blocks/catalog/{block_type.slug}/create"
         else:
-            block_link = (
-                f"{PREFECT_UI_URL.value()}/blocks/catalog/{block_type.slug}/create"
-            )
+            block_link = f"{ui_url}/blocks/catalog/{block_type.slug}/create"
         app.console.print(
             f"Create a {block_type_slug} block: {block_link}",
         )

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/cloud/webhook.py

@@ -14,7 +14,7 @@ from prefect.cli.cloud import cloud_app, confirm_logged_in
 from prefect.cli.root import app, is_interactive
 from prefect.client.cloud import get_cloud_client
 from prefect.exceptions import ObjectNotFound
-from prefect.settings import PREFECT_API_URL
+from prefect.settings import get_current_settings
 
 webhook_app: PrefectTyper = PrefectTyper(
     name="webhook", help="Manage Prefect Cloud Webhooks"
@@ -49,7 +49,7 @@ async def ls():
     confirm_logged_in()
 
     # The /webhooks API lives inside the /accounts/{id}/workspaces/{id} routing tree
-    async with get_cloud_client(host=PREFECT_API_URL.value()) as client:
+    async with get_cloud_client(host=get_current_settings().api.url) as client:
         retrieved_webhooks = await client.request("POST", "/webhooks/filter")
         display_table = _render_webhooks_into_table(retrieved_webhooks)
         app.console.print(display_table)
@@ -63,7 +63,7 @@ async def get(webhook_id: UUID):
     confirm_logged_in()
 
     # The /webhooks API lives inside the /accounts/{id}/workspaces/{id} routing tree
-    async with get_cloud_client(host=PREFECT_API_URL.value()) as client:
+    async with get_cloud_client(host=get_current_settings().api.url) as client:
         webhook = await client.request("GET", f"/webhooks/{webhook_id}")
         display_table = _render_webhooks_into_table([webhook])
         app.console.print(display_table)
@@ -94,7 +94,7 @@ async def create(
     confirm_logged_in()
 
     # The /webhooks API lives inside the /accounts/{id}/workspaces/{id} routing tree
-    async with get_cloud_client(host=PREFECT_API_URL.value()) as client:
+    async with get_cloud_client(host=get_current_settings().api.url) as client:
         response = await client.request(
             "POST",
             "/webhooks/",
@@ -122,7 +122,7 @@ async def rotate(webhook_id: UUID):
         return
 
     # The /webhooks API lives inside the /accounts/{id}/workspaces/{id} routing tree
-    async with get_cloud_client(host=PREFECT_API_URL.value()) as client:
+    async with get_cloud_client(host=get_current_settings().api.url) as client:
         response = await client.request("POST", f"/webhooks/{webhook_id}/rotate")
         app.console.print(f"Successfully rotated webhook URL to {response['slug']}")
 
@@ -138,7 +138,7 @@ async def toggle(
 
     status_lookup = {True: "enabled", False: "disabled"}
 
-    async with get_cloud_client(host=PREFECT_API_URL.value()) as client:
+    async with get_cloud_client(host=get_current_settings().api.url) as client:
         response = await client.request("GET", f"/webhooks/{webhook_id}")
         current_status = response["enabled"]
         new_status = not current_status
@@ -166,7 +166,7 @@ async def update(
     confirm_logged_in()
 
     # The /webhooks API lives inside the /accounts/{id}/workspaces/{id} routing tree
-    async with get_cloud_client(host=PREFECT_API_URL.value()) as client:
+    async with get_cloud_client(host=get_current_settings().api.url) as client:
         response = await client.request("GET", f"/webhooks/{webhook_id}")
         update_payload = {
             "name": webhook_name or response["name"],
@@ -192,7 +192,7 @@ async def delete(webhook_id: UUID):
         exit_with_error("Deletion aborted.")
 
     # The /webhooks API lives inside the /accounts/{id}/workspaces/{id} routing tree
-    async with get_cloud_client(host=PREFECT_API_URL.value()) as client:
+    async with get_cloud_client(host=get_current_settings().api.url) as client:
         try:
             await client.request("DELETE", f"/webhooks/{webhook_id}")
             app.console.print(f"Successfully deleted webhook {webhook_id}")

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/dashboard.py

@@ -3,7 +3,7 @@ import webbrowser
 from prefect.cli._types import PrefectTyper
 from prefect.cli._utilities import exit_with_error, exit_with_success
 from prefect.cli.root import app
-from prefect.settings import PREFECT_UI_URL
+from prefect.settings import get_current_settings
 from prefect.utilities.asyncutils import run_sync_in_worker_thread
 
 dashboard_app: PrefectTyper = PrefectTyper(
@@ -19,7 +19,7 @@ async def open() -> None:
     Open the Prefect UI in the browser.
     """
 
-    if not (ui_url := PREFECT_UI_URL.value()):
+    if not (ui_url := get_current_settings().ui_url):
         exit_with_error(
             "`PREFECT_UI_URL` must be set to the URL of a running Prefect server or Prefect Cloud workspace."
         )

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/deploy/_commands.py

@@ -8,12 +8,11 @@ import yaml
 from rich.table import Table
 
 import prefect
-from prefect.cli._types import SettingsOption
 from prefect.cli._utilities import exit_with_error
 from prefect.cli.root import app, is_interactive
 from prefect.client.schemas.objects import ConcurrencyLimitConfig
 from prefect.deployments import initialize_project
-from prefect.settings import PREFECT_DEFAULT_WORK_POOL_NAME
+from prefect.settings import get_current_settings
 
 from ._config import (
     _load_deploy_configs_and_actions,
@@ -178,11 +177,12 @@ async def deploy(
         "--collision-strategy",
         help="Configure the behavior for runs once the concurrency limit is reached. Falls back to `ENQUEUE` if unset.",
     ),
-    work_pool_name: str = SettingsOption(
-        PREFECT_DEFAULT_WORK_POOL_NAME,
+    work_pool_name: str = typer.Option(
+        lambda: get_current_settings().deployments.default_work_pool_name,
         "-p",
         "--pool",
         help="The work pool that will handle this deployment's runs.",
+        show_default="from PREFECT_DEFAULT_WORK_POOL_NAME",
     ),
     work_queue_name: str = typer.Option(
         None,

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/deploy/_core.py

@@ -25,7 +25,7 @@ from prefect.deployments.runner import RunnerDeployment
 from prefect.deployments.steps.core import run_steps
 from prefect.exceptions import ObjectNotFound
 from prefect.flows import load_flow_from_entrypoint
-from prefect.settings import PREFECT_UI_URL
+from prefect.settings import get_current_settings
 from prefect.utilities.callables import parameter_schema
 from prefect.utilities.collections import get_from_dict
 from prefect.utilities.templating import (
@@ -373,10 +373,10 @@ async def _run_single_deploy(
         style="green",
     )
 
-    if PREFECT_UI_URL:
+    if ui_url := get_current_settings().ui_url:
         message = (
             "\nView Deployment in UI:"
-            f" {PREFECT_UI_URL.value()}/deployments/deployment/{deployment_id}\n"
+            f" {ui_url}/deployments/deployment/{deployment_id}\n"
         )
         app.console.print(message, soft_wrap=True)
 

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/root.py

@@ -19,10 +19,7 @@ from prefect.cli._utilities import with_cli_exception_handling
 from prefect.client.base import determine_server_type
 from prefect.client.constants import SERVER_API_VERSION
 from prefect.logging.configuration import setup_logging
-from prefect.settings import (
-    PREFECT_CLI_WRAP_LINES,
-    PREFECT_TEST_MODE,
-)
+from prefect.settings import get_current_settings
 from prefect.types._datetime import parse_datetime
 
 app: PrefectTyper = PrefectTyper(add_completion=True, no_args_is_help=True)
@@ -77,9 +74,9 @@ def main(
             exit(1)
 
     # Configure the output console after loading the profile
-    app.setup_console(soft_wrap=PREFECT_CLI_WRAP_LINES.value(), prompt=prompt)
+    app.setup_console(soft_wrap=get_current_settings().cli.wrap_lines, prompt=prompt)
 
-    if not PREFECT_TEST_MODE:
+    if not get_current_settings().testing.test_mode:
         # When testing, this entrypoint can be called multiple times per process which
         # can cause logging configuration conflicts. Logging is set up in conftest
         # during tests.
@@ -107,7 +104,6 @@ async def version(
 
     from prefect.server.database.dependencies import provide_database_interface
     from prefect.server.utilities.database import get_dialect
-    from prefect.settings import PREFECT_API_DATABASE_CONNECTION_URL
 
     if build_date_str := prefect.__version_info__.get("date", None):
         build_date = parse_datetime(build_date_str).strftime("%a, %b %d, %Y %I:%M %p")
@@ -134,7 +130,8 @@ async def version(
 
     version_info["Pydantic version"] = pydantic_version
 
-    if connection_url := PREFECT_API_DATABASE_CONNECTION_URL.value():
+    if connection_url_setting := get_current_settings().server.database.connection_url:
+        connection_url = connection_url_setting.get_secret_value()
         database = get_dialect(connection_url).name
         version_info["Server"] = {"Database": database}
         if database == "sqlite":

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/shell.py

@@ -22,7 +22,7 @@ from prefect.context import tags
 from prefect.exceptions import FailedRun
 from prefect.logging.loggers import get_run_logger
 from prefect.runner import Runner
-from prefect.settings import PREFECT_UI_URL
+from prefect.settings import get_current_settings
 from prefect.types.entrypoint import EntrypointType
 
 shell_app: PrefectTyper = PrefectTyper(
@@ -236,10 +236,10 @@ async def serve(
         " command:\n[blue]\n\t$ prefect deployment run"
         f" '{runner_deployment.flow_name}/{deployment_name}'\n[/]"
     )
-    if PREFECT_UI_URL:
+    if ui_url := get_current_settings().ui_url:
         help_message += (
             "\nYou can also run your flow via the Prefect UI:"
-            f" [blue]{PREFECT_UI_URL.value()}/deployments/deployment/{deployment_id}[/]\n"
+            f" [blue]{ui_url}/deployments/deployment/{deployment_id}[/]\n"
         )
 
     app.console.print(help_message, soft_wrap=True)

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/cli/work_pool.py

@@ -36,10 +36,7 @@ from prefect.client.schemas.objects import (
     WorkPoolStorageConfiguration,
 )
 from prefect.exceptions import ObjectAlreadyExists, ObjectNotFound
-from prefect.infrastructure.provisioners import (
-    _provisioners,
-    get_infrastructure_provisioner_for_work_pool_type,
-)
+from prefect.infrastructure import provisioners
 from prefect.settings import update_current_profile
 from prefect.types._datetime import now as now_fn
 from prefect.utilities import urls
@@ -76,7 +73,7 @@ def has_provisioner_for_type(work_pool_type: str) -> bool:
     Returns:
         bool: True if a provisioner exists for the given type, False otherwise.
     """
-    return work_pool_type in _provisioners
+    return work_pool_type in provisioners._provisioners
 
 
 @work_pool_app.command()
@@ -202,7 +199,9 @@ async def create(
 
         if provision_infrastructure:
             try:
-                provisioner = get_infrastructure_provisioner_for_work_pool_type(type)
+                provisioner = (
+                    provisioners.get_infrastructure_provisioner_for_work_pool_type(type)
+                )
                 provisioner.console = app.console
                 template_contents = await provisioner.provision(
                     work_pool_name=name, base_job_template=template_contents
@@ -492,8 +491,10 @@ async def provision_infrastructure(
             exit_with_error(f"Failed to read work pool {name!r}: {exc}")
 
         try:
-            provisioner = get_infrastructure_provisioner_for_work_pool_type(
-                work_pool.type
+            provisioner = (
+                provisioners.get_infrastructure_provisioner_for_work_pool_type(
+                    work_pool.type
+                )
             )
             provisioner.console = app.console
             new_base_job_template = await provisioner.provision(

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/client/schemas/actions.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import datetime
 from copy import deepcopy
-from typing import TYPE_CHECKING, Any, Callable, Optional, TypeVar, Union
+from typing import Any, Callable, Optional, TypeVar, Union
 from uuid import UUID, uuid4
 
 import jsonschema
@@ -22,6 +22,7 @@ from prefect._internal.schemas.validators import (
     validate_name_present_on_nonanonymous_blocks,
     validate_schedule_max_scheduled_runs,
 )
+from prefect._result_records import ResultRecordMetadata
 from prefect.client.schemas.objects import (
     StateDetails,
     StateType,
@@ -57,9 +58,6 @@ from prefect.types.names import (
 from prefect.utilities.collections import visit_collection
 from prefect.utilities.pydantic import get_class_fields_only
 
-if TYPE_CHECKING:
-    from prefect._result_records import ResultRecordMetadata
-
 R = TypeVar("R")
 
 
@@ -70,7 +68,7 @@ class StateCreate(ActionBaseModel):
     name: Optional[str] = Field(default=None)
     message: Optional[str] = Field(default=None, examples=["Run started"])
     state_details: StateDetails = Field(default_factory=StateDetails)
-    data: Union["ResultRecordMetadata", Any] = Field(
+    data: Union[ResultRecordMetadata, Any] = Field(
         default=None,
     )
 

{prefect-3.5.1.dev2 → prefect-3.5.1.dev4}/src/prefect/deployments/flow_runs.py

@@ -7,7 +7,7 @@ from opentelemetry import trace
 
 import prefect
 from prefect._result_records import ResultRecordMetadata
-from prefect.client.schemas import FlowRun
+from prefect.client.schemas import FlowRun, TaskRunResult
 from prefect.client.utilities import inject_client
 from prefect.context import FlowRunContext, TaskRunContext
 from prefect.logging import get_logger
@@ -136,6 +136,21 @@ async def run_deployment(
             k: await collect_task_run_inputs(v) for k, v in parameters.items()
         }
 
+        # Track parent task if this is being called from within a task
+        # This enables the execution graph to properly display the deployment
+        # flow run as nested under the calling task
+        if task_run_ctx:
+            # The task run is only considered a parent if it is in the same
+            # flow run (otherwise the child is in a subflow, so the subflow
+            # serves as the parent) or if there is no flow run
+            if not flow_run_ctx or (
+                task_run_ctx.task_run.flow_run_id
+                == getattr(flow_run_ctx.flow_run, "id", None)
+            ):
+                task_inputs["__parents__"] = [
+                    TaskRunResult(id=task_run_ctx.task_run.id)
+                ]
+
         if deployment_id:
             flow = await client.read_flow(deployment.flow_id)
             deployment_name = f"{flow.name}/{deployment.name}"