poly-kv 0.1.0-alpha.3

Shared compressed KV-cache pool for multi-agent context. Two-tier codec policy (fib-quant cold + turbo-quant hot) with typed receipts.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

import pytest

import poly_kv


def require_native():
    if not poly_kv.native_available():
        pytest.skip("poly_kv._native is not built; run maturin develop or maturin build")


def test_bad_shape_rejected_by_native_sidecar():
    require_native()
    shape = poly_kv.ShapeV2(
        batch=1,
        layers=2,
        num_q_heads=8,
        num_kv_heads=3,
        seq_len=8,
        head_dim=4,
        attention_kind="gqa",
    )
    with pytest.raises(poly_kv.PolyKvShapeError):
        poly_kv.validate_shape(shape)