codetether-rlm 0.1.0

Recursive Language Model processing for CodeTether
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

//! Router unit tests.

use crate::capability::output_capability;
use crate::config::RlmConfig;
use crate::router::extract::extract_final;
use crate::router::{RoutingContext, should_route};

#[test]
fn webfetch_routes_when_oversized() {
    let output = "a".repeat(200_000);
    let ctx = RoutingContext {
        tool_id: "webfetch".into(),
        session_id: "s".into(),
        call_id: None,
        model_context_limit: 200_000,
        current_context_tokens: Some(1000),
    };
    assert!(should_route(&output, &ctx, &RlmConfig::default()).should_route);
}

#[test]
fn unknown_tools_never_route() {
    let output = "a".repeat(200_000);
    let ctx = RoutingContext {
        tool_id: "some_new_tool".into(),
        session_id: "s".into(),
        call_id: None,
        model_context_limit: 200_000,
        current_context_tokens: Some(1000),
    };
    let r = should_route(&output, &ctx, &RlmConfig::default());
    assert!(!r.should_route);
    assert_eq!(r.reason, "tool_unknown_capability_no_route");
}

#[test]
fn exact_content_tools_never_route() {
    let output = "a".repeat(500_000);
    let ctx = RoutingContext {
        tool_id: "read".into(),
        session_id: "s".into(),
        call_id: None,
        model_context_limit: 200_000,
        current_context_tokens: Some(1000),
    };
    let r = should_route(&output, &ctx, &RlmConfig::default());
    assert!(!r.should_route);
    assert_eq!(r.reason, "tool_exact_content_no_route");
}

#[test]
fn extract_final_works() {
    assert_eq!(extract_final(r#"FINAL("hello")"#), Some("hello".into()));
    assert_eq!(extract_final(r#"FINAL('') x"#), None);
    assert_eq!(extract_final("no final"), None);
}