rig-core 0.44.0

An opinionated library for building LLM powered applications.
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

use super::*;

#[test]
fn delimiters_bom_and_partial_frames_are_independent_of_byte_chunking() {
    for (bytes, pending) in [
        (&b"data: hi\n\n"[..], 0),
        (&b"data: hi\r\n\r\n"[..], 0),
        (&b"data: hi\r\r"[..], 0),
        (&b"data: hi\r\n\n"[..], 0),
        (&b"\xef\xbb\xbfdata: hi\n\n"[..], 0),
        (&b"\xef\xbb\xbf"[..], 0),
        (&b"\xef\xbb"[..], 2),
        (&b"data: hi\n\ndata: {"[..], 7),
        (&b"data: hi\r\n"[..], 10),
        (&b": heartbeat\n\ndata: hi\n\n"[..], 0),
        (&b"data: one\ndata: two\n"[..], 20),
    ] {
        for size in 1..=bytes.len() {
            let mut tail = SseTail::default();
            for chunk in bytes.chunks(size) {
                tail.feed(chunk);
            }
            assert_eq!(tail.pending(), pending, "{bytes:?}, chunks {size}");
        }
    }
}