rskit-agent 0.2.0-alpha.1

Agentic loop — Provider + Tools + Hooks in a turn-based execution engine
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

//! Agent context compaction integration.

use rskit_ai::chat::count_tokens_approx;
use rskit_errors::AppResult;
use rskit_llm::types::Message;

use crate::types::{ContextStrategy, FailStrategy};

pub(crate) fn compact_if_needed(
    mut messages: Vec<Message>,
    max_input_tokens: Option<u64>,
    strategy: Option<&dyn ContextStrategy>,
) -> AppResult<Vec<Message>> {
    if let Some(max_input_tokens) = max_input_tokens {
        let limit = usize::try_from(max_input_tokens).unwrap_or(usize::MAX);
        if count_tokens_approx(&messages) > limit {
            let strategy = strategy.unwrap_or(&FailStrategy);
            messages = strategy.compact(messages, limit)?;
        }
    }
    Ok(messages)
}