chalkpy 2.93.3__py3-none-any.whl → 2.93.4__py3-none-any.whl

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.93.3"
+__version__ = "2.93.4"

chalk/features/feature_field.py

@@ -405,12 +405,10 @@ class Feature(Generic[_TPrim, _TRich]):
 
         if offline_ttl is None:
             offline_ttl = timedelta(0)
-        elif offline_ttl is ...:
-            # Should we allow the offline_ttl to be set via the class decorator?
-            offline_ttl = CHALK_MAX_TIMEDELTA
         elif isinstance(offline_ttl, str):
             offline_ttl = parse_chalk_duration(offline_ttl)
-        self.offline_ttl = offline_ttl
+        if offline_ttl is not ...:
+            self.offline_ttl = offline_ttl
 
         self.cache_strategy = cache_strategy
 

chalk/features/feature_set.py

@@ -179,6 +179,7 @@ class FeaturesImpl(metaclass=FeaturesMeta):
     __chalk_is_singleton__: ClassVar[bool]
     __chalk_etl_offline_to_online__: ClassVar[bool]
     __chalk_max_staleness__: ClassVar[timedelta]
+    __chalk_offline_ttl__: ClassVar[timedelta]
     __chalk_cache_strategy__: ClassVar[CacheStrategy]
     __chalk_namespace__: ClassVar[str]
     __chalk_primary__: ClassVar[Feature | None] = None  # The primary key feature

chalk/features/feature_set_decorator.py

@@ -34,7 +34,7 @@ from chalk.streams import Windowed
 from chalk.streams._windows import GroupByWindowed, get_name_with_duration
 from chalk.utils import notebook
 from chalk.utils.collections import ensure_tuple
-from chalk.utils.duration import Duration, parse_chalk_duration, parse_chalk_duration_s
+from chalk.utils.duration import CHALK_MAX_TIMEDELTA, Duration, parse_chalk_duration, parse_chalk_duration_s
 from chalk.utils.metaprogramming import MISSING, set_new_attribute
 from chalk.utils.string import to_snake_case
 from chalk.stores.online_store_config import OnlineStoreConfig
@@ -65,6 +65,7 @@ def features(
     singleton: bool = False,
     cache_nulls: CacheNullsType = True,
     cache_defaults: CacheDefaultsType = True,
+    offline_ttl: Optional[Duration] = None,
 ) -> Callable[[Type[T]], Type[T]]: ...
 
 
@@ -84,6 +85,7 @@ def features(
     online_store_config: Optional[OnlineStoreConfig] = None,
     cache_nulls: CacheNullsType = True,
     cache_defaults: CacheDefaultsType = True,
+    offline_ttl: Optional[Duration] = None,
 ) -> Union[Callable[[Type[T]], Type[T]], Type[T]]:
     """Chalk lets you spell out your features directly in Python.
 
@@ -141,6 +143,8 @@ def features(
         The `cache_nulls` and `cache_defaults` options can be used together on the same feature with the
         following exceptions: if `cache_nulls=False`, then `cache_defaults` cannot be `"evict_defaults"`, and if
         `cache_nulls="evict_defaults"`, then `cache_defaults` cannot be `False`.
+    offline_ttl
+        Sets a maximum age for values eligible to be retrieved from the offline store, defined in relation to the query's current point-in-time.
     Other Parameters
     ----------------
     cls
@@ -186,6 +190,7 @@ def features(
             node=source_info and source_info.tree,
         )
         nonlocal max_staleness
+        nonlocal offline_ttl
         if name is not None and re.sub(r"[^a-z_0-9]", "", namespace) != namespace:
             error_builder.add_diagnostic(
                 message=(
@@ -220,6 +225,19 @@ def features(
                     raise_error=ValueError,
                     code="13",
                 )
+        if offline_ttl is None:
+            offline_ttl = CHALK_MAX_TIMEDELTA
+        else:
+            try:
+                offline_ttl = parse_chalk_duration(offline_ttl)
+            except ValueError as e:
+                error_builder.add_diagnostic(
+                    message=f"Invalid 'offline_ttl'. {e.args[0]}",
+                    label=f"invalid duration {offline_ttl}",
+                    range=error_builder.decorator_kwarg_value_range(kwarg="offline_ttl"),
+                    raise_error=ValueError,
+                    code="13",
+                )
 
         cache_strategy = get_cache_strategy_from_cache_settings(
             cache_nulls=cache_nulls,
@@ -271,6 +289,7 @@ def features(
             tags=ensure_tuple(tags),
             etl_offline_to_online=etl_offline_to_online,
             max_staleness=max_staleness,
+            offline_ttl=offline_ttl,
             cache_strategy=cache_strategy,
             namespace=namespace,
             singleton=singleton,
@@ -419,6 +438,7 @@ def _get_field(
     class_tags: Optional[Tuple[str, ...]],
     class_etl_offline_to_online: bool,
     class_max_staleness: timedelta,
+    class_offline_ttl: timedelta,
     namespace: str,
     is_singleton: bool,
     class_cache_strategy: CacheStrategy = CacheStrategy.ALL,
@@ -521,6 +541,7 @@ def _get_field(
         class_tags=class_tags,
         class_etl_offline_to_online=class_etl_offline_to_online,
         class_max_staleness=class_max_staleness,
+        class_offline_ttl=class_offline_ttl,
         class_cache_strategy=class_cache_strategy,
         error_builder=error_builder,
     )
@@ -534,6 +555,7 @@ def _process_field(
     class_tags: Optional[Tuple[str, ...]],
     class_etl_offline_to_online: bool,
     class_max_staleness: timedelta,
+    class_offline_ttl: timedelta,
     error_builder: FeatureClassErrorBuilder,
     class_cache_strategy: CacheStrategy = CacheStrategy.ALL,
 ) -> Feature:
@@ -596,6 +618,10 @@ def _process_field(
     if not hasattr(f, "max_staleness"):
         f.max_staleness = class_max_staleness
 
+    # The attribute is not defined if the feature intends to use the class default
+    if not hasattr(f, "offline_ttl"):
+        f.offline_ttl = class_offline_ttl
+
     f_cache_nulls, f_cache_defaults = get_cache_settings_from_strategy(f.cache_strategy)
     class_cache_nulls, class_cache_defaults =  get_cache_settings_from_strategy(class_cache_strategy)
 
@@ -800,6 +826,7 @@ def _process_class(
     tags: Tuple[str, ...],
     etl_offline_to_online: bool,
     max_staleness: timedelta,
+    offline_ttl: timedelta,
     namespace: str,
     singleton: bool,
     online_store_config: Optional[OnlineStoreConfig],
@@ -1123,6 +1150,7 @@ def _process_class(
             typ=int,
             pyarrow_dtype=pa.uint8(),
             max_staleness=None,
+            offline_ttl=None,
             cache_strategy=CacheStrategy.ALL,
             etl_offline_to_online=False,
             is_autogenerated=True,
@@ -1204,6 +1232,7 @@ def _process_class(
             class_tags=tuple(cls.__chalk_tags__),
             class_etl_offline_to_online=cls.__chalk_etl_offline_to_online__,
             class_max_staleness=cls.__chalk_max_staleness__,
+            class_offline_ttl=cls.__chalk_offline_ttl__,
             class_cache_strategy=cls.__chalk_cache_strategy__,
         )
 
@@ -1237,6 +1266,7 @@ def _process_class(
     set_new_attribute(cls=cls, name="__chalk_owner__", value=owner)
     set_new_attribute(cls=cls, name="__chalk_tags__", value=list(tags))
     set_new_attribute(cls=cls, name="__chalk_max_staleness__", value=max_staleness)
+    set_new_attribute(cls=cls, name="__chalk_offline_ttl__", value=offline_ttl)
     set_new_attribute(cls=cls, name="__chalk_cache_strategy__", value=cache_strategy)
     set_new_attribute(cls=cls, name="__is_features__", value=True)
     set_new_attribute(cls=cls, name="__len__", value=_len_fn)
@@ -1270,6 +1300,7 @@ def _process_class(
             class_tags=tags,
             class_etl_offline_to_online=etl_offline_to_online,
             class_max_staleness=max_staleness,
+            class_offline_ttl=offline_ttl,
             class_cache_strategy=cache_strategy,
             namespace=namespace,
             is_singleton=singleton,
@@ -1325,6 +1356,7 @@ def _process_class(
                     class_tags=tags,
                     class_etl_offline_to_online=etl_offline_to_online,
                     class_max_staleness=max_staleness,
+                    class_offline_ttl=offline_ttl,
                     class_cache_strategy=cache_strategy,
                     error_builder=error_builder,
                 )
@@ -1532,6 +1564,7 @@ def _class_setattr(
         class_tags=tuple(cls.__chalk_tags__),
         class_etl_offline_to_online=cls.__chalk_etl_offline_to_online__,
         class_max_staleness=cls.__chalk_max_staleness__,
+        class_offline_ttl=cls.__chalk_offline_ttl__,
         class_cache_strategy=cls.__chalk_cache_strategy__,
     )
     if existing_feature is not None:

{chalkpy-2.93.3.dist-info → chalkpy-2.93.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkpy
-Version: 2.93.3
+Version: 2.93.4
 Summary: Python SDK for Chalk
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai

{chalkpy-2.93.3.dist-info → chalkpy-2.93.4.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=9YxwkAt3Im0OCMfpmnIB_4PKjZfBCcRmwATLXdHNRm4,2609
-chalk/_version.py,sha256=a8M8W7TWOUNnBbncydFQJ2EtW4CRYKZRaSSP5FM-rG0,23
+chalk/_version.py,sha256=utCJDEQ9kRnlcF3Y3HW8XTiSD1NNySR-0CrYIG6jZyo,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=hCEo7eqSfXZWklkFB2geeipGhiD0qNjPBpQJvOBW6N0,63083
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -585,9 +585,9 @@ chalk/features/_last.py,sha256=IyYe0PVAKBu8FRhCZiK6Dy5afKbwDHEUTES5qVXbaZU,1163
 chalk/features/_tensor.py,sha256=2SQz9cnIywGjHL6LoWj4lrVBa854McFUttJSNV1hKp8,7111
 chalk/features/_vector.py,sha256=jtJmUTtCaUFru4Fw17PYWozl3pPEdIOCYAuZOqIaN3Y,7205
 chalk/features/feature_cache_strategy.py,sha256=bXV-tjHfPzUotMpZ3h_D9Xxq-V1CLj_UcVtGIGMpMkI,4942
-chalk/features/feature_field.py,sha256=tLP9iybcYa7eM2eJJ0iSRks-ptyabA0l_SAJ83t6nEw,93758
-chalk/features/feature_set.py,sha256=yNi0_J4CylAVkVp1Y67qV6i8vHMdE0p99DnyLPABPHI,12406
-chalk/features/feature_set_decorator.py,sha256=HMGnQdXQevpsTCNGIG2njViHDnm1ltFoZDhABFkp3Dw,65059
+chalk/features/feature_field.py,sha256=A2QU5lKKo1QSKC6wWH6xLLgKjHzT2c_WJ3ZuA4XyHJM,93637
+chalk/features/feature_set.py,sha256=Hhf2PhdaYbdfCdo6DmsXoFKHR-CUcI-FIGLM_pN4HFE,12453
+chalk/features/feature_set_decorator.py,sha256=EqW8qDUZzzcSdHCg8sTjqIA1FmxQ3UuzXuliThr7KU4,66580
 chalk/features/feature_time.py,sha256=iUk8NDelig81jP7QT3tguyzx5eOZ-YC84OVgJRRKVwo,1639
 chalk/features/feature_wrapper.py,sha256=OolNWGGX67IAEMHCObFvOCpH5EmwjbMvMygRSBJJtu0,19259
 chalk/features/filter.py,sha256=2ldMbqvXC-nJ0jc-OZ36qHtrej-Jkx4TNQ1W_NZodAs,11177
@@ -764,8 +764,8 @@ chalk/utils/tracing.py,sha256=Glx8YrtjWy0zE5YbpgfgcsLDshAKnnYm9poiWNeCxXs,11075
 chalk/utils/weak_set_by_identity.py,sha256=VmikA_laYwFeOphCwXJIuyOIkrdlQe0bSzaXq7onoQw,953
 chalk/utils/pydanticutil/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/utils/pydanticutil/pydantic_compat.py,sha256=O575lLYJ5GvZC4HMzR9yATxf9XwjC6NrDUXbNwZidlE,3031
-chalkpy-2.93.3.dist-info/METADATA,sha256=RSXT5f7CE9lqU59166jt7xmFAmch5JkwBKp18rgDgqs,27494
-chalkpy-2.93.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-chalkpy-2.93.3.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.93.3.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.93.3.dist-info/RECORD,,
+chalkpy-2.93.4.dist-info/METADATA,sha256=YmIFo4vQ1F-AsT9gysGlfwvEo_6yD_5QFHPfM-pUDKI,27494
+chalkpy-2.93.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chalkpy-2.93.4.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.93.4.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.93.4.dist-info/RECORD,,