keyclaw 0.2.1

Local MITM proxy that keeps secrets out of LLM traffic
Documentation 
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

use keyclaw::placeholder::EXAMPLE_PLACEHOLDER;
use keyclaw::redaction::{NoticeMode, inject_redaction_notice, inject_redaction_notice_with_mode};

#[test]
fn inject_redaction_notice_uses_shared_placeholder_example() {
    let input = br#"{"messages":[{"role":"user","content":"hello"}]}"#;

    let rewritten = inject_redaction_notice(input, 2).expect("inject notice");
    let output = String::from_utf8(rewritten).expect("utf8");

    assert!(output.contains(EXAMPLE_PLACEHOLDER), "output={output}");
}

#[test]
fn inject_redaction_notice_minimal_mode_uses_shorter_text() {
    let input = br#"{"messages":[{"role":"user","content":"hello"}]}"#;

    let rewritten =
        inject_redaction_notice_with_mode(input, 2, NoticeMode::Minimal).expect("inject notice");
    let output = String::from_utf8(rewritten).expect("utf8");

    assert!(output.contains(EXAMPLE_PLACEHOLDER), "output={output}");
    assert!(output.contains("secret(s)"), "output={output}");
    assert!(!output.contains("IMPORTANT:"), "output={output}");
}

#[test]
fn inject_redaction_notice_off_mode_leaves_json_unchanged() {
    let input = br#"{"messages":[{"role":"user","content":"hello"}]}"#;

    let rewritten =
        inject_redaction_notice_with_mode(input, 2, NoticeMode::Off).expect("inject notice");

    assert_eq!(rewritten, input);
}