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
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

use super::*;
use crate::message::AssistantContent;
use crate::operation::Finish;
use crate::test_utils::{MockFrame, MockScript, MockStreamEvent};

fn whole() -> CompletionResponse {
    CompletionResponse::new(
        vec![
            AssistantContent::reasoning("think"),
            AssistantContent::text("answer"),
        ],
        Default::default(),
        crate::message::Origin::new("test.api", "mock", ""),
        serde_json::Value::Null,
    )
}

#[test]
fn a_whole_reply_and_its_stream_fold_into_one_turn() {
    assert_restated_agrees(
        &MockScript::default(),
        [MockFrame::Response(Box::new(whole()))],
        [
            MockFrame::Event(MockStreamEvent::reasoning_delta("think")),
            MockFrame::Event(MockStreamEvent::text("answer")),
            MockFrame::Event(MockStreamEvent::FinalResponse(Finish::default())),
        ],
    );
}

#[test]
#[should_panic(expected = "the same turn")]
fn a_stream_that_drops_a_block_disagrees() {
    assert_restated_agrees(
        &MockScript::default(),
        [MockFrame::Response(Box::new(whole()))],
        [
            MockFrame::Event(MockStreamEvent::text("answer")),
            MockFrame::Event(MockStreamEvent::FinalResponse(Finish::default())),
        ],
    );
}

#[test]
fn rig_issued_call_ids_are_compared_by_position() {
    let call = crate::message::ToolCall::new(
        crate::message::CallId::from_wire(""),
        crate::message::ToolFunction::new(
            crate::message::ToolName::new("lookup").expect("a name"),
            serde_json::json!({}),
        ),
    );
    let whole = CompletionResponse::new(
        vec![AssistantContent::ToolCall(call)],
        Default::default(),
        crate::message::Origin::new("test.api", "mock", ""),
        serde_json::Value::Null,
    );
    assert_restated_agrees(
        &MockScript::default(),
        [MockFrame::Response(Box::new(whole))],
        [
            MockFrame::Event(MockStreamEvent::ToolCall {
                id: String::new(),
                name: "lookup".into(),
                arguments: serde_json::json!({}),
                call_id: None,
            }),
            MockFrame::Event(MockStreamEvent::FinalResponse(Finish::default())),
        ],
    );
}