vgi-python 0.8.1__tar.gz → 0.8.3__tar.gz

{vgi_python-0.8.1 → vgi_python-0.8.3}/.github/workflows/ci.yml

@@ -27,7 +27,7 @@ jobs:
         python-version: ["3.13", "3.14"]
 
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@v7
 
       - name: Install uv
         uses: astral-sh/setup-uv@v8.2.0
@@ -45,7 +45,7 @@ jobs:
     name: Lint
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@v7
 
       - name: Install uv
         uses: astral-sh/setup-uv@v8.2.0
@@ -78,7 +78,7 @@ jobs:
     env:
       DISABLE_MKDOCS_2_WARNING: "true"
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@v7
 
       - name: Install uv
         uses: astral-sh/setup-uv@v8.2.0
@@ -130,10 +130,10 @@ jobs:
           SERVICES: s3
           AWS_DEFAULT_REGION: us-east-1
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@v7
 
       - name: Checkout vgi-rpc
-        uses: actions/checkout@v6
+        uses: actions/checkout@v7
         with:
           repository: Query-farm/vgi-rpc-python
           path: vgi-rpc

{vgi_python-0.8.1 → vgi_python-0.8.3}/.github/workflows/docs.yml

@@ -15,7 +15,7 @@ jobs:
   build:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@v7
 
       - name: Install uv
         uses: astral-sh/setup-uv@v8.2.0

{vgi_python-0.8.1 → vgi_python-0.8.3}/.github/workflows/integration.yml

@@ -53,10 +53,10 @@ jobs:
     name: integration (${{ matrix.lane }})
     steps:
       - name: Checkout vgi-python
-        uses: actions/checkout@v6
+        uses: actions/checkout@v7
 
       - name: Checkout pinned vgi test suite
-        uses: actions/checkout@v6
+        uses: actions/checkout@v7
         with:
           repository: Query-farm/vgi
           ref: ${{ env.VGI_REF }}

{vgi_python-0.8.1 → vgi_python-0.8.3}/.github/workflows/release.yml

@@ -30,7 +30,7 @@ jobs:
     name: Build distributions
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@v7
 
       - name: Install uv
         uses: astral-sh/setup-uv@v8.2.0

{vgi_python-0.8.1 → vgi_python-0.8.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vgi-python
-Version: 0.8.1
+Version: 0.8.3
 Summary: Vector Gateway Interface - Connect DuckDB to external programs via Apache Arrow
 Project-URL: Homepage, https://query.farm
 Project-URL: Repository, https://github.com/Query-farm/vgi-python
@@ -162,7 +162,7 @@ Requires-Dist: httpx>=0.24
 Requires-Dist: platformdirs
 Requires-Dist: pyarrow
 Requires-Dist: typer>=0.9
-Requires-Dist: vgi-rpc>=0.20.4
+Requires-Dist: vgi-rpc>=0.20.5
 Provides-Extra: azure
 Requires-Dist: azure-identity>=1.16.0; extra == 'azure'
 Requires-Dist: pymssql>=2.3.0; extra == 'azure'

{vgi_python-0.8.1 → vgi_python-0.8.3}/packages/vgi-fixtures/pyproject.toml

@@ -8,7 +8,7 @@
 # and the conformance suite import it by its normal `vgi._test_fixtures` path.
 [project]
 name = "vgi-fixtures"
-version = "0.8.1"
+version = "0.8.2"
 description = "Example/test VGI worker modules and entry points (not for production use)"
 readme = "README.md"
 # Local copy of the repo-root LICENSE (kept in sync): the build can't reach
@@ -43,6 +43,7 @@ vgi-fixture-versioned-worker = "vgi._test_fixtures.versioned:main"
 vgi-fixture-versioned-tables-worker = "vgi._test_fixtures.versioned_tables:main"
 vgi-fixture-attach-options-worker = "vgi._test_fixtures.attach_options:main"
 vgi-fixture-bad-protocol-worker = "vgi._test_fixtures.bad_protocol:main"
+vgi-fixture-bad-enum-worker = "vgi._test_fixtures.bad_enum:main"
 vgi-fixture-writable-worker = "vgi._test_fixtures.writable.worker:main"
 vgi-fixture-simple-writable-worker = "vgi._test_fixtures.simple_writable:main"
 

{vgi_python-0.8.1 → vgi_python-0.8.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "vgi-python"
-version = "0.8.1"
+version = "0.8.3"
 description = "Vector Gateway Interface - Connect DuckDB to external programs via Apache Arrow"
 readme = "README.md"
 keywords = [
@@ -40,7 +40,7 @@ dependencies = [
     "pyarrow",
     "typer>=0.9",
     "platformdirs",
-    "vgi-rpc>=0.20.4",
+    "vgi-rpc>=0.20.5",
     "httpx>=0.24",
 ]
 

vgi_python-0.8.3/tests/test_union_argument.py

@@ -0,0 +1,111 @@
+# Copyright 2025, 2026 Query Farm LLC - https://query.farm
+
+"""Union-typed argument decoding.
+
+A DuckDB ``UNION`` / Arrow union argument is tagged: the discriminator lives in
+``UnionScalar.type_code``, which plain ``Scalar.as_py()`` discards. ``Arguments``
+decodes such scalars to ``TaggedUnion`` so the active member name is preserved.
+"""
+
+from __future__ import annotations
+
+import pyarrow as pa
+
+from vgi.arguments import Arguments, TaggedUnion
+
+_RF = pa.struct([pa.field("n_estimators", pa.list_(pa.int64())), pa.field("max_depth", pa.list_(pa.int64()))])
+_SVC = pa.struct([pa.field("C", pa.list_(pa.float64())), pa.field("kernel", pa.list_(pa.string()))])
+
+
+def _union_scalar(code: int, members: dict) -> pa.UnionScalar:
+    """Build a one-element sparse-union scalar with named members, active = ``code``."""
+    arr = pa.UnionArray.from_sparse(
+        pa.array([code], type=pa.int8()),
+        [
+            pa.array([members.get("random_forest_classifier")], type=_RF),
+            pa.array([members.get("svc")], type=_SVC),
+        ],
+        field_names=["random_forest_classifier", "svc"],
+        type_codes=[0, 1],
+    )
+    return arr[0]
+
+
+def test_union_arg_preserves_tag() -> None:
+    """A union argument decodes to a TaggedUnion carrying the active member name."""
+    scalar = _union_scalar(0, {"random_forest_classifier": {"n_estimators": [100, 300], "max_depth": [3, 5]}})
+    got = Arguments(named={"config": scalar}).get("config")
+    assert isinstance(got, TaggedUnion)
+    assert got.tag == "random_forest_classifier"
+    assert got.value == {"n_estimators": [100, 300], "max_depth": [3, 5]}
+
+
+def test_union_arg_other_member() -> None:
+    """The tag reflects whichever union member is set."""
+    scalar = _union_scalar(1, {"svc": {"C": [1.0, 10.0], "kernel": ["rbf", "linear"]}})
+    got = Arguments(named={"config": scalar}).get("config")
+    assert got.tag == "svc"
+    assert got.value == {"C": [1.0, 10.0], "kernel": ["rbf", "linear"]}
+
+
+def test_union_arg_null_payload() -> None:
+    """A null union slot decodes its value to None (the ``inner is None`` branch).
+
+    Note: pyarrow reports ``type_code == 0`` for any null union slot regardless of
+    the codes buffer, so the active member's tag is not recoverable at the scalar
+    level for a null payload — only the value is. (Tag preservation for a
+    non-null payload of a non-zero-code member is covered above and through the
+    batch-level round-trip in the echo integration suite.)
+    """
+    scalar = _union_scalar(1, {"svc": None})
+    got = Arguments(named={"config": scalar}).get("config")
+    assert isinstance(got, TaggedUnion)
+    assert got.value is None
+
+
+def test_union_arg_non_contiguous_type_codes() -> None:
+    """Tag resolution maps via type_codes, not member position.
+
+    Type codes need not be 0,1,2,...; ``_scalar_to_py`` must locate the active
+    member by ``type_codes.index(code)`` rather than treating the code as a
+    positional index.
+    """
+    arr = pa.UnionArray.from_sparse(
+        pa.array([9], type=pa.int8()),
+        [
+            pa.array([{"n_estimators": [1], "max_depth": [2]}], type=_RF),
+            pa.array([{"C": [1.0], "kernel": ["rbf"]}], type=_SVC),
+        ],
+        field_names=["random_forest_classifier", "svc"],
+        type_codes=[5, 9],
+    )
+    got = Arguments(named={"config": arr[0]}).get("config")
+    assert isinstance(got, TaggedUnion)
+    assert got.tag == "svc"
+    assert got.value == {"C": [1.0], "kernel": ["rbf"]}
+
+
+def test_union_arg_dense_union() -> None:
+    """Dense unions (what DuckDB emits) decode the same as sparse."""
+    arr = pa.UnionArray.from_dense(
+        pa.array([1], type=pa.int8()),
+        pa.array([0], type=pa.int32()),
+        [
+            pa.array([], type=_RF),
+            pa.array([{"C": [2.0], "kernel": ["linear"]}], type=_SVC),
+        ],
+        field_names=["random_forest_classifier", "svc"],
+        type_codes=[0, 1],
+    )
+    got = Arguments(named={"config": arr[0]}).get("config")
+    assert isinstance(got, TaggedUnion)
+    assert got.tag == "svc"
+    assert got.value == {"C": [2.0], "kernel": ["linear"]}
+
+
+def test_non_union_args_unchanged() -> None:
+    """Non-union arguments still decode via plain as_py()."""
+    args = Arguments(named={"n": pa.scalar(5), "s": pa.scalar("hi")}, positional=(pa.scalar(1.5),))
+    assert args.get("n") == 5
+    assert args.get("s") == "hi"
+    assert args.get(0) == 1.5