fuzzy-parser 0.3.0

Fuzzy JSON repair for LLM-generated DSL
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

//! JSON Schema import demo: derive the repair schema from a JSON Schema
//! document instead of hand-building it.
//!
//! The document below is the shape `schemars` emits for an internally
//! tagged enum (`#[serde(tag = "type")]`), but any JSON Schema source
//! (Pydantic, OpenAPI components, hand-written files) works the same way.
//!
//! Run with: `cargo run --example schema_import`

use fuzzy_parser::{repair_tagged_enum_json, FuzzyOptions, TaggedEnumSchema};
use serde_json::json;

fn main() {
    let json_schema = json!({
        "$schema": "https://json-schema.org/draft/2020-12/schema",
        "title": "Intent",
        "oneOf": [
            {
                "type": "object",
                "properties": {
                    "type": {"type": "string", "const": "AddDerive"},
                    "target": {"type": "string"},
                    "derives": {
                        "type": "array",
                        "items": {"type": "string", "enum": ["Debug", "Clone", "Serialize"]}
                    },
                    "count": {"type": "integer"}
                },
                "required": ["type", "target"]
            },
            {
                "type": "object",
                "properties": {
                    "type": {"type": "string", "const": "Rename"},
                    "from": {"type": "string"},
                    "to": {"type": "string"}
                },
                "required": ["type", "from", "to"]
            }
        ]
    });

    let import = TaggedEnumSchema::from_json_schema(&json_schema).expect("import failed");
    for w in &import.warnings {
        println!("import warning at {}: {}", w.path, w.detail);
    }

    // Tag typo, field typo, enum-array typo, and a stringified integer —
    // all repaired using only the imported schema.
    let llm_json = r#"{"type": "AddDeriv", "taget": "User", "derives": ["Debg"], "count": "3"}"#;
    let result = repair_tagged_enum_json(llm_json, &import.schema, &FuzzyOptions::default())
        .expect("repair failed");

    println!("repaired: {}", result.repaired);
    println!("corrections:");
    for c in &result.corrections {
        println!(
            "  {} -> {} (similarity {:.2}, at {})",
            c.original, c.corrected, c.similarity, c.field_path
        );
    }
}