prefect-client 3.1.12__py3-none-any.whl → 3.1.14__py3-none-any.whl

prefect/settings/models/root.py

@@ -11,10 +11,11 @@ from typing import (
 )
 from urllib.parse import urlparse
 
-from pydantic import BeforeValidator, ConfigDict, Field, SecretStr, model_validator
+from pydantic import BeforeValidator, Field, SecretStr, model_validator
+from pydantic_settings import SettingsConfigDict
 from typing_extensions import Self
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 from prefect.settings.models.tasks import TasksSettings
 from prefect.settings.models.testing import TestingSettings
 from prefect.settings.models.worker import WorkerSettings
@@ -44,7 +45,7 @@ class Settings(PrefectBaseSettings):
     See https://docs.pydantic.dev/latest/concepts/pydantic_settings
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config()
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config()
 
     home: Annotated[Path, BeforeValidator(lambda x: Path(x).expanduser())] = Field(
         default=Path("~") / ".prefect",
@@ -214,11 +215,7 @@ class Settings(PrefectBaseSettings):
         if self.server.database.connection_url is None:
             self.server.database.connection_url = _default_database_connection_url(self)
             self.server.database.__pydantic_fields_set__.remove("connection_url")
-        db_url = (
-            self.server.database.connection_url.get_secret_value()
-            if isinstance(self.server.database.connection_url, SecretStr)
-            else self.server.database.connection_url
-        )
+        db_url = self.server.database.connection_url.get_secret_value()
         if (
             "PREFECT_API_DATABASE_PASSWORD" in db_url
             or "PREFECT_SERVER_DATABASE_PASSWORD" in db_url
@@ -275,7 +272,7 @@ class Settings(PrefectBaseSettings):
         # To restore defaults, we need to resolve the setting path and then
         # set the default value on the new settings object. When restoring
         # defaults, all settings sources will be ignored.
-        restore_defaults_obj = {}
+        restore_defaults_obj: dict[str, Any] = {}
         for r in restore_defaults or []:
             path = r.accessor.split(".")
             model = self
@@ -301,11 +298,11 @@ class Settings(PrefectBaseSettings):
         updates = updates or {}
         set_defaults = set_defaults or {}
 
-        set_defaults_obj = {}
+        set_defaults_obj: dict[str, Any] = {}
         for setting, value in set_defaults.items():
             set_in_dict(set_defaults_obj, setting.accessor, value)
 
-        updates_obj = {}
+        updates_obj: dict[str, Any] = {}
         for setting, value in updates.items():
             set_in_dict(updates_obj, setting.accessor, value)
 
@@ -377,7 +374,7 @@ def _warn_on_misconfigured_api_url(settings: "Settings"):
                 "`PREFECT_API_URL` uses `/workspace/` but should use `/workspaces/`."
             ),
         }
-        warnings_list = []
+        warnings_list: list[str] = []
 
         for misconfig, warning in misconfigured_mappings.items():
             if misconfig in api_url:
@@ -399,7 +396,7 @@ def _warn_on_misconfigured_api_url(settings: "Settings"):
 
 
 def _default_database_connection_url(settings: "Settings") -> SecretStr:
-    value = None
+    value: str = f"sqlite+aiosqlite:///{settings.home}/prefect.db"
     if settings.server.database.driver == "postgresql+asyncpg":
         required = [
             "host",
@@ -417,7 +414,7 @@ def _default_database_connection_url(settings: "Settings") -> SecretStr:
 
         from sqlalchemy import URL
 
-        return URL(
+        value = URL(
             drivername=settings.server.database.driver,
             host=settings.server.database.host,
             port=settings.server.database.port or 5432,
@@ -443,6 +440,4 @@ def _default_database_connection_url(settings: "Settings") -> SecretStr:
         raise ValueError(
             f"Unsupported database driver: {settings.server.database.driver}"
         )
-
-    value = value if value else f"sqlite+aiosqlite:///{settings.home}/prefect.db"
     return SecretStr(value)

prefect/settings/models/runner.py

@@ -1,8 +1,9 @@
 from typing import ClassVar, Optional
 
-from pydantic import ConfigDict, Field
+from pydantic import Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 from prefect.types import LogLevel
 
 
@@ -11,7 +12,9 @@ class RunnerServerSettings(PrefectBaseSettings):
     Settings for controlling runner server behavior
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("runner", "server"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("runner", "server")
+    )
 
     enable: bool = Field(
         default=False,
@@ -29,7 +32,7 @@ class RunnerServerSettings(PrefectBaseSettings):
     )
 
     log_level: LogLevel = Field(
-        default="error",
+        default="ERROR",
         description="The log level of the runner's webserver.",
     )
 
@@ -44,7 +47,7 @@ class RunnerSettings(PrefectBaseSettings):
     Settings for controlling runner behavior
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("runner",))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(("runner",))
 
     process_limit: int = Field(
         default=5,

prefect/settings/models/server/api.py

@@ -1,9 +1,10 @@
 from datetime import timedelta
 from typing import ClassVar, Optional
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field, SecretStr
+from pydantic import AliasChoices, AliasPath, Field, SecretStr
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 
 
 class ServerAPISettings(PrefectBaseSettings):
@@ -11,7 +12,9 @@ class ServerAPISettings(PrefectBaseSettings):
     Settings for controlling API server behavior
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("server", "api"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("server", "api")
+    )
 
     auth_string: Optional[SecretStr] = Field(
         default=None,

prefect/settings/models/server/database.py

@@ -1,18 +1,80 @@
 import warnings
-from typing import ClassVar, Optional
+from typing import Any, ClassVar, Optional
 from urllib.parse import quote_plus
 
 from pydantic import (
     AliasChoices,
     AliasPath,
-    ConfigDict,
     Field,
     SecretStr,
     model_validator,
 )
+from pydantic_settings import SettingsConfigDict
 from typing_extensions import Literal, Self
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
+
+
+class SQLAlchemyConnectArgsSettings(PrefectBaseSettings):
+    """
+    Settings for controlling SQLAlchemy connection behavior; note that these settings only take effect when
+    using a PostgreSQL database.
+    """
+
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("server", "database", "sqlalchemy", "connect_args")
+    )
+
+    application_name: Optional[str] = Field(
+        default=None,
+        description="Controls the application_name field for connections opened from the connection pool when using a PostgreSQL database with the Prefect backend.",
+    )
+
+
+class SQLAlchemySettings(PrefectBaseSettings):
+    """
+    Settings for controlling SQLAlchemy behavior; note that these settings only take effect when
+    using a PostgreSQL database.
+    """
+
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("server", "database", "sqlalchemy")
+    )
+
+    connect_args: SQLAlchemyConnectArgsSettings = Field(
+        default_factory=SQLAlchemyConnectArgsSettings,
+        description="Settings for controlling SQLAlchemy connection behavior",
+    )
+
+    pool_size: int = Field(
+        default=5,
+        description="Controls connection pool size of database connection pools from the Prefect backend.",
+        validation_alias=AliasChoices(
+            AliasPath("pool_size"),
+            "prefect_server_database_sqlalchemy_pool_size",
+            "prefect_sqlalchemy_pool_size",
+        ),
+    )
+
+    pool_recycle: int = Field(
+        default=3600,
+        description="This setting causes the pool to recycle connections after the given number of seconds has passed; set it to -1 to avoid recycling entirely.",
+    )
+
+    pool_timeout: Optional[float] = Field(
+        default=30.0,
+        description="Number of seconds to wait before giving up on getting a connection from the pool. Defaults to 30 seconds.",
+    )
+
+    max_overflow: int = Field(
+        default=10,
+        description="Controls maximum overflow of the connection pool. To prevent overflow, set to -1.",
+        validation_alias=AliasChoices(
+            AliasPath("max_overflow"),
+            "prefect_server_database_sqlalchemy_max_overflow",
+            "prefect_sqlalchemy_max_overflow",
+        ),
+    )
 
 
 class ServerDatabaseSettings(PrefectBaseSettings):
@@ -20,7 +82,14 @@ class ServerDatabaseSettings(PrefectBaseSettings):
     Settings for controlling server database behavior
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("server", "database"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("server", "database")
+    )
+
+    sqlalchemy: SQLAlchemySettings = Field(
+        default_factory=SQLAlchemySettings,
+        description="Settings for controlling SQLAlchemy behavior",
+    )
 
     connection_url: Optional[SecretStr] = Field(
         default=None,
@@ -128,7 +197,7 @@ class ServerDatabaseSettings(PrefectBaseSettings):
 
     timeout: Optional[float] = Field(
         default=10.0,
-        description="A statement timeout, in seconds, applied to all database interactions made by the API. Defaults to 10 seconds.",
+        description="A statement timeout, in seconds, applied to all database interactions made by the Prefect backend. Defaults to 10 seconds.",
         validation_alias=AliasChoices(
             AliasPath("timeout"),
             "prefect_server_database_timeout",
@@ -146,25 +215,49 @@ class ServerDatabaseSettings(PrefectBaseSettings):
         ),
     )
 
-    sqlalchemy_pool_size: Optional[int] = Field(
-        default=None,
-        description="Controls connection pool size when using a PostgreSQL database with the Prefect API. If not set, the default SQLAlchemy pool size will be used.",
-        validation_alias=AliasChoices(
-            AliasPath("sqlalchemy_pool_size"),
-            "prefect_server_database_sqlalchemy_pool_size",
-            "prefect_sqlalchemy_pool_size",
-        ),
-    )
+    # handle deprecated fields
 
-    sqlalchemy_max_overflow: Optional[int] = Field(
-        default=None,
-        description="Controls maximum overflow of the connection pool when using a PostgreSQL database with the Prefect API. If not set, the default SQLAlchemy maximum overflow value will be used.",
-        validation_alias=AliasChoices(
-            AliasPath("sqlalchemy_max_overflow"),
-            "prefect_server_database_sqlalchemy_max_overflow",
-            "prefect_sqlalchemy_max_overflow",
-        ),
-    )
+    def __getattribute__(self, name: str) -> Any:
+        if name in ["sqlalchemy_pool_size", "sqlalchemy_max_overflow"]:
+            warnings.warn(
+                f"Setting {name} has been moved to the `sqlalchemy` settings group.",
+                DeprecationWarning,
+            )
+            field_name = name.replace("sqlalchemy_", "")
+            return getattr(super().__getattribute__("sqlalchemy"), field_name)
+        return super().__getattribute__(name)
+
+    # validators
+
+    @model_validator(mode="before")
+    @classmethod
+    def set_deprecated_sqlalchemy_settings_on_child_model_and_warn(
+        cls, values: dict[str, Any]
+    ) -> dict[str, Any]:
+        """
+        Set deprecated settings on the child model.
+        """
+        # Initialize sqlalchemy settings if not present
+        if "sqlalchemy" not in values:
+            values["sqlalchemy"] = SQLAlchemySettings()
+
+        if "sqlalchemy_pool_size" in values:
+            warnings.warn(
+                "`sqlalchemy_pool_size` has been moved to the `sqlalchemy` settings group as `pool_size`.",
+                DeprecationWarning,
+            )
+            if "pool_size" not in values["sqlalchemy"].model_fields_set:
+                values["sqlalchemy"].pool_size = values["sqlalchemy_pool_size"]
+
+        if "sqlalchemy_max_overflow" in values:
+            warnings.warn(
+                "`sqlalchemy_max_overflow` has been moved to the `sqlalchemy` settings group as `max_overflow`.",
+                DeprecationWarning,
+            )
+            if "max_overflow" not in values["sqlalchemy"].model_fields_set:
+                values["sqlalchemy"].max_overflow = values["sqlalchemy_max_overflow"]
+
+        return values
 
     @model_validator(mode="after")
     def emit_warnings(self) -> Self:  # noqa: F821
@@ -173,7 +266,9 @@ class ServerDatabaseSettings(PrefectBaseSettings):
         return self
 
 
-def warn_on_database_password_value_without_usage(settings: ServerDatabaseSettings):
+def warn_on_database_password_value_without_usage(
+    settings: ServerDatabaseSettings,
+) -> None:
     """
     Validator for settings warning if the database password is set but not used.
     """
@@ -201,4 +296,4 @@ def warn_on_database_password_value_without_usage(settings: ServerDatabaseSettin
             "PREFECT_SERVER_DATABASE_CONNECTION_URL. "
             "The provided password will be ignored."
         )
-    return settings
+    return None

prefect/settings/models/server/deployments.py

@@ -1,12 +1,13 @@
 from typing import ClassVar
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field
+from pydantic import AliasChoices, AliasPath, Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 
 
 class ServerDeploymentsSettings(PrefectBaseSettings):
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "deployments")
     )
 

prefect/settings/models/server/ephemeral.py

@@ -1,8 +1,9 @@
 from typing import ClassVar
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field
+from pydantic import AliasChoices, AliasPath, Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 
 
 class ServerEphemeralSettings(PrefectBaseSettings):
@@ -10,7 +11,9 @@ class ServerEphemeralSettings(PrefectBaseSettings):
     Settings for controlling ephemeral server behavior
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("server", "ephemeral"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("server", "ephemeral")
+    )
 
     enabled: bool = Field(
         default=False,
@@ -28,6 +31,6 @@ class ServerEphemeralSettings(PrefectBaseSettings):
         default=20,
         description="""
         The number of seconds to wait for the server to start when ephemeral mode is enabled.
-        Defaults to `10`.
+        Defaults to `20`.
         """,
     )

prefect/settings/models/server/events.py

@@ -1,9 +1,10 @@
 from datetime import timedelta
 from typing import ClassVar
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field
+from pydantic import AliasChoices, AliasPath, Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 
 
 class ServerEventsSettings(PrefectBaseSettings):
@@ -11,7 +12,9 @@ class ServerEventsSettings(PrefectBaseSettings):
     Settings for controlling behavior of the events subsystem
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("server", "events"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("server", "events")
+    )
 
     ###########################################################################
     # Events settings

prefect/settings/models/server/flow_run_graph.py

@@ -1,8 +1,9 @@
 from typing import ClassVar
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field
+from pydantic import AliasChoices, AliasPath, Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 
 
 class ServerFlowRunGraphSettings(PrefectBaseSettings):
@@ -10,7 +11,7 @@ class ServerFlowRunGraphSettings(PrefectBaseSettings):
     Settings for controlling behavior of the flow run graph
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "flow_run_graph")
     )
 

prefect/settings/models/server/root.py

@@ -1,9 +1,10 @@
 from pathlib import Path
 from typing import ClassVar, Optional
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field
+from pydantic import AliasChoices, AliasPath, Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 from prefect.types import LogLevel
 
 from .api import ServerAPISettings
@@ -22,7 +23,7 @@ class ServerSettings(PrefectBaseSettings):
     Settings for controlling server behavior
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("server",))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(("server",))
 
     logging_level: LogLevel = Field(
         default="WARNING",

prefect/settings/models/server/services.py

@@ -1,9 +1,10 @@
 from datetime import timedelta
 from typing import ClassVar
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field
+from pydantic import AliasChoices, AliasPath, Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 
 
 class ServerServicesCancellationCleanupSettings(PrefectBaseSettings):
@@ -11,7 +12,7 @@ class ServerServicesCancellationCleanupSettings(PrefectBaseSettings):
     Settings for controlling the cancellation cleanup service
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "services", "cancellation_cleanup")
     )
 
@@ -41,7 +42,7 @@ class ServerServicesEventPersisterSettings(PrefectBaseSettings):
     Settings for controlling the event persister service
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "services", "event_persister")
     )
 
@@ -83,7 +84,7 @@ class ServerServicesFlowRunNotificationsSettings(PrefectBaseSettings):
     Settings for controlling the flow run notifications service
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "services", "flow_run_notifications")
     )
 
@@ -103,7 +104,7 @@ class ServerServicesForemanSettings(PrefectBaseSettings):
     Settings for controlling the foreman service
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "services", "foreman")
     )
 
@@ -184,7 +185,7 @@ class ServerServicesLateRunsSettings(PrefectBaseSettings):
     Settings for controlling the late runs service
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "services", "late_runs")
     )
 
@@ -228,7 +229,7 @@ class ServerServicesSchedulerSettings(PrefectBaseSettings):
     Settings for controlling the scheduler service
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "services", "scheduler")
     )
 
@@ -353,7 +354,7 @@ class ServerServicesPauseExpirationsSettings(PrefectBaseSettings):
     Settings for controlling the pause expiration service
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "services", "pause_expirations")
     )
 
@@ -389,7 +390,7 @@ class ServerServicesTaskRunRecorderSettings(PrefectBaseSettings):
     Settings for controlling the task run recorder service
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "services", "task_run_recorder")
     )
 
@@ -409,7 +410,7 @@ class ServerServicesTriggersSettings(PrefectBaseSettings):
     Settings for controlling the triggers service
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "services", "triggers")
     )
 
@@ -429,7 +430,9 @@ class ServerServicesSettings(PrefectBaseSettings):
     Settings for controlling server services
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("server", "services"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("server", "services")
+    )
 
     cancellation_cleanup: ServerServicesCancellationCleanupSettings = Field(
         default_factory=ServerServicesCancellationCleanupSettings,

prefect/settings/models/server/tasks.py

@@ -1,9 +1,10 @@
 from datetime import timedelta
 from typing import ClassVar
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field
+from pydantic import AliasChoices, AliasPath, Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 
 
 class ServerTasksSchedulingSettings(PrefectBaseSettings):
@@ -11,7 +12,7 @@ class ServerTasksSchedulingSettings(PrefectBaseSettings):
     Settings for controlling server-side behavior related to task scheduling
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
         ("server", "tasks", "scheduling")
     )
 
@@ -51,7 +52,9 @@ class ServerTasksSettings(PrefectBaseSettings):
     Settings for controlling server-side behavior related to tasks
     """
 
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("server", "tasks"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("server", "tasks")
+    )
 
     tag_concurrency_slot_wait_seconds: float = Field(
         default=30,

prefect/settings/models/server/ui.py

@@ -1,12 +1,13 @@
 from typing import ClassVar, Optional
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field
+from pydantic import AliasChoices, AliasPath, Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 
 
 class ServerUISettings(PrefectBaseSettings):
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("server", "ui"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(("server", "ui"))
 
     enabled: bool = Field(
         default=True,

prefect/settings/models/tasks.py

@@ -1,12 +1,15 @@
 from typing import ClassVar, Optional, Union
 
-from pydantic import AliasChoices, AliasPath, ConfigDict, Field
+from pydantic import AliasChoices, AliasPath, Field
+from pydantic_settings import SettingsConfigDict
 
-from prefect.settings.base import PrefectBaseSettings, _build_settings_config
+from prefect.settings.base import PrefectBaseSettings, build_settings_config
 
 
 class TasksRunnerSettings(PrefectBaseSettings):
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("tasks", "runner"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("tasks", "runner")
+    )
 
     thread_pool_max_workers: Optional[int] = Field(
         default=None,
@@ -21,7 +24,9 @@ class TasksRunnerSettings(PrefectBaseSettings):
 
 
 class TasksSchedulingSettings(PrefectBaseSettings):
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("tasks", "scheduling"))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(
+        ("tasks", "scheduling")
+    )
 
     default_storage_block: Optional[str] = Field(
         default=None,
@@ -45,7 +50,7 @@ class TasksSchedulingSettings(PrefectBaseSettings):
 
 
 class TasksSettings(PrefectBaseSettings):
-    model_config: ClassVar[ConfigDict] = _build_settings_config(("tasks",))
+    model_config: ClassVar[SettingsConfigDict] = build_settings_config(("tasks",))
 
     refresh_cache: bool = Field(
         default=False,