b57c022ea1
E1 unblocked: cost column added back to chat_messages via FieldKind::RealDefault(0.0). save_message passes through f64; search restores via r['cost'].as_f64(). Tests: 6/6 (was 5, +1 cost_roundtrips_via_search asserts 0.00777 f64). Known deferred (flagged for future): 1. chat_sessions TEXT-PK requires engine multi-schema support (Store::open takes single schema) — bespoke DDL retained. 2. kei-crossdomain re-migration deferred — engine TextPairWithMetadata extra_columns field not yet shipped; edge DDL hard-codes src_path/ dst_path vs from_uri/to_uri. Bespoke DDL retained. Follow-up: extend engine edge variant with extra_columns + column name customization. kei-crossdomain tests unchanged 5/5. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
100 lines
3.5 KiB
Rust
100 lines
3.5 KiB
Rust
use kei_chat_store::search::search;
|
|
use kei_chat_store::sessions::{archive_session, get_session, save_message, start_session, ChatMessage};
|
|
use kei_chat_store::stats::stats;
|
|
use kei_chat_store::Store;
|
|
|
|
fn mk() -> Store { Store::open_memory().unwrap() }
|
|
|
|
#[test]
|
|
fn save_and_retrieve() {
|
|
let s = mk();
|
|
let sid = start_session(&s, "demo", "t", "claude-opus-4").unwrap();
|
|
save_message(&s, &ChatMessage {
|
|
session_id: sid.clone(), role: "user".into(),
|
|
content: "hello world".into(), tokens_in: 3, tokens_out: 0, cost: 0.001,
|
|
..Default::default()
|
|
}).unwrap();
|
|
let sess = get_session(&s, &sid).unwrap().unwrap();
|
|
assert_eq!(sess.message_count, 1);
|
|
assert_eq!(sess.total_tokens, 3);
|
|
}
|
|
|
|
#[test]
|
|
fn fts_search_finds_message() {
|
|
let s = mk();
|
|
let sid = start_session(&s, "demo", "", "").unwrap();
|
|
save_message(&s, &ChatMessage {
|
|
session_id: sid, role: "user".into(),
|
|
content: "rust async tokio bench".into(),
|
|
..Default::default()
|
|
}).unwrap();
|
|
let hits = search(&s, "tokio", 10).unwrap();
|
|
assert_eq!(hits.len(), 1);
|
|
}
|
|
|
|
#[test]
|
|
fn archive_session_works() {
|
|
let s = mk();
|
|
let sid = start_session(&s, "p", "", "").unwrap();
|
|
archive_session(&s, &sid).unwrap();
|
|
let sess = get_session(&s, &sid).unwrap().unwrap();
|
|
assert_eq!(sess.status, "archived");
|
|
}
|
|
|
|
#[test]
|
|
fn engine_migration_parity_smoke() {
|
|
// Layer-A convergence parity: fresh session opens cleanly through
|
|
// kei_entity_store::Store + CHAT_SCHEMA, start_session returns a
|
|
// UUID that get_session can retrieve with id > 0 chars and the
|
|
// engine-generated chat_messages table is writeable.
|
|
let s = mk();
|
|
let sid = start_session(&s, "conv-layer-a", "smoke", "claude-opus-4").unwrap();
|
|
assert!(!sid.is_empty(), "session id should be a non-empty UUID");
|
|
let mid = save_message(&s, &ChatMessage {
|
|
session_id: sid.clone(), role: "user".into(),
|
|
content: "migration-parity probe".into(),
|
|
tokens_in: 1, tokens_out: 0, cost: 0.0,
|
|
..Default::default()
|
|
}).unwrap();
|
|
assert!(mid > 0, "engine-backed message id must be > 0");
|
|
let sess = get_session(&s, &sid).unwrap().unwrap();
|
|
assert_eq!(sess.id, sid);
|
|
assert_eq!(sess.message_count, 1);
|
|
}
|
|
|
|
#[test]
|
|
fn cost_roundtrips_via_search() {
|
|
// Wave-8 re-migration: cost is re-instated as engine-managed
|
|
// RealDefault column. The value written via save_message must be
|
|
// visible on the ChatMessage returned from search (no longer 0.0).
|
|
let s = mk();
|
|
let sid = start_session(&s, "demo", "", "").unwrap();
|
|
save_message(&s, &ChatMessage {
|
|
session_id: sid, role: "user".into(),
|
|
content: "rust async tokio bench cost-marker".into(),
|
|
tokens_in: 1, tokens_out: 1, cost: 0.00777,
|
|
..Default::default()
|
|
}).unwrap();
|
|
let hits = search(&s, "cost-marker", 10).unwrap();
|
|
assert_eq!(hits.len(), 1);
|
|
assert!((hits[0].cost - 0.00777).abs() < 1e-9,
|
|
"cost should round-trip via engine RealDefault column, got {}",
|
|
hits[0].cost);
|
|
}
|
|
|
|
#[test]
|
|
fn stats_aggregates() {
|
|
let s = mk();
|
|
let sid = start_session(&s, "p", "", "").unwrap();
|
|
for _ in 0..3 {
|
|
save_message(&s, &ChatMessage {
|
|
session_id: sid.clone(), role: "user".into(),
|
|
content: "x".into(), tokens_in: 5, tokens_out: 5, cost: 0.01,
|
|
..Default::default()
|
|
}).unwrap();
|
|
}
|
|
let st = stats(&s).unwrap();
|
|
assert_eq!(st.total_sessions, 1);
|
|
assert_eq!(st.total_messages, 3);
|
|
assert_eq!(st.total_tokens, 30);
|
|
}
|