1. 1#![allow(clippy::unwrap_used, clippy::expect_used)]
2. 2
3. 3use vak_llm::openai::{OpenAiConfig, build_body};
4. 4use vak_llm::types::ChatRequest;
5. 5use vak_llm::types::{ContentBlock, ImageSource, Message, Role};
6. 6
7. 7#[test]
8. 8fn image_block_serializes_to_anthropic_native_shape() {
9. 9 let msg = Message {
10. 10 role: Role::User,
11. 11 content: vec![
12. 12 ContentBlock::text("what is this?"),
13. 13 ContentBlock::image_base64("image/png", "QUJD"),
14. 14 ],
15. 15 };
16. 16 let v = serde_json::to_value(&msg).unwrap();
17. 17 assert_eq!(v["content"][0]["type"], "text");
18. 18 assert_eq!(v["content"][1]["type"], "image");
19. 19 // Anthropic expects the source nested object verbatim.
20. 20 assert_eq!(v["content"][1]["source"]["type"], "base64");
21. 21 assert_eq!(v["content"][1]["source"]["media_type"], "image/png");
22. 22 assert_eq!(v["content"][1]["source"]["data"], "QUJD");
23. 23
24. 24 // Roundtrip through the ledger format.
25. 25 let back: Message = serde_json::from_value(v).unwrap();
26. 26 assert_eq!(back, msg);
27. 27}
28. 28
29. 29#[test]
30. 30fn openai_body_builds_multimodal_parts_from_images() {
31. 31 let req = ChatRequest {
32. 32 messages: vec![Message {
33. 33 role: Role::User,
34. 34 content: vec![
35. 35 ContentBlock::text("describe"),
36. 36 ContentBlock::image_base64("image/jpeg", "Zm9v"),
37. 37 ],
38. 38 }],
39. 39 ..ChatRequest::new("test-model")
40. 40 };
41. 41 let body = build_body(&OpenAiConfig::default(), &req).unwrap();
42. 42 let content = body["messages"][0]["content"].as_array().unwrap();
43. 43 assert_eq!(content[0]["type"], "text");
44. 44 assert_eq!(content[1]["type"], "image_url");
45. 45 assert_eq!(
46. 46 content[1]["image_url"]["url"],
47. 47 "data:image/jpeg;base64,Zm9v"
48. 48 );
49. 49}
50. 50
51. 51#[test]
52. 52fn openai_text_only_stays_a_plain_string() {
53. 53 let req = ChatRequest {
54. 54 messages: vec![Message::user_text("plain")],
55. 55 ..ChatRequest::new("test-model")
56. 56 };
57. 57 let body = build_body(&OpenAiConfig::default(), &req).unwrap();
58. 58 assert_eq!(body["messages"][0]["content"], "plain");
59. 59}
60. 60
61. 61#[test]
62. 62fn image_in_assistant_message_rejected_for_anthropic() {
63. 63 let req = ChatRequest {
64. 64 messages: vec![Message {
65. 65 role: Role::Assistant,
66. 66 content: vec![ContentBlock::Image {
67. 67 source: ImageSource {
68. 68 r#type: "base64".into(),
69. 69 media_type: "image/png".into(),
70. 70 data: "x".into(),
71. 71 },
72. 72 }],
73. 73 }],
74. 74 ..ChatRequest::new("test-model")
75. 75 };
76. 76 assert!(vak_llm::anthropic::build_body(&req, true, false).is_err());
77. 77}
78. 78