vgi-python 0.25.0__tar.gz → 0.26.0__tar.gz

{vgi_python-0.25.0 → vgi_python-0.26.0}/.github/workflows/ci.yml

@@ -30,7 +30,7 @@ jobs:
       - uses: actions/checkout@v7
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v8.2.0
+        uses: astral-sh/setup-uv@v9.0.0
 
       - name: Set up Python ${{ matrix.python-version }}
         run: uv python install ${{ matrix.python-version }}
@@ -48,7 +48,7 @@ jobs:
       - uses: actions/checkout@v7
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v8.2.0
+        uses: astral-sh/setup-uv@v9.0.0
 
       - name: Install dependencies
         run: uv sync --all-extras
@@ -62,15 +62,18 @@ jobs:
       # Docstring consistency (pydoclint) runs inside the pytest suite via
       # tests/test_docstrings.py, so it's not a separate CI lint step.
 
+      # ty was removed here in 0.25.2. It was already non-blocking, and at
+      # 0.0.66 it stopped terminating on this tree at all -- `ty check vgi/`
+      # ran past ten minutes and had to be killed. A signal nothing gates on,
+      # that also burns a CI job's wall clock, is worse than no signal.
       - name: Type check (mypy)
         run: uv run mypy vgi/
 
-      # ty is pre-1.0 and its inference drifts release-to-release; keep it
-      # visible as a signal but don't gate CI on it (mypy is the blocking
-      # type gate).
-      - name: Type check (ty, non-blocking)
-        run: uv run ty check vgi/
-        continue-on-error: true
+      # tests/ became clean in 0.26.0 and is gated from here on. It was
+      # unchecked for long enough to accumulate 174 errors; the point of the
+      # gate is that it never gets to accumulate a second batch.
+      - name: Type check tests (mypy)
+        run: uv run mypy tests/
 
   docs:
     name: Docs (build + examples + prose)
@@ -81,7 +84,7 @@ jobs:
       - uses: actions/checkout@v7
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v8.2.0
+        uses: astral-sh/setup-uv@v9.0.0
 
       - name: Set up Python 3.14
         run: uv python install 3.14
@@ -139,7 +142,7 @@ jobs:
           path: vgi-rpc
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v8.2.0
+        uses: astral-sh/setup-uv@v9.0.0
 
       - name: Set up Python 3.14
         run: uv python install 3.14

{vgi_python-0.25.0 → vgi_python-0.26.0}/.github/workflows/docs.yml

@@ -18,7 +18,7 @@ jobs:
       - uses: actions/checkout@v7
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v8.2.0
+        uses: astral-sh/setup-uv@v9.0.0
 
       - name: Set up Python
         run: uv python install 3.14

{vgi_python-0.25.0 → vgi_python-0.26.0}/.github/workflows/integration.yml

@@ -76,7 +76,7 @@ jobs:
           token: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v8.2.0
+        uses: astral-sh/setup-uv@v9.0.0
 
       - name: Set up Python 3.14
         run: uv python install 3.14

{vgi_python-0.25.0 → vgi_python-0.26.0}/.github/workflows/release.yml

@@ -33,7 +33,7 @@ jobs:
       - uses: actions/checkout@v7
 
       - name: Install uv
-        uses: astral-sh/setup-uv@v8.2.0
+        uses: astral-sh/setup-uv@v9.0.0
 
       - name: Build vgi (sdist + wheel)
         run: uv build --sdist --wheel --out-dir dist-vgi .
@@ -108,17 +108,17 @@ jobs:
           fi
           echo "tags=$tags" >> "$GITHUB_OUTPUT"
 
-      - uses: docker/setup-buildx-action@v3
+      - uses: docker/setup-buildx-action@v4
 
       - name: Log in to GHCR
-        uses: docker/login-action@v3
+        uses: docker/login-action@v4
         with:
           registry: ghcr.io
           username: ${{ github.actor }}
           password: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Build & push
-        uses: docker/build-push-action@v6
+        uses: docker/build-push-action@v7
         with:
           context: .
           file: packages/vgi-fixtures/docker/Dockerfile

{vgi_python-0.25.0 → vgi_python-0.26.0}/CLAUDE.md

@@ -437,7 +437,7 @@ class MyWorker(Worker):
         try:
             row = api_keys.lookup(token)          # your own store
         except ConnectionError as exc:
-            # "I could not find out" — 503 + Retry-After, not 401.
+            # "I could not find out" -- 503 + Retry-After, not 401.
             raise AuthUnavailableError(str(exc)) from exc
         if row is None:
             return None                           # "the credential is unknown"

{vgi_python-0.25.0 → vgi_python-0.26.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vgi-python
-Version: 0.25.0
+Version: 0.26.0
 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
@@ -158,11 +158,11 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
 Requires-Python: >=3.13
 Requires-Dist: click
-Requires-Dist: httpx>=0.24
+Requires-Dist: httpx2>=2.9.1
 Requires-Dist: platformdirs
 Requires-Dist: pyarrow
 Requires-Dist: typer>=0.9
-Requires-Dist: vgi-rpc>=0.39.1
+Requires-Dist: vgi-rpc>=0.40.0
 Provides-Extra: azure
 Requires-Dist: azure-identity>=1.16.0; extra == 'azure'
 Requires-Dist: pymssql>=2.3.0; extra == 'azure'

{vgi_python-0.25.0 → vgi_python-0.26.0}/docs/api/auth.md

@@ -12,6 +12,12 @@ require `pip install vgi-python[http]`; JWT authentication additionally requires
 
 ::: vgi.auth
 
+## Token introspection
+
+`TokenIdentity` is what [`Worker.resolve_token`][vgi.worker.Worker.resolve_token] returns: a
+`principal` plus an optional `token_name`. It lives in a private `vgi-rpc` module, so `vgi.auth`
+re-exports it — a worker that implements `resolve_token` never has to name a private import path.
+
 ## Secret protocol
 
 ::: vgi.secret_protocol

{vgi_python-0.25.0 → vgi_python-0.26.0}/docs/how-to/state-storage.md

@@ -57,7 +57,7 @@ VGI_WORKER_SHARED_STORAGE=cloudflare-do vgi-serve my_worker.py --http
 |---|---|---|---|
 | SQLite | `sqlite` (default) | local / subprocess | none (stdlib) |
 | Azure SQL | `azure-sql` | Azure deployments | `vgi-python[azure]` |
-| Cloudflare DO | `cloudflare-do` | edge / multi-cloud | none extra — uses `httpx`, which ships with `vgi-python`; needs a Worker endpoint + token |
+| Cloudflare DO | `cloudflare-do` | edge / multi-cloud | none extra — uses `httpx2`, which ships with `vgi-python`; needs a Worker endpoint + token |
 
 The per-backend setup (connection strings, credentials, table provisioning) is documented in the
 [Shared Storage reference](../shared-storage.md).

{vgi_python-0.25.0 → vgi_python-0.26.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "vgi-python"
-version = "0.25.0"
+version = "0.26.0"
 description = "Vector Gateway Interface - Connect DuckDB to external programs via Apache Arrow"
 readme = "README.md"
 keywords = [
@@ -40,8 +40,8 @@ dependencies = [
     "pyarrow",
     "typer>=0.9",
     "platformdirs",
-    "vgi-rpc>=0.39.1",
-    "httpx>=0.24",
+    "vgi-rpc>=0.40.0",
+    "httpx2>=2.9.1",
 ]
 
 [project.optional-dependencies]
@@ -186,6 +186,10 @@ python_version = "3.13"
 strict = true
 warn_return_any = true
 warn_unused_ignores = true
+# tests/test_examples_workers.py puts examples/ on sys.path and imports the
+# doc example workers by bare module name; mirror that for mypy so the imports
+# resolve (and the examples themselves get checked).
+mypy_path = "examples"
 
 [[tool.mypy.overrides]]
 module = "typer.*"
@@ -197,6 +201,13 @@ ignore_missing_imports = true
 module = "granian.*"
 ignore_missing_imports = true
 
+[[tool.mypy.overrides]]
+# jsonschema ships no py.typed marker and its stubs live in the separate
+# types-jsonschema distribution, which we don't depend on. Used only by the
+# landing-page conformance checker in tests/.
+module = "jsonschema.*"
+ignore_missing_imports = true
+
 [[tool.mypy.overrides]]
 # These files have type: ignore comments for ty that mypy doesn't need
 module = ["vgi._test_fixtures.scalar.null_handling", "vgi.table_function"]
@@ -250,7 +261,7 @@ members = ["packages/vgi-fixtures"]
 vgi-fixtures = { workspace = true }
 
 [dependency-groups]
-dev = ["lxml>=6.0.2", "pytest-timeout>=2.4.0", "ty>=0.0.8", "vgi-fixtures"]
+dev = ["lxml>=6.0.2", "pytest-timeout>=2.4.0", "vgi-fixtures"]
 docs = [
     "mkdocs>=1.6",
     "mkdocs-material>=9.5",

{vgi_python-0.25.0 → vgi_python-0.26.0}/tests/catalog/test_declarative.py

@@ -1279,9 +1279,11 @@ class TestSchemaDescriptor:
         """
         schema = Schema(name="main", functions=[UsersFunction])  # only table fn
         info = schema.to_schema_info(AttachOpaqueData(b"x"))
-        assert info.estimated_object_count["scalar_function"] == 0
-        assert info.estimated_object_count["aggregate_function"] == 0
-        assert info.estimated_object_count["table_function"] == 1
+        counts = info.estimated_object_count
+        assert counts is not None
+        assert counts["scalar_function"] == 0
+        assert counts["aggregate_function"] == 0
+        assert counts["table_function"] == 1
 
 
 class TestLegacyFunctionsListSchemaInfo:
@@ -1320,6 +1322,7 @@ class TestLegacyFunctionsListSchemaInfo:
         schemas = ci.schemas(attach_opaque_data=AttachOpaqueData(b"x"), transaction_opaque_data=None)
         assert len(schemas) == 1
         counts = schemas[0].estimated_object_count
+        assert counts is not None
         assert counts["scalar_function"] == 1
         assert counts["table_function"] == 1
 

{vgi_python-0.25.0 → vgi_python-0.26.0}/tests/catalog/test_required_filters.py

@@ -23,23 +23,22 @@ class TestRequiredFiltersField:
 
     def _bbox_columns(self) -> pa.Schema:
         """Return a small schema with a ``bbox`` STRUCT plus flat id/ticker."""
-        return pa.schema(
-            [
-                ("id", pa.int64()),
-                ("ticker", pa.string()),
-                (
-                    "bbox",
-                    pa.struct(
-                        [
-                            ("xmin", pa.float64()),
-                            ("ymin", pa.float64()),
-                            ("xmax", pa.float64()),
-                            ("ymax", pa.float64()),
-                        ]
-                    ),
+        fields: list[tuple[str, pa.DataType]] = [
+            ("id", pa.int64()),
+            ("ticker", pa.string()),
+            (
+                "bbox",
+                pa.struct(
+                    [
+                        ("xmin", pa.float64()),
+                        ("ymin", pa.float64()),
+                        ("xmax", pa.float64()),
+                        ("ymax", pa.float64()),
+                    ]
                 ),
-            ]
-        )
+            ),
+        ]
+        return pa.schema(fields)
 
     def test_default_is_empty_tuple(self) -> None:
         """Default is an empty tuple; TableInfo wire field is an empty list."""

{vgi_python-0.25.0 → vgi_python-0.26.0}/tests/catalog/test_scan_branches.py

@@ -63,7 +63,7 @@ class _LegacyOnlyCatalog(_StubCatalogBase):
 
     # Only the legacy method is overridden; table_scan_branches_get falls back
     # to the CatalogInterface default-impl shim.
-    def table_scan_function_get(  # type: ignore[override]
+    def table_scan_function_get(
         self,
         *,
         attach_opaque_data: AttachOpaqueData,
@@ -89,7 +89,7 @@ class _MultiBranchCatalog(_StubCatalogBase):
     keep them non-overlapping at scan time.
     """
 
-    def table_scan_function_get(  # type: ignore[override]
+    def table_scan_function_get(
         self,
         *,
         attach_opaque_data: AttachOpaqueData,
@@ -108,7 +108,7 @@ class _MultiBranchCatalog(_StubCatalogBase):
             required_extensions=[],
         )
 
-    def table_scan_branches_get(  # type: ignore[override]
+    def table_scan_branches_get(
         self,
         *,
         attach_opaque_data: AttachOpaqueData,
@@ -184,7 +184,7 @@ class TestDefaultImplShim:
                 super().__init__()
                 self.last_at: tuple[str | None, str | None] | None = None
 
-            def table_scan_function_get(  # type: ignore[override]
+            def table_scan_function_get(
                 self,
                 *,
                 attach_opaque_data: AttachOpaqueData,

{vgi_python-0.25.0 → vgi_python-0.26.0}/tests/conformance/test_accumulate.py

@@ -18,16 +18,23 @@ transport is covered by the sqllogictest suite in the C++ repo.
 from __future__ import annotations
 
 import contextlib
+from collections.abc import Callable, Iterator
 from datetime import timedelta
+from pathlib import Path
 from types import SimpleNamespace
+from typing import Any, cast
 
 import pyarrow as pa
 import pytest
+from vgi_rpc.rpc import OutputCollector
 
 from vgi._test_fixtures.accumulate import worker as m
 from vgi.function_storage import BoundStorage, FunctionStorage, FunctionStorageSqlite
+from vgi.table_buffering_function import TableBufferingParams
+from vgi.table_function import BindParams
 
-INPUT_SCHEMA = pa.schema([pa.field("x", pa.int64()), pa.field("label", pa.string())])
+_INPUT_FIELDS: list[pa.Field[Any]] = [pa.field("x", pa.int64()), pa.field("label", pa.string())]
+INPUT_SCHEMA = pa.schema(_INPUT_FIELDS)
 OUTPUT_SCHEMA = m._output_schema(INPUT_SCHEMA)
 
 SCOPE_A = b"session-a"
@@ -43,7 +50,7 @@ def _set_storage(fs: FunctionStorage) -> None:
 
 
 @pytest.fixture(autouse=True)
-def _storage(tmp_path):
+def _storage(tmp_path: Path) -> Iterator[SimpleNamespace]:
     """Inject a throwaway file-backed FunctionStorage for the persistent collection."""
     path = str(tmp_path / "fs.db")
     _set_storage(FunctionStorageSqlite(db_path=path))
@@ -69,51 +76,85 @@ class FakeOut:
         """Mark the stream finished."""
         self.finished = True
 
+    @property
+    def collector(self) -> OutputCollector:
+        """This fake viewed as the ``OutputCollector`` the fixture's API declares.
 
-def _exec_storage():
+        ``OutputCollector`` is a concrete class, not a protocol, so the
+        duck-typed stand-in needs one cast to reach the real signatures.
+        """
+        return cast("OutputCollector", self)
+
+
+def _exec_storage() -> tuple[bytes, BoundStorage]:
     """Build a fresh execution-scoped BoundStorage (mimics a distinct query)."""
     _exec_counter[0] += 1
     eid = f"exec-{_exec_counter[0]}".encode()
     return eid, BoundStorage(m.AccumulateFunction.storage, eid)
 
 
-def _bind_params(name, input_schema=INPUT_SCHEMA, opaque=SCOPE_A):
-    return SimpleNamespace(
-        args=SimpleNamespace(name=name, data=None, ttl=None, max_row_size=0, result="all"),
-        bind_call=SimpleNamespace(input_schema=input_schema),
-        attach_opaque_data=opaque,
+# The two param builders return duck-typed SimpleNamespaces — the fixture only
+# reads a handful of fields off them — cast to the declared param types so the
+# call sites below type-check against the real fixture signatures.
+def _bind_params(name: str, input_schema: pa.Schema = INPUT_SCHEMA, opaque: bytes = SCOPE_A) -> BindParams[Any]:
+    return cast(
+        "BindParams[Any]",
+        SimpleNamespace(
+            args=SimpleNamespace(name=name, data=None, ttl=None, max_row_size=0, result="all"),
+            bind_call=SimpleNamespace(input_schema=input_schema),
+            attach_opaque_data=opaque,
+        ),
     )
 
 
-def _buf_params(name, *, ttl=None, max_row_size=0, result="all", output_schema=OUTPUT_SCHEMA, opaque=SCOPE_A):
+def _buf_params(
+    name: str,
+    *,
+    ttl: Any = None,
+    max_row_size: int = 0,
+    result: str = "all",
+    output_schema: pa.Schema = OUTPUT_SCHEMA,
+    opaque: bytes = SCOPE_A,
+) -> TableBufferingParams[Any]:
     eid, storage = _exec_storage()
-    return SimpleNamespace(
-        args=SimpleNamespace(name=name, data=None, ttl=ttl, max_row_size=max_row_size, result=result),
-        output_schema=output_schema,
-        attach_opaque_data=opaque,
-        storage=storage,
-        execution_id=eid,
+    return cast(
+        "TableBufferingParams[Any]",
+        SimpleNamespace(
+            args=SimpleNamespace(name=name, data=None, ttl=ttl, max_row_size=max_row_size, result=result),
+            output_schema=output_schema,
+            attach_opaque_data=opaque,
+            storage=storage,
+            execution_id=eid,
+        ),
     )
 
 
-def _input_batch(xs):
+def _input_batch(xs: list[int]) -> pa.RecordBatch:
     return pa.RecordBatch.from_arrays(
         [pa.array(xs, pa.int64()), pa.array([f"r{x}" for x in xs], pa.string())],
         schema=INPUT_SCHEMA,
     )
 
 
-def _drain(fn) -> FakeOut:
+def _drain(fn: Callable[[OutputCollector], None]) -> FakeOut:
     out = FakeOut()
     guard = 0
     while not out.finished:
-        fn(out)
+        fn(out.collector)
         guard += 1
         assert guard < 1_000_000, "stream did not terminate"
     return out
 
 
-def _run(name, *, batches, ttl=None, max_row_size=0, result="all", opaque=SCOPE_A) -> pa.Table:
+def _run(
+    name: str,
+    *,
+    batches: list[pa.RecordBatch],
+    ttl: Any = None,
+    max_row_size: int = 0,
+    result: str = "all",
+    opaque: bytes = SCOPE_A,
+) -> pa.Table:
     resp = m.AccumulateFunction.on_bind(_bind_params(name, opaque=opaque))
     pp = _buf_params(
         name, ttl=ttl, max_row_size=max_row_size, result=result, output_schema=resp.output_schema, opaque=opaque
@@ -131,38 +172,47 @@ def _run(name, *, batches, ttl=None, max_row_size=0, result="all", opaque=SCOPE_
     )
 
 
-def _call(name, xs, *, ttl=None, max_row_size=0, result="all", opaque=SCOPE_A) -> pa.Table:
+def _call(
+    name: str,
+    xs: list[int],
+    *,
+    ttl: Any = None,
+    max_row_size: int = 0,
+    result: str = "all",
+    opaque: bytes = SCOPE_A,
+) -> pa.Table:
     return _run(name, batches=[_input_batch(xs)], ttl=ttl, max_row_size=max_row_size, result=result, opaque=opaque)
 
 
-def _read_batches(name, *, opaque=SCOPE_A) -> list[pa.RecordBatch]:
+def _read_batches(name: str, *, opaque: bytes = SCOPE_A) -> list[pa.RecordBatch]:
     resp = m.AccumulateReadFunction.on_bind(_bind_params(name, opaque=opaque))
     pp = _buf_params(name, output_schema=resp.output_schema, opaque=opaque)
     state = m.AccumulateReadFunction.initial_state(pp)
     return _drain(lambda o: m.AccumulateReadFunction.process(pp, state, o)).batches
 
 
-def _read(name, *, opaque=SCOPE_A) -> pa.Table:
+def _read(name: str, *, opaque: bytes = SCOPE_A) -> pa.Table:
     resp = m.AccumulateReadFunction.on_bind(_bind_params(name, opaque=opaque))
     batches = _read_batches(name, opaque=opaque)
     return pa.Table.from_batches(batches, schema=resp.output_schema) if batches else resp.output_schema.empty_table()
 
 
-def _clear(name, *, opaque=SCOPE_A) -> pa.RecordBatch:
+def _clear(name: str, *, opaque: bytes = SCOPE_A) -> pa.RecordBatch:
     pp = _buf_params(name, output_schema=m.CLEAR_SCHEMA, opaque=opaque)
     state = m.AccumulateClearFunction.initial_state(pp)
     out = FakeOut()
-    m.AccumulateClearFunction.process(pp, state, out)
-    m.AccumulateClearFunction.process(pp, state, out)  # second tick -> finish
+    m.AccumulateClearFunction.process(pp, state, out.collector)
+    m.AccumulateClearFunction.process(pp, state, out.collector)  # second tick -> finish
     assert out.finished and len(out.batches) == 1
     return out.batches[0]
 
 
-def _xs(table) -> list[int]:
-    return sorted(table.column("x").to_pylist())
+def _xs(table: pa.Table) -> list[int]:
+    # `x` is declared non-null int64; to_pylist() types every element optional.
+    return sorted(cast("list[int]", table.column("x").to_pylist()))
 
 
-def _total(name, opaque=SCOPE_A) -> int:
+def _total(name: str, opaque: bytes = SCOPE_A) -> int:
     ps = m._store(m.AccumulateFunction.storage, opaque)
     sch = m._get_schema(ps, name.encode())
     return 0 if sch is None else m._read_collection(ps, name.encode(), sch).num_rows
@@ -209,7 +259,7 @@ def test_on_bind_rejects_schema_mismatch_for_same_name() -> None:
 
 
 @pytest.mark.parametrize("name", ["", "   "])
-def test_on_bind_rejects_blank_name(name) -> None:
+def test_on_bind_rejects_blank_name(name: str) -> None:
     """Empty or blank collection names are rejected by all three functions."""
     with pytest.raises(ValueError, match="non-empty"):
         m.AccumulateFunction.on_bind(_bind_params(name))
@@ -310,7 +360,7 @@ def test_read_is_scoped_to_attach_session() -> None:
 # --- persistence & scoping -------------------------------------------------
 
 
-def test_persists_across_reconnect(tmp_path) -> None:
+def test_persists_across_reconnect(tmp_path: Path) -> None:
     """Collections survive a worker restart (fresh handle on the same DB file)."""
     _call("a", [1, 2])
     # Simulate a worker restart: a fresh storage handle on the same DB file.
@@ -372,7 +422,7 @@ def test_large_input_round_trips_all_rows() -> None:
     assert len(set(t.column("_timestamp").to_pylist())) == 1
 
 
-def test_output_returned_in_bounded_batches(monkeypatch) -> None:
+def test_output_returned_in_bounded_batches(monkeypatch: pytest.MonkeyPatch) -> None:
     """Output is staged and drained in batches bounded by OUT_BATCH_ROWS."""
     monkeypatch.setattr(m, "OUT_BATCH_ROWS", 100)
     _call("big", list(range(1000)), result="none")

{vgi_python-0.25.0 → vgi_python-0.26.0}/tests/conformance/test_const_param_enforcement.py

@@ -17,7 +17,8 @@ import pyarrow as pa
 import pytest
 
 from vgi.arguments import Arguments, ArgumentValidationError, ConstParam, Param, Returns
-from vgi.protocol import BindRequest, FunctionType
+from vgi.invocation import FunctionType
+from vgi.protocol import BindRequest
 from vgi.scalar_function import BindResult, ScalarFunction
 
 

{vgi_python-0.25.0 → vgi_python-0.26.0}/tests/conformance/test_resumable_scan.py

@@ -11,6 +11,7 @@ dropped rows. Also checks the capability gate on the non-resumable transport.
 from __future__ import annotations
 
 from collections.abc import Iterator
+from typing import cast
 
 import pyarrow as pa
 import pytest
@@ -45,13 +46,22 @@ def http_base_url() -> Iterator[str]:
         stack.close()
 
 
+def _ns(batch: pa.RecordBatch) -> list[int]:
+    """The batch's ``n`` column as ints.
+
+    ``sequence`` declares ``n`` non-null; ``to_pylist()`` types every element
+    optional regardless, so narrow once here instead of at each call site.
+    """
+    return cast("list[int]", batch.column("n").to_pylist())
+
+
 def _reference_rows(base_url: str) -> list[int]:
     """Read the full, ordered ``sequence`` result the normal way."""
     with Client.from_http(base_url) as client:
         return [
             v
             for b in client.table_function(function_name="sequence", schema_name="main", arguments=_ARGS)
-            for v in b.column("n").to_pylist()
+            for v in _ns(b)
         ]
 
 
@@ -67,7 +77,7 @@ def test_resumable_scan_matches_table_function(http_base_url: str) -> None:
             batch, _token = cur.next()
             if batch is None:
                 break
-            rows.extend(batch.column("n").to_pylist())
+            rows.extend(_ns(batch))
         cur.close()
 
     assert rows == expected
@@ -85,7 +95,7 @@ def test_resume_on_fresh_client_after_node_hop(http_base_url: str) -> None:
         for _ in range(3):
             batch, token = cur.next()
             assert batch is not None
-            rows.extend(batch.column("n").to_pylist())
+            rows.extend(_ns(batch))
         assert token is not None  # per-batch tokens are the default
         cur.close()
 
@@ -98,7 +108,7 @@ def test_resume_on_fresh_client_after_node_hop(http_base_url: str) -> None:
             batch, token = cur2.next()
             if batch is None:
                 break
-            rows.extend(batch.column("n").to_pylist())
+            rows.extend(_ns(batch))
         cur2.close()
 
     assert rows == expected  # contiguous, complete, no duplicates
@@ -122,7 +132,7 @@ def test_continue_on_fresh_client_skips_rebind(http_base_url: str) -> None:
         for _ in range(3):
             batch, token = cur.next()
             assert batch is not None
-            rows.extend(batch.column("n").to_pylist())
+            rows.extend(_ns(batch))
         assert token is not None
         cur.close()
 
@@ -133,7 +143,7 @@ def test_continue_on_fresh_client_skips_rebind(http_base_url: str) -> None:
             batch, token = cur2.next()
             if batch is None:
                 break
-            rows.extend(batch.column("n").to_pylist())
+            rows.extend(_ns(batch))
         cur2.close()
 
     assert rows == expected

{vgi_python-0.25.0 → vgi_python-0.26.0}/tests/conformance/test_scalar_attach_opaque_data.py

@@ -15,13 +15,12 @@ from typing import Annotated, Any
 
 import pyarrow as pa
 
-from vgi.arguments import Arguments
-from vgi.protocol import BindRequest, FunctionType
+from vgi.arguments import Arguments, Param, Returns
+from vgi.invocation import FunctionType
+from vgi.protocol import BindRequest
 from vgi.scalar_function import (
     BindParameters,
     BindResult,
-    Param,
-    Returns,
     ScalarFunction,
 )