openAgent

1import type { AssistantMessage, Part, Provider, UserMessage } from "@opencode-ai/sdk/v2"
2import { Locale } from "./locale"
3import * as Model from "./model"
4 
5export type TranscriptOptions = {
6  thinking: boolean
7  toolDetails: boolean
8  assistantMetadata: boolean
9  providers?: Provider[]
10}
11 
12export type SessionInfo = {
13  id: string
14  title: string
15  time: {
16    created: number
17    updated: number
18  }
19}
20 
21export type MessageWithParts = {
22  info: UserMessage | AssistantMessage
23  parts: Part[]
24}
25 
26export function formatTranscript(
27  session: SessionInfo,
28  messages: MessageWithParts[],
29  options: TranscriptOptions,
30): string {
31  const providers = Model.index(options.providers)
32  let transcript = `# ${session.title}\n\n`
33  transcript += `**Session ID:** ${session.id}\n`
34  transcript += `**Created:** ${new Date(session.time.created).toLocaleString()}\n`
35  transcript += `**Updated:** ${new Date(session.time.updated).toLocaleString()}\n\n`
36  transcript += `---\n\n`
37 
38  for (const msg of messages.toSorted(
39    (a, b) => a.info.time.created - b.info.time.created || a.info.id.localeCompare(b.info.id),
40  )) {
41    transcript += formatMessage(msg.info, msg.parts, options, providers)
42    transcript += `---\n\n`
43  }
44 
45  return transcript
46}
47 
48export function formatMessage(
49  msg: UserMessage | AssistantMessage,
50  parts: Part[],
51  options: TranscriptOptions,
52  providers?: Provider[] | ReadonlyMap<string, Provider>,
53): string {
54  let result = ""
55 
56  if (msg.role === "user") {
57    result += `## User\n\n`
58  } else {
59    result += formatAssistantHeader(msg, options.assistantMetadata, providers ?? options.providers)
60  }
61 
62  for (const part of parts) {
63    result += formatPart(part, options)
64  }
65 
66  return result
67}
68 
69export function formatAssistantHeader(
70  msg: AssistantMessage,
71  includeMetadata: boolean,
72  providers?: Provider[] | ReadonlyMap<string, Provider>,
73): string {
74  if (!includeMetadata) {
75    return `## Assistant\n\n`
76  }
77 
78  const duration =
79    msg.time.completed && msg.time.created ? ((msg.time.completed - msg.time.created) / 1000).toFixed(1) + "s" : ""
80 
81  const modelName = Model.name(providers, msg.providerID, msg.modelID)
82 
83  return `## Assistant (${Locale.titlecase(msg.agent)} · ${modelName}${duration ? ` · ${duration}` : ""})\n\n`
84}
85 
86export function formatPart(part: Part, options: TranscriptOptions): string {
87  if (part.type === "text" && !part.synthetic) {
88    return `${part.text}\n\n`
89  }
90 
91  if (part.type === "reasoning") {
92    if (options.thinking) {
93      return `_Thinking:_\n\n${part.text}\n\n`
94    }
95    return ""
96  }
97 
98  if (part.type === "tool") {
99    let result = `**Tool: ${part.tool}**\n`
100    if (options.toolDetails && part.state.input) {
101      result += `\n**Input:**\n\`\`\`json\n${JSON.stringify(part.state.input, null, 2)}\n\`\`\`\n`
102    }
103    if (options.toolDetails && part.state.status === "completed" && part.state.output) {
104      result += `\n**Output:**\n\`\`\`\n${part.state.output}\n\`\`\`\n`
105    }
106    if (options.toolDetails && part.state.status === "error" && part.state.error) {
107      result += `\n**Error:**\n\`\`\`\n${part.state.error}\n\`\`\`\n`
108    }
109    result += `\n`
110    return result
111  }
112 
113  return ""
114}
115