tea-context 0.1.0

Deterministic prompt modules, context providers, and compiler for tea-rs
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
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

use std::str::FromStr;

use tea_context::{
    CacheScope, ConflictKey, ContextProviderId, PromptAuthority, PromptModuleId, PromptProvenance,
    PromptSegmentId, SkillId, TrustLevel,
};

#[test]
fn canonical_context_ids_round_trip_and_reject_invalid_values() {
    for value in ["kernel.identity", "workspace/main", "tool-read_hint"] {
        let id = PromptModuleId::from_str(value).unwrap();
        assert_eq!(id.as_str(), value);
        assert_eq!(
            serde_json::from_value::<PromptModuleId>(serde_json::to_value(&id).unwrap()).unwrap(),
            id
        );
    }
    for value in ["", "Upper", "../escape", "double..dot", "has space"] {
        assert!(PromptSegmentId::from_str(value).is_err());
        assert!(ContextProviderId::from_str(value).is_err());
        assert!(ConflictKey::from_str(value).is_err());
        assert!(SkillId::from_str(value).is_err());
    }
}

#[test]
fn authority_order_is_fixed_high_to_low() {
    assert!(PromptAuthority::Kernel < PromptAuthority::Organization);
    assert!(PromptAuthority::Organization < PromptAuthority::Product);
    assert!(PromptAuthority::Product < PromptAuthority::Workspace);
    assert!(PromptAuthority::Workspace < PromptAuthority::Tool);
    assert!(PromptAuthority::Tool < PromptAuthority::Skill);
    assert!(PromptAuthority::Skill < PromptAuthority::Session);
    assert!(PromptAuthority::Session < PromptAuthority::UserAddition);
}

#[test]
fn provenance_trust_and_cache_round_trip() {
    let provenance = PromptProvenance::new(
        ContextProviderId::from_str("workspace.instructions").unwrap(),
        "workspace_file",
        Some("/workspace/AGENTS.md".to_owned()),
    )
    .unwrap();
    assert_eq!(provenance.source_kind(), "workspace_file");
    assert_eq!(provenance.locator(), Some("/workspace/AGENTS.md"));
    assert_eq!(
        serde_json::from_value::<PromptProvenance>(serde_json::to_value(&provenance).unwrap())
            .unwrap(),
        provenance
    );
    for trust in [
        TrustLevel::Trusted,
        TrustLevel::Delegated,
        TrustLevel::Untrusted,
    ] {
        serde_json::to_value(trust).unwrap();
    }
    for scope in [
        CacheScope::None,
        CacheScope::Run,
        CacheScope::Session,
        CacheScope::Profile,
        CacheScope::Global,
    ] {
        serde_json::to_value(scope).unwrap();
    }
}

#[test]
fn provenance_bounds_fail_closed() {
    let provider = ContextProviderId::from_str("test.provider").unwrap();
    assert!(PromptProvenance::new(provider.clone(), "Invalid", None).is_err());
    assert!(PromptProvenance::new(provider, "source", Some("x".repeat(2049))).is_err());
}