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

//! Query extraction from router context.

use crate::router::CrateAutoProcessContext;

/// Build the user-facing question for an RLM request.
pub fn extract(ctx: &CrateAutoProcessContext<'_>) -> String {
    if let Some(q) = ctx.tool_args.get("query").and_then(|v| v.as_str()) {
        return q.to_string();
    }
    if let Some(p) = ctx.tool_args.get("pattern").and_then(|v| v.as_str()) {
        return format!("Find all occurrences of '{p}'");
    }
    default_query(ctx.tool_id)
}

fn default_query(tool_id: &str) -> String {
    match tool_id {
        "grep" => "Summarize and group the search matches.".into(),
        "read" => "Summarize the file structure and important logic.".into(),
        "bash" => "Summarize errors, warnings, and key command output.".into(),
        "session_context" | "context_reset" => "Create a continuation briefing.".into(),
        _ => "Summarize the most important information.".into(),
    }
}