chalkpy 2.96.0__py3-none-any.whl → 2.96.1__py3-none-any.whl

chalk/_gen/chalk/engine/v2/introspection_service_pb2.py

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/engine/v2/introspection_service.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.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n+chalk/engine/v2/introspection_service.proto\x12\x0f\x63halk.engine.v2\x1a\x1f\x63halk/auth/v1/permissions.proto"8\n\x17ParseSQLResolverRequest\x12\x1d\n\nsql_string\x18\x01 \x01(\tR\tsqlString"L\n\x18ParseSQLResolverResponse\x12\x16\n\x06inputs\x18\x01 \x03(\tR\x06inputs\x12\x18\n\x07outputs\x18\x02 \x03(\tR\x07outputs2\x87\x01\n\x14IntrospectionService\x12o\n\x10ParseSQLResolver\x12(.chalk.engine.v2.ParseSQLResolverRequest\x1a).chalk.engine.v2.ParseSQLResolverResponse"\x06\x90\x02\x01\x80}\x02\x42\x8e\x01\n\x13\x63om.chalk.engine.v2B\x19IntrospectionServiceProtoP\x01\xa2\x02\x03\x43\x45X\xaa\x02\x0f\x43halk.Engine.V2\xca\x02\x0f\x43halk\\Engine\\V2\xe2\x02\x1b\x43halk\\Engine\\V2\\GPBMetadata\xea\x02\x11\x43halk::Engine::V2b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.engine.v2.introspection_service_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\023com.chalk.engine.v2B\031IntrospectionServiceProtoP\001\242\002\003CEX\252\002\017Chalk.Engine.V2\312\002\017Chalk\\Engine\\V2\342\002\033Chalk\\Engine\\V2\\GPBMetadata\352\002\021Chalk::Engine::V2"
+    _globals["_INTROSPECTIONSERVICE"].methods_by_name["ParseSQLResolver"]._options = None
+    _globals["_INTROSPECTIONSERVICE"].methods_by_name["ParseSQLResolver"]._serialized_options = b"\220\002\001\200}\002"
+    _globals["_PARSESQLRESOLVERREQUEST"]._serialized_start = 97
+    _globals["_PARSESQLRESOLVERREQUEST"]._serialized_end = 153
+    _globals["_PARSESQLRESOLVERRESPONSE"]._serialized_start = 155
+    _globals["_PARSESQLRESOLVERRESPONSE"]._serialized_end = 231
+    _globals["_INTROSPECTIONSERVICE"]._serialized_start = 234
+    _globals["_INTROSPECTIONSERVICE"]._serialized_end = 369
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/engine/v2/introspection_service_pb2.pyi

@@ -0,0 +1,21 @@
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Optional as _Optional
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class ParseSQLResolverRequest(_message.Message):
+    __slots__ = ("sql_string",)
+    SQL_STRING_FIELD_NUMBER: _ClassVar[int]
+    sql_string: str
+    def __init__(self, sql_string: _Optional[str] = ...) -> None: ...
+
+class ParseSQLResolverResponse(_message.Message):
+    __slots__ = ("inputs", "outputs")
+    INPUTS_FIELD_NUMBER: _ClassVar[int]
+    OUTPUTS_FIELD_NUMBER: _ClassVar[int]
+    inputs: _containers.RepeatedScalarFieldContainer[str]
+    outputs: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, inputs: _Optional[_Iterable[str]] = ..., outputs: _Optional[_Iterable[str]] = ...) -> None: ...

chalk/_gen/chalk/engine/v2/introspection_service_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.v2 import (
+    introspection_service_pb2 as chalk_dot_engine_dot_v2_dot_introspection__service__pb2,
+)
+
+
+class IntrospectionServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.ParseSQLResolver = channel.unary_unary(
+            "/chalk.engine.v2.IntrospectionService/ParseSQLResolver",
+            request_serializer=chalk_dot_engine_dot_v2_dot_introspection__service__pb2.ParseSQLResolverRequest.SerializeToString,
+            response_deserializer=chalk_dot_engine_dot_v2_dot_introspection__service__pb2.ParseSQLResolverResponse.FromString,
+        )
+
+
+class IntrospectionServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def ParseSQLResolver(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_IntrospectionServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "ParseSQLResolver": grpc.unary_unary_rpc_method_handler(
+            servicer.ParseSQLResolver,
+            request_deserializer=chalk_dot_engine_dot_v2_dot_introspection__service__pb2.ParseSQLResolverRequest.FromString,
+            response_serializer=chalk_dot_engine_dot_v2_dot_introspection__service__pb2.ParseSQLResolverResponse.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler("chalk.engine.v2.IntrospectionService", rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+# This class is part of an EXPERIMENTAL API.
+class IntrospectionService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def ParseSQLResolver(
+        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.v2.IntrospectionService/ParseSQLResolver",
+            chalk_dot_engine_dot_v2_dot_introspection__service__pb2.ParseSQLResolverRequest.SerializeToString,
+            chalk_dot_engine_dot_v2_dot_introspection__service__pb2.ParseSQLResolverResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/engine/v2/introspection_service_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.v2.introspection_service_pb2 import (
+    ParseSQLResolverRequest,
+    ParseSQLResolverResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class IntrospectionServiceStub:
+    def __init__(self, channel: Channel) -> None: ...
+    ParseSQLResolver: UnaryUnaryMultiCallable[
+        ParseSQLResolverRequest,
+        ParseSQLResolverResponse,
+    ]
+
+class IntrospectionServiceServicer(metaclass=ABCMeta):
+    @abstractmethod
+    def ParseSQLResolver(
+        self,
+        request: ParseSQLResolverRequest,
+        context: ServicerContext,
+    ) -> ParseSQLResolverResponse: ...
+
+def add_IntrospectionServiceServicer_to_server(servicer: IntrospectionServiceServicer, server: Server) -> None: ...

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.96.0"
+__version__ = "2.96.1"

chalk/sql/_internal/sql_file_resolver.py

@@ -837,10 +837,98 @@ class EscapedSqlString:
     errors: List[ResolverError]
 
 
+@dataclasses.dataclass
+class ParsedSqlResolver:
+    inputs: list[str]
+    outputs: list[str]
+
+
+def parse_sql_resolver(sql_string: str) -> ParsedSqlResolver:
+    """
+    Parse a SQL resolver string to extract input and output features.
+
+    Args:
+        sql_string: SQL string with ${feature.name} syntax for inputs and directive comments
+
+    Returns:
+        ParsedSqlResolver with inputs (feature names) and outputs (fully qualified column names)
+    """
+    try:
+        import sqlglot
+        import sqlglot.expressions
+    except ImportError:
+        raise missing_dependency_exception("chalkpy[runtime]")
+
+    from chalk._lsp.error_builder import SQLFileResolverErrorBuilder
+
+    error_builder = SQLFileResolverErrorBuilder(
+        uri="<parse-request>",
+        sql_string=sql_string,
+        has_import_errors=False,
+    )
+
+    # Extract inputs
+    escaped = escape_sql_params(
+        sql_string=sql_string,
+        path="<parse-request>",
+        error_builder=error_builder,
+        parse_only=True,
+    )
+    inputs = list(escaped.args.values())
+    sql_string = escaped.escaped_sql_string
+
+    # Parse directive comments
+    comments = ""
+    docstring = ""
+    for comment in sql_string.splitlines():
+        if comment.strip().startswith("--"):
+            stripped_comment = comment.strip().replace("--", "")
+            if stripped_comment.strip().startswith("-"):
+                comments += f"{stripped_comment}\n"
+            else:
+                if count_colons(stripped_comment) != 1:
+                    docstring += f"{stripped_comment.strip()}\n"
+                else:
+                    comments += f"{stripped_comment}\n"
+        else:
+            break
+
+    # Extract resolves and namespace directives to build the output namespace
+    namespace_str = None
+    if len(comments) != 0:
+        try:
+            comment_dict: Dict[str, Any] = yaml.safe_load(comments)
+        except Exception:
+            comment_dict = {}
+
+        if comment_dict and "resolves" in comment_dict:
+            resolves = comment_dict["resolves"]
+            namespace_from_comment = comment_dict.get("namespace")
+            namespace_str = build_namespaced_name(namespace=namespace_from_comment, name=to_snake_case(resolves))
+
+    # Parse SELECT columns and qualify with namespace if available
+    outputs = []
+    try:
+        glots = sqlglot.parse(sql=sql_string)
+        if glots:
+            glot = glots[0]
+            if isinstance(glot, (sqlglot.expressions.Select, sqlglot.expressions.Union)):
+                raw_outputs = list(glot.named_selects)
+                if namespace_str:
+                    outputs = [f"{namespace_str}.{col}" for col in raw_outputs]
+                else:
+                    outputs = raw_outputs
+    except Exception:
+        pass
+
+    return ParsedSqlResolver(inputs=inputs, outputs=outputs)
+
+
 def escape_sql_params(
     sql_string: str,
     path: str,
     error_builder: SQLFileResolverErrorBuilder,
+    parse_only: bool = False,
 ) -> EscapedSqlString:
     """
     Chalk allows people to write SQL resolvers that reference features using ${} syntax, such as:
@@ -853,6 +941,7 @@ def escape_sql_params(
     :param sql_string: String of hte original SQL resolver
     :param path: For error reporting, filepath of the SQL resolver
     :param error_builder: For LSP errors
+    :param parse_only: If True, skip feature registry lookups for default values
     """
     args = {}  # sql string -> input feature string
 
@@ -891,8 +980,9 @@ def escape_sql_params(
                     variable = split[0].strip()
                     default_arg = split[1].strip()
                     default_arg = json.loads(default_arg)
-                    f = Feature.from_root_fqn(variable)
-                    default_arg = f.converter.from_json_to_rich(default_arg)
+                    if not parse_only:
+                        f = Feature.from_root_fqn(variable)
+                        default_arg = f.converter.from_json_to_rich(default_arg)
                     default_args.append(default_arg)
                     has_default_arg = True
         if not has_default_arg:

{chalkpy-2.96.0.dist-info → chalkpy-2.96.1.dist-info}/METADATA

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

{chalkpy-2.96.0.dist-info → chalkpy-2.96.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
-chalk/_version.py,sha256=VFVEmaCPCelx8c52gLAGDX2mvkTzbt9BnxqellL0CWM,23
+chalk/_version.py,sha256=53oisDKLL-9FrciuE997yxhzB_eVulphF1oc5zt1rD8,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -193,6 +193,10 @@ chalk/_gen/chalk/engine/v2/feature_values_pb2.py,sha256=enBKvU-D_eVeY3at_g29_-tu
 chalk/_gen/chalk/engine/v2/feature_values_pb2.pyi,sha256=pVmjsH9_KUKS7SCF7PVwO6d7R2aMPn28RTU262mmbL0,4781
 chalk/_gen/chalk/engine/v2/feature_values_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
 chalk/_gen/chalk/engine/v2/feature_values_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
+chalk/_gen/chalk/engine/v2/introspection_service_pb2.py,sha256=CO4-cB9glhblVhG8omar9ZYcPAIJQBAdOKTp9t2jD8s,2621
+chalk/_gen/chalk/engine/v2/introspection_service_pb2.pyi,sha256=lGJA1i5WZs-Y9eCjiBKqyoc0Q0xf5dwMo79Axi-W9wM,987
+chalk/_gen/chalk/engine/v2/introspection_service_pb2_grpc.py,sha256=_R0b7rari7VXNjKOwfYPfj4U9z80rAibkRH0VcyP1-s,3000
+chalk/_gen/chalk/engine/v2/introspection_service_pb2_grpc.pyi,sha256=cuIlTQVMgH-Ve5dJGz6AsgO9bxfgJdnTXjigUuKFCLM,909
 chalk/_gen/chalk/engine/v2/offline_store_service_pb2.py,sha256=ESIxkWWz3YDLVa8sBrCqzxzpRu80sF7iRtu1jwnY_Zg,3999
 chalk/_gen/chalk/engine/v2/offline_store_service_pb2.pyi,sha256=hKEmA909OMJaS4CYHJShkuUUua05hHWp1R11IhK3-VQ,684
 chalk/_gen/chalk/engine/v2/offline_store_service_pb2_grpc.py,sha256=hQBnBhagqeqVbg8uFDEVRDS5cTuH7BDkQN5TkdFSHKM,10022
@@ -743,7 +747,7 @@ chalk/sql/_internal/chalk_query.py,sha256=O2Mv0JHWvSrP3iDfkooy22yoDuWAH57m1i8qcy
 chalk/sql/_internal/incremental.py,sha256=xqnohrZIoqE3y92edUl_oEkHQoFZw9GN8GIOUynyyEI,2900
 chalk/sql/_internal/query_execution_parameters.py,sha256=FT0GixOryGeKR1x7UrQNoGFtySd_8-B4Xrp38nFi0FU,3679
 chalk/sql/_internal/query_registry.py,sha256=jOk2x9NvW2pZocI_iTvgd9DyOlr4-ajz8rTi0kkLSaI,3005
-chalk/sql/_internal/sql_file_resolver.py,sha256=BCGTpe3tVLQTWNSrfyUFsg8VfO-pupeN8yxGpiSmy5E,77817
+chalk/sql/_internal/sql_file_resolver.py,sha256=0Ui8wRmkyzIBa929qYXwEqjrZ72N_QpVeH8foIGtfJ0,80822
 chalk/sql/_internal/sql_settings.py,sha256=9lcpHNrmEhr1Zxl_Ct7U0p0AbLUvlpSayxHEPmyqu8E,543
 chalk/sql/_internal/sql_source.py,sha256=083tGM7YzSsL-AhwTAtCzvGSjKxGVH_R5pILEpe5g1c,43825
 chalk/sql/_internal/sql_source_group.py,sha256=RIMEjEG3oSGwign37w9avpCRdVfVQepR8SPVv6JoOx0,5562
@@ -819,8 +823,8 @@ chalk/utils/tracing.py,sha256=NiiM-9dbuJhSCv6R1npR1uYNKWlkqTR6Ygm0Voi2NrY,13078
 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.96.0.dist-info/METADATA,sha256=h9BF0cHEBJti7iDvmLElN_TK1jm8NdIhpCaY5vez3_E,27754
-chalkpy-2.96.0.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-chalkpy-2.96.0.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.96.0.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.96.0.dist-info/RECORD,,
+chalkpy-2.96.1.dist-info/METADATA,sha256=Yju64p2JSKrDw5doqpVwJk2-ePFHdTKDN6oKULBj3As,27754
+chalkpy-2.96.1.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+chalkpy-2.96.1.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.96.1.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.96.1.dist-info/RECORD,,