vv-agent 0.6.2

VectorVein agent runtime, SDK, CLI, tools, and workspace backends
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

use std::path::PathBuf;
use std::sync::Arc;

const DEFAULT_MIN_TOKENS: u64 = 10_000;
const DEFAULT_MAX_TOKENS: u64 = 40_000;
const DEFAULT_MIN_TEXT_MESSAGES: usize = 5;
const DEFAULT_GROWTH_RATIO: f64 = 0.5;

pub type SessionMemoryExtractionCallback =
    Arc<dyn Fn(&str, Option<&str>, Option<&str>) -> Option<String> + Send + Sync + 'static>;

#[derive(Clone)]
pub struct SessionMemoryConfig {
    pub min_tokens_before_extraction: u64,
    pub max_tokens: u64,
    pub min_text_messages: usize,
    pub growth_ratio: f64,
    pub storage_dir: PathBuf,
    pub extraction_callback: Option<SessionMemoryExtractionCallback>,
    pub extraction_backend: Option<String>,
    pub extraction_model: Option<String>,
    pub token_model: String,
}

impl std::fmt::Debug for SessionMemoryConfig {
    fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        formatter
            .debug_struct("SessionMemoryConfig")
            .field(
                "min_tokens_before_extraction",
                &self.min_tokens_before_extraction,
            )
            .field("max_tokens", &self.max_tokens)
            .field("min_text_messages", &self.min_text_messages)
            .field("growth_ratio", &self.growth_ratio)
            .field("storage_dir", &self.storage_dir)
            .field(
                "extraction_callback",
                &self.extraction_callback.as_ref().map(|_| "<callback>"),
            )
            .field("extraction_backend", &self.extraction_backend)
            .field("extraction_model", &self.extraction_model)
            .field("token_model", &self.token_model)
            .finish()
    }
}

impl Default for SessionMemoryConfig {
    fn default() -> Self {
        Self {
            min_tokens_before_extraction: DEFAULT_MIN_TOKENS,
            max_tokens: DEFAULT_MAX_TOKENS,
            min_text_messages: DEFAULT_MIN_TEXT_MESSAGES,
            growth_ratio: DEFAULT_GROWTH_RATIO,
            storage_dir: PathBuf::from(".memory/session"),
            extraction_callback: None,
            extraction_backend: None,
            extraction_model: None,
            token_model: String::new(),
        }
    }
}