Compare commits
2 Commits
dev
...
kit/repro-
| Author | SHA1 | Date |
|---|---|---|
|
|
ce8e9b1d84 | |
|
|
2d6d788db5 |
|
|
@ -48,8 +48,8 @@ export namespace SessionProcessor {
|
||||||
needsCompaction = false
|
needsCompaction = false
|
||||||
const shouldBreak = (await Config.get()).experimental?.continue_loop_on_deny !== true
|
const shouldBreak = (await Config.get()).experimental?.continue_loop_on_deny !== true
|
||||||
while (true) {
|
while (true) {
|
||||||
|
let currentText: MessageV2.TextPart | undefined
|
||||||
try {
|
try {
|
||||||
let currentText: MessageV2.TextPart | undefined
|
|
||||||
let reasoningMap: Record<string, MessageV2.ReasoningPart> = {}
|
let reasoningMap: Record<string, MessageV2.ReasoningPart> = {}
|
||||||
const stream = await LLM.stream(streamInput)
|
const stream = await LLM.stream(streamInput)
|
||||||
|
|
||||||
|
|
@ -385,6 +385,15 @@ export namespace SessionProcessor {
|
||||||
SessionStatus.set(input.sessionID, { type: "idle" })
|
SessionStatus.set(input.sessionID, { type: "idle" })
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if (currentText) {
|
||||||
|
currentText.text = currentText.text.trimEnd()
|
||||||
|
currentText.time = {
|
||||||
|
start: currentText.time?.start ?? Date.now(),
|
||||||
|
end: Date.now(),
|
||||||
|
}
|
||||||
|
await Session.updatePart(currentText)
|
||||||
|
currentText = undefined
|
||||||
|
}
|
||||||
if (snapshot) {
|
if (snapshot) {
|
||||||
const patch = await Snapshot.patch(snapshot)
|
const patch = await Snapshot.patch(snapshot)
|
||||||
if (patch.files.length) {
|
if (patch.files.length) {
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,160 @@
|
||||||
|
import { describe, expect, test, spyOn } from "bun:test"
|
||||||
|
import { Bus } from "../../src/bus"
|
||||||
|
import { Instance } from "../../src/project/instance"
|
||||||
|
import type { Provider } from "../../src/provider/provider"
|
||||||
|
import { ModelID, ProviderID } from "../../src/provider/schema"
|
||||||
|
import { Session } from "../../src/session"
|
||||||
|
import { LLM } from "../../src/session/llm"
|
||||||
|
import { MessageV2 } from "../../src/session/message-v2"
|
||||||
|
import { SessionProcessor } from "../../src/session/processor"
|
||||||
|
import { MessageID } from "../../src/session/schema"
|
||||||
|
import { Log } from "../../src/util/log"
|
||||||
|
import { tmpdir } from "../fixture/fixture"
|
||||||
|
|
||||||
|
Log.init({ print: false })
|
||||||
|
|
||||||
|
const model: Provider.Model = {
|
||||||
|
id: ModelID.make("gpt-5.4"),
|
||||||
|
providerID: ProviderID.make("openai"),
|
||||||
|
api: {
|
||||||
|
id: "openai",
|
||||||
|
url: "https://api.openai.com/v1",
|
||||||
|
npm: "@ai-sdk/openai",
|
||||||
|
},
|
||||||
|
name: "GPT-5.4",
|
||||||
|
capabilities: {
|
||||||
|
temperature: true,
|
||||||
|
reasoning: true,
|
||||||
|
attachment: false,
|
||||||
|
toolcall: false,
|
||||||
|
input: {
|
||||||
|
text: true,
|
||||||
|
audio: false,
|
||||||
|
image: false,
|
||||||
|
video: false,
|
||||||
|
pdf: false,
|
||||||
|
},
|
||||||
|
output: {
|
||||||
|
text: true,
|
||||||
|
audio: false,
|
||||||
|
image: false,
|
||||||
|
video: false,
|
||||||
|
pdf: false,
|
||||||
|
},
|
||||||
|
interleaved: false,
|
||||||
|
},
|
||||||
|
cost: {
|
||||||
|
input: 0,
|
||||||
|
output: 0,
|
||||||
|
cache: { read: 0, write: 0 },
|
||||||
|
},
|
||||||
|
limit: {
|
||||||
|
context: 128_000,
|
||||||
|
output: 8_000,
|
||||||
|
},
|
||||||
|
status: "active",
|
||||||
|
options: {},
|
||||||
|
headers: {},
|
||||||
|
release_date: "2026-01-01",
|
||||||
|
}
|
||||||
|
|
||||||
|
describe("interrupted streamed text reproducers", () => {
|
||||||
|
test("persists streamed assistant text on abort so reconnect sees the same partial reply", async () => {
|
||||||
|
await using tmp = await tmpdir({ git: true })
|
||||||
|
|
||||||
|
await Instance.provide({
|
||||||
|
directory: tmp.path,
|
||||||
|
fn: async () => {
|
||||||
|
const session = await Session.create({ title: "manual" })
|
||||||
|
const user = await Session.updateMessage({
|
||||||
|
id: MessageID.ascending(),
|
||||||
|
role: "user",
|
||||||
|
sessionID: session.id,
|
||||||
|
agent: "build",
|
||||||
|
model: {
|
||||||
|
providerID: model.providerID,
|
||||||
|
modelID: model.id,
|
||||||
|
},
|
||||||
|
time: {
|
||||||
|
created: Date.now(),
|
||||||
|
},
|
||||||
|
})
|
||||||
|
const assistant: MessageV2.Assistant = {
|
||||||
|
id: MessageID.ascending(),
|
||||||
|
role: "assistant",
|
||||||
|
sessionID: session.id,
|
||||||
|
parentID: user.id,
|
||||||
|
providerID: model.providerID,
|
||||||
|
modelID: model.id,
|
||||||
|
mode: "build",
|
||||||
|
agent: "build",
|
||||||
|
path: {
|
||||||
|
cwd: tmp.path,
|
||||||
|
root: tmp.path,
|
||||||
|
},
|
||||||
|
cost: 0,
|
||||||
|
tokens: {
|
||||||
|
input: 0,
|
||||||
|
output: 0,
|
||||||
|
reasoning: 0,
|
||||||
|
cache: { read: 0, write: 0 },
|
||||||
|
},
|
||||||
|
time: {
|
||||||
|
created: Date.now(),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
await Session.updateMessage(assistant)
|
||||||
|
|
||||||
|
const seen = Promise.withResolvers<void>()
|
||||||
|
let live = ""
|
||||||
|
const off = Bus.subscribe(MessageV2.Event.PartDelta, (evt) => {
|
||||||
|
if (evt.properties.messageID !== assistant.id) return
|
||||||
|
if (evt.properties.field !== "text") return
|
||||||
|
live += evt.properties.delta
|
||||||
|
seen.resolve()
|
||||||
|
})
|
||||||
|
|
||||||
|
const mock = spyOn(LLM, "stream").mockImplementation(async (input) => {
|
||||||
|
return {
|
||||||
|
fullStream: (async function* () {
|
||||||
|
yield { type: "start" }
|
||||||
|
yield { type: "text-start" }
|
||||||
|
yield { type: "text-delta", text: "Here is a long one:" }
|
||||||
|
await new Promise((resolve) => input.abort.addEventListener("abort", resolve, { once: true }))
|
||||||
|
throw new DOMException("Aborted", "AbortError")
|
||||||
|
})(),
|
||||||
|
} as never
|
||||||
|
})
|
||||||
|
|
||||||
|
const abort = new AbortController()
|
||||||
|
const proc = SessionProcessor.create({
|
||||||
|
assistantMessage: assistant,
|
||||||
|
sessionID: session.id,
|
||||||
|
model,
|
||||||
|
abort: abort.signal,
|
||||||
|
})
|
||||||
|
const run = proc.process({} as never)
|
||||||
|
|
||||||
|
await seen.promise
|
||||||
|
abort.abort()
|
||||||
|
|
||||||
|
expect(await run).toBe("stop")
|
||||||
|
|
||||||
|
const msg = await MessageV2.get({
|
||||||
|
sessionID: session.id,
|
||||||
|
messageID: assistant.id,
|
||||||
|
})
|
||||||
|
const text = msg.parts.find((part) => part.type === "text")
|
||||||
|
|
||||||
|
expect(live).toBe("Here is a long one:")
|
||||||
|
expect(text?.type).toBe("text")
|
||||||
|
expect(text && text.type === "text" ? text.text : undefined).toBe(live)
|
||||||
|
expect(msg.info.role === "assistant" ? msg.info.time.completed : undefined).toBeDefined()
|
||||||
|
|
||||||
|
off()
|
||||||
|
mock.mockRestore()
|
||||||
|
await Session.remove(session.id)
|
||||||
|
},
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
Loading…
Reference in New Issue