prefect-client 2.14.11__tar.gz → 2.14.13__tar.gz

{prefect-client-2.14.11 → prefect-client-2.14.13}/MANIFEST.in

@@ -21,6 +21,7 @@ include src/prefect/.prefectignore
 include src/prefect/logging/logging.yml
 include src/prefect/cli/templates/*.yaml
 include src/prefect/server/collection_blocks_data.json
+include src/prefect/server/api/collections_data/views/*.json
 
 # Migrations
 include src/prefect/server/database/alembic.ini

{prefect-client-2.14.11/src/prefect_client.egg-info → prefect-client-2.14.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 2.14.11
+Version: 2.14.13
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.
@@ -32,7 +32,26 @@ Description: <p align="center"><img src="https://github.com/PrefectHQ/prefect/as
         
         # Prefect
         
-        Prefect is an orchestrator for data-intensive workflows. It's the simplest way to transform any Python function into a unit of work that can be observed and orchestrated. With Prefect, you can build resilient, dynamic workflows that react to the world around them and recover from unexpected changes. With just a few decorators, Prefect supercharges your code with features like automatic retries, distributed execution, scheduling, caching, and much more. Every activity is tracked and can be monitored with the Prefect server or Prefect Cloud dashboard.
+        Prefect is an orchestration and observability platform for building, observing, and triaging workflows. 
+        It's the simplest way to transform Python code into an interactive workflow application.
+        
+        Prefect allows you to expose your workflows through an API so teams dependent on you can programmatically access your pipelines, business logic, and more.
+        Prefect also allows you to standardize workflow development and deployment across your organization.
+        
+        With Prefect, you can build resilient, dynamic workflows that react to the world around them and recover from unexpected changes.
+        With just a few decorators, Prefect supercharges your code with features like automatic retries, distributed execution, scheduling, caching, and much more.
+        
+        Every activity is tracked and can be monitored with a self-hosted [Prefect server](https://docs.prefect.io/latest/guides/host/) instance or managed [Prefect Cloud](https://www.prefect.io/cloud-vs-oss?utm_source=oss&utm_medium=oss&utm_campaign=oss_gh_repo&utm_term=none&utm_content=none) dashboard.
+        
+        ## Getting started
+        
+        Prefect requires Python 3.8 or later. To [install Prefect](https://docs.prefect.io/getting-started/installation/), run the following command:
+        
+        ```bash
+        pip install prefect
+        ```
+        
+        Then create a Python file the uses Prefect `flow` and `task` decorators to orchestrate and observe your workflow that fetches the number of GitHub stars from a repository.
         
         ```python
         from prefect import flow, task
@@ -40,7 +59,7 @@ Description: <p align="center"><img src="https://github.com/PrefectHQ/prefect/as
         import httpx
         
         
-        @task(retries=3)
+        @task(log_prints=True)
         def get_stars(repo: str):
             url = f"https://api.github.com/repos/{repo}"
             count = httpx.get(url).json()["stargazers_count"]
@@ -54,10 +73,11 @@ Description: <p align="center"><img src="https://github.com/PrefectHQ/prefect/as
         
         
         # run the flow!
-        github_stars(["PrefectHQ/Prefect"])
+        if __name__=="__main__":
+            github_stars(["PrefectHQ/Prefect"])
         ```
         
-        After running some flows, fire up the Prefect UI to see what happened:
+        Fire up the Prefect UI to see what happened:
         
         ```bash
         prefect server start
@@ -65,30 +85,32 @@ Description: <p align="center"><img src="https://github.com/PrefectHQ/prefect/as
         
         ![Prefect UI dashboard](/docs/img/ui/cloud-dashboard.png)
         
-        From here, you can continue to use Prefect interactively or [deploy your flows](https://docs.prefect.io/concepts/deployments) to remote environments, running on a scheduled or event-driven basis.
-        
-        ## Getting Started
-        
-        Prefect requires Python 3.8 or later. To [install Prefect](https://docs.prefect.io/getting-started/installation/), run the following command in a shell or terminal session:
+        To run your workflow on a schedule, turn it into a deployment and schedule it to run every minute by changing the last line of your script to the following:
         
-        ```bash
-        pip install prefect
+        ```python
+            github_stars.serve(name="first-deployment", cron="* * * * *")
         ```
         
-        Start by exploring the [core concepts of Prefect workflows](https://docs.prefect.io/concepts/), then follow one of our [friendly tutorials](https://docs.prefect.io/tutorials/first-steps) to learn by example.
+        You now have a server running locally that is looking for scheduled deployments!
+        Additionally you can run your workflow manually from the UI or CLI - and if you're using Prefect Cloud, you can even run deployments in response to [events](https://docs.prefect.io/latest/concepts/automations/).
         
         ## Prefect Cloud
         
-        Stop worrying about your workflows. Prefect Cloud allows you to centrally deploy, monitor, and manage the data workflows you support. With managed orchestration, automations, and webhooks, all backed by enterprise-class security, build production-ready code quickly and reliably.
+        Stop worrying about your workflows.
+        Prefect Cloud allows you to centrally deploy, monitor, and manage the data workflows you support. With managed orchestration, automations, and webhooks, all backed by enterprise-class security, build production-ready code quickly and reliably.
         
-        Read more about Prefect Cloud [here](https://www.prefect.io/cloud-vs-oss?utm_source=oss&utm_medium=oss&utm_campaign=oss_gh_repo&utm_term=none&utm_content=none) or signup to [try it for yourself](https://app.prefect.cloud?utm_source=oss&utm_medium=oss&utm_campaign=oss_gh_repo&utm_term=none&utm_content=none).
+        Read more about Prefect Cloud [here](https://www.prefect.io/cloud-vs-oss?utm_source=oss&utm_medium=oss&utm_campaign=oss_gh_repo&utm_term=none&utm_content=none) or sign up to [try it for yourself](https://app.prefect.cloud?utm_source=oss&utm_medium=oss&utm_campaign=oss_gh_repo&utm_term=none&utm_content=none).
         
         ![Prefect Automations](/docs/img/ui/automations.png)
         
+        ## Next steps
+        
+        There's lots more you can do to orchestrate and observe your workflows with Prefect!
+        Start with our [friendly tutorial](https://docs.prefect.io/tutorials) or explore the [core concepts of Prefect workflows](https://docs.prefect.io/concepts/).
         
         ## Join the community
         
-        Prefect is made possible by the fastest growing community of thousands of friendly data engineers. Join us in building a new kind of workflow system. The [Prefect Slack community](https://prefect.io/slack) is a fantastic place to learn more about Prefect, ask questions, or get help with workflow design. The [Prefect Discourse](https://discourse.prefect.io/) is a community-driven knowledge base to find answers to your Prefect-related questions. All community forums, including code contributions, issue discussions, and slack messages are subject to our [Code of Conduct](https://discourse.prefect.io/faq).
+        Prefect is made possible by the fastest growing community of thousands of friendly data engineers. Join us in building a new kind of workflow system. The [Prefect Slack community](https://prefect.io/slack) is a fantastic place to learn more about Prefect, ask questions, or get help with workflow design. All community forums, including code contributions, issue discussions, and slack messages are subject to our [Code of Conduct](https://discourse.prefect.io/faq).
         
         ## Contribute
         

prefect-client-2.14.13/README.md

@@ -0,0 +1,107 @@
+<p align="center"><img src="https://github.com/PrefectHQ/prefect/assets/3407835/c654cbc6-63e8-4ada-a92a-efd2f8f24b85" width=1000></p>
+
+<p align="center">
+    <a href="https://pypi.python.org/pypi/prefect/" alt="PyPI version">
+        <img alt="PyPI" src="https://img.shields.io/pypi/v/prefect?color=0052FF&labelColor=090422"></a>
+    <a href="https://github.com/prefecthq/prefect/" alt="Stars">
+        <img src="https://img.shields.io/github/stars/prefecthq/prefect?color=0052FF&labelColor=090422" /></a>
+    <a href="https://pepy.tech/badge/prefect/" alt="Downloads">
+        <img src="https://img.shields.io/pypi/dm/prefect?color=0052FF&labelColor=090422" /></a>
+    <a href="https://github.com/prefecthq/prefect/pulse" alt="Activity">
+        <img src="https://img.shields.io/github/commit-activity/m/prefecthq/prefect?color=0052FF&labelColor=090422" /></a>
+    <br>
+    <a href="https://prefect.io/slack" alt="Slack">
+        <img src="https://img.shields.io/badge/slack-join_community-red.svg?color=0052FF&labelColor=090422&logo=slack" /></a>
+    <a href="https://discourse.prefect.io/" alt="Discourse">
+        <img src="https://img.shields.io/badge/discourse-browse_forum-red.svg?color=0052FF&labelColor=090422&logo=discourse" /></a>
+    <a href="https://www.youtube.com/c/PrefectIO/" alt="YouTube">
+        <img src="https://img.shields.io/badge/youtube-watch_videos-red.svg?color=0052FF&labelColor=090422&logo=youtube" /></a>
+</p>
+
+# Prefect
+
+Prefect is an orchestration and observability platform for building, observing, and triaging workflows. 
+It's the simplest way to transform Python code into an interactive workflow application.
+
+Prefect allows you to expose your workflows through an API so teams dependent on you can programmatically access your pipelines, business logic, and more.
+Prefect also allows you to standardize workflow development and deployment across your organization.
+
+With Prefect, you can build resilient, dynamic workflows that react to the world around them and recover from unexpected changes.
+With just a few decorators, Prefect supercharges your code with features like automatic retries, distributed execution, scheduling, caching, and much more.
+
+Every activity is tracked and can be monitored with a self-hosted [Prefect server](https://docs.prefect.io/latest/guides/host/) instance or managed [Prefect Cloud](https://www.prefect.io/cloud-vs-oss?utm_source=oss&utm_medium=oss&utm_campaign=oss_gh_repo&utm_term=none&utm_content=none) dashboard.
+
+## Getting started
+
+Prefect requires Python 3.8 or later. To [install Prefect](https://docs.prefect.io/getting-started/installation/), run the following command:
+
+```bash
+pip install prefect
+```
+
+Then create a Python file the uses Prefect `flow` and `task` decorators to orchestrate and observe your workflow that fetches the number of GitHub stars from a repository.
+
+```python
+from prefect import flow, task
+from typing import List
+import httpx
+
+
+@task(log_prints=True)
+def get_stars(repo: str):
+    url = f"https://api.github.com/repos/{repo}"
+    count = httpx.get(url).json()["stargazers_count"]
+    print(f"{repo} has {count} stars!")
+
+
+@flow(name="GitHub Stars")
+def github_stars(repos: List[str]):
+    for repo in repos:
+        get_stars(repo)
+
+
+# run the flow!
+if __name__=="__main__":
+    github_stars(["PrefectHQ/Prefect"])
+```
+
+Fire up the Prefect UI to see what happened:
+
+```bash
+prefect server start
+```
+
+![Prefect UI dashboard](/docs/img/ui/cloud-dashboard.png)
+
+To run your workflow on a schedule, turn it into a deployment and schedule it to run every minute by changing the last line of your script to the following:
+
+```python
+    github_stars.serve(name="first-deployment", cron="* * * * *")
+```
+
+You now have a server running locally that is looking for scheduled deployments!
+Additionally you can run your workflow manually from the UI or CLI - and if you're using Prefect Cloud, you can even run deployments in response to [events](https://docs.prefect.io/latest/concepts/automations/).
+
+## Prefect Cloud
+
+Stop worrying about your workflows.
+Prefect Cloud allows you to centrally deploy, monitor, and manage the data workflows you support. With managed orchestration, automations, and webhooks, all backed by enterprise-class security, build production-ready code quickly and reliably.
+
+Read more about Prefect Cloud [here](https://www.prefect.io/cloud-vs-oss?utm_source=oss&utm_medium=oss&utm_campaign=oss_gh_repo&utm_term=none&utm_content=none) or sign up to [try it for yourself](https://app.prefect.cloud?utm_source=oss&utm_medium=oss&utm_campaign=oss_gh_repo&utm_term=none&utm_content=none).
+
+![Prefect Automations](/docs/img/ui/automations.png)
+
+## Next steps
+
+There's lots more you can do to orchestrate and observe your workflows with Prefect!
+Start with our [friendly tutorial](https://docs.prefect.io/tutorials) or explore the [core concepts of Prefect workflows](https://docs.prefect.io/concepts/).
+
+## Join the community
+
+Prefect is made possible by the fastest growing community of thousands of friendly data engineers. Join us in building a new kind of workflow system. The [Prefect Slack community](https://prefect.io/slack) is a fantastic place to learn more about Prefect, ask questions, or get help with workflow design. All community forums, including code contributions, issue discussions, and slack messages are subject to our [Code of Conduct](https://discourse.prefect.io/faq).
+
+## Contribute
+
+See our [documentation on contributing to Prefect](https://docs.prefect.io/contributing/overview/).
+
+Thanks for being part of the mission to build a new kind of workflow system and, of course, **happy engineering!**

{prefect-client-2.14.11 → prefect-client-2.14.13}/src/prefect/client/orchestration.py

@@ -92,6 +92,7 @@ from prefect.client.schemas.objects import (
     Worker,
     WorkPool,
     WorkQueue,
+    WorkQueueStatusDetail,
 )
 from prefect.client.schemas.responses import (
     DeploymentResponse,
@@ -1036,6 +1037,32 @@ class PrefectClient:
                 raise
         return WorkQueue.parse_obj(response.json())
 
+    async def read_work_queue_status(
+        self,
+        id: UUID,
+    ) -> WorkQueueStatusDetail:
+        """
+        Read a work queue status.
+
+        Args:
+            id: the id of the work queue to load
+
+        Raises:
+            prefect.exceptions.ObjectNotFound: If request returns 404
+            httpx.RequestError: If request fails
+
+        Returns:
+            WorkQueueStatus: an instantiated WorkQueueStatus object
+        """
+        try:
+            response = await self._client.get(f"/work_queues/{id}/status")
+        except httpx.HTTPStatusError as e:
+            if e.response.status_code == status.HTTP_404_NOT_FOUND:
+                raise prefect.exceptions.ObjectNotFound(http_exc=e) from e
+            else:
+                raise
+        return WorkQueueStatusDetail.parse_obj(response.json())
+
     async def match_work_queues(
         self,
         prefixes: List[str],

{prefect-client-2.14.11 → prefect-client-2.14.13}/src/prefect/client/schemas/objects.py

@@ -30,7 +30,7 @@ from prefect._internal.schemas.validators import (
     raise_on_name_with_banned_characters,
 )
 from prefect.client.schemas.schedules import SCHEDULE_TYPES
-from prefect.settings import PREFECT_CLOUD_API_URL
+from prefect.settings import PREFECT_CLOUD_API_URL, PREFECT_CLOUD_UI_URL
 from prefect.utilities.collections import AutoEnum, listrepr
 from prefect.utilities.names import generate_slug
 
@@ -110,6 +110,7 @@ class StateDetails(PrefectBaseModel):
     pause_key: str = None
     run_input_keyset: Optional[Dict[str, str]] = None
     refresh_cache: bool = None
+    retriable: bool = None
 
 
 class State(ObjectBaseModel, Generic[R]):
@@ -777,6 +778,16 @@ class Workspace(PrefectBaseModel):
             f"/workspaces/{self.workspace_id}"
         )
 
+    def ui_url(self) -> str:
+        """
+        Generate the UI URL for accessing this workspace
+        """
+        return (
+            f"{PREFECT_CLOUD_UI_URL.value()}"
+            f"/account/{self.account_id}"
+            f"/workspace/{self.workspace_id}"
+        )
+
     def __hash__(self):
         return hash(self.handle)
 

{prefect-client-2.14.11 → prefect-client-2.14.13}/src/prefect/engine.py

@@ -952,7 +952,7 @@ async def orchestrate_flow_run(
 async def pause_flow_run(
     wait_for_input: None = None,
     flow_run_id: UUID = None,
-    timeout: int = 300,
+    timeout: int = 3600,
     poll_interval: int = 10,
     reschedule: bool = False,
     key: str = None,
@@ -964,7 +964,7 @@ async def pause_flow_run(
 async def pause_flow_run(
     wait_for_input: Type[T],
     flow_run_id: UUID = None,
-    timeout: int = 300,
+    timeout: int = 3600,
     poll_interval: int = 10,
     reschedule: bool = False,
     key: str = None,
@@ -988,7 +988,7 @@ async def pause_flow_run(
 async def pause_flow_run(
     wait_for_input: Optional[Type[T]] = None,
     flow_run_id: UUID = None,
-    timeout: int = 300,
+    timeout: int = 3600,
     poll_interval: int = 10,
     reschedule: bool = False,
     key: str = None,
@@ -1011,7 +1011,7 @@ async def pause_flow_run(
             have an associated deployment and results need to be configured with the
             `persist_results` option.
         timeout: the number of seconds to wait for the flow to be resumed before
-            failing. Defaults to 5 minutes (300 seconds). If the pause timeout exceeds
+            failing. Defaults to 1 hour (3600 seconds). If the pause timeout exceeds
             any configured flow-level timeout, the flow might fail even after resuming.
         poll_interval: The number of seconds between checking whether the flow has been
             resumed. Defaults to 10 seconds.
@@ -1028,6 +1028,24 @@ async def pause_flow_run(
             resumed without providing the input, the flow will fail. If the flow is
             resumed with the input, the flow will resume and the input will be loaded
             and returned from this function.
+
+    Example:
+    ```python
+    @task
+    def task_one():
+        for i in range(3):
+            sleep(1)
+
+    @flow
+    def my_flow():
+        terminal_state = task_one.submit(return_state=True)
+        if terminal_state.type == StateType.COMPLETED:
+            print("Task one succeeded! Pausing flow run..")
+            pause_flow_run(timeout=2)
+        else:
+            print("Task one failed. Skipping pause flow run..")
+    ```
+
     """
     if flow_run_id:
         if wait_for_input is not None:
@@ -1050,11 +1068,8 @@ async def pause_flow_run(
 
 
 @inject_client
-@experimental_parameter(
-    "wait_for_input", group="flow_run_input", when=lambda y: y is not None
-)
 async def _in_process_pause(
-    timeout: int = 300,
+    timeout: int = 3600,
     poll_interval: int = 10,
     reschedule=False,
     key: str = None,
@@ -1149,7 +1164,7 @@ async def _in_process_pause(
 @inject_client
 async def _out_of_process_pause(
     flow_run_id: UUID,
-    timeout: int = 300,
+    timeout: int = 3600,
     reschedule: bool = True,
     key: str = None,
     client=None,
@@ -1172,7 +1187,7 @@ async def _out_of_process_pause(
 async def suspend_flow_run(
     wait_for_input: None = None,
     flow_run_id: Optional[UUID] = None,
-    timeout: Optional[int] = 300,
+    timeout: Optional[int] = 3600,
     key: Optional[str] = None,
     client: PrefectClient = None,
 ) -> None:
@@ -1183,7 +1198,7 @@ async def suspend_flow_run(
 async def suspend_flow_run(
     wait_for_input: Type[T],
     flow_run_id: Optional[UUID] = None,
-    timeout: Optional[int] = 300,
+    timeout: Optional[int] = 3600,
     key: Optional[str] = None,
     client: PrefectClient = None,
 ) -> T:
@@ -1192,10 +1207,13 @@ async def suspend_flow_run(
 
 @sync_compatible
 @inject_client
+@experimental_parameter(
+    "wait_for_input", group="flow_run_input", when=lambda y: y is not None
+)
 async def suspend_flow_run(
     wait_for_input: Optional[Type[T]] = None,
     flow_run_id: Optional[UUID] = None,
-    timeout: Optional[int] = 300,
+    timeout: Optional[int] = 3600,
     key: Optional[str] = None,
     client: PrefectClient = None,
 ):
@@ -1214,7 +1232,7 @@ async def suspend_flow_run(
             suspend the specified flow run. If not supplied will attempt to
             suspend the current flow run.
         timeout: the number of seconds to wait for the flow to be resumed before
-            failing. Defaults to 5 minutes (300 seconds). If the pause timeout
+            failing. Defaults to 1 hour (3600 seconds). If the pause timeout
             exceeds any configured flow-level timeout, the flow might fail even
             after resuming.
         key: An optional key to prevent calling suspend more than once. This
@@ -2075,7 +2093,13 @@ async def orchestrate_task_run(
                     )
                     terminal_state.state_details.cache_key = cache_key
 
+            if terminal_state.is_failed():
+                # Defer to user to decide whether failure is retriable
+                terminal_state.state_details.retriable = (
+                    await _check_task_failure_retriable(task, task_run, terminal_state)
+                )
             state = await propose_state(client, terminal_state, task_run_id=task_run.id)
+
             last_event = _emit_task_run_state_change_event(
                 task_run=task_run,
                 initial_state=last_state,
@@ -2672,6 +2696,48 @@ async def _run_task_hooks(task: Task, task_run: TaskRun, state: State) -> None:
                 logger.info(f"Hook {hook_name!r} finished running successfully")
 
 
+async def _check_task_failure_retriable(
+    task: Task, task_run: TaskRun, state: State
+) -> bool:
+    """Run the `retry_condition_fn` callable for a task, making sure to catch and log any errors
+    that occur. If None, return True. If not callable, logs an error and returns False.
+    """
+    if task.retry_condition_fn is None:
+        return True
+
+    logger = task_run_logger(task_run)
+
+    try:
+        logger.debug(
+            f"Running `retry_condition_fn` check {task.retry_condition_fn!r} for task"
+            f" {task.name!r}"
+        )
+        if is_async_fn(task.retry_condition_fn):
+            return bool(
+                await task.retry_condition_fn(task=task, task_run=task_run, state=state)
+            )
+        else:
+            return bool(
+                await from_async.call_in_new_thread(
+                    create_call(
+                        task.retry_condition_fn,
+                        task=task,
+                        task_run=task_run,
+                        state=state,
+                    )
+                )
+            )
+    except Exception:
+        logger.error(
+            (
+                "An error was encountered while running `retry_condition_fn` check"
+                f" '{task.retry_condition_fn!r}' for task {task.name!r}"
+            ),
+            exc_info=True,
+        )
+        return False
+
+
 async def _run_flow_hooks(flow: Flow, flow_run: FlowRun, state: State) -> None:
     """Run the on_failure, on_completion, on_cancellation, and on_crashed hooks for a flow, making sure to
     catch and log any errors that occur.

{prefect-client-2.14.11 → prefect-client-2.14.13}/src/prefect/events/actions.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Optional
+from typing import Any, Dict, Optional, Union
 from uuid import UUID
 
 from prefect._internal.pydantic import HAS_PYDANTIC_V2
@@ -34,4 +34,15 @@ class RunDeployment(Action):
     deployment_id: UUID = Field(..., description="The identifier of the deployment")
 
 
-ActionTypes = RunDeployment
+class SendNotification(Action):
+    """Send a notification with the given parameters"""
+
+    type: Literal["send-notification"] = "send-notification"
+    block_document_id: UUID = Field(
+        ..., description="The identifier of the notification block"
+    )
+    body: str = Field(..., description="Notification body")
+    subject: Optional[str] = Field(None, description="Notification subject")
+
+
+ActionTypes = Union[RunDeployment, SendNotification]

{prefect-client-2.14.11 → prefect-client-2.14.13}/src/prefect/events/schemas.py

@@ -341,7 +341,7 @@ class DeploymentTrigger(ResourceTrigger):
     def owner_resource(self) -> Optional[str]:
         return f"prefect.deployment.{self._deployment_id}"
 
-    def actions(self) -> List[ActionTypes]:
+    def actions(self) -> List[RunDeployment]:
         assert self._deployment_id
         return [
             RunDeployment(

{prefect-client-2.14.11 → prefect-client-2.14.13}/src/prefect/flows.py

@@ -33,6 +33,7 @@ from typing import (
 from uuid import UUID
 
 from prefect._vendor.fastapi.encoders import jsonable_encoder
+from typing_extensions import Self
 
 from prefect._internal.concurrency.api import create_call, from_async
 from prefect._internal.pydantic import HAS_PYDANTIC_V2
@@ -117,6 +118,7 @@ from prefect.utilities.visualization import (
 T = TypeVar("T")  # Generic type var for capturing the inner return type of async funcs
 R = TypeVar("R")  # The return type of the user's function
 P = ParamSpec("P")  # The parameters of the flow
+F = TypeVar("F", bound="Flow")  # The type of the flow
 
 logger = get_logger("flows")
 
@@ -373,7 +375,7 @@ class Flow(Generic[P, R]):
             List[Callable[[FlowSchema, FlowRun, State], None]]
         ] = None,
         on_crashed: Optional[List[Callable[[FlowSchema, FlowRun, State], None]]] = None,
-    ):
+    ) -> Self:
         """
         Create a new flow from the current object, updating provided options.
 
@@ -428,7 +430,7 @@ class Flow(Generic[P, R]):
             >>> assert state.result() == 4
 
         """
-        return Flow(
+        new_flow = Flow(
             fn=self.fn,
             name=name or self.name,
             description=description or self.description,
@@ -467,6 +469,9 @@ class Flow(Generic[P, R]):
             on_cancellation=on_cancellation or self.on_cancellation,
             on_crashed=on_crashed or self.on_crashed,
         )
+        new_flow._storage = self._storage
+        new_flow._entrypoint = self._entrypoint
+        return new_flow
 
     def validate_parameters(self, parameters: Dict[str, Any]) -> Dict[str, Any]:
         """
@@ -618,6 +623,8 @@ class Flow(Generic[P, R]):
         """
         from prefect.deployments.runner import RunnerDeployment
 
+        if not name.endswith(".py"):
+            raise_on_name_with_banned_characters(name)
         if self._storage and self._entrypoint:
             return await RunnerDeployment.from_storage(
                 storage=self._storage,
@@ -775,10 +782,10 @@ class Flow(Generic[P, R]):
     @classmethod
     @sync_compatible
     async def from_source(
-        cls,
+        cls: Type[F],
         source: Union[str, RunnerStorage, ReadableDeploymentStorage],
         entrypoint: str,
-    ) -> "Flow":
+    ) -> F:
         """
         Loads a flow from a remote s ource.
 

{prefect-client-2.14.11 → prefect-client-2.14.13}/src/prefect/infrastructure/provisioners/container_instance.py

@@ -669,8 +669,9 @@ class ContainerInstancePushProvisioner:
             registry: The registry to grant access to.
         """
         command = (
-            f"az role assignment create --assignee {identity['principalId']} --scope"
-            f" {registry['id']} --role AcrPull --subscription {subscription_id}"
+            "az role assignment create --assignee-object-id"
+            f" {identity['principalId']} --assignee-principal-type ServicePrincipal"
+            f" --scope {registry['id']} --role AcrPull --subscription {subscription_id}"
         )
         await self.azure_cli.run_command(
             command,
@@ -830,18 +831,10 @@ class ContainerInstancePushProvisioner:
                     "The prefix must be alphanumeric and between 3-50 characters.",
                     style="red",
                 )
-        while True:
-            self._identity_name = prompt(
-                "Please enter a name for the identity (used for ACR access)",
-                default=self._identity_name,
-            )
-            if self._validate_user_input(self._identity_name):
-                break
-            else:
-                self._console.print(
-                    "The identity name must be alphanumeric and at least 3 characters.",
-                    style="red",
-                )
+        self._identity_name = prompt(
+            "Please enter a name for the identity (used for ACR access)",
+            default=self._identity_name,
+        )
         self._credentials_block_name = prompt(
             "Please enter a name for the ACI credentials block",
             default=self._credentials_block_name,

{prefect-client-2.14.11 → prefect-client-2.14.13}/src/prefect/input/run_input.py

@@ -56,9 +56,6 @@ class RunInput(pydantic.BaseModel):
     class Config:
         extra = "forbid"
 
-    title: str = "Run is asking for input"
-    description: Optional[str] = None
-
     @classmethod
     @sync_compatible
     async def save(cls, keyset: Keyset, flow_run_id: Optional[UUID] = None):

{prefect-client-2.14.11 → prefect-client-2.14.13}/src/prefect/runner/runner.py

@@ -86,7 +86,7 @@ from prefect.settings import (
 )
 from prefect.states import Crashed, Pending, exception_to_failed_state
 from prefect.utilities.asyncutils import is_async_fn, sync_compatible
-from prefect.utilities.processutils import run_process
+from prefect.utilities.processutils import _register_signal, run_process
 from prefect.utilities.services import critical_service_loop
 
 __all__ = ["Runner", "serve"]
@@ -337,7 +337,7 @@ class Runner:
                 runner.start()
             ```
         """
-        signal.signal(signal.SIGTERM, self.handle_sigterm)
+        _register_signal(signal.SIGTERM, self.handle_sigterm)
 
         webserver = webserver if webserver is not None else self.webserver