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
27
28
29
30
31
32
33
34
35

use super::{TreeSitterOracle, TreeSitterVerification, answer_names, compare};

impl TreeSitterOracle {
    pub(crate) fn verify_functions(&mut self, answer: &str) -> TreeSitterVerification {
        let functions = match self.get_functions() {
            Ok(functions) => functions,
            Err(error) => return parse_failure("functions", error),
        };
        let actual = functions.into_iter().map(|f| f.name).collect();
        compare::compare_names(
            answer_names::functions(answer).unwrap_or_default(),
            actual,
            "Function",
        )
    }

    pub(crate) fn verify_structs(&mut self, answer: &str) -> TreeSitterVerification {
        let structs = match self.get_structs() {
            Ok(structs) => structs,
            Err(error) => return parse_failure("structs", error),
        };
        let actual = structs.into_iter().map(|s| s.name).collect();
        compare::compare_names(
            answer_names::structs(answer).unwrap_or_default(),
            actual,
            "Struct",
        )
    }
}

fn parse_failure(kind: &str, error: anyhow::Error) -> TreeSitterVerification {
    TreeSitterVerification::CannotVerify {
        reason: format!("Failed to parse {kind}: {error}"),
    }
}