1. 1//! Markdown transcript renderer (docs/design/29-personal-os.md P4). One
2. 2//! shared renderer for the TUI export and the server-side export so both
3. 3//! surfaces emit byte-identical markdown for the same session. Pure:
4. 4//! messages in, string out.
5. 5
6. 6use vak_llm::{ContentBlock, Message, Role};
7. 7
8. 8/// Render a projected message list exactly as `export_transcript` has
9. 9/// always written it: `# Vakyartha transcript`, then one `## {idx} · {role}`
10. 10/// section per message with block-level lines between.
11. 11pub fn render_markdown(msgs: &[Message]) -> String {
12. 12 let mut out = String::from("# Vakyartha transcript\n\n");
13. 13 for (idx, m) in msgs.iter().enumerate() {
14. 14 let role = match m.role {
15. 15 Role::User => "user",
16. 16 _ => "assistant",
17. 17 };
18. 18 out.push_str(&format!("## {idx} · {role}\n\n"));
19. 19 for block in &m.content {
20. 20 match block {
21. 21 ContentBlock::Text { text } => {
22. 22 out.push_str(text.trim());
23. 23 out.push_str("\n\n");
24. 24 }
25. 25 ContentBlock::Thinking { text, .. } => {
26. 26 out.push_str(&format!("> thinking: {}\n\n", text.trim()));
27. 27 }
28. 28 ContentBlock::ToolUse { name, input, .. } => {
29. 29 out.push_str(&format!("- tool `{name}` `{input}`\n"));
30. 30 }
31. 31 ContentBlock::ToolResult {
32. 32 content, is_error, ..
33. 33 } => {
34. 34 let mark = if *is_error { "✗" } else { "→" };
35. 35 out.push_str(&format!(
36. 36 "- {mark} result: {}\n",
37. 37 content.replace('\n', " ")
38. 38 ));
39. 39 }
40. 40 ContentBlock::Image { source } => {
41. 41 out.push_str(&format!("- image ({})\n", source.media_type));
42. 42 }
43. 43 ContentBlock::Provider { kind, .. } => {
44. 44 out.push_str(&format!("- provider block `{kind}`\n"));
45. 45 }
46. 46 }
47. 47 }
48. 48 out.push('\n');
49. 49 }
50. 50 out
51. 51}
52. 52
53. 53#[cfg(test)]
54. 54mod tests {
55. 55 #![allow(clippy::unwrap_used, clippy::expect_used)]
56. 56 use super::*;
57. 57
58. 58 #[test]
59. 59 fn output_equals_legacy_tui_export_byte_for_byte() {
60. 60 let msgs = vec![
61. 61 Message {
62. 62 role: Role::User,
63. 63 content: vec![
64. 64 ContentBlock::text(" fix the flaky test \nsecond line "),
65. 65 ContentBlock::image_base64("image/png", "aGVsbG8="),
66. 66 ],
67. 67 },
68. 68 Message {
69. 69 role: Role::Assistant,
70. 70 content: vec![
71. 71 ContentBlock::Thinking {
72. 72 text: " consider retries ".into(),
73. 73 signature: Some("sig".into()),
74. 74 },
75. 75 ContentBlock::ToolUse {
76. 76 id: "t1".into(),
77. 77 name: "bash".into(),
78. 78 input: serde_json::json!({"command": "cargo test"}),
79. 79 },
80. 80 ContentBlock::ToolResult {
81. 81 tool_use_id: "t1".into(),
82. 82 content: "ok\nwith newline".into(),
83. 83 is_error: false,
84. 84 },
85. 85 ContentBlock::tool_error("t2", "boom"),
86. 86 ContentBlock::text("done."),
87. 87 ],
88. 88 },
89. 89 ];
90. 90
91. 91 // Expected string transcribed from the TUI export path this module
92. 92 // replaces (note: `text.trim()` keeps interior line-trailing
93. 93 // spaces, and each message section ends with a blank line); any
94. 94 // divergence from it is a parity bug.
95. 95 let expected = "# Vakyartha transcript\n\
96. 96\n\
97. 97## 0 · user\n\
98. 98\n\
99. 99fix the flaky test \nsecond line\n\
100. 100\n\
101. 101- image (image/png)\n\
102. 102\n\
103. 103## 1 · assistant\n\
104. 104\n\
105. 105> thinking: consider retries\n\
106. 106\n\
107. 107- tool `bash` `{\"command\":\"cargo test\"}`\n\
108. 108- → result: ok with newline\n\
109. 109- ✗ result: boom\n\
110. 110done.\n\
111. 111\n\
112. 112\n";
113. 113 assert_eq!(render_markdown(&msgs), expected);
114. 114 }
115. 115
116. 116 #[test]
117. 117 fn empty_transcript_renders_header_only() {
118. 118 assert_eq!(render_markdown(&[]), "# Vakyartha transcript\n\n");
119. 119 }
120. 120}
121. 121