rxgraph 0.7.0__tar.gz → 0.9.0__tar.gz

{rxgraph-0.7.0 → rxgraph-0.9.0}/Cargo.lock

@@ -2,6 +2,12 @@
 # It is not intended for manual editing.
 version = 4
 
+[[package]]
+name = "adler2"
+version = "2.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "320119579fcad9c21884f5c4861d16174d0e06250625266f50fe6898340abefa"
+
 [[package]]
 name = "ahash"
 version = "0.8.12"
@@ -36,6 +42,21 @@ dependencies = [
  "num-traits",
 ]
 
+[[package]]
+name = "alloc-no-stdlib"
+version = "2.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cc7bb162ec39d46ab1ca8c77bf72e890535becd1751bb45f64c597edb4c8c6b3"
+
+[[package]]
+name = "alloc-stdlib"
+version = "0.2.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0e76a019e91224d279006ff972f1e984179a6e9feb050adba6ce8274aef23195"
+dependencies = [
+ "alloc-no-stdlib",
+]
+
 [[package]]
 name = "alloca"
 version = "0.4.0"
@@ -329,12 +350,39 @@ version = "2.11.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "c4512299f36f043ab09a583e57bceb5a5aab7a73db1805848e8fef3c9e8c78b3"
 
+[[package]]
+name = "brotli"
+version = "8.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5cc91aac060a7a1e25823bdccbfb6af1875b88f17c6daac97894eed8207166b3"
+dependencies = [
+ "alloc-no-stdlib",
+ "alloc-stdlib",
+ "brotli-decompressor",
+]
+
+[[package]]
+name = "brotli-decompressor"
+version = "5.0.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3a32acac15fe1967bc3986b2a6347dffc965602354ea6f450ad07e8bfd253583"
+dependencies = [
+ "alloc-no-stdlib",
+ "alloc-stdlib",
+]
+
 [[package]]
 name = "bumpalo"
 version = "3.20.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "72f5acc6cb2ba439de613abc23857ec3d78374d8ed5ac84e9d11336e87da8649"
 
+[[package]]
+name = "byteorder"
+version = "1.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1fd0f2584146f6f2ef48085050886acf353beff7305ebd1ae69500e27c67f64b"
+
 [[package]]
 name = "bytes"
 version = "1.11.1"
@@ -354,6 +402,8 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "556e016178bb5662a08681bbe0f00f8e17631781a4dfc8c45e466e4b185ec27f"
 dependencies = [
  "find-msvc-tools",
+ "jobserver",
+ "libc",
  "shlex",
 ]
 
@@ -615,6 +665,16 @@ dependencies = [
  "rustc_version",
 ]
 
+[[package]]
+name = "flate2"
+version = "1.1.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "843fba2746e448b37e26a819579957415c8cef339bf08564fe8b7ddbd959573c"
+dependencies = [
+ "miniz_oxide",
+ "zlib-rs",
+]
+
 [[package]]
 name = "foldhash"
 version = "0.1.5"
@@ -763,6 +823,12 @@ dependencies = [
  "serde_core",
 ]
 
+[[package]]
+name = "integer-encoding"
+version = "3.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8bb03732005da905c88227371639bf1ad885cc712789c011c31c5fb3ab3ccf02"
+
 [[package]]
 name = "inventory"
 version = "0.3.24"
@@ -787,6 +853,16 @@ version = "1.0.18"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "8f42a60cbdf9a97f5d2305f08a87dc4e09308d1276d28c869c684d7777685682"
 
+[[package]]
+name = "jobserver"
+version = "0.1.34"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9afb3de4395d6b3e67a780b6de64b51c978ecf11cb9a462c66be7d4ca9039d33"
+dependencies = [
+ "getrandom 0.3.4",
+ "libc",
+]
+
 [[package]]
 name = "js-sys"
 version = "0.3.99"
@@ -880,6 +956,15 @@ version = "0.4.30"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "616ec5685824bcc94416c6d4a7a446eea774a31efd7062c8480ba6fd06d7a6e5"
 
+[[package]]
+name = "lz4_flex"
+version = "0.13.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7ef0d4ed8669f8f8826eb00dc878084aa8f253506c4fd5e8f58f5bce72ddb97e"
+dependencies = [
+ "twox-hash",
+]
+
 [[package]]
 name = "matrixmultiply"
 version = "0.3.10"
@@ -896,6 +981,16 @@ version = "2.8.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "6b947ae49db0d222b1dbc6b113ce7248a3fc3a6ca21b696717bfc000ba4484d8"
 
+[[package]]
+name = "miniz_oxide"
+version = "0.8.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1fa76a2c86f704bdb222d66965fb3d63269ce38518b83cb0575fca855ebb6316"
+dependencies = [
+ "adler2",
+ "simd-adler32",
+]
+
 [[package]]
 name = "ndarray"
 version = "0.17.2"
@@ -998,6 +1093,15 @@ version = "11.1.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "d6790f58c7ff633d8771f42965289203411a5e5c68388703c06e14f24770b41e"
 
+[[package]]
+name = "ordered-float"
+version = "2.10.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "68f19d67e5a2795c94e73e0bb1cc1a7edeb2e28efd39e2e1c9b7a40c1108b11c"
+dependencies = [
+ "num-traits",
+]
+
 [[package]]
 name = "page_size"
 version = "0.6.0"
@@ -1008,6 +1112,45 @@ dependencies = [
  "winapi",
 ]
 
+[[package]]
+name = "parquet"
+version = "58.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5dafa7d01085b62a47dd0c1829550a0a36710ea9c4fe358a05a85477cec8a908"
+dependencies = [
+ "ahash",
+ "arrow-array",
+ "arrow-buffer",
+ "arrow-data",
+ "arrow-ipc",
+ "arrow-schema",
+ "arrow-select",
+ "base64",
+ "brotli",
+ "bytes",
+ "chrono",
+ "flate2",
+ "half",
+ "hashbrown 0.17.1",
+ "lz4_flex",
+ "num-bigint",
+ "num-integer",
+ "num-traits",
+ "paste",
+ "seq-macro",
+ "simdutf8",
+ "snap",
+ "thrift",
+ "twox-hash",
+ "zstd",
+]
+
+[[package]]
+name = "paste"
+version = "1.0.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "57c0d7b74b563b49d38dae00a0c37d4d6de9b432382b2892f0574ddcae73fd0a"
+
 [[package]]
 name = "phf"
 version = "0.12.1"
@@ -1032,6 +1175,12 @@ version = "0.2.17"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "a89322df9ebe1c1578d689c92318e070967d1042b512afbe49518723f4e6d5cd"
 
+[[package]]
+name = "pkg-config"
+version = "0.3.33"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "19f132c84eca552bf34cab8ec81f1c1dcc229b811638f9d283dceabe58c5569e"
+
 [[package]]
 name = "plotters"
 version = "0.3.7"
@@ -1302,13 +1451,14 @@ checksum = "b39cdef0fa800fc44525c84ccb54a029961a8215f9619753635a9c0d2538d46d"
 
 [[package]]
 name = "rxgraph"
-version = "0.7.0"
+version = "0.9.0"
 dependencies = [
  "anyhow",
  "arrow",
  "arrow-schema",
  "criterion",
  "inventory",
+ "parquet",
  "pretty_assertions",
  "pyo3",
  "pyo3-arrow",
@@ -1323,7 +1473,7 @@ dependencies = [
 
 [[package]]
 name = "rxgraph-python"
-version = "0.7.0"
+version = "0.9.0"
 dependencies = [
  "pyo3",
  "rxgraph",
@@ -1350,6 +1500,12 @@ version = "1.0.28"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "8a7852d02fc848982e0c167ef163aaff9cd91dc640ba85e263cb1ce46fae51cd"
 
+[[package]]
+name = "seq-macro"
+version = "0.3.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1bc711410fbe7399f390ca1c3b60ad0f53f80e95c5eb935e52268a0e2cd49acc"
+
 [[package]]
 name = "serde"
 version = "1.0.228"
@@ -1399,6 +1555,12 @@ version = "2.0.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "f8fadd59c855ef2080decdef8ff161eb6661b86933c9d82e5ba29dc602a55aba"
 
+[[package]]
+name = "simd-adler32"
+version = "0.3.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "703d5c7ef118737c72f1af64ad2f6f8c5e1921f818cdcb97b8fe6fc69bf66214"
+
 [[package]]
 name = "simdutf8"
 version = "0.1.5"
@@ -1423,6 +1585,12 @@ version = "1.15.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
 
+[[package]]
+name = "snap"
+version = "1.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1b6b67fb9a61334225b5b790716f609cd58395f895b3fe8b328786812a40bc3b"
+
 [[package]]
 name = "sprs"
 version = "0.11.4"
@@ -1481,6 +1649,17 @@ dependencies = [
  "syn",
 ]
 
+[[package]]
+name = "thrift"
+version = "0.17.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7e54bc85fc7faa8bc175c4bab5b92ba8d9a3ce893d0e9f42cc455c8ab16a9e09"
+dependencies = [
+ "byteorder",
+ "integer-encoding",
+ "ordered-float",
+]
+
 [[package]]
 name = "tiny-keccak"
 version = "2.0.2"
@@ -1500,6 +1679,12 @@ dependencies = [
  "serde_json",
 ]
 
+[[package]]
+name = "twox-hash"
+version = "2.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9ea3136b675547379c4bd395ca6b938e5ad3c3d20fad76e7fe85f9e0d011419c"
+
 [[package]]
 name = "unicode-ident"
 version = "1.0.24"
@@ -1872,8 +2057,42 @@ dependencies = [
  "syn",
 ]
 
+[[package]]
+name = "zlib-rs"
+version = "0.6.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3be3d40e40a133f9c916ee3f9f4fa2d9d63435b5fbe1bfc6d9dae0aa0ada1513"
+
 [[package]]
 name = "zmij"
 version = "1.0.21"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "b8848ee67ecc8aedbaf3e4122217aff892639231befc6a1b58d29fff4c2cabaa"
+
+[[package]]
+name = "zstd"
+version = "0.13.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e91ee311a569c327171651566e07972200e76fcfe2242a4fa446149a3881c08a"
+dependencies = [
+ "zstd-safe",
+]
+
+[[package]]
+name = "zstd-safe"
+version = "7.2.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8f49c4d5f0abb602a93fb8736af2a4f4dd9512e36f7f570d66e65ff867ed3b9d"
+dependencies = [
+ "zstd-sys",
+]
+
+[[package]]
+name = "zstd-sys"
+version = "2.0.16+zstd.1.5.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "91e19ebc2adc8f83e43039e79776e3fda8ca919132d68a1fed6a5faca2683748"
+dependencies = [
+ "cc",
+ "pkg-config",
+]

{rxgraph-0.7.0 → rxgraph-0.9.0}/Cargo.toml

@@ -14,6 +14,7 @@ arrow = "~58.3.0"
 arrow-schema = "~58.3.0"
 criterion = { version = "~0.8.2", features = ["html_reports"] }
 inventory = "~0.3.21"
+parquet = "~58.3.0"
 pretty_assertions = "~1.4.1"
 pyo3 = "~0.28.3"
 pyo3-arrow = "~0.17.0"

{rxgraph-0.7.0 → rxgraph-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rxgraph
-Version: 0.7.0
+Version: 0.9.0
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
@@ -185,6 +185,17 @@ result = graph.search(
 )
 ```
 
+File-backed graphs use the same plugin path and keep Python out of the search
+loop:
+
+```python
+graph = rxg.Graph.from_parquet(
+    nodes="nodes.parquet",
+    edges="edges.parquet",
+    payloads="lazy",
+)
+```
+
 The consumer's crate depends on `rxgraph` with the `python` feature (their own
 crate is the `cdylib` that needs pyo3's `extension-module`; rxgraph's `python`
 feature deliberately does not enable it):
@@ -195,17 +206,18 @@ rxgraph = { version = "0.6", features = ["python"] }
 pyo3 = { version = "0.28", features = ["extension-module"] }
 ```
 
-The Rust side implements `rxgraph::Kernel` and ends with a small macro call:
+The Rust side implements `rxgraph::TypedKernel`, decodes payload structs with
+`TryFrom<rxgraph::ArrowRow<'_>>`, and ends with a small macro call:
 
 ```rust
-rxgraph::plugin! {
+rxgraph::typed_plugin! {
     module = _native;
     "hop_budget" => HopBudget::from_params,
 }
 ```
 
 See [`examples/rust-kernel-plugin/`](examples/rust-kernel-plugin/README.md) for
-the full guide, including the `EdgeCtx` accessor reference and the maturin build.
+the full guide, including the typed payload shape and the maturin build.
 
 ## Architecture
 

{rxgraph-0.7.0 → rxgraph-0.9.0}/README.md

@@ -171,6 +171,17 @@ result = graph.search(
 )
 ```
 
+File-backed graphs use the same plugin path and keep Python out of the search
+loop:
+
+```python
+graph = rxg.Graph.from_parquet(
+    nodes="nodes.parquet",
+    edges="edges.parquet",
+    payloads="lazy",
+)
+```
+
 The consumer's crate depends on `rxgraph` with the `python` feature (their own
 crate is the `cdylib` that needs pyo3's `extension-module`; rxgraph's `python`
 feature deliberately does not enable it):
@@ -181,17 +192,18 @@ rxgraph = { version = "0.6", features = ["python"] }
 pyo3 = { version = "0.28", features = ["extension-module"] }
 ```
 
-The Rust side implements `rxgraph::Kernel` and ends with a small macro call:
+The Rust side implements `rxgraph::TypedKernel`, decodes payload structs with
+`TryFrom<rxgraph::ArrowRow<'_>>`, and ends with a small macro call:
 
 ```rust
-rxgraph::plugin! {
+rxgraph::typed_plugin! {
     module = _native;
     "hop_budget" => HopBudget::from_params,
 }
 ```
 
 See [`examples/rust-kernel-plugin/`](examples/rust-kernel-plugin/README.md) for
-the full guide, including the `EdgeCtx` accessor reference and the maturin build.
+the full guide, including the typed payload shape and the maturin build.
 
 ## Architecture
 

{rxgraph-0.7.0 → rxgraph-0.9.0}/crates/rxgraph/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "rxgraph"
-version = "0.7.0"
+version = "0.9.0"
 edition.workspace = true
 rust-version.workspace = true
 description = "High-performance graph traversal engine"
@@ -22,6 +22,7 @@ anyhow.workspace = true
 arrow.workspace = true
 arrow-schema.workspace = true
 inventory.workspace = true
+parquet.workspace = true
 # Optional: only pulled in by the `python` feature for the PyO3 binding layer.
 # `extension-module` is intentionally NOT enabled here; the final extension
 # crate (cdylib) enables it, so this stays linkable/testable as an rlib.

{rxgraph-0.7.0 → rxgraph-0.9.0}/crates/rxgraph/examples/native_row_lazy.rs

@@ -6,7 +6,7 @@ use std::{
 
 use anyhow::{Context, Result};
 use rxgraph::{
-    GraphId, NodeId, RunOptions, TraversalStrategy, search_native,
+    GraphId, NodeId, RunOptions, TraversalStrategy, Value, search_native,
     traversal::native::{self, GraphStore, OutgoingEdge},
 };
 
@@ -113,7 +113,10 @@ impl native::Kernel for RiskKernel {
         })
     }
 
-    fn visit(&self, cx: &native::EdgeCtx<'_, Self::Node, Self::Edge, Self::State>) -> Result<bool> {
+    fn visit(
+        &self,
+        cx: &native::EdgeCtx<'_, '_, Self::Node, Self::Edge, Self::State>,
+    ) -> Result<bool> {
         let transfer = cx.edge()?;
         let dest = cx.dest()?;
         Ok(transfer.allowed
@@ -123,7 +126,7 @@ impl native::Kernel for RiskKernel {
 
     fn next_state(
         &self,
-        cx: &native::EdgeCtx<'_, Self::Node, Self::Edge, Self::State>,
+        cx: &native::EdgeCtx<'_, '_, Self::Node, Self::Edge, Self::State>,
     ) -> Result<Self::State> {
         let transfer = cx.edge()?;
         let dest = cx.dest()?;
@@ -136,9 +139,16 @@ impl native::Kernel for RiskKernel {
         Ok(next)
     }
 
-    fn stop(&self, cx: &native::EdgeCtx<'_, Self::Node, Self::Edge, Self::State>) -> Result<bool> {
+    fn stop(
+        &self,
+        cx: &native::EdgeCtx<'_, '_, Self::Node, Self::Edge, Self::State>,
+    ) -> Result<bool> {
         Ok(cx.dest()?.target && (!self.require_checkpoint || !cx.state().checkpoints.is_empty()))
     }
+
+    fn state_row(&self, state: &Self::State) -> rxgraph::StateRow {
+        vec![("total_risk".to_string(), Value::U64(state.total_risk))]
+    }
 }
 
 struct FraudStore {

{rxgraph-0.7.0 → rxgraph-0.9.0}/crates/rxgraph/src/dsl/arrow_value.rs

@@ -300,7 +300,7 @@ pub(crate) fn ipc_list_literal_to_value(bytes: &[u8]) -> Result<Value> {
         .map(Value::List)
 }
 
-fn array_row_to_value(array: &dyn Array, row: usize) -> Result<Value> {
+pub(crate) fn array_row_to_value(array: &dyn Array, row: usize) -> Result<Value> {
     macro_rules! primitive {
         ($array:ty, $value:expr) => {
             if let Some(array) = array.as_any().downcast_ref::<$array>() {

{rxgraph-0.7.0 → rxgraph-0.9.0}/crates/rxgraph/src/dsl/value.rs

@@ -113,7 +113,7 @@ impl Value {
         }
     }
 
-    pub(crate) fn as_list(&self) -> Result<Option<&[Value]>> {
+    pub fn as_list(&self) -> Result<Option<&[Value]>> {
         match self {
             Self::Null => Ok(None),
             Self::List(values) => Ok(Some(values)),
@@ -121,7 +121,23 @@ impl Value {
         }
     }
 
-    pub(crate) fn into_struct(self) -> Result<Option<Vec<(String, Value)>>> {
+    pub fn into_list(self) -> Result<Option<Vec<Value>>> {
+        match self {
+            Self::Null => Ok(None),
+            Self::List(values) => Ok(Some(values)),
+            other => bail!("expected list, got {other:?}"),
+        }
+    }
+
+    pub fn as_struct(&self) -> Result<Option<&[(String, Value)]>> {
+        match self {
+            Self::Null => Ok(None),
+            Self::Struct(fields) => Ok(Some(fields)),
+            other => bail!("expected struct, got {other:?}"),
+        }
+    }
+
+    pub fn into_struct(self) -> Result<Option<Vec<(String, Value)>>> {
         match self {
             Self::Null => Ok(None),
             Self::Struct(fields) => Ok(Some(fields)),

{rxgraph-0.7.0 → rxgraph-0.9.0}/crates/rxgraph/src/examples/kernels.rs

@@ -10,7 +10,7 @@ use anyhow::{Result, anyhow};
 use crate::{
     dsl::{StateRow, Value},
     graph::{Graph, NodeId, OwnedGraphId},
-    traversal::{EdgeCtx, Kernel},
+    traversal::{ArrowRow, EdgeCtx, Kernel, PayloadField, TypedKernel, native},
 };
 
 /// Per-path state for [`WeightedBudget`]: the weight accumulated so far.
@@ -118,3 +118,87 @@ crate::inventory::submit! {
         make: |params| Ok(crate::boxed_run(WeightedBudget::from_params(params)?)),
     }
 }
+
+#[derive(Clone, Debug)]
+pub struct WeightedBudgetTyped {
+    pub weight_col: String,
+    pub budget: u64,
+    pub target: OwnedGraphId,
+}
+
+#[derive(Clone, Debug, PartialEq, Eq)]
+pub struct BudgetEdge {
+    pub weight: u64,
+}
+
+impl TryFrom<ArrowRow<'_>> for BudgetEdge {
+    type Error = anyhow::Error;
+
+    fn try_from(row: ArrowRow<'_>) -> Result<Self> {
+        Ok(Self {
+            weight: row.u64("weight")?.unwrap_or(0),
+        })
+    }
+}
+
+impl TypedKernel for WeightedBudgetTyped {
+    type Node = ();
+    type Edge = BudgetEdge;
+    type State = BudgetState;
+
+    fn edge_fields(&self) -> Vec<PayloadField> {
+        vec![PayloadField::aliased(self.weight_col.clone(), "weight")]
+    }
+
+    fn initial_state(
+        &self,
+        _cx: &native::StartCtx<'_, Self::Node, Self::Edge>,
+    ) -> Result<Self::State> {
+        Ok(BudgetState { spent: 0 })
+    }
+
+    fn visit(
+        &self,
+        cx: &native::EdgeCtx<'_, '_, Self::Node, Self::Edge, Self::State>,
+    ) -> Result<bool> {
+        Ok(cx.state().spent.saturating_add(cx.edge()?.weight) <= self.budget)
+    }
+
+    fn next_state(
+        &self,
+        cx: &native::EdgeCtx<'_, '_, Self::Node, Self::Edge, Self::State>,
+    ) -> Result<Self::State> {
+        Ok(BudgetState {
+            spent: cx.state().spent.saturating_add(cx.edge()?.weight),
+        })
+    }
+
+    fn stop(
+        &self,
+        cx: &native::EdgeCtx<'_, '_, Self::Node, Self::Edge, Self::State>,
+    ) -> Result<bool> {
+        Ok(cx.dest_external_id()? == Some(self.target.as_ref()))
+    }
+
+    fn state_row(&self, state: &Self::State) -> StateRow {
+        vec![("spent".to_string(), Value::U64(state.spent))]
+    }
+}
+
+impl WeightedBudgetTyped {
+    fn from_params(params: &serde_json::Value) -> Result<Self> {
+        let base = WeightedBudget::from_params(params)?;
+        Ok(Self {
+            weight_col: base.weight_col,
+            budget: base.budget,
+            target: base.target,
+        })
+    }
+}
+
+crate::inventory::submit! {
+    crate::TypedKernelEntry {
+        name: "weighted_budget",
+        make: |params| Ok(crate::boxed_typed_run(WeightedBudgetTyped::from_params(params)?)),
+    }
+}