prefect-client 3.4.7.dev5__tar.gz → 3.4.7.dev6__tar.gz

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/PKG-INFO

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

prefect_client-3.4.7.dev6/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.4.7.dev6"
+__build_date__ = "2025-06-21 08:08:06.814878+00:00"
+__git_commit__ = "db5b5e24387f38ce4aa55b62ce5e3993beacee9c"
+__dirty__ = False

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/client/orchestration/__init__.py

@@ -101,10 +101,11 @@ from prefect.client.schemas.filters import (
     WorkQueueFilterName,
 )
 from prefect.client.schemas.objects import (
-    Constant,
+    TaskRunResult,
+    FlowRunResult,
     Parameter,
+    Constant,
     TaskRunPolicy,
-    TaskRunResult,
     WorkQueue,
     WorkQueueStatusDetail,
 )
@@ -778,13 +779,7 @@ class PrefectClient(
         task_inputs: Optional[
             dict[
                 str,
-                list[
-                    Union[
-                        TaskRunResult,
-                        Parameter,
-                        Constant,
-                    ]
-                ],
+                list[Union[TaskRunResult, FlowRunResult, Parameter, Constant]],
             ]
         ] = None,
     ) -> TaskRun:
@@ -1440,6 +1435,7 @@ class SyncPrefectClient(
                 list[
                     Union[
                         TaskRunResult,
+                        FlowRunResult,
                         Parameter,
                         Constant,
                     ]

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/client/schemas/__init__.py

@@ -11,11 +11,12 @@ if TYPE_CHECKING:
         BlockType,
         FlowRun,
         FlowRunPolicy,
+        FlowRunResult,
         State,
         StateDetails,
         StateType,
         TaskRun,
-        TaskRunInput,
+        RunInput,
         TaskRunPolicy,
         TaskRunResult,
         Workspace,
@@ -36,6 +37,7 @@ _public_api = {
     "DEFAULT_BLOCK_SCHEMA_VERSION": (__package__, ".objects"),
     "FlowRun": (__package__, ".objects"),
     "FlowRunPolicy": (__package__, ".objects"),
+    "FlowRunResult": (__package__, ".objects"),
     "OrchestrationResult": (__package__, ".responses"),
     "SetStateStatus": (__package__, ".responses"),
     "State": (__package__, ".objects"),
@@ -60,6 +62,7 @@ __all__ = [
     "DEFAULT_BLOCK_SCHEMA_VERSION",
     "FlowRun",
     "FlowRunPolicy",
+    "FlowRunResult",
     "OrchestrationResult",
     "SetStateStatus",
     "State",
@@ -70,7 +73,7 @@ __all__ = [
     "StateRejectDetails",
     "StateType",
     "TaskRun",
-    "TaskRunInput",
+    "RunInput",
     "TaskRunPolicy",
     "TaskRunResult",
     "Workspace",

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/client/schemas/actions.py

@@ -443,6 +443,7 @@ class TaskRunCreate(ActionBaseModel):
         list[
             Union[
                 objects.TaskRunResult,
+                objects.FlowRunResult,
                 objects.Parameter,
                 objects.Constant,
             ]

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/client/schemas/objects.py

@@ -81,6 +81,11 @@ DEFAULT_BLOCK_SCHEMA_VERSION: Literal["non-versioned"] = "non-versioned"
 DEFAULT_AGENT_WORK_POOL_NAME: Literal["default-agent-pool"] = "default-agent-pool"
 
 
+class RunType(AutoEnum):
+    FLOW_RUN = "flow_run"
+    TASK_RUN = "task_run"
+
+
 class StateType(AutoEnum):
     """Enumeration of state types."""
 
@@ -164,7 +169,6 @@ class ConcurrencyLimitConfig(PrefectBaseModel):
 class StateDetails(PrefectBaseModel):
     flow_run_id: Optional[UUID] = None
     task_run_id: Optional[UUID] = None
-    # for task runs that represent subflows, the subflow's run ID
     child_flow_run_id: Optional[UUID] = None
     scheduled_time: Optional[DateTime] = None
     cache_key: Optional[str] = None
@@ -182,6 +186,16 @@ class StateDetails(PrefectBaseModel):
     # Captures the trace_id and span_id of the span where this state was created
     traceparent: Optional[str] = None
 
+    def to_run_result(
+        self, run_type: RunType
+    ) -> Optional[Union[FlowRunResult, TaskRunResult]]:
+        if run_type == run_type.FLOW_RUN and self.flow_run_id:
+            return FlowRunResult(id=self.flow_run_id)
+        elif run_type == run_type.TASK_RUN and self.task_run_id:
+            return TaskRunResult(id=self.task_run_id)
+        else:
+            return None
+
 
 def data_discriminator(x: Any) -> str:
     if isinstance(x, dict) and "storage_key" in x:
@@ -734,7 +748,7 @@ class TaskRunPolicy(PrefectBaseModel):
         return validate_not_negative(v)
 
 
-class TaskRunInput(PrefectBaseModel):
+class RunInput(PrefectBaseModel):
     """
     Base class for classes that represent inputs to task runs, which
     could include, constants, parameters, or other task runs.
@@ -747,21 +761,26 @@ class TaskRunInput(PrefectBaseModel):
         input_type: str
 
 
-class TaskRunResult(TaskRunInput):
+class TaskRunResult(RunInput):
     """Represents a task run result input to another task run."""
 
     input_type: Literal["task_run"] = "task_run"
     id: UUID
 
 
-class Parameter(TaskRunInput):
+class FlowRunResult(RunInput):
+    input_type: Literal["flow_run"] = "flow_run"
+    id: UUID
+
+
+class Parameter(RunInput):
     """Represents a parameter input to a task run."""
 
     input_type: Literal["parameter"] = "parameter"
     name: str
 
 
-class Constant(TaskRunInput):
+class Constant(RunInput):
     """Represents constant input value to a task run."""
 
     input_type: Literal["constant"] = "constant"
@@ -811,7 +830,9 @@ class TaskRun(TimeSeriesBaseModel, ObjectBaseModel):
     state_id: Optional[UUID] = Field(
         default=None, description="The id of the current task run state."
     )
-    task_inputs: dict[str, list[Union[TaskRunResult, Parameter, Constant]]] = Field(
+    task_inputs: dict[
+        str, list[Union[TaskRunResult, FlowRunResult, Parameter, Constant]]
+    ] = Field(
         default_factory=dict,
         description=(
             "Tracks the source of inputs to a task run. Used for internal bookkeeping. "

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/context.py

@@ -34,6 +34,7 @@ from prefect._internal.compatibility.migration import getattr_migration
 from prefect.assets import Asset
 from prefect.client.orchestration import PrefectClient, SyncPrefectClient, get_client
 from prefect.client.schemas import FlowRun, TaskRun
+from prefect.client.schemas.objects import RunType
 from prefect.events.worker import EventsWorker
 from prefect.exceptions import MissingContextError
 from prefect.results import (
@@ -363,7 +364,7 @@ class EngineContext(RunContext):
         flow: The flow instance associated with the run
         flow_run: The API metadata for the flow run
         task_runner: The task runner instance being used for the flow run
-        task_run_results: A mapping of result ids to task run states for this flow run
+        run_results: A mapping of result ids to run states for this flow run
         log_prints: Whether to log print statements from the flow run
         parameters: The parameters passed to the flow run
         detached: Flag indicating if context has been serialized and sent to remote infrastructure
@@ -394,9 +395,10 @@ class EngineContext(RunContext):
     # Counter for flow pauses
     observed_flow_pauses: dict[str, int] = Field(default_factory=dict)
 
-    # Tracking for result from task runs in this flow run for dependency tracking
-    # Holds the ID of the object returned by the task run and task run state
-    task_run_results: dict[int, State] = Field(default_factory=dict)
+    # Tracking for result from task runs and sub flows in this flow run for
+    # dependency tracking. Holds the ID of the object returned by
+    # the run and state
+    run_results: dict[int, tuple[State, RunType]] = Field(default_factory=dict)
 
     # Tracking information needed to track asset linage between
     # tasks and materialization

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/flow_engine.py

@@ -105,7 +105,7 @@ from prefect.utilities.callables import (
 from prefect.utilities.collections import visit_collection
 from prefect.utilities.engine import (
     capture_sigterm,
-    link_state_to_result,
+    link_state_to_flow_run_result,
     propose_state,
     propose_state_sync,
     resolve_to_final_result,
@@ -338,6 +338,7 @@ class FlowRunEngine(BaseFlowRunEngine[P, R]):
             self._return_value, State
         ):
             _result = self._return_value
+            link_state_to_flow_run_result(self.state, _result)
 
             if asyncio.iscoroutine(_result):
                 # getting the value for a BaseResult may return an awaitable
@@ -373,6 +374,7 @@ class FlowRunEngine(BaseFlowRunEngine[P, R]):
         self.set_state(terminal_state)
         self._return_value = resolved_result
 
+        link_state_to_flow_run_result(terminal_state, resolved_result)
         self._telemetry.end_span_on_success()
 
         return result
@@ -903,6 +905,7 @@ class AsyncFlowRunEngine(BaseFlowRunEngine[P, R]):
             self._return_value, State
         ):
             _result = self._return_value
+            link_state_to_flow_run_result(self.state, _result)
 
             if asyncio.iscoroutine(_result):
                 # getting the value for a BaseResult may return an awaitable
@@ -1426,7 +1429,7 @@ def run_generator_flow_sync(
                     while True:
                         gen_result = next(gen)
                         # link the current state to the result for dependency tracking
-                        link_state_to_result(engine.state, gen_result)
+                        link_state_to_flow_run_result(engine.state, gen_result)
                         yield gen_result
                 except StopIteration as exc:
                     engine.handle_success(exc.value)
@@ -1468,7 +1471,7 @@ async def run_generator_flow_async(
                         # can't use anext in Python < 3.10
                         gen_result = await gen.__anext__()
                         # link the current state to the result for dependency tracking
-                        link_state_to_result(engine.state, gen_result)
+                        link_state_to_flow_run_result(engine.state, gen_result)
                         yield gen_result
                 except (StopAsyncIteration, GeneratorExit) as exc:
                     await engine.handle_success(None)

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/runner/submit.py

@@ -17,7 +17,13 @@ from prefect.client.schemas.filters import (
     FlowRunFilterParentFlowRunId,
     TaskRunFilter,
 )
-from prefect.client.schemas.objects import Constant, FlowRun, Parameter, TaskRunResult
+from prefect.client.schemas.objects import (
+    Constant,
+    FlowRun,
+    FlowRunResult,
+    Parameter,
+    TaskRunResult,
+)
 from prefect.context import FlowRunContext
 from prefect.flows import Flow
 from prefect.logging import get_logger
@@ -66,9 +72,9 @@ async def _submit_flow_to_runner(
 
         parent_flow_run_context = FlowRunContext.get()
 
-        task_inputs: dict[str, list[TaskRunResult | Parameter | Constant]] = {
-            k: list(await collect_task_run_inputs(v)) for k, v in parameters.items()
-        }
+        task_inputs: dict[
+            str, list[Union[TaskRunResult, FlowRunResult, Parameter, Constant]]
+        ] = {k: list(await collect_task_run_inputs(v)) for k, v in parameters.items()}
         parameters = await resolve_inputs(parameters)
         dummy_task = Task(name=flow.name, fn=flow.fn, version=flow.version)
         parent_task_run = await client.create_task_run(

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/task_engine.py

@@ -37,7 +37,7 @@ import prefect.types._datetime
 from prefect.cache_policies import CachePolicy
 from prefect.client.orchestration import PrefectClient, SyncPrefectClient, get_client
 from prefect.client.schemas import TaskRun
-from prefect.client.schemas.objects import State, TaskRunInput
+from prefect.client.schemas.objects import RunInput, State
 from prefect.concurrency.context import ConcurrencyContext
 from prefect.concurrency.v1.asyncio import concurrency as aconcurrency
 from prefect.concurrency.v1.context import ConcurrencyContext as ConcurrencyContextV1
@@ -96,7 +96,7 @@ from prefect.utilities.callables import call_with_parameters, parameters_to_args
 from prefect.utilities.collections import visit_collection
 from prefect.utilities.engine import (
     emit_task_run_state_change_event,
-    link_state_to_result,
+    link_state_to_task_run_result,
     resolve_to_final_result,
 )
 from prefect.utilities.math import clamped_poisson_interval
@@ -453,7 +453,7 @@ class SyncTaskRunEngine(BaseTaskRunEngine[P, R]):
             else:
                 result = state.data
 
-            link_state_to_result(new_state, result)
+            link_state_to_task_run_result(new_state, result)
 
         # emit a state change event
         self._last_event = emit_task_run_state_change_event(
@@ -683,7 +683,7 @@ class SyncTaskRunEngine(BaseTaskRunEngine[P, R]):
     def initialize_run(
         self,
         task_run_id: Optional[UUID] = None,
-        dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+        dependencies: Optional[dict[str, set[RunInput]]] = None,
     ) -> Generator[Self, Any, Any]:
         """
         Enters a client context and creates a task run if needed.
@@ -777,7 +777,7 @@ class SyncTaskRunEngine(BaseTaskRunEngine[P, R]):
     def start(
         self,
         task_run_id: Optional[UUID] = None,
-        dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+        dependencies: Optional[dict[str, set[RunInput]]] = None,
     ) -> Generator[None, None, None]:
         with self.initialize_run(task_run_id=task_run_id, dependencies=dependencies):
             with (
@@ -1038,7 +1038,7 @@ class AsyncTaskRunEngine(BaseTaskRunEngine[P, R]):
             else:
                 result = new_state.data
 
-            link_state_to_result(new_state, result)
+            link_state_to_task_run_result(new_state, result)
 
         # emit a state change event
         self._last_event = emit_task_run_state_change_event(
@@ -1267,7 +1267,7 @@ class AsyncTaskRunEngine(BaseTaskRunEngine[P, R]):
     async def initialize_run(
         self,
         task_run_id: Optional[UUID] = None,
-        dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+        dependencies: Optional[dict[str, set[RunInput]]] = None,
     ) -> AsyncGenerator[Self, Any]:
         """
         Enters a client context and creates a task run if needed.
@@ -1359,7 +1359,7 @@ class AsyncTaskRunEngine(BaseTaskRunEngine[P, R]):
     async def start(
         self,
         task_run_id: Optional[UUID] = None,
-        dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+        dependencies: Optional[dict[str, set[RunInput]]] = None,
     ) -> AsyncGenerator[None, None]:
         async with self.initialize_run(
             task_run_id=task_run_id, dependencies=dependencies
@@ -1465,7 +1465,7 @@ def run_task_sync(
     parameters: Optional[dict[str, Any]] = None,
     wait_for: Optional["OneOrManyFutureOrResult[Any]"] = None,
     return_type: Literal["state", "result"] = "result",
-    dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+    dependencies: Optional[dict[str, set[RunInput]]] = None,
     context: Optional[dict[str, Any]] = None,
 ) -> Union[R, State, None]:
     engine = SyncTaskRunEngine[P, R](
@@ -1496,7 +1496,7 @@ async def run_task_async(
     parameters: Optional[dict[str, Any]] = None,
     wait_for: Optional["OneOrManyFutureOrResult[Any]"] = None,
     return_type: Literal["state", "result"] = "result",
-    dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+    dependencies: Optional[dict[str, set[RunInput]]] = None,
     context: Optional[dict[str, Any]] = None,
 ) -> Union[R, State, None]:
     engine = AsyncTaskRunEngine[P, R](
@@ -1527,7 +1527,7 @@ def run_generator_task_sync(
     parameters: Optional[dict[str, Any]] = None,
     wait_for: Optional["OneOrManyFutureOrResult[Any]"] = None,
     return_type: Literal["state", "result"] = "result",
-    dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+    dependencies: Optional[dict[str, set[RunInput]]] = None,
     context: Optional[dict[str, Any]] = None,
 ) -> Generator[R, None, None]:
     if return_type != "result":
@@ -1568,7 +1568,7 @@ def run_generator_task_sync(
                             # dictionary in an unbounded way, so finding a
                             # way to periodically clean it up (using
                             # weakrefs or similar) would be good
-                            link_state_to_result(engine.state, gen_result)
+                            link_state_to_task_run_result(engine.state, gen_result)
                             yield gen_result
                     except StopIteration as exc:
                         engine.handle_success(exc.value, transaction=txn)
@@ -1586,7 +1586,7 @@ async def run_generator_task_async(
     parameters: Optional[dict[str, Any]] = None,
     wait_for: Optional["OneOrManyFutureOrResult[Any]"] = None,
     return_type: Literal["state", "result"] = "result",
-    dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+    dependencies: Optional[dict[str, set[RunInput]]] = None,
     context: Optional[dict[str, Any]] = None,
 ) -> AsyncGenerator[R, None]:
     if return_type != "result":
@@ -1627,7 +1627,7 @@ async def run_generator_task_async(
                             # dictionary in an unbounded way, so finding a
                             # way to periodically clean it up (using
                             # weakrefs or similar) would be good
-                            link_state_to_result(engine.state, gen_result)
+                            link_state_to_task_run_result(engine.state, gen_result)
                             yield gen_result
                     except (StopAsyncIteration, GeneratorExit) as exc:
                         await engine.handle_success(None, transaction=txn)
@@ -1647,7 +1647,7 @@ def run_task(
     parameters: Optional[dict[str, Any]] = None,
     wait_for: Optional["OneOrManyFutureOrResult[Any]"] = None,
     return_type: Literal["state"] = "state",
-    dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+    dependencies: Optional[dict[str, set[RunInput]]] = None,
     context: Optional[dict[str, Any]] = None,
 ) -> State[R]: ...
 
@@ -1660,7 +1660,7 @@ def run_task(
     parameters: Optional[dict[str, Any]] = None,
     wait_for: Optional["OneOrManyFutureOrResult[Any]"] = None,
     return_type: Literal["result"] = "result",
-    dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+    dependencies: Optional[dict[str, set[RunInput]]] = None,
     context: Optional[dict[str, Any]] = None,
 ) -> R: ...
 
@@ -1672,7 +1672,7 @@ def run_task(
     parameters: Optional[dict[str, Any]] = None,
     wait_for: Optional["OneOrManyFutureOrResult[Any]"] = None,
     return_type: Literal["state", "result"] = "result",
-    dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+    dependencies: Optional[dict[str, set[RunInput]]] = None,
     context: Optional[dict[str, Any]] = None,
 ) -> Union[R, State, None, Coroutine[Any, Any, Union[R, State, None]]]:
     """

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/task_runners.py

@@ -19,7 +19,7 @@ from typing import (
 from typing_extensions import ParamSpec, Self, TypeVar
 
 from prefect._internal.uuid7 import uuid7
-from prefect.client.schemas.objects import TaskRunInput
+from prefect.client.schemas.objects import RunInput
 from prefect.exceptions import MappingLengthMismatch, MappingMissingIterable
 from prefect.futures import (
     PrefectConcurrentFuture,
@@ -81,7 +81,7 @@ class TaskRunner(abc.ABC, Generic[F]):
         task: "Task[P, Coroutine[Any, Any, R]]",
         parameters: dict[str, Any],
         wait_for: Iterable[PrefectFuture[Any]] | None = None,
-        dependencies: dict[str, set[TaskRunInput]] | None = None,
+        dependencies: dict[str, set[RunInput]] | None = None,
     ) -> F: ...
 
     @overload
@@ -91,7 +91,7 @@ class TaskRunner(abc.ABC, Generic[F]):
         task: "Task[Any, R]",
         parameters: dict[str, Any],
         wait_for: Iterable[PrefectFuture[Any]] | None = None,
-        dependencies: dict[str, set[TaskRunInput]] | None = None,
+        dependencies: dict[str, set[RunInput]] | None = None,
     ) -> F: ...
 
     @abc.abstractmethod
@@ -100,7 +100,7 @@ class TaskRunner(abc.ABC, Generic[F]):
         task: "Task[P, R]",
         parameters: dict[str, Any],
         wait_for: Iterable[PrefectFuture[Any]] | None = None,
-        dependencies: dict[str, set[TaskRunInput]] | None = None,
+        dependencies: dict[str, set[RunInput]] | None = None,
     ) -> F: ...
 
     def map(
@@ -262,7 +262,7 @@ class ThreadPoolTaskRunner(TaskRunner[PrefectConcurrentFuture[R]]):
         task: "Task[P, Coroutine[Any, Any, R]]",
         parameters: dict[str, Any],
         wait_for: Iterable[PrefectFuture[Any]] | None = None,
-        dependencies: dict[str, set[TaskRunInput]] | None = None,
+        dependencies: dict[str, set[RunInput]] | None = None,
     ) -> PrefectConcurrentFuture[R]: ...
 
     @overload
@@ -271,7 +271,7 @@ class ThreadPoolTaskRunner(TaskRunner[PrefectConcurrentFuture[R]]):
         task: "Task[Any, R]",
         parameters: dict[str, Any],
         wait_for: Iterable[PrefectFuture[Any]] | None = None,
-        dependencies: dict[str, set[TaskRunInput]] | None = None,
+        dependencies: dict[str, set[RunInput]] | None = None,
     ) -> PrefectConcurrentFuture[R]: ...
 
     def submit(
@@ -279,7 +279,7 @@ class ThreadPoolTaskRunner(TaskRunner[PrefectConcurrentFuture[R]]):
         task: "Task[P, R | Coroutine[Any, Any, R]]",
         parameters: dict[str, Any],
         wait_for: Iterable[PrefectFuture[Any]] | None = None,
-        dependencies: dict[str, set[TaskRunInput]] | None = None,
+        dependencies: dict[str, set[RunInput]] | None = None,
     ) -> PrefectConcurrentFuture[R]:
         """
         Submit a task to the task run engine running in a separate thread.
@@ -415,7 +415,7 @@ class PrefectTaskRunner(TaskRunner[PrefectDistributedFuture[R]]):
         task: "Task[P, Coroutine[Any, Any, R]]",
         parameters: dict[str, Any],
         wait_for: Iterable[PrefectFuture[Any]] | None = None,
-        dependencies: dict[str, set[TaskRunInput]] | None = None,
+        dependencies: dict[str, set[RunInput]] | None = None,
     ) -> PrefectDistributedFuture[R]: ...
 
     @overload
@@ -424,7 +424,7 @@ class PrefectTaskRunner(TaskRunner[PrefectDistributedFuture[R]]):
         task: "Task[Any, R]",
         parameters: dict[str, Any],
         wait_for: Iterable[PrefectFuture[Any]] | None = None,
-        dependencies: dict[str, set[TaskRunInput]] | None = None,
+        dependencies: dict[str, set[RunInput]] | None = None,
     ) -> PrefectDistributedFuture[R]: ...
 
     def submit(
@@ -432,7 +432,7 @@ class PrefectTaskRunner(TaskRunner[PrefectDistributedFuture[R]]):
         task: "Task[P, R]",
         parameters: dict[str, Any],
         wait_for: Iterable[PrefectFuture[Any]] | None = None,
-        dependencies: dict[str, set[TaskRunInput]] | None = None,
+        dependencies: dict[str, set[RunInput]] | None = None,
     ) -> PrefectDistributedFuture[R]:
         """
         Submit a task to the task run engine running in a separate thread.

{prefect_client-3.4.7.dev5 → prefect_client-3.4.7.dev6}/src/prefect/tasks.py

@@ -47,8 +47,8 @@ from prefect.cache_policies import DEFAULT, NO_CACHE, CachePolicy
 from prefect.client.orchestration import get_client
 from prefect.client.schemas import TaskRun
 from prefect.client.schemas.objects import (
+    RunInput,
     StateDetails,
-    TaskRunInput,
     TaskRunPolicy,
     TaskRunResult,
 )
@@ -244,12 +244,17 @@ def _infer_parent_task_runs(
     # tracked within the same flow run.
     if flow_run_context:
         for v in parameters.values():
+            upstream_state = None
+
             if isinstance(v, State):
                 upstream_state = v
             elif isinstance(v, PrefectFuture):
                 upstream_state = v.state
             else:
-                upstream_state = flow_run_context.task_run_results.get(id(v))
+                res = flow_run_context.run_results.get(id(v))
+                if res:
+                    upstream_state, _ = res
+
             if upstream_state and upstream_state.is_running():
                 parents.append(
                     TaskRunResult(id=upstream_state.state_details.task_run_id)
@@ -840,7 +845,7 @@ class Task(Generic[P, R]):
         flow_run_context: Optional[FlowRunContext] = None,
         parent_task_run_context: Optional[TaskRunContext] = None,
         wait_for: Optional[OneOrManyFutureOrResult[Any]] = None,
-        extra_task_inputs: Optional[dict[str, set[TaskRunInput]]] = None,
+        extra_task_inputs: Optional[dict[str, set[RunInput]]] = None,
         deferred: bool = False,
     ) -> TaskRun:
         from prefect.utilities._engine import dynamic_key_for_task_run
@@ -943,7 +948,7 @@ class Task(Generic[P, R]):
         flow_run_context: Optional[FlowRunContext] = None,
         parent_task_run_context: Optional[TaskRunContext] = None,
         wait_for: Optional[OneOrManyFutureOrResult[Any]] = None,
-        extra_task_inputs: Optional[dict[str, set[TaskRunInput]]] = None,
+        extra_task_inputs: Optional[dict[str, set[RunInput]]] = None,
         deferred: bool = False,
     ) -> TaskRun:
         from prefect.utilities._engine import dynamic_key_for_task_run
@@ -1530,7 +1535,7 @@ class Task(Generic[P, R]):
         args: Optional[tuple[Any, ...]] = None,
         kwargs: Optional[dict[str, Any]] = None,
         wait_for: Optional[Iterable[PrefectFuture[R]]] = None,
-        dependencies: Optional[dict[str, set[TaskRunInput]]] = None,
+        dependencies: Optional[dict[str, set[RunInput]]] = None,
     ) -> PrefectDistributedFuture[R]:
         """
         Create a pending task run for a task worker to execute.