Add typed ParameterPath as a replacement for KeyPath

src/draive/__init__.py

@@ -79,7 +79,7 @@
  openai_lmm_completion,
  openai_tokenize_text,
 )
-from draive.parameters import Argument, Field
+from draive.parameters import Argument, Field, ParameterPath
 from draive.scope import (
  ScopeDependencies,
  ScopeDependency,
@@ -219,6 +219,7 @@
  "OpenAIEmbeddingConfig",
  "OpenAIException",
  "OpenAIImageGenerationConfig",
+ "ParameterPath",
  "ReadOnlyMemory",
  "ScopeDependencies",
  "ScopeDependency",

src/draive/parameters/__init__.py

@@ -1,6 +1,7 @@
 from draive.parameters.data import Field, ParametrizedData
 from draive.parameters.definition import ParametersDefinition
 from draive.parameters.function import Argument, Function
+from draive.parameters.path import ParameterPath
 from draive.parameters.specification import ParametersSpecification, ToolSpecification
 from draive.parameters.tool import ParametrizedTool
 
@@ -14,4 +15,5 @@
  "ParametrizedData",
  "ParametrizedData",
  "ParametrizedTool",
+ "ParameterPath",
 ]

src/draive/parameters/data.py

@@ -6,11 +6,11 @@
 from dataclasses import dataclass
 from dataclasses import field as dataclass_field
 from dataclasses import fields as dataclass_fields
-from typing import Any, Self, cast, dataclass_transform, overload
+from typing import Any, ClassVar, Self, cast, dataclass_transform, overload
 
-from draive.keypaths import KeyPath
 from draive.parameters.definition import ParameterDefinition, ParametersDefinition
 from draive.parameters.missing import MISSING_PARAMETER, MissingParameter
+from draive.parameters.path import ParameterPath
 from draive.parameters.specification import ParameterSpecification
 from draive.parameters.validation import parameter_validator
 
@@ -203,17 +203,36 @@ def _field_parameter(
 
 
 class ParametrizedData(metaclass=ParametrizedDataMeta):
+ _: ClassVar[Self]
+
+ def __init_subclass__(cls) -> None:
+ super().__init_subclass__()
+ cls._: Self = cast(
+ Self,
+ ParameterPath(cls, cls), # type: ignore
+ )
+
  @classmethod
- def key_path(
+ def path(
  cls,
- path: str | None = None,
  /,
- ) -> KeyPath[Self]:
- return KeyPath(
- cls,
- path=path,
+ ) -> Self:
+ return cast(
+ Self,
+ ParameterPath(cls, cls), # type: ignore
  )
 
+ @classmethod
+ def path_cast[Parameter](
+ cls,
+ path: Parameter,
+ /,
+ ) -> ParameterPath[Self, Parameter]:
+ assert isinstance( # nosec: B101
+ path, ParameterPath
+ ), "Prepare parameter path by using Self._.path.to.property"
+ return cast(ParameterPath[Self, Parameter], path)
+
  @classmethod
  def validated(
  cls,
@@ -229,8 +248,10 @@ def validator(
  ) -> Self:
  if isinstance(value, cls):
  return value
+
  elif isinstance(value, dict):
  return cls.validated(**value)
+
  else:
  raise TypeError("Invalid value %s", value)