prefect-client 3.4.8.dev6__tar.gz → 3.4.8.dev8__tar.gz

{prefect_client-3.4.8.dev6 → prefect_client-3.4.8.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.4.8.dev6
+Version: 3.4.8.dev8
 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.8.dev8/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.4.8.dev8"
+__build_date__ = "2025-07-10 08:09:14.278171+00:00"
+__git_commit__ = "2313d0422c7f1f527cc0c450f13c3cac707cabbc"
+__dirty__ = False

{prefect_client-3.4.8.dev6 → prefect_client-3.4.8.dev8}/src/prefect/_internal/websockets.py

@@ -7,6 +7,7 @@ to avoid duplication between events and logs clients.
 
 import os
 import ssl
+import warnings
 from typing import Any, Generator, Optional
 from urllib.parse import urlparse
 from urllib.request import proxy_bypass
@@ -86,6 +87,40 @@ class WebsocketProxyConnect(connect):
         if ssl_context:
             self._kwargs.setdefault("ssl", ssl_context)
 
+        # Add custom headers from settings
+        custom_headers = get_current_settings().client.custom_headers
+        if custom_headers:
+            # Get existing extra_headers or create new dict
+            extra_headers = self._kwargs.get("extra_headers", {})
+            if not isinstance(extra_headers, dict):
+                extra_headers = {}
+
+            for header_name, header_value in custom_headers.items():
+                # Check for protected headers that shouldn't be overridden
+                if header_name.lower() in {
+                    "user-agent",
+                    "sec-websocket-key",
+                    "sec-websocket-version",
+                    "sec-websocket-extensions",
+                    "sec-websocket-protocol",
+                    "connection",
+                    "upgrade",
+                    "host",
+                }:
+                    warnings.warn(
+                        f"Custom header '{header_name}' is ignored because it conflicts with "
+                        f"a protected WebSocket header. Protected headers include: "
+                        f"User-Agent, Sec-WebSocket-Key, Sec-WebSocket-Version, "
+                        f"Sec-WebSocket-Extensions, Sec-WebSocket-Protocol, Connection, "
+                        f"Upgrade, Host",
+                        UserWarning,
+                        stacklevel=2,
+                    )
+                else:
+                    extra_headers[header_name] = header_value
+
+            self._kwargs["extra_headers"] = extra_headers
+
     async def _proxy_connect(self: Self) -> ClientConnection:
         if self._proxy_url:
             # Create proxy in the current event loop context

{prefect_client-3.4.8.dev6 → prefect_client-3.4.8.dev8}/src/prefect/client/base.py

@@ -29,6 +29,7 @@ from prefect.settings import (
     PREFECT_CLIENT_RETRY_JITTER_FACTOR,
     PREFECT_CLOUD_API_URL,
     PREFECT_SERVER_ALLOW_EPHEMERAL_MODE,
+    get_current_settings,
 )
 from prefect.utilities.collections import AutoEnum
 from prefect.utilities.math import bounded_poisson_interval, clamped_poisson_interval
@@ -206,6 +207,23 @@ class PrefectHttpxAsyncClient(httpx.AsyncClient):
         )
         self.headers["User-Agent"] = user_agent
 
+        # Add custom headers from settings
+        custom_headers = get_current_settings().client.custom_headers
+        for header_name, header_value in custom_headers.items():
+            # Prevent overriding critical headers
+            if header_name.lower() in {
+                "user-agent",
+                "prefect-csrf-token",
+                "prefect-csrf-client",
+            }:
+                logger.warning(
+                    f"Custom header '{header_name}' is ignored because it conflicts with "
+                    f"a protected header managed by Prefect. Protected headers include: "
+                    f"User-Agent, Prefect-Csrf-Token, Prefect-Csrf-Client"
+                )
+            else:
+                self.headers[header_name] = header_value
+
     async def _send_with_retry(
         self,
         request: Request,
@@ -432,6 +450,23 @@ class PrefectHttpxSyncClient(httpx.Client):
         )
         self.headers["User-Agent"] = user_agent
 
+        # Add custom headers from settings
+        custom_headers = get_current_settings().client.custom_headers
+        for header_name, header_value in custom_headers.items():
+            # Prevent overriding critical headers
+            if header_name.lower() in {
+                "user-agent",
+                "prefect-csrf-token",
+                "prefect-csrf-client",
+            }:
+                logger.warning(
+                    f"Custom header '{header_name}' is ignored because it conflicts with "
+                    f"a protected header managed by Prefect. Protected headers include: "
+                    f"User-Agent, Prefect-Csrf-Token, Prefect-Csrf-Client"
+                )
+            else:
+                self.headers[header_name] = header_value
+
     def _send_with_retry(
         self,
         request: Request,

{prefect_client-3.4.8.dev6 → prefect_client-3.4.8.dev8}/src/prefect/events/cli/automations.py

@@ -3,6 +3,7 @@ Command line interface for working with automations.
 """
 
 import functools
+from pathlib import Path
 from typing import Any, Callable, Optional, Type
 from uuid import UUID
 
@@ -18,7 +19,7 @@ from prefect.cli._types import PrefectTyper
 from prefect.cli._utilities import exit_with_error, exit_with_success
 from prefect.cli.root import app, is_interactive
 from prefect.client.orchestration import get_client
-from prefect.events.schemas.automations import Automation
+from prefect.events.schemas.automations import Automation, AutomationCore
 from prefect.exceptions import PrefectHTTPStatusError
 
 automations_app: PrefectTyper = PrefectTyper(
@@ -345,3 +346,95 @@ async def delete(
                 exit_with_error("Deletion aborted.")
             await client.delete_automation(automation[0].id)
             exit_with_success(f"Deleted automation with name {name!r}")
+
+
+@automations_app.command()
+@requires_automations
+async def create(
+    from_file: Optional[str] = typer.Option(
+        None,
+        "--from-file",
+        "-f",
+        help="Path to YAML or JSON file containing automation(s)",
+    ),
+    from_json: Optional[str] = typer.Option(
+        None,
+        "--from-json",
+        "-j",
+        help="JSON string containing automation(s)",
+    ),
+):
+    """Create one or more automations from a file or JSON string.
+
+    Examples:
+        $ prefect automation create --from-file automation.yaml
+        $ prefect automation create -f automation.json
+        $ prefect automation create --from-json '{"name": "my-automation", "trigger": {...}, "actions": [...]}'
+        $ prefect automation create -j '[{"name": "auto1", ...}, {"name": "auto2", ...}]'
+    """
+    if from_file and from_json:
+        exit_with_error("Please provide either --from-file or --from-json, not both.")
+
+    if not from_file and not from_json:
+        exit_with_error("Please provide either --from-file or --from-json.")
+
+    if from_file:
+        file_path = Path(from_file)
+        if not file_path.exists():
+            exit_with_error(f"File not found: {from_file}")
+
+        with open(file_path, "r") as f:
+            content = f.read()
+
+        if file_path.suffix.lower() in [".yaml", ".yml"]:
+            data = pyyaml.safe_load(content)
+        elif file_path.suffix.lower() == ".json":
+            data = orjson.loads(content)
+        else:
+            exit_with_error(
+                "File extension not recognized. Please use .yaml, .yml, or .json"
+            )
+    else:  # from_json
+        try:
+            data = orjson.loads(from_json)
+        except orjson.JSONDecodeError as e:
+            exit_with_error(f"Invalid JSON: {e}")
+
+    automations_data = []
+    if isinstance(data, dict) and "automations" in data:
+        automations_data = data["automations"]
+    elif isinstance(data, list):
+        automations_data = data
+    else:
+        automations_data = [data]
+
+    created = []
+    failed = []
+    async with get_client() as client:
+        for i, automation_data in enumerate(automations_data):
+            try:
+                automation = AutomationCore.model_validate(automation_data)
+                automation_id = await client.create_automation(automation)
+                created.append((automation.name, automation_id))
+            except Exception as e:
+                name = automation_data.get("name", f"automation at index {i}")
+                failed.append((name, str(e)))
+
+    if failed:
+        app.console.print(f"[red]Failed to create {len(failed)} automation(s):[/red]")
+        for name, error in failed:
+            app.console.print(f"  - {name}: {error}")
+
+    if created:
+        if len(created) == 1 and not failed:
+            name, automation_id = created[0]
+            exit_with_success(f"Created automation '{name}' with id {automation_id}")
+        else:
+            app.console.print(f"[green]Created {len(created)} automation(s):[/green]")
+            for name, automation_id in created:
+                app.console.print(f"  - '{name}' with id {automation_id}")
+
+    if failed:
+        raise typer.Exit(1)
+    else:
+        raise typer.Exit(0)

{prefect_client-3.4.8.dev6 → prefect_client-3.4.8.dev8}/src/prefect/settings/models/client.py

@@ -1,4 +1,4 @@
-from typing import ClassVar
+from typing import ClassVar, Dict
 
 from pydantic import AliasChoices, AliasPath, Field
 from pydantic_settings import SettingsConfigDict
@@ -87,6 +87,17 @@ class ClientSettings(PrefectBaseSettings):
         """,
     )
 
+    # this needs to be typing.Dict for now as dict[str, str] is not compatible with pydantic < 2.11
+    custom_headers: Dict[str, str] = Field(
+        default_factory=dict,
+        description="""
+        Custom HTTP headers to include with every API request to the Prefect server.
+        Headers are specified as key-value pairs. Note that headers like 'User-Agent'
+        and CSRF-related headers are managed by Prefect and cannot be overridden.
+        """,
+        examples=[{"X-Custom-Header": "value"}, {"Authorization": "Bearer token"}],
+    )
+
     metrics: ClientMetricsSettings = Field(
         default_factory=ClientMetricsSettings,
         description="Settings for controlling metrics reporting from the client",

{prefect_client-3.4.8.dev6 → prefect_client-3.4.8.dev8}/src/prefect/utilities/engine.py

@@ -690,6 +690,7 @@ def emit_task_run_state_change_event(
         resource={
             "prefect.resource.id": f"prefect.task-run.{task_run.id}",
             "prefect.resource.name": task_run.name,
+            "prefect.run-count": str(task_run.run_count),
             "prefect.state-message": truncated_to(
                 state_message_truncation_length, validated_state.message
             ),

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

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.4.8.dev6"
-__build_date__ = "2025-07-08 08:09:34.772687+00:00"
-__git_commit__ = "4a2db7097d87714e8d401d9ebfe5954a4fa77cb2"
-__dirty__ = False