mirror of
https://github.com/headroomlabs-ai/headroom.git
synced 2026-08-27 14:17:10 -04:00
e530de5ad2
Adds crates/headroom-core/src/transforms/code_compressor.rs (1,882 lines): the AST-aware CodeCompressor ported to Rust on tree-sitter, with grammars for Python, JavaScript, TypeScript, Go, Rust, Java, C and C++. Parity-only, like #1153. Nothing calls it: the only references outside the module are the pub mod / pub use declarations in transforms/mod.rs, and live_zone.rs still routes SourceCode to a no-op. The pyo3 bridge is untouched and no Python source changes, so the engine is unreachable from the shipped package. #1155 wires it into live-zone dispatch. Every grammar is pinned with '=' to the exact version of the corresponding Python tree-sitter-<lang> PyPI wheel. Same version on crates.io and PyPI means the same grammar.js, hence the same generated parser.c, hence node-for-node identical ASTs — the precondition for byte-parity. A canary over 9 samples x 8 languages confirmed identical node-type and line-span trees at these pins; bumping any pin requires re-running it and re-recording the fixtures. Ships 30 recorded parity fixtures, a CodeCompressorComparator in headroom-parity, and scripts/record_code_compressor_fixtures.py. Verified byte-identical to the recorded Python output: [code_aware_compressor] total=30 matched=30 skipped=0 diffed=0 Full harness on the merge result: 227 fixtures, 182 matched, 45 skipped (cache_aligner + ccr stubs), 0 diffed, exit 0 — with kompress at 21/21 under ONNX Runtime 1.24.4 (see #2591). Also verified cargo check -p headroom-core --no-default-features passes, so the static-musl path stays intact.
42 lines
3.4 KiB
JSON
42 lines
3.4 KiB
JSON
{
|
|
"config": {
|
|
"ccr_ttl": 300,
|
|
"compress_comments": true,
|
|
"docstring_mode": "first_line",
|
|
"enable_ccr": false,
|
|
"fallback_to_kompress": false,
|
|
"language_hint": null,
|
|
"max_body_lines": 5,
|
|
"min_tokens_for_compression": 100,
|
|
"preserve_decorators": true,
|
|
"preserve_imports": true,
|
|
"preserve_signatures": true,
|
|
"preserve_type_annotations": true,
|
|
"semantic_analysis": true,
|
|
"target_compression_rate": 0.2
|
|
},
|
|
"input": "use std::collections::HashMap;\n\npub struct Processor {\n name: String,\n count: u32,\n}\n\nimpl Processor {\n pub fn new(name: String) -> Self {\n Self { name, count: 0 }\n }\n\n pub fn process(&mut self, items: Vec<String>) -> Vec<String> {\n let mut results = Vec::new();\n for item in items {\n if item.is_empty() {\n continue;\n }\n let clean = item.trim().to_lowercase();\n results.push(clean);\n self.count += 1;\n }\n results\n }\n}\n\n#[derive(Debug)]\nenum Color {\n Red,\n Green,\n Blue,\n}\n\nfn main() {\n let mut p = Processor::new(\"main\".to_string());\n println!(\"{:?}\", p.process(vec![]));\n}\n\n// variant 1",
|
|
"input_sha256": "b8fa00d165cd059a0549473920e4d61bcbde3980257d328f2cbc76fc3a3db787",
|
|
"output": {
|
|
"cache_key": null,
|
|
"compressed": "use std::collections::HashMap;\n\npub struct Processor {\n name: String,\n count: u32,\n}\nenum Color {\n Red,\n Green,\n Blue,\n}\n\nimpl Processor {\n pub fn new(name: String) -> Self {\n Self { name, count: 0 }\n }\n\n pub fn process(&mut self, items: Vec<String>) -> Vec<String> {\n let mut results = Vec::new();\n for item in items {\n if item.is_empty() {\n continue;\n }\n let clean = item.trim().to_lowercase();\n results.push(clean);\n self.count += 1;\n }\n results\n }\n}\n\nfn main() {\n let mut p = Processor::new(\"main\".to_string());\n // [1 lines omitted; calls: Processor, new, process]\n}\n\n#[derive(Debug)]\n// variant 1",
|
|
"compressed_bodies": 0,
|
|
"compressed_tokens": 185,
|
|
"compression_ratio": 1.022099447513812,
|
|
"language": "rust",
|
|
"language_confidence": 1.0,
|
|
"original": "use std::collections::HashMap;\n\npub struct Processor {\n name: String,\n count: u32,\n}\n\nimpl Processor {\n pub fn new(name: String) -> Self {\n Self { name, count: 0 }\n }\n\n pub fn process(&mut self, items: Vec<String>) -> Vec<String> {\n let mut results = Vec::new();\n for item in items {\n if item.is_empty() {\n continue;\n }\n let clean = item.trim().to_lowercase();\n results.push(clean);\n self.count += 1;\n }\n results\n }\n}\n\n#[derive(Debug)]\nenum Color {\n Red,\n Green,\n Blue,\n}\n\nfn main() {\n let mut p = Processor::new(\"main\".to_string());\n println!(\"{:?}\", p.process(vec![]));\n}\n\n// variant 1",
|
|
"original_tokens": 181,
|
|
"preserved_imports": 1,
|
|
"preserved_signatures": 1,
|
|
"symbol_scores": {
|
|
"Processor": 1.0,
|
|
"main": 0.0,
|
|
"new": 0.333,
|
|
"process": 0.0
|
|
},
|
|
"syntax_valid": true
|
|
},
|
|
"recorded_at": "2026-06-19T00:15:16.345389+00:00",
|
|
"transform": "code_aware_compressor"
|
|
}
|