prefect 3.4.23.dev3__tar.gz → 3.4.24.dev2__tar.gz

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/PKG-INFO

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

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/pyproject.toml

@@ -158,7 +158,7 @@ dev = [
     "watchfiles>=1.0.4",
     "respx>=0.22.0",
     "codespell==2.4.1",
-    "ruff==0.13.2",
+    "ruff==0.14.0",
     "mypy==1.15.0",
     "types-cachetools>=5.3.1",
     "types-PyYAML>=6.0.12",

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

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.4.24.dev2"
+__build_date__ = "2025-10-13 08:09:52.505435+00:00"
+__git_commit__ = "2c75d8cd6ca1f17e98e98c7335c17201c6343c43"
+__dirty__ = False

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/cli/__init__.py

@@ -4,6 +4,7 @@ from prefect.cli.root import app
 # Import CLI submodules to register them to the app
 # isort: split
 
+import prefect.cli.api
 import prefect.cli.artifact
 import prefect.cli.block
 import prefect.cli.cloud

prefect-3.4.24.dev2/src/prefect/cli/api.py

@@ -0,0 +1,277 @@
+"""
+Command line interface for making direct API requests.
+"""
+
+from __future__ import annotations
+
+import json
+import sys
+from pathlib import Path
+from typing import Any, Optional
+
+import httpx
+import typer
+from rich.console import Console
+from rich.syntax import Syntax
+
+from prefect.cli._utilities import exit_with_error
+from prefect.cli.root import app
+from prefect.client.cloud import get_cloud_client
+from prefect.client.orchestration import get_client
+from prefect.settings import get_current_settings
+
+console: Console = Console()
+console_err: Console = Console(stderr=True)
+
+
+def parse_headers(header_list: list[str]) -> dict[str, str]:
+    """Parse header strings in format 'Key: Value' into a dict."""
+    headers = {}
+    for header in header_list:
+        if ":" not in header:
+            exit_with_error(
+                f"Invalid header format: {header!r}. Use 'Key: Value' format."
+            )
+        key, value = header.split(":", 1)
+        headers[key.strip()] = value.strip()
+    return headers
+
+
+def parse_data(data: str | None) -> dict[str, Any] | str | None:
+    """Parse data input - can be JSON string, @filename, or None."""
+    if data is None:
+        return None
+
+    if data.startswith("@"):
+        filepath = Path(data[1:])
+        if not filepath.exists():
+            exit_with_error(f"File not found: {filepath}")
+        try:
+            content = filepath.read_text()
+            return json.loads(content)
+        except json.JSONDecodeError as e:
+            exit_with_error(f"Invalid JSON in file {filepath}: {e}")
+    else:
+        try:
+            return json.loads(data)
+        except json.JSONDecodeError as e:
+            exit_with_error(f"Invalid JSON data: {e}")
+
+
+def read_stdin_data() -> dict[str, Any] | None:
+    """Read and parse JSON data from stdin if available."""
+    if not sys.stdin.isatty():
+        try:
+            content = sys.stdin.read()
+            if content.strip():
+                return json.loads(content)
+        except json.JSONDecodeError as e:
+            exit_with_error(f"Invalid JSON from stdin: {e}")
+    return None
+
+
+def format_output(response: httpx.Response, verbose: bool) -> None:
+    """Format and print the response using rich."""
+    if verbose:
+        req = response.request
+        console.print(f"[dim]> {req.method} {req.url}[/dim]")
+        for key, value in req.headers.items():
+            if key.lower() == "authorization":
+                value = "Bearer ***"
+            console.print(f"[dim]> {key}: {value}[/dim]")
+        console.print()
+
+        console.print(f"[dim]< {response.status_code} {response.reason_phrase}[/dim]")
+        for key, value in response.headers.items():
+            console.print(f"[dim]< {key}: {value}[/dim]")
+        console.print()
+
+    if response.text:
+        try:
+            data = response.json()
+            json_str = json.dumps(data, indent=2)
+            if sys.stdout.isatty():
+                syntax = Syntax(json_str, "json", theme="monokai", word_wrap=True)
+                console.print(syntax)
+            else:
+                print(json.dumps(data))
+        except (json.JSONDecodeError, ValueError):
+            console.print(response.text)
+
+
+def get_exit_code(error: Exception) -> int:
+    """Determine the appropriate exit code for an error."""
+    if isinstance(error, httpx.HTTPStatusError):
+        status = error.response.status_code
+        if status in (401, 403):
+            return 3
+        elif 400 <= status < 500:
+            return 4
+        elif 500 <= status < 600:
+            return 5
+    elif isinstance(
+        error, (httpx.ConnectError, httpx.TimeoutException, httpx.NetworkError)
+    ):
+        return 7
+
+    return 1
+
+
+@app.command(name="api")
+async def api_request(
+    method: str = typer.Argument(
+        ..., help="HTTP method (GET, POST, PUT, PATCH, DELETE)"
+    ),
+    path: str = typer.Argument(..., help="API path (e.g., /flows, /flows/filter)"),
+    data: Optional[str] = typer.Option(
+        None,
+        "--data",
+        help="Request body as JSON string or @filename",
+    ),
+    headers: list[str] = typer.Option(
+        None,
+        "-H",
+        "--header",
+        help="Custom header in 'Key: Value' format",
+    ),
+    verbose: bool = typer.Option(
+        False,
+        "-v",
+        "--verbose",
+        help="Show request/response headers",
+    ),
+    root: bool = typer.Option(
+        False,
+        "--root",
+        help="Access API root level (e.g., /api/me)",
+    ),
+    account: bool = typer.Option(
+        False,
+        "--account",
+        help="Access account level (Cloud only)",
+    ),
+):
+    """
+    Make a direct request to the Prefect API.
+
+    Examples:
+        ```bash
+        # GET request
+        $ prefect api GET /flows/abc-123
+
+        # POST request with data
+        $ prefect api POST /flows/filter --data '{"limit": 10}'
+
+        # POST to filter endpoint (defaults to empty object)
+        $ prefect api POST /flows/filter
+
+        # Custom headers
+        $ prefect api POST /flows/filter -H "X-Custom: value" --data '{}'
+
+        # Verbose output
+        $ prefect api GET /flows --verbose
+
+        # Account-level operation (Cloud)
+        $ prefect api GET /workspaces --account
+
+        # API root level (Cloud only)
+        $ prefect api GET /me --root
+        ```
+    """
+    if headers is None:
+        headers = []
+
+    http_methods = {"GET", "POST", "PUT", "PATCH", "DELETE", "HEAD", "OPTIONS"}
+    http_method = method.upper()
+
+    if http_method not in http_methods:
+        exit_with_error(
+            f"Invalid HTTP method: {method!r}. "
+            f"Must be one of: {', '.join(sorted(http_methods))}"
+        )
+
+    settings = get_current_settings()
+    configured_api_url = settings.api.url
+    if not configured_api_url:
+        exit_with_error(
+            "No API URL configured. Set PREFECT_API_URL or run 'prefect cloud login'."
+        )
+
+    cloud_api_url = settings.cloud.api_url
+    is_cloud = cloud_api_url and configured_api_url.startswith(cloud_api_url)
+
+    if (root or account) and not is_cloud:
+        exit_with_error(
+            "--root and --account flags are only valid for Prefect Cloud. "
+            "For self-hosted servers, paths are relative to your configured API URL."
+        )
+
+    body_data = parse_data(data)
+
+    if body_data is None and data is None:
+        stdin_data = read_stdin_data()
+        if stdin_data is not None:
+            body_data = stdin_data
+        elif http_method in ("POST", "PUT", "PATCH"):
+            body_data = {}
+
+    custom_headers = parse_headers(headers)
+
+    try:
+        if root or account:
+            async with get_cloud_client() as client:
+                if account:
+                    route = f"{client.account_base_url}{path}"
+                else:
+                    route = path
+
+                response = await client.raw_request(
+                    http_method,
+                    route,
+                    json=body_data if body_data is not None else None,
+                    headers=custom_headers if custom_headers else None,
+                )
+                response.raise_for_status()
+        else:
+            async with get_client() as client:
+                response = await client.request(
+                    http_method,
+                    path,
+                    json=body_data if body_data is not None else None,
+                    headers=custom_headers if custom_headers else None,
+                )
+                response.raise_for_status()
+
+        format_output(response, verbose)
+
+    except httpx.HTTPStatusError as e:
+        if verbose:
+            format_output(e.response, verbose)
+        else:
+            console_err.print(
+                f"[red]Error: {e.response.status_code} {e.response.reason_phrase}[/red]"
+            )
+            if e.response.text:
+                try:
+                    error_data = e.response.json()
+                    json_str = json.dumps(error_data, indent=2)
+                    syntax = Syntax(json_str, "json", theme="monokai")
+                    console_err.print(syntax)
+                except (json.JSONDecodeError, ValueError):
+                    console_err.print(e.response.text)
+
+            if e.response.status_code == 401:
+                console_err.print(
+                    "\n[yellow]Check your API key configuration with:[/yellow] prefect config view"
+                )
+
+        raise typer.Exit(get_exit_code(e))
+
+    except (httpx.ConnectError, httpx.TimeoutException, httpx.NetworkError) as e:
+        console_err.print(f"[red]Error: Network error - {e}[/red]")
+        console_err.print(f"\nCould not connect to API at: {configured_api_url}")
+        raise typer.Exit(7)
+
+    except Exception as e:
+        console_err.print(f"[red]Error: {e}[/red]")
+        raise typer.Exit(1)

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/client/cloud.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import re
 from typing import Any, NoReturn, Optional, cast
 
@@ -186,6 +188,35 @@ class CloudClient:
     async def get(self, route: str, **kwargs: Any) -> Any:
         return await self.request("GET", route, **kwargs)
 
+    async def raw_request(
+        self,
+        method: str,
+        path: str,
+        params: dict[str, Any] | None = None,
+        path_params: dict[str, Any] | None = None,
+        **kwargs: Any,
+    ) -> httpx.Response:
+        """
+        Make a raw HTTP request and return the Response object.
+
+        Unlike request(), this does not parse JSON or raise special exceptions,
+        returning the raw httpx.Response for direct access to headers, status, etc.
+
+        Args:
+            method: HTTP method (GET, POST, etc.)
+            path: API path/route
+            params: Query parameters
+            path_params: Path parameters for formatting
+            **kwargs: Additional arguments passed to httpx (json, headers, etc.)
+
+        Returns:
+            Raw httpx.Response object
+        """
+        if path_params:
+            path = path.format(**path_params)
+        request = self._client.build_request(method, path, params=params, **kwargs)
+        return await self._client.send(request)
+
     async def request(self, method: str, route: str, **kwargs: Any) -> Any:
         try:
             res = await self._client.request(method, route, **kwargs)

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/client/orchestration/__init__.py

@@ -314,19 +314,21 @@ class PrefectClient(
         httpx_settings = httpx_settings.copy() if httpx_settings else {}
         httpx_settings.setdefault("headers", {})
 
-        if PREFECT_API_TLS_INSECURE_SKIP_VERIFY:
-            # Create an unverified context for insecure connections
-            ctx = ssl.create_default_context()
-            ctx.check_hostname = False
-            ctx.verify_mode = ssl.CERT_NONE
-            httpx_settings.setdefault("verify", ctx)
-        else:
-            cert_file = PREFECT_API_SSL_CERT_FILE.value()
-            if not cert_file:
-                cert_file = certifi.where()
-            # Create a verified context with the certificate file
-            ctx = ssl.create_default_context(cafile=cert_file)
-            httpx_settings.setdefault("verify", ctx)
+        tls_verify = httpx_settings.get("verify")
+        if not tls_verify or not isinstance(tls_verify, ssl.SSLContext):
+            if PREFECT_API_TLS_INSECURE_SKIP_VERIFY:
+                # Create an unverified context for insecure connections
+                ctx = ssl.create_default_context()
+                ctx.check_hostname = False
+                ctx.verify_mode = ssl.CERT_NONE
+                httpx_settings.setdefault("verify", ctx)
+            else:
+                cert_file = PREFECT_API_SSL_CERT_FILE.value()
+                if not cert_file:
+                    cert_file = certifi.where()
+                # Create a verified context with the certificate file
+                ctx = ssl.create_default_context(cafile=cert_file)
+                httpx_settings.setdefault("verify", ctx)
 
         if api_version is None:
             api_version = SERVER_API_VERSION
@@ -1177,19 +1179,21 @@ class SyncPrefectClient(
         httpx_settings = httpx_settings.copy() if httpx_settings else {}
         httpx_settings.setdefault("headers", {})
 
-        if PREFECT_API_TLS_INSECURE_SKIP_VERIFY:
-            # Create an unverified context for insecure connections
-            ctx = ssl.create_default_context()
-            ctx.check_hostname = False
-            ctx.verify_mode = ssl.CERT_NONE
-            httpx_settings.setdefault("verify", ctx)
-        else:
-            cert_file = PREFECT_API_SSL_CERT_FILE.value()
-            if not cert_file:
-                cert_file = certifi.where()
-            # Create a verified context with the certificate file
-            ctx = ssl.create_default_context(cafile=cert_file)
-            httpx_settings.setdefault("verify", ctx)
+        tls_verify = httpx_settings.get("verify")
+        if not tls_verify or not isinstance(tls_verify, ssl.SSLContext):
+            if PREFECT_API_TLS_INSECURE_SKIP_VERIFY:
+                # Create an unverified context for insecure connections
+                ctx = ssl.create_default_context()
+                ctx.check_hostname = False
+                ctx.verify_mode = ssl.CERT_NONE
+                httpx_settings.setdefault("verify", ctx)
+            else:
+                cert_file = PREFECT_API_SSL_CERT_FILE.value()
+                if not cert_file:
+                    cert_file = certifi.where()
+                # Create a verified context with the certificate file
+                ctx = ssl.create_default_context(cafile=cert_file)
+                httpx_settings.setdefault("verify", ctx)
 
         if api_version is None:
             api_version = SERVER_API_VERSION

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/server/events/services/actions.py

@@ -51,5 +51,6 @@ class Actions(RunInEphemeralServers, Service):
         except asyncio.CancelledError:
             pass
         finally:
+            await self.consumer.cleanup()
             self.consumer_task = None
         logger.debug("Actions stopped")

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/server/events/services/event_logger.py

@@ -71,5 +71,6 @@ class EventLogger(RunInEphemeralServers, Service):
         except asyncio.CancelledError:
             pass
         finally:
+            await self.consumer.cleanup()
             self.consumer_task = None
         logger.debug("Event logger stopped")

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/server/events/services/event_persister.py

@@ -134,6 +134,7 @@ class EventPersister(RunInEphemeralServers, Service):
         except asyncio.CancelledError:
             pass
         finally:
+            await self.consumer.cleanup()
             self.consumer_task = None
             if self.started_event:
                 self.started_event.clear()

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/server/events/services/triggers.py

@@ -57,6 +57,7 @@ class ReactiveTriggers(RunInEphemeralServers, Service):
         except asyncio.CancelledError:
             pass
         finally:
+            await self.consumer.cleanup()
             self.consumer_task = None
         logger.debug("Reactive triggers stopped")
 

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/server/schemas/schedules.py

@@ -497,7 +497,7 @@ class RRuleSchedule(PrefectBaseModel):
     model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
 
     rrule: str
-    timezone: Optional[TimeZone] = Field(default="UTC", examples=["America/New_York"])
+    timezone: Optional[TimeZone] = "UTC"
 
     @field_validator("rrule")
     @classmethod

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/server/utilities/messaging/memory.py

@@ -110,8 +110,8 @@ class Subscription:
             if dead_letter_queue_path
             else get_current_settings().home / "dlq"
         )
-        self._queue: asyncio.Queue[MemoryMessage] = asyncio.Queue()
-        self._retry: asyncio.Queue[MemoryMessage] = asyncio.Queue()
+        self._queue: asyncio.Queue[MemoryMessage] = asyncio.Queue(maxsize=10000)
+        self._retry: asyncio.Queue[MemoryMessage] = asyncio.Queue(maxsize=1000)
 
     async def deliver(self, message: MemoryMessage) -> None:
         """
@@ -120,14 +120,22 @@ class Subscription:
         Args:
             message: The message to deliver.
         """
-        await self._queue.put(message)
-        await update_metric(self.topic.name, "published")
-        logger.debug(
-            "Delivered message to topic=%r queue_size=%d retry_queue_size=%d",
-            self.topic.name,
-            self._queue.qsize(),
-            self._retry.qsize(),
-        )
+        try:
+            self._queue.put_nowait(message)
+            await update_metric(self.topic.name, "published")
+            logger.debug(
+                "Delivered message to topic=%r queue_size=%d retry_queue_size=%d",
+                self.topic.name,
+                self._queue.qsize(),
+                self._retry.qsize(),
+            )
+        except asyncio.QueueFull:
+            logger.warning(
+                "Subscription queue is full, dropping message for topic=%r queue_size=%d retry_queue_size=%d",
+                self.topic.name,
+                self._queue.qsize(),
+                self._retry.qsize(),
+            )
 
     async def retry(self, message: MemoryMessage) -> None:
         """
@@ -350,6 +358,16 @@ class Consumer(_Consumer):
             # Re-raise if any non-StopConsumer exceptions
             raise group
 
+    async def cleanup(self) -> None:
+        """
+        Cleanup resources by unsubscribing from the topic.
+
+        This should be called when the consumer is no longer needed to prevent
+        memory leaks from orphaned subscriptions.
+        """
+        self.topic.unsubscribe(self.subscription)
+        logger.debug("Unsubscribed from topic=%r", self.topic.name)
+
     async def _consume_loop(self, handler: MessageHandler) -> None:
         while True:
             message = await self.subscription.get()

{prefect-3.4.23.dev3 → prefect-3.4.24.dev2}/src/prefect/types/__init__.py

@@ -53,10 +53,10 @@ NonNegativeTimeDelta = Annotated[
 TimeZone = Annotated[
     str,
     Field(
-        default="UTC",
         pattern="|".join(
             [z for z in sorted(available_timezones()) if "localtime" not in z]
         ),
+        examples=["America/New_York"],
     ),
 ]
 

prefect-3.4.23.dev3/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.4.23.dev3"
-__build_date__ = "2025-10-09 08:09:15.700743+00:00"
-__git_commit__ = "2bff0eb2206fb852d380539365ffdf6d680b18db"
-__dirty__ = False