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

use crate::common;

use common::{module, segment};
use tea_context::{
    BudgetBehavior, CacheScope, PromptAuthority, PromptBudget, PromptCompiler, SegmentDisposition,
    TrustLevel,
};

#[test]
fn included_byte_ranges_slice_exact_content_and_preserve_provenance() {
    let prompt = PromptCompiler
        .compile(
            [module(
                "inspect",
                PromptAuthority::Workspace,
                0,
                vec![
                    segment("one", "alpha", BudgetBehavior::Required),
                    segment("two", "βeta", BudgetBehavior::Required),
                ],
            )],
            PromptBudget::new(100, 100).unwrap(),
        )
        .unwrap();
    assert_eq!(prompt.text(), "alpha\n\nβeta");
    for entry in prompt.inspection() {
        let range = entry.byte_range().unwrap().clone();
        let slice = &prompt.text()[range];
        assert_eq!(slice.len(), entry.rendered_bytes());
        assert_eq!(entry.disposition(), SegmentDisposition::Included);
        assert_eq!(entry.trust(), TrustLevel::Delegated);
        assert_eq!(entry.cache_scope(), CacheScope::Run);
        assert_eq!(entry.provenance().source_kind(), "test");
        assert!(!slice.is_empty());
    }
    assert_eq!(prompt.inspection()[0].byte_range().unwrap(), &(0..5));
    assert_eq!(prompt.inspection()[1].byte_range().unwrap(), &(7..12));
}

#[test]
fn inspection_is_identical_for_permuted_module_inputs() {
    let modules = vec![
        module(
            "later",
            PromptAuthority::Session,
            0,
            vec![segment("later", "later", BudgetBehavior::Required)],
        ),
        module(
            "earlier",
            PromptAuthority::Product,
            0,
            vec![segment("earlier", "earlier", BudgetBehavior::Required)],
        ),
    ];
    let forward = PromptCompiler
        .compile(modules.clone(), PromptBudget::new(100, 100).unwrap())
        .unwrap();
    let reverse = PromptCompiler
        .compile(
            modules.into_iter().rev(),
            PromptBudget::new(100, 100).unwrap(),
        )
        .unwrap();
    assert_eq!(forward.inspection(), reverse.inspection());
    assert_eq!(forward.diagnostics(), reverse.diagnostics());
}