1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
//! Host implementation of `murmur:runtime/tokens@0.4.0#count`.
//!
//! A hook that imports this interface measures text with the host's own
//! `cl100k_base` tokenizer — [`crate::agent::count_tokens`], the counter behind
//! [`crate::agent::ContextOccupancy`] and the seed budget. A hook sizing a
//! payload with it agrees with every estimate the host makes. It does not agree
//! with a driver's reported usage, which steers compaction and spend once a
//! call reports it.
//!
//! Counting is pure computation — no side effect, no host resource, nothing to
//! withhold — so the import is registered on every hook linker unconditionally
//! and no capability grant gates it. See `wit/hook/tokens.wit`.
use wasmtime::component::Linker;
/// The versioned instance name the host provides `count` under. Hook components
/// that do not import it simply ignore the registration.
pub(crate) const TOKENS_IFACE_VERSIONED: &str = "murmur:runtime/tokens@0.4.0";
/// Register `murmur:runtime/tokens@0.4.0#count` on a hook linker.
///
/// Must be called on **every** linker a hook component instantiates against.
/// The `hook` world declares this import, so a linker missing it turns any hook
/// that actually calls `count` into an instantiation failure — the one outcome
/// a purely additive import must not produce.
pub(crate) fn add_tokens_to_linker<T: 'static>(linker: &mut Linker<T>) -> Result<(), String> {
linker
.instance(TOKENS_IFACE_VERSIONED)
.map_err(|e| format!("failed to define {TOKENS_IFACE_VERSIONED}: {e}"))?
.func_wrap(
"count",
|_store: wasmtime::StoreContextMut<'_, T>, (text,): (String,)| {
Ok((u64::from(crate::agent::count_tokens(&text)),))
},
)
.map_err(|e| format!("failed to register {TOKENS_IFACE_VERSIONED}#count: {e}"))?;
Ok(())
}