saya-cli 0.4.1

Database-aware AI agent for the terminal: full-screen TUI, schema discovery, and bounded read-only SQL over PostgreSQL, MySQL, SQLite, DuckDB, and Snowflake.
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

//! Tests for the KnowledgeOverridden / KnowledgeLearningSkipped receipt lines, driven through `super::apply_event` (moved verbatim
//! from the inline `tests` module in `stream_events/apply.rs`).

use super::{BlockKind, Transcript, apply_event};
use saya_agent::AgentEvent;
use saya_agent::{LearningSkipReason, OverrideFindingDto};
use saya_types::ClaimId;

fn last_block_text(transcript: &Transcript) -> Option<&str> {
    transcript.blocks().last().map(|b| b.text.as_str())
}

/// KnowledgeOverridden pushes a System block whose text names the referenced
/// column and the specified value. Trails the answer — the
/// block lands below the assistant text in the transcript.
#[test]
fn knowledge_overridden_pushes_a_system_block_naming_the_finding() {
    let mut t = Transcript::new();
    apply_event(
        &mut t,
        AgentEvent::knowledge_overridden(vec![OverrideFindingDto {
            claim_id: ClaimId::parse("c-rental-time").unwrap(),
            kind: "default_time_column".into(),
            claimed_value: "return_date".into(),
            observed_columns: vec!["rental_date".into()],
        }]),
        false,
    );
    let block = last_block_text(&t).expect("a block was pushed");
    assert!(
        block.contains("memory overridden · 1 finding"),
        "header: {block}"
    );
    assert!(
        block.contains("referenced rental_date"),
        "names the referenced column: {block}"
    );
    assert!(
        block.contains("where you specified return_date"),
        "names the specified value: {block}"
    );
    // The wording constraint: no causal "used" about the time column.
    assert!(
        !block.contains("used"),
        "the TUI block must not assert a causal 'used': {block}"
    );
    assert_eq!(t.blocks().last().unwrap().kind, BlockKind::System);
}

/// An empty finding set pushes nothing — silence.
#[test]
fn an_empty_knowledge_overridden_event_pushes_nothing() {
    let mut t = Transcript::new();
    apply_event(&mut t, AgentEvent::knowledge_overridden(Vec::new()), false);
    assert!(
        t.blocks().is_empty(),
        "no findings → no block: {:?}",
        t.blocks()
    );
}

/// KnowledgeLearningSkipped pushes a System block whose text names the skip
/// reason (packet-54 decision 4 — the TUI renders it, it does not fall
/// through to the catch-all that would drop it). Trails the answer.
#[test]
fn knowledge_learning_skipped_pushes_a_system_block_naming_the_reason() {
    let mut t = Transcript::new();
    apply_event(
        &mut t,
        AgentEvent::knowledge_learning_skipped(LearningSkipReason::TimedOut),
        false,
    );
    let block = last_block_text(&t).expect("a block was pushed");
    assert!(
        block.contains("memory not recorded · extraction timed out"),
        "TUI block names the timeout: {block}"
    );
    assert_eq!(t.blocks().last().unwrap().kind, BlockKind::System);

    let mut t = Transcript::new();
    apply_event(
        &mut t,
        AgentEvent::knowledge_learning_skipped(LearningSkipReason::Failed),
        false,
    );
    let block = last_block_text(&t).expect("a block was pushed");
    assert!(
        block.contains("memory not recorded · extraction failed"),
        "TUI block names the failure: {block}"
    );
}