Dax Raad
120 lines
3.6 KiB
TypeScript
120 lines
3.6 KiB
TypeScript
import { Provider } from "@/provider/provider"
|
|
import { fn } from "@/util/fn"
|
|
import z from "zod"
|
|
import { Session } from "."
|
|
import { generateText } from "ai"
|
|
import { MessageV2 } from "./message-v2"
|
|
import { Flag } from "@/flag/flag"
|
|
import { Identifier } from "@/id/id"
|
|
import { Snapshot } from "@/snapshot"
|
|
import type { UserMessage } from "@opencode-ai/sdk"
|
|
|
|
export namespace SessionSummary {
|
|
export const summarize = fn(
|
|
z.object({
|
|
sessionID: z.string(),
|
|
messageID: z.string(),
|
|
providerID: z.string(),
|
|
}),
|
|
async (input) => {
|
|
const all = await Session.messages(input.sessionID)
|
|
await Promise.all([
|
|
summarizeSession({ sessionID: input.sessionID, messages: all }),
|
|
summarizeMessage({ messageID: input.messageID, messages: all }),
|
|
])
|
|
},
|
|
)
|
|
|
|
async function summarizeSession(input: { sessionID: string; messages: MessageV2.WithParts[] }) {
|
|
const diffs = await computeDiff({ messages: input.messages })
|
|
await Session.update(input.sessionID, (draft) => {
|
|
draft.summary = {
|
|
diffs,
|
|
}
|
|
})
|
|
}
|
|
|
|
async function summarizeMessage(input: { messageID: string; messages: MessageV2.WithParts[] }) {
|
|
const messages = input.messages.filter(
|
|
(m) => m.info.id === input.messageID || (m.info.role === "assistant" && m.info.parentID === input.messageID),
|
|
)
|
|
const userMsg = messages.find((m) => m.info.id === input.messageID)!.info as UserMessage
|
|
const diffs = await computeDiff({ messages })
|
|
userMsg.summary = {
|
|
diffs,
|
|
text: userMsg.summary?.text ?? "",
|
|
}
|
|
if (
|
|
Flag.OPENCODE_EXPERIMENTAL_TURN_SUMMARY &&
|
|
messages.every((m) => m.info.role !== "assistant" || m.info.time.completed)
|
|
) {
|
|
const assistantMsg = messages.find((m) => m.info.role === "assistant")!.info as MessageV2.Assistant
|
|
const small = await Provider.getSmallModel(assistantMsg.providerID)
|
|
if (!small) return
|
|
const result = await generateText({
|
|
model: small.language,
|
|
maxOutputTokens: 100,
|
|
messages: [
|
|
{
|
|
role: "user",
|
|
content: `
|
|
Summarize the following conversation into 2 sentences MAX explaining what the assistant did and why. Do not explain the user's input.
|
|
<conversation>
|
|
${JSON.stringify(MessageV2.toModelMessage(messages))}
|
|
</conversation>
|
|
`,
|
|
},
|
|
],
|
|
})
|
|
userMsg.summary.text = result.text
|
|
}
|
|
await Session.updateMessage(userMsg)
|
|
}
|
|
|
|
export const diff = fn(
|
|
z.object({
|
|
sessionID: Identifier.schema("session"),
|
|
messageID: Identifier.schema("message").optional(),
|
|
}),
|
|
async (input) => {
|
|
let all = await Session.messages(input.sessionID)
|
|
if (input.messageID)
|
|
all = all.filter(
|
|
(x) => x.info.id === input.messageID || (x.info.role === "assistant" && x.info.parentID === input.messageID),
|
|
)
|
|
|
|
return computeDiff({
|
|
messages: all,
|
|
})
|
|
},
|
|
)
|
|
|
|
async function computeDiff(input: { messages: MessageV2.WithParts[] }) {
|
|
let from: string | undefined
|
|
let to: string | undefined
|
|
|
|
// scan assistant messages to find earliest from and latest to
|
|
// snapshot
|
|
for (const item of input.messages) {
|
|
if (!from) {
|
|
for (const part of item.parts) {
|
|
if (part.type === "step-start" && part.snapshot) {
|
|
from = part.snapshot
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
for (const part of item.parts) {
|
|
if (part.type === "step-finish" && part.snapshot) {
|
|
to = part.snapshot
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if (from && to) return Snapshot.diffFull(from, to)
|
|
return []
|
|
}
|
|
}
|