moose-lib 0.6.140__tar.gz → 0.6.142__tar.gz

{moose_lib-0.6.140 → moose_lib-0.6.142}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: moose_lib
-Version: 0.6.140
+Version: 0.6.142
 Home-page: https://www.fiveonefour.com/moose
 Author: Fiveonefour Labs Inc.
 Author-email: support@fiveonefour.com

{moose_lib-0.6.140 → moose_lib-0.6.142}/moose_lib/data_models.py

@@ -35,6 +35,11 @@ def clickhouse_default(expression: str) -> ClickhouseDefault:
     return ClickhouseDefault(expression=expression)
 
 
+@dataclasses.dataclass(frozen=True)
+class ClickHouseTTL:
+    expression: str
+
+
 def clickhouse_decimal(precision: int, scale: int) -> Type[Decimal]:
     return Annotated[Decimal, Field(max_digits=precision, decimal_places=scale)]
 
@@ -194,6 +199,7 @@ class Column(BaseModel):
     primary_key: bool
     default: str | None = None
     annotations: list[Tuple[str, Any]] = []
+    ttl: str | None = None
 
     def to_expr(self):
         # Lazy import to avoid circular dependency at import time
@@ -409,6 +415,12 @@ def _to_columns(model: type[BaseModel]) -> list[Column]:
             None,
         )
 
+        # Extract TTL expression from metadata, if provided
+        ttl_expr = next(
+            (md.expression for md in mds if isinstance(md, ClickHouseTTL)),
+            None,
+        )
+
         columns.append(
             Column(
                 name=column_name,
@@ -418,6 +430,7 @@ def _to_columns(model: type[BaseModel]) -> list[Column]:
                 primary_key=primary_key,
                 default=default_expr,
                 annotations=annotations,
+                ttl=ttl_expr,
             )
         )
     return columns

{moose_lib-0.6.140 → moose_lib-0.6.142}/moose_lib/dmv2/olap_table.py

@@ -112,6 +112,8 @@ class OlapConfig(BaseModel):
                              `order_by_expression="(id, name)"` is equivalent to order_by_fields=["id", "name"], or
                              "tuple()" for no sorting.
         partition_by: Optional PARTITION BY expression (single ClickHouse SQL expression).
+        sample_by_expression: Optional SAMPLE BY expression for data sampling (single ClickHouse SQL expression).
+                              Used to enable efficient approximate query processing with SAMPLE clause.
         engine: The ClickHouse table engine to use. Can be either a ClickHouseEngines enum value
                 (for backward compatibility) or an EngineConfig instance (recommended).
         version: Optional version string for tracking configuration changes.
@@ -123,11 +125,15 @@ class OlapConfig(BaseModel):
     order_by_fields: list[str] = []
     order_by_expression: Optional[str] = None
     partition_by: Optional[str] = None
+    sample_by_expression: Optional[str] = None
     engine: Optional[Union[ClickHouseEngines, EngineConfig]] = None
     version: Optional[str] = None
     metadata: Optional[dict] = None
     life_cycle: Optional[LifeCycle] = None
     settings: Optional[dict[str, str]] = None
+    # Optional table-level TTL expression (without leading 'TTL')
+    ttl: Optional[str] = None
+
     # Optional secondary/data-skipping indexes
     class TableIndex(BaseModel):
         name: str

{moose_lib-0.6.140 → moose_lib-0.6.142}/moose_lib/internal.py

@@ -150,6 +150,7 @@ class TableConfig(BaseModel):
         columns: List of columns with their types and attributes.
         order_by: List of columns used for the ORDER BY clause.
         partition_by: The column name used for the PARTITION BY clause.
+        sample_by_expression: Optional SAMPLE BY expression for data sampling.
         engine_config: Engine configuration with type-safe, engine-specific parameters.
         version: Optional version string of the table configuration.
         metadata: Optional metadata for the table.
@@ -162,12 +163,14 @@ class TableConfig(BaseModel):
     columns: List[Column]
     order_by: List[str] | str
     partition_by: Optional[str]
+    sample_by_expression: Optional[str] = None
     engine_config: Optional[EngineConfigDict] = Field(None, discriminator='engine')
     version: Optional[str] = None
     metadata: Optional[dict] = None
     life_cycle: Optional[str] = None
     table_settings: Optional[dict[str, str]] = None
     indexes: list[OlapConfig.TableIndex] = []
+    ttl: Optional[str] = None
 
 
 class TopicConfig(BaseModel):
@@ -601,6 +604,7 @@ def to_infra_map() -> dict:
             columns=table._column_list,
             order_by=order_by_value,
             partition_by=table.config.partition_by,
+            sample_by_expression=table.config.sample_by_expression,
             engine_config=engine_config,
             version=table.config.version,
             metadata=getattr(table, "metadata", None),
@@ -608,6 +612,7 @@ def to_infra_map() -> dict:
             # Map 'settings' to 'table_settings' for internal use
             table_settings=table_settings if table_settings else None,
             indexes=table.config.indexes,
+            ttl=table.config.ttl,
         )
 
     for name, stream in get_streams().items():

{moose_lib-0.6.140 → moose_lib-0.6.142}/moose_lib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: moose_lib
-Version: 0.6.140
+Version: 0.6.142
 Home-page: https://www.fiveonefour.com/moose
 Author: Fiveonefour Labs Inc.
 Author-email: support@fiveonefour.com