diff --git a/crates/code2prompt-core/src/template.rs b/crates/code2prompt-core/src/template.rs index 362fb15..0047e39 100644 --- a/crates/code2prompt-core/src/template.rs +++ b/crates/code2prompt-core/src/template.rs @@ -37,7 +37,16 @@ pub fn handlebars_setup(template_str: &str, template_name: &str) -> Result` - A vector of undefined variable names. pub fn extract_undefined_variables(template: &str) -> Vec { - let registered_identifiers = ["path", "code", "git_diff"]; + let registered_identifiers = [ + "absolute_code_path", + "source_tree", + "files", + "path", + "code", + "git_diff", + "git_diff_branch", + "git_log_branch" + ]; let re = Regex::new(r"\{\{\s*(?P[a-zA-Z_][a-zA-Z_0-9]*)\s*\}\}").unwrap(); re.captures_iter(template) .map(|cap| cap["var"].to_string())