ceos/opencode
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(session): finalize native text before request finish

Browse source
This commit is contained in:
Kit Langton 2026-05-11 20:11:16 -04:00
commit 4c75b463e7
1 changed files with 34 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

63
packages/opencode/src/session/processor.ts
View file

@ -365,6 +365,38 @@ export const layer: Layer.Layer<
}
})
const finishText = Effect.fn("SessionProcessor.finishText")(function* (
providerMetadata?: Extract<StreamEvent, { type: "text-end" }>["providerMetadata"],
) {
if (!ctx.currentText) return
// oxlint-disable-next-line no-self-assign -- reactivity trigger
ctx.currentText.text = ctx.currentText.text
ctx.currentText.text = (yield* plugin.trigger(
"experimental.text.complete",
{
sessionID: ctx.sessionID,
messageID: ctx.assistantMessage.id,
partID: ctx.currentText.id,
},
{ text: ctx.currentText.text },
)).text
if (!ctx.assistantMessage.summary) {
// TODO(v2): Temporary dual-write while migrating session messages to v2 events.
if (Flag.OPENCODE_EXPERIMENTAL_EVENT_SYSTEM) {
yield* sync.run(SessionEvent.Text.Ended.Sync, {
sessionID: ctx.sessionID,
text: ctx.currentText.text,
timestamp: DateTime.makeUnsafe(Date.now()),
})
}
}
const end = Date.now()
ctx.currentText.time = { start: ctx.currentText.time?.start ?? end, end }
if (providerMetadata) ctx.currentText.metadata = providerMetadata
yield* session.updatePart(ctx.currentText)
ctx.currentText = undefined
})
const handleEvent = Effect.fnUntraced(function* (value: StreamEvent) {
switch (value.type) {
case "request-start":
@ -696,38 +728,11 @@ export const layer: Layer.Layer<
return
case "text-end":
if (!ctx.currentText) return
// oxlint-disable-next-line no-self-assign -- reactivity trigger
ctx.currentText.text = ctx.currentText.text
ctx.currentText.text = (yield* plugin.trigger(
"experimental.text.complete",
{
sessionID: ctx.sessionID,
messageID: ctx.assistantMessage.id,
partID: ctx.currentText.id,
},
{ text: ctx.currentText.text },
)).text
if (!ctx.assistantMessage.summary) {
// TODO(v2): Temporary dual-write while migrating session messages to v2 events.
if (Flag.OPENCODE_EXPERIMENTAL_EVENT_SYSTEM) {
yield* sync.run(SessionEvent.Text.Ended.Sync, {
sessionID: ctx.sessionID,
text: ctx.currentText.text,
timestamp: DateTime.makeUnsafe(Date.now()),
})
}
}
{
const end = Date.now()
ctx.currentText.time = { start: ctx.currentText.time?.start ?? end, end }
}
if (value.providerMetadata) ctx.currentText.metadata = value.providerMetadata
yield* session.updatePart(ctx.currentText)
ctx.currentText = undefined
yield* finishText(value.providerMetadata)
return
case "request-finish":
yield* finishText()
if (!ctx.assistantMessage.finish) yield* finishStep(value)
return
}