GameCult/ModelConnector
Watch
9
0
Fork
You've already forked ModelConnector
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

lease-persona-callers: the identity bound keeps room for the Persona sequence

Browse source 
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
Meta Vens 2026-10-31 01:59:39 +01:00
commit d2b82a6c50
1 changed files with 26 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

28
crates/model-connector/src/service/lease.rs
View file

@ -129,7 +129,11 @@ impl TransportService {
let spend_window_ms = caller.spend_window_ms;
let connection_key_epoch = caller.connection_key_epoch;
require_id(invocation_id, "invocation_id")?;
if invocation_id.len() + LEASE_PREFIX.len() + SEQUENCE_SUFFIX_BYTES > MAX_IDENTITY_BYTES {
let suffix = match turn {
LeaseTurn::SelfRun => 0,
LeaseTurn::PersonaTurn => SEQUENCE_SUFFIX_BYTES,
};
if invocation_id.len() + LEASE_PREFIX.len() + suffix > MAX_IDENTITY_BYTES {
return Err(ContractError::Invalid("invocation_id").into());
}
if max_cost_micro_usd == 0 {
@ -724,11 +728,31 @@ mod tests {
.unwrap();
assert!(
service
.begin_lease(CALLER, LeaseTurn::SelfRun, &format!("{longest}r"), 100, NOW + 1, NOW)
.begin_lease(
CALLER,
LeaseTurn::SelfRun,
&format!("{longest}r"),
100,
NOW + 1,
NOW
)
.is_err()
);
}
#[test]
fn a_persona_invocation_id_leaves_room_for_its_sequence() {
let store = TestStore::new();
let mut service = open(&store, 1 << 40);
let longest = "r".repeat(MAX_IDENTITY_BYTES - LEASE_PREFIX.len() - SEQUENCE_SUFFIX_BYTES);
let persona = |service: &mut TransportService, id: &str| {
service.begin_lease(CALLER, LeaseTurn::PersonaTurn, id, 100, NOW + 60_000, NOW)
};
let grant = granted(persona(&mut service, &longest).unwrap());
assert!(grant.lease_id.len() <= MAX_IDENTITY_BYTES);
assert!(persona(&mut service, &format!("{longest}r")).is_err());
}
#[test]
fn a_settled_lease_leaves_its_charge_not_its_reservation_across_a_restart() {
let store = TestStore::new();