vgi-python 0.31.3__py3-none-any.whl → 0.32.0__py3-none-any.whl

vgi/__init__.py

@@ -26,6 +26,7 @@ from vgi.argument_spec import (
     VGI_VARARGS_TRUE,
     ArgumentSpec,
     argument_specs_to_schema,
+    parameter_default_values_from_specs,
     schema_to_argument_specs,
 )
 from vgi.arguments import (
@@ -42,13 +43,54 @@ from vgi.arguments import (
     TaggedUnion,
 )
 from vgi.auth import AuthContext, CallContext
+from vgi.filter_v2 import (
+    Arithmetic,
+    ArithmeticOperator,
+    BooleanExpression,
+    Call,
+    Cast,
+    ColumnRef,
+    Comparison,
+    ComparisonOperator,
+    EvaluationContext,
+    ExternalSet,
+    FieldRef,
+    FilterPredicate,
+    FilterState,
+    FilterV2Error,
+    FunctionIdentity,
+    In,
+    IsNull,
+    Literal,
+    LiteralSet,
+    Negate,
+    Not,
+    PredicateMode,
+    PredicateSource,
+    RuntimeFilter,
+    StandardFilterFunction,
+)
+from vgi.filter_v2_builder import (
+    FilterPayload,
+    artifact_payload,
+    build_filter_batch,
+    deserialize_filter_batch,
+    serialize_filter_batch,
+    type_payload,
+    value_payload,
+)
 from vgi.metadata import (
+    ArgumentMonotonicity,
     CatalogFunctionType,
+    EvaluationContextCapability,
+    FilterFunctionCapability,
+    FilterSemanticProfile,
     FunctionExample,
     FunctionStability,
     OrderPreservation,
     ParameterInfo,
     ResolvedMetadata,
+    RuntimeFilterAlgorithmCapability,
     TableInputValidationError,
     functions_to_arrow,
 )
@@ -65,6 +107,7 @@ from vgi.scalar_function import (
     ScalarFunctionGenerator,
     TypeMismatchError,
 )
+from vgi.schema_path import SchemaPath
 from vgi.schema_utils import schema, schema_like
 from vgi.table_filter_pushdown import (
     ColumnBounds,
@@ -86,20 +129,27 @@ from vgi.table_in_out_function import (
     TableInOutGenerator,
 )
 from vgi.worker import Worker
+from vgi.write_results import WriteOperation, WriteResultMode, write_changes_batch, write_result_schema
 
 __all__ = [
     "AggregateBindParams",
     "AggregateFunction",
+    "WriteOperation",
+    "WriteResultMode",
+    "write_result_schema",
+    "write_changes_batch",
     "AnyArrow",
     "AnyArrowValue",
     "Arg",
     "ArgumentSpec",
     "ArgumentValidationError",
     "Arguments",
+    "ArgumentMonotonicity",
     "Auth",
     "AuthContext",
     "CallContext",
     "argument_specs_to_schema",
+    "parameter_default_values_from_specs",
     "BindRequest",
     "ColumnBounds",
     "ColumnRefNode",
@@ -130,6 +180,42 @@ __all__ = [
     "FunctionExample",
     "FunctionStability",
     "CatalogFunctionType",
+    "FilterSemanticProfile",
+    "FilterFunctionCapability",
+    "RuntimeFilterAlgorithmCapability",
+    "EvaluationContextCapability",
+    "FilterV2Error",
+    "FilterState",
+    "FilterPredicate",
+    "PredicateMode",
+    "PredicateSource",
+    "EvaluationContext",
+    "ColumnRef",
+    "FieldRef",
+    "Literal",
+    "Comparison",
+    "ComparisonOperator",
+    "BooleanExpression",
+    "Not",
+    "IsNull",
+    "LiteralSet",
+    "ExternalSet",
+    "In",
+    "Cast",
+    "Arithmetic",
+    "ArithmeticOperator",
+    "Negate",
+    "Call",
+    "FunctionIdentity",
+    "StandardFilterFunction",
+    "RuntimeFilter",
+    "FilterPayload",
+    "value_payload",
+    "type_payload",
+    "artifact_payload",
+    "build_filter_batch",
+    "serialize_filter_batch",
+    "deserialize_filter_batch",
     "InitRequest",
     "Level",
     "VgiOutputCollector",
@@ -138,6 +224,7 @@ __all__ = [
     "ParameterInfo",
     "ResolvedMetadata",
     "RowCountMismatchError",
+    "SchemaPath",
     "ScalarFunction",
     "ScalarFunctionGenerator",
     "TableInOutFunction",

vgi/_test_fixtures/accumulate/worker.py

@@ -694,7 +694,7 @@ _ACCUMULATE_CATALOG = Catalog(
     default_schema="main",
     schemas=[
         Schema(
-            name="main",
+            path=["main"],
             comment="Row accumulation keyed by name, persisted via FunctionStorage and scoped per ATTACH",
             functions=[
                 AccumulateFunction,

vgi/_test_fixtures/attach_options.py

@@ -238,7 +238,7 @@ _CATALOG_DESCRIPTOR = Catalog(
     name=CATALOG_NAME,
     schemas=[
         Schema(
-            name="main",
+            path=["main"],
             tables=(),
             views=(),
             functions=(EchoAttachOptionsFunction,),

vgi/_test_fixtures/bad_enum.py

@@ -49,8 +49,8 @@ class _BogusNullHandling(Enum):
 class BadEnumCatalog(ExampleCatalog):
     """ExampleCatalog that advertises a bogus null_handling for one function."""
 
-    def _function_to_info(self, func_cls: type, schema_name: str) -> FunctionInfo:
-        info = super()._function_to_info(func_cls, schema_name)
+    def _function_to_info(self, func_cls: type, schema_path: list[str]) -> FunctionInfo:
+        info = super()._function_to_info(func_cls, schema_path)
         if info.name == BAD_ENUM_FUNCTION and info.null_handling is not None:
             # FunctionInfo is frozen; replace() returns a corrupted copy.
             return replace(info, null_handling=_BogusNullHandling.WEIRD)  # type: ignore[arg-type]