chalkpy 2.94.0__tar.gz → 2.94.2__tar.gz

{chalkpy-2.94.0 → chalkpy-2.94.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkpy
-Version: 2.94.0
+Version: 2.94.2
 Summary: Python SDK for Chalk
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai
@@ -24,7 +24,7 @@ Requires-Dist: cattrs<25,>=22.1.0
 Requires-Dist: dataclasses_json>=0.5.7
 Requires-Dist: executing<3,>=1.2.0
 Requires-Dist: googleapis-common-protos>=1.56.0
-Requires-Dist: grpcio<2,>=1.59.2
+Requires-Dist: grpcio<2,>=1.63.0
 Requires-Dist: ipywidgets>=8.0.6
 Requires-Dist: isodate<0.8,>=0.6.1
 Requires-Dist: numpy<3

chalkpy-2.94.2/chalk/_gen/chalk/engine/v1/query_planning_server_pb2.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/engine/v1/query_planning_server.proto
+# Protobuf Python Version: 4.25.3
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from chalk._gen.chalk.enginestorage.v1 import adhoc_query_pb2 as chalk_dot_enginestorage_dot_v1_dot_adhoc__query__pb2
+from chalk._gen.chalk.planner.v1 import batch_operator_pb2 as chalk_dot_planner_dot_v1_dot_batch__operator__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n+chalk/engine/v1/query_planning_server.proto\x12\x0f\x63halk.engine.v1\x1a(chalk/enginestorage/v1/adhoc_query.proto\x1a%chalk/planner/v1/batch_operator.proto"x\n\x16\x43reateBatchPlanRequest\x12^\n\x13\x61ll_planning_inputs\x18\x01 \x01(\x0b\x32..chalk.enginestorage.v1.AllPlanningInputsProtoR\x11\x61llPlanningInputs"r\n\x17\x43reateBatchPlanResponse\x12:\n\nbatch_plan\x18\x01 \x01(\x0b\x32\x1b.chalk.planner.v1.BatchPlanR\tbatchPlan\x12\x1b\n\tcache_hit\x18\x02 \x01(\x08R\x08\x63\x61\x63heHit2~\n\x14QueryPlanningService\x12\x66\n\x0f\x43reateBatchPlan\x12\'.chalk.engine.v1.CreateBatchPlanRequest\x1a(.chalk.engine.v1.CreateBatchPlanResponse"\x00\x42\x8d\x01\n\x13\x63om.chalk.engine.v1B\x18QueryPlanningServerProtoP\x01\xa2\x02\x03\x43\x45X\xaa\x02\x0f\x43halk.Engine.V1\xca\x02\x0f\x43halk\\Engine\\V1\xe2\x02\x1b\x43halk\\Engine\\V1\\GPBMetadata\xea\x02\x11\x43halk::Engine::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.engine.v1.query_planning_server_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\023com.chalk.engine.v1B\030QueryPlanningServerProtoP\001\242\002\003CEX\252\002\017Chalk.Engine.V1\312\002\017Chalk\\Engine\\V1\342\002\033Chalk\\Engine\\V1\\GPBMetadata\352\002\021Chalk::Engine::V1"
+    _globals["_CREATEBATCHPLANREQUEST"]._serialized_start = 145
+    _globals["_CREATEBATCHPLANREQUEST"]._serialized_end = 265
+    _globals["_CREATEBATCHPLANRESPONSE"]._serialized_start = 267
+    _globals["_CREATEBATCHPLANRESPONSE"]._serialized_end = 381
+    _globals["_QUERYPLANNINGSERVICE"]._serialized_start = 383
+    _globals["_QUERYPLANNINGSERVICE"]._serialized_end = 509
+# @@protoc_insertion_point(module_scope)

chalkpy-2.94.2/chalk/_gen/chalk/engine/v1/query_planning_server_pb2.pyi

@@ -0,0 +1,25 @@
+from chalk._gen.chalk.enginestorage.v1 import adhoc_query_pb2 as _adhoc_query_pb2
+from chalk._gen.chalk.planner.v1 import batch_operator_pb2 as _batch_operator_pb2
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class CreateBatchPlanRequest(_message.Message):
+    __slots__ = ("all_planning_inputs",)
+    ALL_PLANNING_INPUTS_FIELD_NUMBER: _ClassVar[int]
+    all_planning_inputs: _adhoc_query_pb2.AllPlanningInputsProto
+    def __init__(
+        self, all_planning_inputs: _Optional[_Union[_adhoc_query_pb2.AllPlanningInputsProto, _Mapping]] = ...
+    ) -> None: ...
+
+class CreateBatchPlanResponse(_message.Message):
+    __slots__ = ("batch_plan", "cache_hit")
+    BATCH_PLAN_FIELD_NUMBER: _ClassVar[int]
+    CACHE_HIT_FIELD_NUMBER: _ClassVar[int]
+    batch_plan: _batch_operator_pb2.BatchPlan
+    cache_hit: bool
+    def __init__(
+        self, batch_plan: _Optional[_Union[_batch_operator_pb2.BatchPlan, _Mapping]] = ..., cache_hit: bool = ...
+    ) -> None: ...

chalkpy-2.94.2/chalk/_gen/chalk/engine/v1/query_planning_server_pb2_grpc.py

@@ -0,0 +1,80 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+
+import grpc
+
+from chalk._gen.chalk.engine.v1 import (
+    query_planning_server_pb2 as chalk_dot_engine_dot_v1_dot_query__planning__server__pb2,
+)
+
+
+class QueryPlanningServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateBatchPlan = channel.unary_unary(
+            "/chalk.engine.v1.QueryPlanningService/CreateBatchPlan",
+            request_serializer=chalk_dot_engine_dot_v1_dot_query__planning__server__pb2.CreateBatchPlanRequest.SerializeToString,
+            response_deserializer=chalk_dot_engine_dot_v1_dot_query__planning__server__pb2.CreateBatchPlanResponse.FromString,
+        )
+
+
+class QueryPlanningServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def CreateBatchPlan(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+
+def add_QueryPlanningServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "CreateBatchPlan": grpc.unary_unary_rpc_method_handler(
+            servicer.CreateBatchPlan,
+            request_deserializer=chalk_dot_engine_dot_v1_dot_query__planning__server__pb2.CreateBatchPlanRequest.FromString,
+            response_serializer=chalk_dot_engine_dot_v1_dot_query__planning__server__pb2.CreateBatchPlanResponse.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler("chalk.engine.v1.QueryPlanningService", rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+# This class is part of an EXPERIMENTAL API.
+class QueryPlanningService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def CreateBatchPlan(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.engine.v1.QueryPlanningService/CreateBatchPlan",
+            chalk_dot_engine_dot_v1_dot_query__planning__server__pb2.CreateBatchPlanRequest.SerializeToString,
+            chalk_dot_engine_dot_v1_dot_query__planning__server__pb2.CreateBatchPlanResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalkpy-2.94.2/chalk/_gen/chalk/engine/v1/query_planning_server_pb2_grpc.pyi

@@ -0,0 +1,36 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from abc import (
+    ABCMeta,
+    abstractmethod,
+)
+from chalk._gen.chalk.engine.v1.query_planning_server_pb2 import (
+    CreateBatchPlanRequest,
+    CreateBatchPlanResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class QueryPlanningServiceStub:
+    def __init__(self, channel: Channel) -> None: ...
+    CreateBatchPlan: UnaryUnaryMultiCallable[
+        CreateBatchPlanRequest,
+        CreateBatchPlanResponse,
+    ]
+
+class QueryPlanningServiceServicer(metaclass=ABCMeta):
+    @abstractmethod
+    def CreateBatchPlan(
+        self,
+        request: CreateBatchPlanRequest,
+        context: ServicerContext,
+    ) -> CreateBatchPlanResponse: ...
+
+def add_QueryPlanningServiceServicer_to_server(servicer: QueryPlanningServiceServicer, server: Server) -> None: ...

chalkpy-2.94.2/chalk/_version.py

@@ -0,0 +1 @@
+__version__ = "2.94.2"

{chalkpy-2.94.0 → chalkpy-2.94.2}/chalk/importer.py

@@ -304,6 +304,19 @@ def _parse_agg_function_call(expr: Underscore | None) -> Tuple[str, Underscore,
                 f"expecting 'int' type argument for 'k', but received arg of type '{type(call_expr._chalk__kwargs.get('k'))}'"
             )
         opts = FrozenOrderedSet(call_expr._chalk__kwargs.items())
+    elif aggregation == "approx_percentile":
+        if len(call_expr._chalk__args) > 0:
+            raise ChalkParseError("should not have any positional arguments")
+        elif {"quantile"} != call_expr._chalk__kwargs.keys():
+            raise ChalkParseError("expecting exactly one required keyword argument 'quantile'")
+        elif not isinstance(call_expr._chalk__kwargs.get("quantile"), float):
+            raise ChalkParseError(
+                f"expecting 'float' type argument for 'quantile', but received arg of type '{type(call_expr._chalk__kwargs.get('quantile'))}'"
+            )
+        # TODO: expand proto definition to accept kwargs that are not necessarily `k`
+        quantile = call_expr._chalk__kwargs["quantile"]
+        nano_quantile = int(round(quantile * 1_000_000_000))
+        opts = FrozenOrderedSet([("k", nano_quantile)])
     elif aggregation in ("min_by_n", "max_by_n"):
         if len(call_expr._chalk__kwargs) > 0:
             raise ChalkParseError("should not have any keyword arguments")
@@ -570,39 +583,51 @@ def parse_grouped_window(f: Feature) -> WindowConfigResolved:
         aggregation_kwargs=aggregation_kwargs,
         pyarrow_dtype=pyarrow_dtype,
         filters=parsed_filters,
-        backfill_resolver=_try_parse_resolver_fqn(
-            "backfill_resolver",
-            f.window_materialization.get("backfill_resolver", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_schedule=f.window_materialization.get("backfill_schedule", None)
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_lookback_duration_seconds=_try_parse_duration(
-            "backfill_lookback_duration",
-            f.window_materialization.get("backfill_lookback_duration", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_start_time=_try_parse_datetime(
-            "backfill_start_time",
-            f.window_materialization.get("backfill_start_time", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        continuous_resolver=_try_parse_resolver_fqn(
-            "continuous_resolver",
-            f.window_materialization.get("continuous_resolver", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        continuous_buffer_duration_seconds=_try_parse_duration(
-            "continuous_buffer_duration",
-            f.window_materialization.get("continuous_buffer_duration", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
+        backfill_resolver=(
+            _try_parse_resolver_fqn(
+                "backfill_resolver",
+                f.window_materialization.get("backfill_resolver", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_schedule=(
+            f.window_materialization.get("backfill_schedule", None)
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_lookback_duration_seconds=(
+            _try_parse_duration(
+                "backfill_lookback_duration",
+                f.window_materialization.get("backfill_lookback_duration", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_start_time=(
+            _try_parse_datetime(
+                "backfill_start_time",
+                f.window_materialization.get("backfill_start_time", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        continuous_resolver=(
+            _try_parse_resolver_fqn(
+                "continuous_resolver",
+                f.window_materialization.get("continuous_resolver", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        continuous_buffer_duration_seconds=(
+            _try_parse_duration(
+                "continuous_buffer_duration",
+                f.window_materialization.get("continuous_buffer_duration", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
     )
 
     return cfg
@@ -798,39 +823,51 @@ def parse_windowed_materialization(f: Feature) -> WindowConfigResolved | None:
         aggregation_kwargs=aggregation_kwargs,
         pyarrow_dtype=f.converter.pyarrow_dtype,
         filters=parsed_filters,
-        backfill_resolver=_try_parse_resolver_fqn(
-            "backfill_resolver",
-            f.window_materialization.get("backfill_resolver", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_schedule=f.window_materialization.get("backfill_schedule", None)
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_lookback_duration_seconds=_try_parse_duration(
-            "backfill_lookback_duration",
-            f.window_materialization.get("backfill_lookback_duration", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        backfill_start_time=_try_parse_datetime(
-            "backfill_start_time",
-            f.window_materialization.get("backfill_start_time", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        continuous_resolver=_try_parse_resolver_fqn(
-            "continuous_resolver",
-            f.window_materialization.get("continuous_resolver", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
-        continuous_buffer_duration_seconds=_try_parse_duration(
-            "continuous_buffer_duration",
-            f.window_materialization.get("continuous_buffer_duration", None),
-        )
-        if isinstance(f.window_materialization, dict)
-        else None,
+        backfill_resolver=(
+            _try_parse_resolver_fqn(
+                "backfill_resolver",
+                f.window_materialization.get("backfill_resolver", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_schedule=(
+            f.window_materialization.get("backfill_schedule", None)
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_lookback_duration_seconds=(
+            _try_parse_duration(
+                "backfill_lookback_duration",
+                f.window_materialization.get("backfill_lookback_duration", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        backfill_start_time=(
+            _try_parse_datetime(
+                "backfill_start_time",
+                f.window_materialization.get("backfill_start_time", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        continuous_resolver=(
+            _try_parse_resolver_fqn(
+                "continuous_resolver",
+                f.window_materialization.get("continuous_resolver", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
+        continuous_buffer_duration_seconds=(
+            _try_parse_duration(
+                "continuous_buffer_duration",
+                f.window_materialization.get("continuous_buffer_duration", None),
+            )
+            if isinstance(f.window_materialization, dict)
+            else None
+        ),
     )
 
 

{chalkpy-2.94.0 → chalkpy-2.94.2}/chalk/parsed/to_proto.py

@@ -900,7 +900,7 @@ class ToProtoConverter:
                     else None,
                     backfill_schedule=mat.backfill_schedule,
                     approx_top_k_arg_k=aggregation_kwargs.get("k")
-                    if mat.aggregation in ("approx_top_k", "min_by_n", "max_by_n")
+                    if mat.aggregation in ("approx_top_k", "approx_percentile", "min_by_n", "max_by_n")
                     else None,
                 ),
                 tags=f.tags,
@@ -996,7 +996,7 @@ class ToProtoConverter:
                             else None,
                             continuous_resolver=wmp.continuous_resolver,
                             approx_top_k_arg_k=aggregation_kwargs.get("k")
-                            if wmp.aggregation in ("approx_top_k", "min_by_n", "max_by_n")
+                            if wmp.aggregation in ("approx_top_k", "approx_percentile", "min_by_n", "max_by_n")
                             else None,
                         )
                         if wmp is not None

{chalkpy-2.94.0 → chalkpy-2.94.2}/chalkpy.egg-info/SOURCES.txt

@@ -180,6 +180,10 @@ chalk/_gen/chalk/engine/v1/plan_server_pb2.py
 chalk/_gen/chalk/engine/v1/plan_server_pb2.pyi
 chalk/_gen/chalk/engine/v1/plan_server_pb2_grpc.py
 chalk/_gen/chalk/engine/v1/plan_server_pb2_grpc.pyi
+chalk/_gen/chalk/engine/v1/query_planning_server_pb2.py
+chalk/_gen/chalk/engine/v1/query_planning_server_pb2.pyi
+chalk/_gen/chalk/engine/v1/query_planning_server_pb2_grpc.py
+chalk/_gen/chalk/engine/v1/query_planning_server_pb2_grpc.pyi
 chalk/_gen/chalk/engine/v1/query_server_pb2.py
 chalk/_gen/chalk/engine/v1/query_server_pb2.pyi
 chalk/_gen/chalk/engine/v1/query_server_pb2_grpc.py

{chalkpy-2.94.0 → chalkpy-2.94.2}/pyproject.toml

@@ -33,7 +33,7 @@ dependencies = [
     "dataclasses_json>=0.5.7",
     "executing>=1.2.0,<3",
     "googleapis-common-protos>=1.56.0",
-    "grpcio>=1.59.2,<2",
+    "grpcio>=1.63.0,<2",
     "ipywidgets>=8.0.6",
     "isodate>=0.6.1,<0.8",
     "numpy<3",

chalkpy-2.94.0/chalk/_version.py

@@ -1 +0,0 @@
-__version__ = "2.94.0"