mdstream 0.2.0

Streaming-first Markdown middleware for LLM output (committed + pending blocks, render-agnostic).
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

mod support;

use mdstream::{BlockKind, Options};

#[test]
fn streamdown_benchmark_simple_html_block_chunking_invariance() {
    // From Streamdown's parse-blocks benchmark ("HTML Blocks").
    let markdown = include_str!("fixtures/streamdown_bench/html_simple.md");

    let opts = Options::default();
    let blocks_whole = support::collect_final_blocks(support::chunk_whole(markdown), opts.clone());
    let blocks_lines = support::collect_final_blocks(support::chunk_lines(markdown), opts.clone());
    let blocks_rand = support::collect_final_blocks(
        support::chunk_pseudo_random(
            markdown,
            "streamdown_benchmark_simple_html_block_chunking_invariance",
            0,
            40,
        ),
        opts,
    );

    assert_eq!(blocks_lines, blocks_whole);
    assert_eq!(blocks_rand, blocks_whole);

    assert_eq!(blocks_whole.len(), 1);
    assert_eq!(blocks_whole[0].0, BlockKind::HtmlBlock);
    assert!(
        blocks_whole[0]
            .1
            .contains("<div>\n  <p>HTML content</p>\n</div>\n")
    );
}

#[test]
fn streamdown_benchmark_nested_html_block_chunking_invariance() {
    // From Streamdown's parse-blocks benchmark ("HTML Blocks").
    let markdown = include_str!("fixtures/streamdown_bench/html_nested.md");

    let opts = Options::default();
    let blocks_whole = support::collect_final_blocks(support::chunk_whole(markdown), opts.clone());
    let blocks_lines = support::collect_final_blocks(support::chunk_lines(markdown), opts.clone());
    let blocks_rand = support::collect_final_blocks(
        support::chunk_pseudo_random(
            markdown,
            "streamdown_benchmark_nested_html_block_chunking_invariance",
            0,
            40,
        ),
        opts,
    );

    assert_eq!(blocks_lines, blocks_whole);
    assert_eq!(blocks_rand, blocks_whole);

    assert_eq!(blocks_whole.len(), 1);
    assert_eq!(blocks_whole[0].0, BlockKind::HtmlBlock);
    assert!(blocks_whole[0].1.contains("<p>Nested content</p>"));
    assert!(blocks_whole[0].1.trim_end().ends_with("</div>"));
}

#[test]
fn streamdown_benchmark_multiple_html_blocks_chunking_invariance() {
    // From Streamdown's parse-blocks benchmark ("HTML Blocks").
    let markdown = include_str!("fixtures/streamdown_bench/html_multiple_blocks.md");

    let opts = Options::default();
    let blocks_whole = support::collect_final_blocks(support::chunk_whole(markdown), opts.clone());
    let blocks_lines = support::collect_final_blocks(support::chunk_lines(markdown), opts.clone());
    let blocks_rand = support::collect_final_blocks(
        support::chunk_pseudo_random(
            markdown,
            "streamdown_benchmark_multiple_html_blocks_chunking_invariance",
            0,
            40,
        ),
        opts,
    );

    assert_eq!(blocks_lines, blocks_whole);
    assert_eq!(blocks_rand, blocks_whole);

    let kinds: Vec<BlockKind> = blocks_whole.iter().map(|(k, _)| *k).collect();
    assert_eq!(
        kinds,
        vec![
            BlockKind::HtmlBlock,
            BlockKind::Paragraph,
            BlockKind::HtmlBlock,
            BlockKind::Paragraph,
        ]
    );

    assert!(blocks_whole[0].1.contains("<div>First block</div>"));
    assert!(blocks_whole[1].1.contains("Some markdown"));
    assert!(blocks_whole[2].1.contains("<section>"));
    assert!(blocks_whole[2].1.contains("</section>"));
    assert!(blocks_whole[3].1.contains("More markdown"));
}