moose-lib 0.6.139__tar.gz → 0.6.141__tar.gz

{moose_lib-0.6.139 → moose_lib-0.6.141}/PKG-INFO

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

{moose_lib-0.6.139 → moose_lib-0.6.141}/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,21 @@ 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 secondary/data-skipping indexes
+    class TableIndex(BaseModel):
+        name: str
+        expression: str
+        type: str
+        arguments: Optional[list[str]] = None
+        granularity: int
+
+    indexes: list[TableIndex] = []
 
     def model_post_init(self, __context):
         has_fields = bool(self.order_by_fields)

{moose_lib-0.6.139 → moose_lib-0.6.141}/moose_lib/internal.py

@@ -21,8 +21,7 @@ from moose_lib.dmv2 import (
     get_workflows,
     get_web_apps,
     OlapTable,
-    View,
-    MaterializedView,
+    OlapConfig,
     SqlResource
 )
 from moose_lib.dmv2.stream import KafkaSchemaConfig
@@ -151,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.
@@ -163,11 +163,13 @@ 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
+    table_settings: Optional[dict[str, str]] = None
+    indexes: list[OlapConfig.TableIndex] = []
 
 
 class TopicConfig(BaseModel):
@@ -601,12 +603,14 @@ 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),
             life_cycle=table.config.life_cycle.value if table.config.life_cycle else None,
             # Map 'settings' to 'table_settings' for internal use
             table_settings=table_settings if table_settings else None,
+            indexes=table.config.indexes,
         )
 
     for name, stream in get_streams().items():

{moose_lib-0.6.139 → moose_lib-0.6.141}/moose_lib.egg-info/PKG-INFO

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