loctree 0.13.1

Structural code intelligence for AI agents. Scan once, query everything.
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

use super::{AicxIntent, ProjectScope};
use std::sync::mpsc;
use std::time::Duration;

#[derive(Debug)]
pub enum AicxInProcessError {
    TimedOut,
    Other(anyhow::Error),
}

#[derive(Debug)]
pub struct AicxInProcessClient {
    client: aicx::api::Aicx,
}

impl AicxInProcessClient {
    pub fn from_env() -> Result<Self, anyhow::Error> {
        let base = aicx::api::Aicx::from_env()?;
        let store_root = base.store_root().join("store");
        Ok(Self {
            client: aicx::api::Aicx::with_store_root(store_root),
        })
    }

    pub fn intents(
        &self,
        scope: &ProjectScope,
        window_hours: u64,
        limit: usize,
        cap: Option<Duration>,
    ) -> Result<Vec<AicxIntent>, AicxInProcessError> {
        let Some(cap) = cap else {
            return self
                .intents_unbounded(scope, window_hours, limit)
                .map_err(AicxInProcessError::Other);
        };

        let scope = scope.clone();
        let (tx, rx) = mpsc::channel();
        std::thread::spawn(move || {
            let result = Self::from_env()
                .and_then(|client| client.intents_unbounded(&scope, window_hours, limit));
            let _ = tx.send(result);
        });

        match rx.recv_timeout(cap) {
            Ok(Ok(rows)) => Ok(rows),
            Ok(Err(error)) => Err(AicxInProcessError::Other(error)),
            Err(mpsc::RecvTimeoutError::Timeout) => Err(AicxInProcessError::TimedOut),
            Err(mpsc::RecvTimeoutError::Disconnected) => Err(AicxInProcessError::Other(
                anyhow::anyhow!("in-process AICX worker disconnected"),
            )),
        }
    }

    fn intents_unbounded(
        &self,
        scope: &ProjectScope,
        window_hours: u64,
        limit: usize,
    ) -> Result<Vec<AicxIntent>, anyhow::Error> {
        let mut config = aicx::intents::IntentsConfig {
            project: scope.primary().to_string(),
            hours: window_hours,
            strict: false,
            min_confidence: None,
            kind_filter: None,
            frame_kind: Some(aicx::timeline::FrameKind::UserMsg),
        };
        let projects = scope.projects();
        let extraction = if projects.len() > 1 {
            let projects = projects.to_vec();
            self.client
                .extract_intents_for_projects(&config, &projects)?
        } else {
            if config.project.is_empty() {
                config.project = String::new();
            }
            self.client.extract_intents(&config)?
        };

        Ok(extraction
            .records
            .into_iter()
            .take(limit)
            .map(intent_record_to_loctree)
            .collect())
    }
}

fn intent_record_to_loctree(record: aicx::intents::IntentRecord) -> AicxIntent {
    AicxIntent {
        kind: intent_kind_label(record.kind).to_string(),
        text: record.summary,
        agent: record.agent,
        date: record.date,
        timestamp: record.timestamp,
        session_id: record.session_id,
        project: record.project,
        source_chunk_path: record.source_chunk,
        frame_kind: Some("user_msg".to_string()),
        oracle_status: None,
    }
}

fn intent_kind_label(kind: aicx::intents::IntentKind) -> &'static str {
    match kind {
        aicx::intents::IntentKind::Decision => "decision",
        aicx::intents::IntentKind::Intent => "intent",
        aicx::intents::IntentKind::Outcome => "outcome",
        aicx::intents::IntentKind::Task => "task",
    }
}