prefect-client 2.14.21__tar.gz → 2.15.0__tar.gz

{prefect-client-2.14.21 → prefect-client-2.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 2.14.21
+Version: 2.15.0
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.

{prefect-client-2.14.21 → prefect-client-2.15.0}/requirements-client.txt

@@ -6,6 +6,7 @@ coolname >= 1.0.4, < 3.0.0
 croniter >= 1.0.12, < 3.0.0
 fsspec >= 2022.5.0
 graphviz >= 0.20.1
+greenback >= 1.2.0
 griffe >= 0.20.0
 httpcore >=0.15.0, < 2.0.0
 httpx[http2] >= 0.23, != 0.23.2
@@ -26,9 +27,12 @@ pyyaml >= 5.4.1, < 7.0.0
 rich >= 11.0, < 14.0
 ruamel.yaml >= 0.17.0
 sniffio >=1.3.0, < 2.0.0
-starlette >= 0.27.0, < 0.33.0
 toml >= 0.10.0
 typing_extensions >= 4.5.0, < 5.0.0
 ujson >= 5.8.0, < 6.0.0
 uvicorn >= 0.14.0
 websockets >= 10.4, < 13.0
+
+# additional dependencies of starlette, which we're currently vendoring
+itsdangerous
+python-multipart>=0.0.7

{prefect-client-2.14.21 → prefect-client-2.15.0}/requirements.txt

@@ -10,6 +10,7 @@ dateparser >= 1.1.1, < 2.0.0
 docker >= 4.0, < 7.0
 graphviz >= 0.20.1
 griffe >= 0.20.0
+greenback >= 1.2.0
 jinja2 >= 3.0.0, < 4.0.0
 kubernetes >= 24.2.0, < 30.0.0
 pytz >= 2021.1, < 2025

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_internal/concurrency/api.py

@@ -16,8 +16,11 @@ from typing import (
     Union,
 )
 
+import greenback
 from typing_extensions import ParamSpec
 
+from prefect._internal.concurrency import logger
+from prefect._internal.concurrency.event_loop import get_running_loop
 from prefect._internal.concurrency.threads import (
     WorkerThread,
     get_global_loop,
@@ -179,7 +182,7 @@ class from_async(_base):
             for context in contexts or []:
                 stack.enter_context(context)
             await waiter.wait()
-            return call.result()
+            return await call.aresult()
 
     @staticmethod
     async def wait_for_call_in_new_thread(
@@ -193,7 +196,7 @@ class from_async(_base):
             waiter.add_done_callback(callback)
         _base.call_soon_in_new_thread(call, timeout=timeout)
         await waiter.wait()
-        return call.result()
+        return await call.aresult()
 
     @staticmethod
     def call_in_waiting_thread(
@@ -231,6 +234,25 @@ class from_sync(_base):
         done_callbacks: Optional[Iterable[Call]] = None,
         contexts: Optional[Iterable[ContextManager]] = None,
     ) -> Awaitable[T]:
+        # Check for a running event loop to prevent blocking
+        if get_running_loop():
+            if greenback.has_portal():
+                # Use greenback to avoid blocking the event loop while waiting
+                return greenback.await_(
+                    from_async.wait_for_call_in_loop_thread(
+                        __call,
+                        timeout=timeout,
+                        done_callbacks=done_callbacks,
+                        contexts=contexts,
+                    )
+                )
+            else:
+                logger.error(
+                    "Detected unsafe call to `from_sync` from thread with event loop. "
+                    "Use `await greenback.ensure_portal()` to allow call to run "
+                    "without blocking the event loop."
+                )
+
         call = _cast_to_call(__call)
         waiter = SyncWaiter(call)
         _base.call_soon_in_loop_thread(call, timeout=timeout)
@@ -262,6 +284,10 @@ class from_sync(_base):
         thread: threading.Thread,
         timeout: Optional[float] = None,
     ) -> T:
+        if get_running_loop():
+            raise RuntimeError(
+                "Detected unsafe call to `from_sync` from thread with event loop."
+            )
         call = _base.call_soon_in_waiting_thread(__call, thread, timeout=timeout)
         return call.result()
 
@@ -269,6 +295,10 @@ class from_sync(_base):
     def call_in_new_thread(
         __call: Union[Callable[[], T], Call[T]], timeout: Optional[float] = None
     ) -> T:
+        if get_running_loop():
+            raise RuntimeError(
+                "Detected unsafe call to `from_sync` from thread with event loop."
+            )
         call = _base.call_soon_in_new_thread(__call, timeout=timeout)
         return call.result()
 
@@ -283,5 +313,10 @@ class from_sync(_base):
             call = _cast_to_call(__call)
             return call()
 
+        if get_running_loop():
+            raise RuntimeError(
+                "Detected unsafe call to `from_sync` from thread with event loop."
+            )
+
         call = _base.call_soon_in_loop_thread(__call, timeout=timeout)
         return call.result()

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_internal/concurrency/calls.py

@@ -18,6 +18,7 @@ from concurrent.futures._base import (
 )
 from typing import Any, Awaitable, Callable, Dict, Generic, Optional, Tuple, TypeVar
 
+import greenback
 from typing_extensions import ParamSpec
 
 from prefect._internal.concurrency import logger
@@ -347,6 +348,14 @@ class Call(Generic[T]):
             logger.debug("Finished call %r", self)  # noqa: F821
 
     async def _run_async(self, coro):
+        from prefect._internal.concurrency.threads import in_global_loop
+
+        # Ensure the greenback portal is shimmed for this task so we can safely call
+        # back into async code from sync code if the user does something silly; avoid
+        # this overhead for our internal event loop
+        if not in_global_loop():
+            await greenback.ensure_portal()
+
         cancel_scope = None
         try:
             with set_current_call(self):

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_internal/concurrency/cancellation.py

@@ -230,7 +230,9 @@ class CancelScope(abc.ABC):
                 else (
                     "cancelled"
                     if self._cancelled
-                    else "running" if self._started else "pending"
+                    else "running"
+                    if self._started
+                    else "pending"
                 )
             ).upper()
             timeout = f", timeout={self._timeout:.2f}" if self._timeout else ""

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_internal/concurrency/event_loop.py

@@ -29,7 +29,7 @@ def call_in_loop(
     __loop: asyncio.AbstractEventLoop,
     __fn: Callable[P, T],
     *args: P.args,
-    **kwargs: P.kwargs
+    **kwargs: P.kwargs,
 ) -> T:
     """
     Run a synchronous call in event loop's thread from another thread.
@@ -49,7 +49,7 @@ def call_soon_in_loop(
     __loop: asyncio.AbstractEventLoop,
     __fn: Callable[P, T],
     *args: P.args,
-    **kwargs: P.kwargs
+    **kwargs: P.kwargs,
 ) -> concurrent.futures.Future:
     """
     Run a synchronous call in an event loop's thread from another thread.

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_internal/concurrency/threads.py

@@ -257,11 +257,12 @@ def in_global_loop() -> bool:
     """
     Check if called from the global loop.
     """
-    if GLOBAL_LOOP is None:
+    loop = get_running_loop()
+    if GLOBAL_LOOP is None or not loop:
         # Avoid creating a global loop if there isn't one
         return False
 
-    return get_global_loop()._loop == get_running_loop()
+    return GLOBAL_LOOP._loop == loop
 
 
 def wait_for_global_loop_exit(timeout: Optional[float] = None) -> None:

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_internal/pydantic/annotations/pendulum.py

@@ -13,9 +13,9 @@ from typing_extensions import Annotated
 
 
 class _PendulumDateTimeAnnotation:
-    _pendulum_type: t.Type[t.Union[pendulum.DateTime, pendulum.Date, pendulum.Time]] = (
-        pendulum.DateTime
-    )
+    _pendulum_type: t.Type[
+        t.Union[pendulum.DateTime, pendulum.Date, pendulum.Time]
+    ] = pendulum.DateTime
 
     _pendulum_types_to_schemas = {
         pendulum.DateTime: core_schema.datetime_schema(),
@@ -35,7 +35,7 @@ class _PendulumDateTimeAnnotation:
             return pendulum.parse(value)
 
         def to_str(
-            value: t.Union[pendulum.DateTime, pendulum.Date, pendulum.Time]
+            value: t.Union[pendulum.DateTime, pendulum.Date, pendulum.Time],
         ) -> str:
             return value.isoformat()
 

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_internal/pydantic/v2_schema.py

@@ -59,7 +59,7 @@ def process_v2_params(
     *,
     position: int,
     docstrings: t.Dict[str, str],
-    aliases: t.Dict
+    aliases: t.Dict,
 ) -> t.Tuple[str, t.Any, "pydantic.Field"]:
     """
     Generate a sanitized name, type, and pydantic.Field for a given parameter.
@@ -98,7 +98,7 @@ def create_v2_schema(
     name_: str,
     model_cfg: t.Optional[ConfigDict] = None,
     model_base: t.Optional[t.Type[V2BaseModel]] = None,
-    **model_fields
+    **model_fields,
 ):
     """
     Create a pydantic v2 model and craft a v1 compatible schema from it.

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/__init__.py

@@ -2,7 +2,7 @@
 
 __version__ = "0.99.1"
 
-from starlette import status as status
+from prefect._vendor.starlette import status as status
 
 from .applications import FastAPI as FastAPI
 from .background import BackgroundTasks as BackgroundTasks

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/applications.py

@@ -36,17 +36,17 @@ from prefect._vendor.fastapi.openapi.utils import get_openapi
 from prefect._vendor.fastapi.params import Depends
 from prefect._vendor.fastapi.types import DecoratedCallable
 from prefect._vendor.fastapi.utils import generate_unique_id
-from starlette.applications import Starlette
-from starlette.datastructures import State
-from starlette.exceptions import HTTPException
-from starlette.middleware import Middleware
-from starlette.middleware.base import BaseHTTPMiddleware
-from starlette.middleware.errors import ServerErrorMiddleware
-from starlette.middleware.exceptions import ExceptionMiddleware
-from starlette.requests import Request
-from starlette.responses import HTMLResponse, JSONResponse, Response
-from starlette.routing import BaseRoute
-from starlette.types import ASGIApp, Lifespan, Receive, Scope, Send
+from prefect._vendor.starlette.applications import Starlette
+from prefect._vendor.starlette.datastructures import State
+from prefect._vendor.starlette.exceptions import HTTPException
+from prefect._vendor.starlette.middleware import Middleware
+from prefect._vendor.starlette.middleware.base import BaseHTTPMiddleware
+from prefect._vendor.starlette.middleware.errors import ServerErrorMiddleware
+from prefect._vendor.starlette.middleware.exceptions import ExceptionMiddleware
+from prefect._vendor.starlette.requests import Request
+from prefect._vendor.starlette.responses import HTMLResponse, JSONResponse, Response
+from prefect._vendor.starlette.routing import BaseRoute
+from prefect._vendor.starlette.types import ASGIApp, Lifespan, Receive, Scope, Send
 
 AppType = TypeVar("AppType", bound="FastAPI")
 
@@ -150,7 +150,7 @@ class FastAPI(Starlette):
         )
         self.exception_handlers: Dict[
             Any, Callable[[Request, Any], Union[Response, Awaitable[Response]]]
-        ] = ({} if exception_handlers is None else dict(exception_handlers))
+        ] = {} if exception_handlers is None else dict(exception_handlers)
         self.exception_handlers.setdefault(HTTPException, http_exception_handler)
         self.exception_handlers.setdefault(
             RequestValidationError, request_validation_exception_handler
@@ -168,7 +168,7 @@ class FastAPI(Starlette):
         self.setup()
 
     def build_middleware_stack(self) -> ASGIApp:
-        # Duplicate/override from Starlette to add AsyncExitStackMiddleware
+        # Duplicate/override from prefect._vendor.starlette to add AsyncExitStackMiddleware
         # inside of ExceptionMiddleware, inside of custom user middlewares
         debug = self.debug
         error_handler = None

prefect-client-2.15.0/src/prefect/_vendor/fastapi/background.py

@@ -0,0 +1,3 @@
+from prefect._vendor.starlette.background import (
+    BackgroundTasks as BackgroundTasks,
+)

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/concurrency.py

@@ -4,9 +4,13 @@ from typing import AsyncGenerator, ContextManager, TypeVar
 
 import anyio
 from anyio import CapacityLimiter
-from starlette.concurrency import iterate_in_threadpool as iterate_in_threadpool  # noqa
-from starlette.concurrency import run_in_threadpool as run_in_threadpool  # noqa
-from starlette.concurrency import (  # noqa
+from prefect._vendor.starlette.concurrency import (
+    iterate_in_threadpool as iterate_in_threadpool,
+)  # noqa
+from prefect._vendor.starlette.concurrency import (
+    run_in_threadpool as run_in_threadpool,
+)  # noqa
+from prefect._vendor.starlette.concurrency import (  # noqa
     run_until_first_complete as run_until_first_complete,
 )
 

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/datastructures.py

@@ -1,12 +1,14 @@
 from typing import Any, Callable, Dict, Iterable, Type, TypeVar
 
-from starlette.datastructures import URL as URL  # noqa: F401
-from starlette.datastructures import Address as Address  # noqa: F401
-from starlette.datastructures import FormData as FormData  # noqa: F401
-from starlette.datastructures import Headers as Headers  # noqa: F401
-from starlette.datastructures import QueryParams as QueryParams  # noqa: F401
-from starlette.datastructures import State as State  # noqa: F401
-from starlette.datastructures import UploadFile as StarletteUploadFile
+from prefect._vendor.starlette.datastructures import URL as URL  # noqa: F401
+from prefect._vendor.starlette.datastructures import Address as Address  # noqa: F401
+from prefect._vendor.starlette.datastructures import FormData as FormData  # noqa: F401
+from prefect._vendor.starlette.datastructures import Headers as Headers  # noqa: F401
+from prefect._vendor.starlette.datastructures import (
+    QueryParams as QueryParams,  # noqa: F401
+)
+from prefect._vendor.starlette.datastructures import State as State  # noqa: F401
+from prefect._vendor.starlette.datastructures import UploadFile as StarletteUploadFile
 
 
 class UploadFile(StarletteUploadFile):

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/dependencies/utils.py

@@ -75,12 +75,17 @@ else:
     from pydantic.typing import evaluate_forwardref, get_args, get_origin
     from pydantic.utils import lenient_issubclass
 
-from starlette.background import BackgroundTasks
-from starlette.concurrency import run_in_threadpool
-from starlette.datastructures import FormData, Headers, QueryParams, UploadFile
-from starlette.requests import HTTPConnection, Request
-from starlette.responses import Response
-from starlette.websockets import WebSocket
+from prefect._vendor.starlette.background import BackgroundTasks
+from prefect._vendor.starlette.concurrency import run_in_threadpool
+from prefect._vendor.starlette.datastructures import (
+    FormData,
+    Headers,
+    QueryParams,
+    UploadFile,
+)
+from prefect._vendor.starlette.requests import HTTPConnection, Request
+from prefect._vendor.starlette.responses import Response
+from prefect._vendor.starlette.websockets import WebSocket
 from typing_extensions import Annotated
 
 sequence_shapes = {
@@ -798,7 +803,7 @@ async def request_body_to_args(
                 results: List[Union[bytes, str]] = []
 
                 async def process_fn(
-                    fn: Callable[[], Coroutine[Any, Any, Any]]
+                    fn: Callable[[], Coroutine[Any, Any, Any]],
                 ) -> None:
                     result = await fn()
                     results.append(result)  # noqa: B023

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/encoders.py

@@ -19,7 +19,7 @@ DictIntStrAny = Dict[Union[int, str], Any]
 
 
 def generate_encoders_by_class_tuples(
-    type_encoder_map: Dict[Any, Callable[[Any], Any]]
+    type_encoder_map: Dict[Any, Callable[[Any], Any]],
 ) -> Dict[Callable[[Any], Any], Tuple[Any, ...]]:
     encoders_by_class_tuples: Dict[Callable[[Any], Any], Tuple[Any, ...]] = defaultdict(
         tuple

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/exception_handlers.py

@@ -5,10 +5,13 @@ from prefect._vendor.fastapi.exceptions import (
 )
 from prefect._vendor.fastapi.utils import is_body_allowed_for_status_code
 from prefect._vendor.fastapi.websockets import WebSocket
-from starlette.exceptions import HTTPException
-from starlette.requests import Request
-from starlette.responses import JSONResponse, Response
-from starlette.status import HTTP_422_UNPROCESSABLE_ENTITY, WS_1008_POLICY_VIOLATION
+from prefect._vendor.starlette.exceptions import HTTPException
+from prefect._vendor.starlette.requests import Request
+from prefect._vendor.starlette.responses import JSONResponse, Response
+from prefect._vendor.starlette.status import (
+    HTTP_422_UNPROCESSABLE_ENTITY,
+    WS_1008_POLICY_VIOLATION,
+)
 
 
 async def http_exception_handler(request: Request, exc: HTTPException) -> Response:

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/exceptions.py

@@ -9,8 +9,10 @@ else:
     from pydantic import BaseModel, ValidationError, create_model
     from pydantic.error_wrappers import ErrorList
 
-from starlette.exceptions import HTTPException as StarletteHTTPException
-from starlette.exceptions import WebSocketException as WebSocketException  # noqa: F401
+from prefect._vendor.starlette.exceptions import HTTPException as StarletteHTTPException
+from prefect._vendor.starlette.exceptions import (
+    WebSocketException as WebSocketException,  # noqa: F401
+)
 
 
 class HTTPException(StarletteHTTPException):

prefect-client-2.15.0/src/prefect/_vendor/fastapi/middleware/__init__.py

@@ -0,0 +1 @@
+from prefect._vendor.starlette.middleware import Middleware as Middleware

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/middleware/asyncexitstack.py

@@ -1,7 +1,7 @@
 from typing import Optional
 
 from prefect._vendor.fastapi.concurrency import AsyncExitStack
-from starlette.types import ASGIApp, Receive, Scope, Send
+from prefect._vendor.starlette.types import ASGIApp, Receive, Scope, Send
 
 
 class AsyncExitStackMiddleware:

prefect-client-2.15.0/src/prefect/_vendor/fastapi/middleware/cors.py

@@ -0,0 +1,3 @@
+from prefect._vendor.starlette.middleware.cors import (
+    CORSMiddleware as CORSMiddleware,
+)

prefect-client-2.15.0/src/prefect/_vendor/fastapi/middleware/gzip.py

@@ -0,0 +1,3 @@
+from prefect._vendor.starlette.middleware.gzip import (
+    GZipMiddleware as GZipMiddleware,
+)

prefect-client-2.15.0/src/prefect/_vendor/fastapi/middleware/httpsredirect.py

@@ -0,0 +1,3 @@
+from prefect._vendor.starlette.middleware.httpsredirect import (  # noqa
+    HTTPSRedirectMiddleware as HTTPSRedirectMiddleware,
+)

prefect-client-2.15.0/src/prefect/_vendor/fastapi/middleware/trustedhost.py

@@ -0,0 +1,3 @@
+from prefect._vendor.starlette.middleware.trustedhost import (  # noqa
+    TrustedHostMiddleware as TrustedHostMiddleware,
+)

prefect-client-2.15.0/src/prefect/_vendor/fastapi/middleware/wsgi.py

@@ -0,0 +1,3 @@
+from prefect._vendor.starlette.middleware.wsgi import (
+    WSGIMiddleware as WSGIMiddleware,
+)

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/openapi/docs.py

@@ -2,7 +2,7 @@ import json
 from typing import Any, Dict, Optional
 
 from prefect._vendor.fastapi.encoders import jsonable_encoder
-from starlette.responses import HTMLResponse
+from prefect._vendor.starlette.responses import HTMLResponse
 
 swagger_ui_default_parameters = {
     "dom_id": "#swagger-ui",

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/openapi/utils.py

@@ -44,9 +44,9 @@ else:
     )
     from pydantic.utils import lenient_issubclass
 
-from starlette.responses import JSONResponse
-from starlette.routing import BaseRoute
-from starlette.status import HTTP_422_UNPROCESSABLE_ENTITY
+from prefect._vendor.starlette.responses import JSONResponse
+from prefect._vendor.starlette.routing import BaseRoute
+from prefect._vendor.starlette.status import HTTP_422_UNPROCESSABLE_ENTITY
 
 validation_error_definition = {
     "title": "ValidationError",

prefect-client-2.15.0/src/prefect/_vendor/fastapi/requests.py

@@ -0,0 +1,4 @@
+from prefect._vendor.starlette.requests import (
+    HTTPConnection as HTTPConnection,  # noqa: F401
+)
+from prefect._vendor.starlette.requests import Request as Request  # noqa: F401

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/responses.py

@@ -1,12 +1,18 @@
 from typing import Any
 
-from starlette.responses import FileResponse as FileResponse  # noqa
-from starlette.responses import HTMLResponse as HTMLResponse  # noqa
-from starlette.responses import JSONResponse as JSONResponse  # noqa
-from starlette.responses import PlainTextResponse as PlainTextResponse  # noqa
-from starlette.responses import RedirectResponse as RedirectResponse  # noqa
-from starlette.responses import Response as Response  # noqa
-from starlette.responses import StreamingResponse as StreamingResponse  # noqa
+from prefect._vendor.starlette.responses import FileResponse as FileResponse  # noqa
+from prefect._vendor.starlette.responses import HTMLResponse as HTMLResponse  # noqa
+from prefect._vendor.starlette.responses import JSONResponse as JSONResponse  # noqa
+from prefect._vendor.starlette.responses import (
+    PlainTextResponse as PlainTextResponse,  # noqa
+)
+from prefect._vendor.starlette.responses import (
+    RedirectResponse as RedirectResponse,  # noqa
+)
+from prefect._vendor.starlette.responses import Response as Response  # noqa
+from prefect._vendor.starlette.responses import (
+    StreamingResponse as StreamingResponse,  # noqa
+)
 
 try:
     import ujson

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/routing.py

@@ -57,12 +57,12 @@ else:
     from pydantic.fields import ModelField, Undefined
     from pydantic.utils import lenient_issubclass
 
-from starlette import routing
-from starlette.concurrency import run_in_threadpool
-from starlette.exceptions import HTTPException
-from starlette.requests import Request
-from starlette.responses import JSONResponse, Response
-from starlette.routing import (
+from prefect._vendor.starlette import routing
+from prefect._vendor.starlette.concurrency import run_in_threadpool
+from prefect._vendor.starlette.exceptions import HTTPException
+from prefect._vendor.starlette.requests import Request
+from prefect._vendor.starlette.responses import JSONResponse, Response
+from prefect._vendor.starlette.routing import (
     BaseRoute,
     Match,
     compile_path,
@@ -70,9 +70,9 @@ from starlette.routing import (
     request_response,
     websocket_session,
 )
-from starlette.routing import Mount as Mount  # noqa
-from starlette.types import ASGIApp, Lifespan, Scope
-from starlette.websockets import WebSocket
+from prefect._vendor.starlette.routing import Mount as Mount  # noqa
+from prefect._vendor.starlette.types import ASGIApp, Lifespan, Scope
+from prefect._vendor.starlette.websockets import WebSocket
 
 
 def _prepare_response_content(
@@ -408,9 +408,9 @@ class APIRoute(routing.Route):
             methods = ["GET"]
         self.methods: Set[str] = {method.upper() for method in methods}
         if isinstance(generate_unique_id_function, DefaultPlaceholder):
-            current_generate_unique_id: Callable[["APIRoute"], str] = (
-                generate_unique_id_function.value
-            )
+            current_generate_unique_id: Callable[
+                ["APIRoute"], str
+            ] = generate_unique_id_function.value
         else:
             current_generate_unique_id = generate_unique_id_function
         self.unique_id = self.operation_id or current_generate_unique_id(self)
@@ -433,9 +433,9 @@ class APIRoute(routing.Route):
             # would pass the validation and be returned as is.
             # By being a new field, no inheritance will be passed as is. A new model
             # will be always created.
-            self.secure_cloned_response_field: Optional[ModelField] = (
-                create_cloned_field(self.response_field)
-            )
+            self.secure_cloned_response_field: Optional[
+                ModelField
+            ] = create_cloned_field(self.response_field)
         else:
             self.response_field = None  # type: ignore
             self.secure_cloned_response_field = None

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/security/api_key.py

@@ -2,9 +2,9 @@ from typing import Optional
 
 from prefect._vendor.fastapi.openapi.models import APIKey, APIKeyIn
 from prefect._vendor.fastapi.security.base import SecurityBase
-from starlette.exceptions import HTTPException
-from starlette.requests import Request
-from starlette.status import HTTP_403_FORBIDDEN
+from prefect._vendor.starlette.exceptions import HTTPException
+from prefect._vendor.starlette.requests import Request
+from prefect._vendor.starlette.status import HTTP_403_FORBIDDEN
 
 
 class APIKeyBase(SecurityBase):

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/security/http.py

@@ -15,8 +15,8 @@ if HAS_PYDANTIC_V2:
 else:
     from pydantic import BaseModel
 
-from starlette.requests import Request
-from starlette.status import HTTP_401_UNAUTHORIZED, HTTP_403_FORBIDDEN
+from prefect._vendor.starlette.requests import Request
+from prefect._vendor.starlette.status import HTTP_401_UNAUTHORIZED, HTTP_403_FORBIDDEN
 
 
 class HTTPBasicCredentials(BaseModel):

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/security/oauth2.py

@@ -6,8 +6,8 @@ from prefect._vendor.fastapi.openapi.models import OAuthFlows as OAuthFlowsModel
 from prefect._vendor.fastapi.param_functions import Form
 from prefect._vendor.fastapi.security.base import SecurityBase
 from prefect._vendor.fastapi.security.utils import get_authorization_scheme_param
-from starlette.requests import Request
-from starlette.status import HTTP_401_UNAUTHORIZED, HTTP_403_FORBIDDEN
+from prefect._vendor.starlette.requests import Request
+from prefect._vendor.starlette.status import HTTP_401_UNAUTHORIZED, HTTP_403_FORBIDDEN
 
 
 class OAuth2PasswordRequestForm:

{prefect-client-2.14.21 → prefect-client-2.15.0}/src/prefect/_vendor/fastapi/security/open_id_connect_url.py

@@ -2,9 +2,9 @@ from typing import Optional
 
 from prefect._vendor.fastapi.openapi.models import OpenIdConnect as OpenIdConnectModel
 from prefect._vendor.fastapi.security.base import SecurityBase
-from starlette.exceptions import HTTPException
-from starlette.requests import Request
-from starlette.status import HTTP_403_FORBIDDEN
+from prefect._vendor.starlette.exceptions import HTTPException
+from prefect._vendor.starlette.requests import Request
+from prefect._vendor.starlette.status import HTTP_403_FORBIDDEN
 
 
 class OpenIdConnect(SecurityBase):

prefect-client-2.15.0/src/prefect/_vendor/fastapi/staticfiles.py

@@ -0,0 +1 @@
+from prefect._vendor.starlette.staticfiles import StaticFiles as StaticFiles  # noqa

prefect-client-2.15.0/src/prefect/_vendor/fastapi/templating.py

@@ -0,0 +1,3 @@
+from prefect._vendor.starlette.templating import (
+    Jinja2Templates as Jinja2Templates,
+)

prefect-client-2.15.0/src/prefect/_vendor/fastapi/testclient.py

@@ -0,0 +1 @@
+from prefect._vendor.starlette.testclient import TestClient as TestClient  # noqa