mlrun 1.10.0rc2__py3-none-any.whl → 1.10.0rc4__py3-none-any.whl

mlrun/common/runtimes/constants.py

@@ -12,11 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import enum
 import typing
 
-from deprecated import deprecated
-
 import mlrun.common.constants as mlrun_constants
 import mlrun_pipelines.common.models
 
@@ -239,24 +236,6 @@ class RunStates:
         }[pipeline_run_status]
 
 
-# TODO: remove this class in 1.11.0 - use only MLRunInternalLabels
-@deprecated(
-    version="1.9.0",
-    reason="This class is deprecated and will be removed in 1.11.0. Use MLRunInternalLabels instead.",
-    category=FutureWarning,
-)
-class RunLabels(enum.Enum):
-    owner = mlrun_constants.MLRunInternalLabels.owner
-    v3io_user = mlrun_constants.MLRunInternalLabels.v3io_user
-
-    @staticmethod
-    def all():
-        return [
-            RunLabels.owner,
-            RunLabels.v3io_user,
-        ]
-
-
 class SparkApplicationStates:
     """
     https://github.com/GoogleCloudPlatform/spark-on-k8s-operator/blob/master/pkg/apis/sparkoperator.k8s.io/v1beta2/types.go#L321

mlrun/common/schemas/artifact.py

@@ -15,7 +15,6 @@
 import typing
 
 import pydantic.v1
-from deprecated import deprecated
 
 import mlrun.common.types
 
@@ -26,6 +25,7 @@ class ArtifactCategories(mlrun.common.types.StrEnum):
     model = "model"
     dataset = "dataset"
     document = "document"
+    llm_prompt = "llm-prompt"
     other = "other"
 
     # we define the link as a category to prevent import cycles, but it's not a real category
@@ -41,19 +41,27 @@ class ArtifactCategories(mlrun.common.types.StrEnum):
             return [ArtifactCategories.dataset.value, link_kind], False
         if self.value == ArtifactCategories.document.value:
             return [ArtifactCategories.document.value, link_kind], False
+        if self.value == ArtifactCategories.llm_prompt.value:
+            return [ArtifactCategories.llm_prompt.value, link_kind], False
         if self.value == ArtifactCategories.other.value:
             return (
                 [
                     ArtifactCategories.model.value,
                     ArtifactCategories.dataset.value,
                     ArtifactCategories.document.value,
+                    ArtifactCategories.llm_prompt.value,
                 ],
                 True,
             )
 
     @classmethod
     def from_kind(cls, kind: str) -> "ArtifactCategories":
-        if kind in [cls.model.value, cls.dataset.value, cls.document.value]:
+        if kind in [
+            cls.model.value,
+            cls.dataset.value,
+            cls.document.value,
+            cls.llm_prompt.value,
+        ]:
             return cls(kind)
         return cls.other
 
@@ -64,6 +72,7 @@ class ArtifactCategories(mlrun.common.types.StrEnum):
             ArtifactCategories.model,
             ArtifactCategories.dataset,
             ArtifactCategories.document,
+            ArtifactCategories.llm_prompt,
         ]
 
 
@@ -78,16 +87,6 @@ class ArtifactIdentifier(pydantic.v1.BaseModel):
     # hash: typing.Optional[str]
 
 
-@deprecated(
-    version="1.7.0",
-    reason="mlrun.common.schemas.ArtifactsFormat is deprecated and will be removed in 1.9.0. "
-    "Use mlrun.common.formatters.ArtifactFormat instead.",
-    category=FutureWarning,
-)
-class ArtifactsFormat(mlrun.common.types.StrEnum):
-    full = "full"
-
-
 class ArtifactMetadata(pydantic.v1.BaseModel):
     key: str
     project: str
@@ -108,6 +107,7 @@ class ArtifactSpec(pydantic.v1.BaseModel):
     db_key: typing.Optional[str]
     extra_data: typing.Optional[dict[str, typing.Any]]
     unpackaging_instructions: typing.Optional[dict[str, typing.Any]]
+    parent_uri: typing.Optional[str]
 
     class Config:
         extra = pydantic.v1.Extra.allow

mlrun/common/schemas/pipeline.py

@@ -15,22 +15,6 @@
 import typing
 
 import pydantic.v1
-from deprecated import deprecated
-
-import mlrun.common.types
-
-
-@deprecated(
-    version="1.7.0",
-    reason="mlrun.common.schemas.PipelinesFormat is deprecated and will be removed in 1.9.0. "
-    "Use mlrun.common.formatters.PipelineFormat instead.",
-    category=FutureWarning,
-)
-class PipelinesFormat(mlrun.common.types.StrEnum):
-    full = "full"
-    metadata_only = "metadata_only"
-    summary = "summary"
-    name_only = "name_only"
 
 
 class PipelinesPagination(str):

mlrun/common/schemas/project.py

@@ -16,7 +16,6 @@ import datetime
 import typing
 
 import pydantic.v1
-from deprecated import deprecated
 
 import mlrun.common.types
 
@@ -24,22 +23,6 @@ from .common import ImageBuilder
 from .object import ObjectKind, ObjectStatus
 
 
-@deprecated(
-    version="1.7.0",
-    reason="mlrun.common.schemas.ProjectsFormat is deprecated and will be removed in 1.9.0. "
-    "Use mlrun.common.formatters.ProjectFormat instead.",
-    category=FutureWarning,
-)
-class ProjectsFormat(mlrun.common.types.StrEnum):
-    full = "full"
-    name_only = "name_only"
-    # minimal format removes large fields from the response (e.g. functions, workflows, artifacts)
-    # and is used for faster response times (in the UI)
-    minimal = "minimal"
-    # internal - allowed only in follower mode, only for the leader for upgrade purposes
-    leader = "leader"
-
-
 class ProjectMetadata(pydantic.v1.BaseModel):
     name: str
     created: typing.Optional[datetime.datetime] = None

mlrun/common/schemas/runs.py

@@ -15,26 +15,9 @@
 import typing
 
 import pydantic.v1
-from deprecated import deprecated
-
-import mlrun.common.types
 
 
 class RunIdentifier(pydantic.v1.BaseModel):
     kind: typing.Literal["run"] = "run"
     uid: typing.Optional[str]
     iter: typing.Optional[int]
-
-
-@deprecated(
-    version="1.7.0",
-    reason="mlrun.common.schemas.RunsFormat is deprecated and will be removed in 1.9.0. "
-    "Use mlrun.common.formatters.RunFormat instead.",
-    category=FutureWarning,
-)
-class RunsFormat(mlrun.common.types.StrEnum):
-    # No enrichment, data is pulled as-is from the database.
-    standard = "standard"
-
-    # Performs run enrichment, including the run's artifacts. Only available for the `get` run API.
-    full = "full"

mlrun/config.py

@@ -79,7 +79,7 @@ default_config = {
     # comma separated list of images that are in the specified images_registry, and therefore will be enriched with this
     # registry when used. default to mlrun/* which means any image which is of the mlrun repository (mlrun/mlrun,
     # mlrun/ml-base, etc...)
-    "images_to_enrich_registry": "^mlrun/*,python:3.9",
+    "images_to_enrich_registry": "^mlrun/*,^python:3.(9|11)$",
     "kfp_url": "",
     "kfp_ttl": "14400",  # KFP ttl in sec, after that completed PODs will be deleted
     "kfp_image": "mlrun/mlrun-kfp",  # image to use for KFP runner
@@ -94,7 +94,7 @@ default_config = {
     "default_base_image": "mlrun/mlrun",  # default base image when doing .deploy()
     # template for project default image name. Parameter {name} will be replaced with project name
     "default_project_image_name": ".mlrun-project-image-{name}",
-    "default_project": "default",  # default project name
+    "active_project": "",  # active project name
     "default_archive": "",  # default remote archive URL (for build tar.gz)
     "mpijob_crd_version": "",  # mpijob crd version (e.g: "v1alpha1". must be in: mlrun.runtime.MPIJobCRDVersions)
     "ipython_widget": True,
@@ -286,7 +286,7 @@ default_config = {
             "remote": "mlrun/mlrun",
             "dask": "mlrun/ml-base",
             "mpijob": "mlrun/mlrun",
-            "application": "python:3.9",
+            "application": "python",
         },
         # see enrich_function_preemption_spec for more info,
         # and mlrun.common.schemas.function.PreemptionModes for available options

mlrun/datastore/base.py

@@ -104,10 +104,10 @@ class DataStore:
         """Whether the data store supports isdir"""
         return True
 
-    def _get_secret_or_env(self, key, default=None, prefix=None):
+    def _get_secret_or_env(self, key, default=None):
         # Project-secrets are mounted as env variables whose name can be retrieved from SecretsStore
         return mlrun.get_secret_or_env(
-            key, secret_provider=self._get_secret, default=default, prefix=prefix
+            key, secret_provider=self._get_secret, default=default
         )
 
     def get_storage_options(self):

mlrun/datastore/datastore.py

@@ -54,7 +54,7 @@ def parse_url(url):
     return schema, endpoint, parsed_url
 
 
-def schema_to_store(schema):
+def schema_to_store(schema) -> DataStore.__subclasses__():
     # import store classes inside to enable making their dependencies optional (package extras)
 
     if not schema or schema in get_local_file_schema():

mlrun/datastore/datastore_profile.py

@@ -141,19 +141,11 @@ class ConfigProfile(DatastoreProfile):
 class DatastoreProfileKafkaTarget(DatastoreProfile):
     type: str = pydantic.v1.Field("kafka_target")
     _private_attributes = "kwargs_private"
-    brokers: typing.Optional[str] = None
+    brokers: str
     topic: str
     kwargs_public: typing.Optional[dict]
     kwargs_private: typing.Optional[dict]
 
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
-
-        if not self.brokers:
-            raise mlrun.errors.MLRunInvalidArgumentError(
-                "DatastoreProfileKafkaTarget requires the 'brokers' field to be set"
-            )
-
     def get_topic(self) -> typing.Optional[str]:
         return self.topic
 
@@ -555,7 +547,7 @@ def datastore_profile_read(url, project_name="", secrets: typing.Optional[dict]
         url (str): A URL with 'ds' scheme pointing to the datastore profile
             (e.g., 'ds://profile-name').
         project_name (str, optional): The project name where the profile is stored.
-            Defaults to MLRun's default project.
+            Defaults to MLRun's active project.
         secrets (dict, optional): Dictionary containing secrets needed for profile retrieval.
 
     Returns:
@@ -580,7 +572,7 @@ def datastore_profile_read(url, project_name="", secrets: typing.Optional[dict]
         )
 
     profile_name = parsed_url.hostname
-    project_name = project_name or mlrun.mlconf.default_project
+    project_name = project_name or mlrun.mlconf.active_project
     datastore = TemporaryClientDatastoreProfiles().get(profile_name)
     if datastore:
         return datastore

mlrun/datastore/redis.py

@@ -48,9 +48,8 @@ class RedisStore(DataStore):
             raise mlrun.errors.MLRunInvalidArgumentError(
                 "Provide Redis username and password only via secrets"
             )
-        credentials_prefix = self._get_secret_or_env("CREDENTIALS_PREFIX")
-        user = self._get_secret_or_env("REDIS_USER", "", credentials_prefix)
-        password = self._get_secret_or_env("REDIS_PASSWORD", "", credentials_prefix)
+        user = self._get_secret_or_env("REDIS_USER", "")
+        password = self._get_secret_or_env("REDIS_PASSWORD", "")
         host = parsed_endpoint.hostname
         port = parsed_endpoint.port if parsed_endpoint.port else redis_default_port
         schema = parsed_endpoint.scheme

mlrun/datastore/sources.py

@@ -768,7 +768,6 @@ class SnowflakeSource(BaseSourceDriver):
     :parameter url: URL of the snowflake cluster
     :parameter user: snowflake user
     :parameter database: snowflake database
-    :parameter schema: snowflake schema - deprecated, use db_schema
     :parameter db_schema: snowflake schema
     :parameter warehouse: snowflake warehouse
     """
@@ -790,18 +789,10 @@ class SnowflakeSource(BaseSourceDriver):
         url: Optional[str] = None,
         user: Optional[str] = None,
         database: Optional[str] = None,
-        schema: Optional[str] = None,
         db_schema: Optional[str] = None,
         warehouse: Optional[str] = None,
         **kwargs,
     ):
-        # TODO: Remove in 1.9.0
-        if schema:
-            warnings.warn(
-                "schema is deprecated in 1.7.0, and will be removed in 1.9.0, please use db_schema"
-            )
-        db_schema = db_schema or schema  # TODO: Remove in 1.9.0
-
         attributes = attributes or {}
         if url:
             attributes["url"] = url

mlrun/datastore/store_resources.py

@@ -152,19 +152,19 @@ def get_store_resource(
         )
     elif kind == StorePrefix.FeatureSet:
         project, name, tag, uid = parse_versioned_object_uri(
-            uri, project or config.default_project
+            uri, project or config.active_project
         )
         return db.get_feature_set(name, project, tag, uid)
 
     elif kind == StorePrefix.FeatureVector:
         project, name, tag, uid = parse_versioned_object_uri(
-            uri, project or config.default_project
+            uri, project or config.active_project
         )
         return db.get_feature_vector(name, project, tag, uid)
 
     elif StorePrefix.is_artifact(kind):
         project, key, iteration, tag, tree, uid = parse_artifact_uri(
-            uri, project or config.default_project
+            uri, project or config.active_project
         )
         resource = db.read_artifact(
             key,

mlrun/datastore/storeytargets.py

@@ -152,9 +152,7 @@ class KafkaStoreyTarget(storey.KafkaTarget):
                 parsed.path.strip("/") if parsed.path else datastore_profile.get_topic()
             )
         else:
-            brokers = attributes.pop(
-                "brokers", attributes.pop("bootstrap_servers", None)
-            )
+            brokers = attributes.pop("brokers", None)
             topic, brokers = parse_kafka_url(path, brokers)
 
         if not topic:
@@ -175,8 +173,7 @@ class RedisNoSqlStoreyTarget(storey.NoSqlTarget):
     def __init__(self, *args, **kwargs):
         path = kwargs.pop("path")
         endpoint, uri = mlrun.datastore.targets.RedisNoSqlTarget.get_server_endpoint(
-            path,
-            kwargs.pop("credentials_prefix", None),
+            path
         )
         kwargs["path"] = endpoint + "/" + uri
         super().__init__(*args, **kwargs)

mlrun/datastore/targets.py

@@ -17,7 +17,6 @@ import os
 import random
 import sys
 import time
-import warnings
 from collections import Counter
 from copy import copy
 from typing import Any, Optional, Union
@@ -409,7 +408,6 @@ class BaseStoreTarget(DataTargetBase):
         flush_after_seconds: Optional[int] = None,
         storage_options: Optional[dict[str, str]] = None,
         schema: Optional[dict[str, Any]] = None,
-        credentials_prefix=None,
     ):
         super().__init__(
             self.kind,
@@ -424,7 +422,6 @@ class BaseStoreTarget(DataTargetBase):
             max_events,
             flush_after_seconds,
             schema=schema,
-            credentials_prefix=credentials_prefix,
         )
 
         self.name = name or self.kind
@@ -440,13 +437,6 @@ class BaseStoreTarget(DataTargetBase):
         self.flush_after_seconds = flush_after_seconds
         self.storage_options = storage_options
         self.schema = schema or {}
-        self.credentials_prefix = credentials_prefix
-        if credentials_prefix:
-            warnings.warn(
-                "The 'credentials_prefix' parameter is deprecated and will be removed in "
-                "1.9.0. Please use datastore profiles instead.",
-                FutureWarning,
-            )
 
         self._target = None
         self._resource = None
@@ -457,18 +447,11 @@ class BaseStoreTarget(DataTargetBase):
             key,
             secret_provider=self._secrets,
             default=default_value,
-            prefix=self.credentials_prefix,
         )
 
     def _get_store_and_path(self):
-        credentials_prefix_secrets = (
-            {"CREDENTIALS_PREFIX": self.credentials_prefix}
-            if self.credentials_prefix
-            else None
-        )
         store, resolved_store_path, url = mlrun.store_manager.get_or_create_store(
-            self.get_target_path(),
-            credentials_prefix_secrets,
+            self.get_target_path()
         )
         return store, resolved_store_path, url
 
@@ -621,7 +604,6 @@ class BaseStoreTarget(DataTargetBase):
         driver.path = spec.path
         driver.attributes = spec.attributes
         driver.schema = spec.schema
-        driver.credentials_prefix = spec.credentials_prefix
 
         if hasattr(spec, "columns"):
             driver.columns = spec.columns
@@ -638,7 +620,6 @@ class BaseStoreTarget(DataTargetBase):
         driver.max_events = spec.max_events
         driver.flush_after_seconds = spec.flush_after_seconds
         driver.storage_options = spec.storage_options
-        driver.credentials_prefix = spec.credentials_prefix
 
         driver._resource = resource
         driver.run_id = spec.run_id
@@ -720,7 +701,6 @@ class BaseStoreTarget(DataTargetBase):
         target.key_bucketing_number = self.key_bucketing_number
         target.partition_cols = self.partition_cols
         target.time_partitioning_granularity = self.time_partitioning_granularity
-        target.credentials_prefix = self.credentials_prefix
 
         self._resource.status.update_target(target)
         return target
@@ -1213,7 +1193,6 @@ class SnowflakeTarget(BaseStoreTarget):
         flush_after_seconds: Optional[int] = None,
         storage_options: Optional[dict[str, str]] = None,
         schema: Optional[dict[str, Any]] = None,
-        credentials_prefix=None,
         url: Optional[str] = None,
         user: Optional[str] = None,
         db_schema: Optional[str] = None,
@@ -1249,7 +1228,6 @@ class SnowflakeTarget(BaseStoreTarget):
             flush_after_seconds=flush_after_seconds,
             storage_options=storage_options,
             schema=schema,
-            credentials_prefix=credentials_prefix,
         )
 
     def get_spark_options(self, key_column=None, timestamp_key=None, overwrite=True):
@@ -1488,7 +1466,7 @@ class RedisNoSqlTarget(NoSqlBaseTarget):
     writer_step_name = "RedisNoSqlTarget"
 
     @staticmethod
-    def get_server_endpoint(path, credentials_prefix=None):
+    def get_server_endpoint(path):
         endpoint, uri = parse_path(path)
         endpoint = endpoint or mlrun.mlconf.redis.url
         if endpoint.startswith("ds://"):
@@ -1506,15 +1484,8 @@ class RedisNoSqlTarget(NoSqlBaseTarget):
                 raise mlrun.errors.MLRunInvalidArgumentError(
                     "Provide Redis username and password only via secrets"
                 )
-            credentials_prefix = credentials_prefix or mlrun.get_secret_or_env(
-                key="CREDENTIALS_PREFIX"
-            )
-            user = mlrun.get_secret_or_env(
-                "REDIS_USER", default="", prefix=credentials_prefix
-            )
-            password = mlrun.get_secret_or_env(
-                "REDIS_PASSWORD", default="", prefix=credentials_prefix
-            )
+            user = mlrun.get_secret_or_env("REDIS_USER", default="")
+            password = mlrun.get_secret_or_env("REDIS_PASSWORD", default="")
             host = parsed_endpoint.hostname
             port = parsed_endpoint.port if parsed_endpoint.port else "6379"
             scheme = parsed_endpoint.scheme
@@ -1528,9 +1499,7 @@ class RedisNoSqlTarget(NoSqlBaseTarget):
         from storey import Table
         from storey.redis_driver import RedisDriver
 
-        endpoint, uri = self.get_server_endpoint(
-            self.get_target_path(), self.credentials_prefix
-        )
+        endpoint, uri = self.get_server_endpoint(self.get_target_path())
 
         return Table(
             uri,
@@ -1539,9 +1508,7 @@ class RedisNoSqlTarget(NoSqlBaseTarget):
         )
 
     def get_spark_options(self, key_column=None, timestamp_key=None, overwrite=True):
-        endpoint, uri = self.get_server_endpoint(
-            self.get_target_path(), self.credentials_prefix
-        )
+        endpoint, uri = self.get_server_endpoint(self.get_target_path())
         parsed_endpoint = urlparse(endpoint)
         store, path_in_store, path = self._get_store_and_path()
         return {
@@ -1592,7 +1559,6 @@ class RedisNoSqlTarget(NoSqlBaseTarget):
             class_name="mlrun.datastore.storeytargets.RedisNoSqlStoreyTarget",
             columns=column_list,
             table=table,
-            credentials_prefix=self.credentials_prefix,
             **self.attributes,
         )
 
@@ -1648,7 +1614,6 @@ class KafkaTarget(BaseStoreTarget):
     :param path:                topic name e.g. "my_topic"
     :param after_step:          optional, after what step in the graph to add the target
     :param columns:             optional, which columns from data to write
-    :param bootstrap_servers:   Deprecated. Use the brokers parameter instead
     :param producer_options:    additional configurations for kafka producer
     :param brokers:             kafka broker as represented by a host:port pair, or a list of kafka brokers, e.g.
         "localhost:9092", or ["kafka-broker-1:9092", "kafka-broker-2:9092"]
@@ -1664,27 +1629,12 @@ class KafkaTarget(BaseStoreTarget):
     def __init__(
         self,
         *args,
-        bootstrap_servers=None,
         producer_options=None,
         brokers=None,
         **kwargs,
     ):
         attrs = {}
 
-        # TODO: Remove this in 1.9.0
-        if bootstrap_servers:
-            if brokers:
-                raise mlrun.errors.MLRunInvalidArgumentError(
-                    "KafkaTarget cannot be created with both the 'brokers' parameter and the deprecated "
-                    "'bootstrap_servers' parameter. Please use 'brokers' only."
-                )
-            warnings.warn(
-                "'bootstrap_servers' parameter is deprecated in 1.7.0 and will be removed in 1.9.0, "
-                "use 'brokers' instead.",
-                FutureWarning,
-            )
-            brokers = bootstrap_servers
-
         if brokers:
             attrs["brokers"] = brokers
         if producer_options is not None:
@@ -2239,7 +2189,7 @@ def _get_target_path(driver, resource, run_id_mode=False, netloc=None, scheme=""
         else "vectors"
     )
     name = resource.metadata.name
-    project = resource.metadata.project or mlrun.mlconf.default_project
+    project = resource.metadata.project or mlrun.mlconf.active_project
 
     default_kind_name = kind
     if scheme == "ds":

mlrun/datastore/utils.py

@@ -16,7 +16,6 @@ import math
 import tarfile
 import tempfile
 import typing
-import warnings
 from urllib.parse import parse_qs, urlparse
 
 import pandas as pd
@@ -171,16 +170,7 @@ def _generate_sql_query_with_time_filter(
 def get_kafka_brokers_from_dict(options: dict, pop=False) -> typing.Optional[str]:
     get_or_pop = options.pop if pop else options.get
     kafka_brokers = get_or_pop("kafka_brokers", None)
-    if kafka_brokers:
-        return kafka_brokers
-    kafka_bootstrap_servers = get_or_pop("kafka_bootstrap_servers", None)
-    if kafka_bootstrap_servers:
-        warnings.warn(
-            "The 'kafka_bootstrap_servers' parameter is deprecated and will be removed in "
-            "1.9.0. Please pass the 'kafka_brokers' parameter instead.",
-            FutureWarning,
-        )
-    return kafka_bootstrap_servers
+    return kafka_brokers
 
 
 def transform_list_filters_to_tuple(additional_filters):

mlrun/db/base.py

@@ -185,6 +185,7 @@ class RunDBInterface(ABC):
         kind: Optional[str] = None,
         category: Union[str, mlrun.common.schemas.ArtifactCategories] = None,
         tree: Optional[str] = None,
+        parent: Optional[str] = None,
         format_: mlrun.common.formatters.ArtifactFormat = mlrun.common.formatters.ArtifactFormat.full,
         limit: Optional[int] = None,
         partition_by: Optional[
@@ -441,10 +442,10 @@ class RunDBInterface(ABC):
     ) -> dict:
         pass
 
-    # TODO: remove in 1.9.0
+    # TODO: remove in 1.10.0
     @deprecated(
-        version="1.9.0",
-        reason="'list_features' will be removed in 1.9.0, use 'list_features_v2' instead",
+        version="1.7.0",
+        reason="'list_features' will be removed in 1.10.0, use 'list_features_v2' instead",
         category=FutureWarning,
     )
     @abstractmethod
@@ -469,10 +470,10 @@ class RunDBInterface(ABC):
     ) -> mlrun.common.schemas.FeaturesOutputV2:
         pass
 
-    # TODO: remove in 1.9.0
+    # TODO: remove in 1.10.0
     @deprecated(
-        version="1.9.0",
-        reason="'list_entities' will be removed in 1.9.0, use 'list_entities_v2' instead",
+        version="1.7.0",
+        reason="'list_entities' will be removed in 1.10.0, use 'list_entities_v2' instead",
         category=FutureWarning,
     )
     @abstractmethod