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

//! Bus completion emission for auto_process.

use uuid::Uuid;

use crate::events::{RlmCompletion, RlmOutcome};
use crate::result::RlmResult;

use super::types::{CrateAutoProcessContext, LoopOutcome};

/// Emit completion event on the bus, if present.
pub(super) fn emit_bus(
    ctx: &CrateAutoProcessContext<'_>,
    result: &RlmResult,
    outcome: &LoopOutcome,
    trace_id: Uuid,
) {
    if let Some(ref bus) = ctx.bus {
        bus.emit_completion(RlmCompletion {
            trace_id,
            outcome: classify_outcome(outcome),
            iterations: result.stats.iterations,
            subcalls: result.stats.subcalls,
            input_tokens: result.stats.input_tokens,
            output_tokens: result.stats.output_tokens,
            elapsed_ms: result.stats.elapsed_ms,
            reason: outcome.last_error.clone(),
            root_model: ctx.model.clone(),
            subcall_model_used: ctx.subcall_model.clone(),
        });
    }
}

fn classify_outcome(o: &LoopOutcome) -> RlmOutcome {
    if o.aborted {
        return RlmOutcome::Aborted;
    }
    if o.last_error.is_some() {
        return RlmOutcome::Failed;
    }
    if o.final_answer.is_some() {
        RlmOutcome::Converged
    } else {
        RlmOutcome::Exhausted
    }
}