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

use crate::oracle::QueryType;

use super::TreeSitterOracle;

const STRUCTURAL_TERMS: &[&str] = &[
    "signature",
    "parameters",
    "return type",
    "fields of",
    "what fields",
    "struct definition",
    "enum variants",
    "implements",
    "methods",
];

impl TreeSitterOracle {
    pub(crate) fn classify_query(query: &str) -> QueryType {
        let lower = query.to_lowercase();
        if STRUCTURAL_TERMS.iter().any(|term| lower.contains(term)) {
            QueryType::Structural
        } else {
            QueryType::Semantic
        }
    }
}