File size: 2,360 Bytes
afa0cbf | 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 | use super::ContentItemKind;
use crate::models::ContentItem;
use crate::models::InternalChatMessageMetadataPassthrough;
use crate::models::ResponseItem;
use anyhow::Result;
use pretty_assertions::assert_eq;
use serde_json::Value;
use serde_json::json;
#[test]
fn content_kinds_round_trip_without_restricting_future_values() -> Result<()> {
let item = response_item(InternalChatMessageMetadataPassthrough {
turn_id: Some("turn-1".to_string()),
content_item_kinds: Some(vec![ContentItemKind("future_content_kind".to_string())]),
..Default::default()
});
let serialized = serde_json::to_value(&item)?;
assert_eq!(
serialized,
json!({
"type": "message",
"role": "user",
"content": [{"type": "input_text", "text": "hello"}],
"internal_chat_message_metadata_passthrough": {
"turn_id": "turn-1",
"content_item_kinds": ["future_content_kind"],
},
})
);
assert_eq!(serde_json::from_value::<ResponseItem>(serialized)?, item);
Ok(())
}
#[test]
fn malformed_content_kinds_do_not_prevent_loading_response_item() -> Result<()> {
for malformed_kinds in [
json!("user_text"),
json!({"kind": "user_text"}),
json!(["user_text", 123]),
] {
let item = serde_json::from_value::<ResponseItem>(response_item_json(json!({
"turn_id": "turn-1",
"content_item_kinds": malformed_kinds,
})))?;
assert_eq!(
item,
response_item(InternalChatMessageMetadataPassthrough {
turn_id: Some("turn-1".to_string()),
..Default::default()
})
);
}
Ok(())
}
fn response_item(metadata: InternalChatMessageMetadataPassthrough) -> ResponseItem {
ResponseItem::Message {
id: None,
role: "user".to_string(),
content: vec![ContentItem::InputText {
text: "hello".to_string(),
}],
phase: None,
internal_chat_message_metadata_passthrough: Some(metadata),
}
}
fn response_item_json(metadata: Value) -> Value {
json!({
"type": "message",
"role": "user",
"content": [{"type": "input_text", "text": "hello"}],
"internal_chat_message_metadata_passthrough": metadata,
})
}
|