vtcode-core 0.104.1

Core library for VT Code - a Rust-based terminal coding agent
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

pub mod data;
pub mod feedback;

pub use data::CompletionLearningData;
pub use feedback::FeedbackProcessor;

use hashbrown::HashMap;

/// Learning system for improving completion suggestions
pub struct LearningSystem {
    learning_data: CompletionLearningData,
    feedback_processor: FeedbackProcessor,
}

impl Default for LearningSystem {
    fn default() -> Self {
        Self::new()
    }
}

impl LearningSystem {
    pub fn new() -> Self {
        Self {
            learning_data: CompletionLearningData::default(),
            feedback_processor: FeedbackProcessor::new(),
        }
    }

    /// Process user feedback to improve future suggestions
    pub fn process_feedback(&mut self, suggestion_text: &str, accepted: bool, context: &str) {
        self.feedback_processor
            .process(suggestion_text, accepted, context);
        self.learning_data
            .update_from_feedback(suggestion_text, accepted);
    }

    /// Get learning insights for a given context
    pub fn get_insights(&self, context: &str) -> HashMap<String, f64> {
        self.learning_data.get_context_insights(context)
    }
}