batch-mode-batch-schema 0.2.4

Defines the schema and structures for batch processing, including batch choices, message roles, error handling, and token management, with support for JSON deserialization and validation.
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

// ---------------- [ File: batch-mode-batch-schema/src/batch_request_record.rs ]
crate::ix!();

#[derive(Builder,Getters,Clone,Debug,Serialize,Deserialize)]
#[builder(setter(into))]
#[getset(get="pub")]
pub struct BatchRequestRecord {
    id:        BatchRequestId,
    custom_id: CustomRequestId,
    prompt:    Option<String>,
    messages:  Option<Vec<String>>,
}

#[cfg(test)]
mod batch_request_record_tests {
    use super::*;
    use serde_json::json;

    #[traced_test]
    fn test_batch_request_record_deserialization_full() {
        info!("Starting test: test_batch_request_record_deserialization_full");

        let json_data = json!({
            "id": "batch_req_123",
            "custom_id": "custom_456",
            "prompt": "Test prompt",
            "messages": ["Message 1", "Message 2"]
        });

        let record: BatchRequestRecord = serde_json::from_value(json_data).unwrap();

        pretty_assert_eq!(record.id().as_str(), "batch_req_123");
        pretty_assert_eq!(record.custom_id().as_str(), "custom_456");
        pretty_assert_eq!(*record.prompt(), Some("Test prompt".to_string()));

        // FIX: do not unwrap() by value. We do as_ref().unwrap() to avoid moving
        pretty_assert_eq!(
            record.messages().as_ref().unwrap(),
            &["Message 1".to_string(), "Message 2".to_string()]
        );

        info!("Finished test: test_batch_request_record_deserialization_full");
    }

    #[test]
    fn test_batch_request_record_deserialization_partial() {
        let json_data = json!({
            "id": "batch_req_789",
            "custom_id": "custom_012"
            // 'prompt' and 'messages' are missing
        });

        let record: BatchRequestRecord = serde_json::from_value(json_data).unwrap();

        pretty_assert_eq!(record.id().as_str(), "batch_req_789");
        pretty_assert_eq!(record.custom_id().as_str(), "custom_012");
        assert!(record.prompt().is_none());
        assert!(record.messages().is_none());
    }

    #[test]
    fn test_batch_request_record_missing_required_fields() {
        let json_data = json!({
            "custom_id": "custom_345"
            // 'id' is missing
        });

        let result: Result<BatchRequestRecord, _> = serde_json::from_value(json_data);
        assert!(result.is_err(), "Deserialization should fail if required fields are missing");
    }
}