|
|
@@ -24,13 +24,7 @@ import { errorMessage } from "@/util/error"
|
|
|
import { isRecord } from "@/util/record"
|
|
|
import { EventV2Bridge } from "@/event-v2-bridge"
|
|
|
import { Database } from "@opencode-ai/core/database/database"
|
|
|
-import { SessionEvent } from "@opencode-ai/core/session/event"
|
|
|
-import { SessionMessage } from "@opencode-ai/core/session/message"
|
|
|
-import { ModelV2 } from "@opencode-ai/core/model"
|
|
|
-import { ProviderV2 } from "@opencode-ai/core/provider"
|
|
|
-import * as DateTime from "effect/DateTime"
|
|
|
-import { RuntimeFlags } from "@/effect/runtime-flags"
|
|
|
-import { ToolOutput, Usage, type LLMEvent } from "@opencode-ai/llm"
|
|
|
+import { Usage, type LLMEvent } from "@opencode-ai/llm"
|
|
|
|
|
|
const DOOM_LOOP_THRESHOLD = 3
|
|
|
export type Result = "compact" | "stop" | "continue"
|
|
|
@@ -64,13 +58,10 @@ export interface Interface {
|
|
|
}
|
|
|
|
|
|
type ToolCall = {
|
|
|
- assistantMessageID?: SessionMessage.ID
|
|
|
partID: SessionV1.ToolPart["id"]
|
|
|
messageID: SessionV1.ToolPart["messageID"]
|
|
|
sessionID: SessionV1.ToolPart["sessionID"]
|
|
|
done: Deferred.Deferred<void>
|
|
|
- inputEnded: boolean
|
|
|
- raw: string
|
|
|
}
|
|
|
|
|
|
interface ProcessorContext extends Input {
|
|
|
@@ -80,9 +71,7 @@ interface ProcessorContext extends Input {
|
|
|
blocked: boolean
|
|
|
needsCompaction: boolean
|
|
|
currentText: SessionV1.TextPart | undefined
|
|
|
- currentTextID: string | undefined
|
|
|
reasoningMap: Record<string, SessionV1.ReasoningPart>
|
|
|
- v2AssistantMessageID: SessionMessage.ID | undefined
|
|
|
}
|
|
|
|
|
|
type StreamEvent = LLMEvent
|
|
|
@@ -104,7 +93,6 @@ export const layer = Layer.effect(
|
|
|
const status = yield* SessionStatus.Service
|
|
|
const image = yield* Image.Service
|
|
|
const events = yield* EventV2Bridge.Service
|
|
|
- const flags = yield* RuntimeFlags.Service
|
|
|
const database = yield* Database.Service
|
|
|
|
|
|
const create = Effect.fn("SessionProcessor.create")(function* (input: Input) {
|
|
|
@@ -122,11 +110,8 @@ export const layer = Layer.effect(
|
|
|
blocked: false,
|
|
|
needsCompaction: false,
|
|
|
currentText: undefined,
|
|
|
- currentTextID: undefined,
|
|
|
reasoningMap: {},
|
|
|
- v2AssistantMessageID: undefined,
|
|
|
}
|
|
|
- const mirrorAssistant = flags.experimentalEventSystem && !input.assistantMessage.summary
|
|
|
let aborted = false
|
|
|
|
|
|
const parse = (e: unknown) =>
|
|
|
@@ -141,34 +126,6 @@ export const layer = Layer.effect(
|
|
|
if (done) yield* Deferred.succeed(done, undefined).pipe(Effect.ignore)
|
|
|
})
|
|
|
|
|
|
- const ensureV2AssistantMessage = Effect.fn("SessionProcessor.ensureV2AssistantMessage")(function* () {
|
|
|
- if (ctx.v2AssistantMessageID) return ctx.v2AssistantMessageID
|
|
|
- ctx.v2AssistantMessageID = SessionMessage.ID.create()
|
|
|
- yield* events.publish(SessionEvent.Step.Started, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: ctx.v2AssistantMessageID,
|
|
|
- agent: input.assistantMessage.agent,
|
|
|
- model: {
|
|
|
- id: ModelV2.ID.make(ctx.model.id),
|
|
|
- providerID: ProviderV2.ID.make(ctx.model.providerID),
|
|
|
- variant: ModelV2.VariantID.make(input.assistantMessage.variant ?? "default"),
|
|
|
- },
|
|
|
- snapshot: ctx.snapshot,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- return ctx.v2AssistantMessageID
|
|
|
- })
|
|
|
-
|
|
|
- const requireV2AssistantMessage = (toolCall?: ToolCall) =>
|
|
|
- toolCall?.assistantMessageID === undefined
|
|
|
- ? Effect.die("V2 tool settlement has no owning assistant message")
|
|
|
- : Effect.succeed(toolCall.assistantMessageID)
|
|
|
-
|
|
|
- const currentV2AssistantMessage = () =>
|
|
|
- ctx.v2AssistantMessageID === undefined
|
|
|
- ? Effect.die("V2 step settlement has no owning assistant message")
|
|
|
- : Effect.succeed(ctx.v2AssistantMessageID)
|
|
|
-
|
|
|
const readToolCall = Effect.fn("SessionProcessor.readToolCall")(function* (toolCallID: string) {
|
|
|
const call = ctx.toolcalls[toolCallID]
|
|
|
if (!call) return undefined
|
|
|
@@ -247,17 +204,6 @@ export const layer = Layer.effect(
|
|
|
|
|
|
const finishReasoning = Effect.fn("SessionProcessor.finishReasoning")(function* (reasoningID: string) {
|
|
|
if (!(reasoningID in ctx.reasoningMap)) return
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- yield* events.publish(SessionEvent.Reasoning.Ended, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: yield* currentV2AssistantMessage(),
|
|
|
- reasoningID,
|
|
|
- text: ctx.reasoningMap[reasoningID].text,
|
|
|
- providerMetadata: ctx.reasoningMap[reasoningID].metadata,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
// oxlint-disable-next-line no-self-assign -- reactivity trigger
|
|
|
ctx.reasoningMap[reasoningID].text = ctx.reasoningMap[reasoningID].text
|
|
|
ctx.reasoningMap[reasoningID].time = { ...ctx.reasoningMap[reasoningID].time, end: Date.now() }
|
|
|
@@ -265,33 +211,6 @@ export const layer = Layer.effect(
|
|
|
delete ctx.reasoningMap[reasoningID]
|
|
|
})
|
|
|
|
|
|
- const flushV2Fragments = Effect.fn("SessionProcessor.flushV2Fragments")(function* () {
|
|
|
- if (!mirrorAssistant) return
|
|
|
- if (!ctx.assistantMessage.summary && ctx.currentText && ctx.currentTextID) {
|
|
|
- yield* events.publish(SessionEvent.Text.Ended, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: yield* currentV2AssistantMessage(),
|
|
|
- textID: ctx.currentTextID,
|
|
|
- text: ctx.currentText.text,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
- yield* Effect.forEach(Object.entries(ctx.reasoningMap), ([reasoningID, part]) =>
|
|
|
- currentV2AssistantMessage().pipe(
|
|
|
- Effect.flatMap((assistantMessageID) =>
|
|
|
- events.publish(SessionEvent.Reasoning.Ended, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- reasoningID,
|
|
|
- text: part.text,
|
|
|
- providerMetadata: part.metadata,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- }),
|
|
|
- ),
|
|
|
- ),
|
|
|
- )
|
|
|
- })
|
|
|
-
|
|
|
const ensureToolCall = Effect.fn("SessionProcessor.ensureToolCall")(function* (input: {
|
|
|
id: string
|
|
|
name: string
|
|
|
@@ -312,17 +231,6 @@ export const layer = Layer.effect(
|
|
|
}
|
|
|
return { call: ctx.toolcalls[input.id], part }
|
|
|
}
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- const assistantMessageID = mirrorAssistant ? yield* ensureV2AssistantMessage() : undefined
|
|
|
- if (assistantMessageID) {
|
|
|
- yield* events.publish(SessionEvent.Tool.Input.Started, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- callID: input.id,
|
|
|
- name: input.name,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
const part = yield* session.updatePart({
|
|
|
id: PartID.ascending(),
|
|
|
messageID: ctx.assistantMessage.id,
|
|
|
@@ -334,13 +242,10 @@ export const layer = Layer.effect(
|
|
|
metadata: input.providerExecuted ? { providerExecuted: true } : undefined,
|
|
|
} satisfies SessionV1.ToolPart)
|
|
|
ctx.toolcalls[input.id] = {
|
|
|
- assistantMessageID,
|
|
|
done: yield* Deferred.make<void>(),
|
|
|
partID: part.id,
|
|
|
messageID: part.messageID,
|
|
|
sessionID: part.sessionID,
|
|
|
- inputEnded: false,
|
|
|
- raw: "",
|
|
|
}
|
|
|
return { call: ctx.toolcalls[input.id], part }
|
|
|
})
|
|
|
@@ -372,16 +277,6 @@ export const layer = Layer.effect(
|
|
|
switch (value.type) {
|
|
|
case "reasoning-start":
|
|
|
if (value.id in ctx.reasoningMap) return
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- yield* events.publish(SessionEvent.Reasoning.Started, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: yield* ensureV2AssistantMessage(),
|
|
|
- reasoningID: value.id,
|
|
|
- providerMetadata: value.providerMetadata,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
ctx.reasoningMap[value.id] = {
|
|
|
id: PartID.ascending(),
|
|
|
messageID: ctx.assistantMessage.id,
|
|
|
@@ -399,15 +294,6 @@ export const layer = Layer.effect(
|
|
|
if (!(value.id in ctx.reasoningMap)) return
|
|
|
ctx.reasoningMap[value.id].text += value.text
|
|
|
if (value.providerMetadata) ctx.reasoningMap[value.id].metadata = value.providerMetadata
|
|
|
- if (mirrorAssistant) {
|
|
|
- yield* events.publish(SessionEvent.Reasoning.Delta, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: yield* currentV2AssistantMessage(),
|
|
|
- reasoningID: value.id,
|
|
|
- delta: value.text,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
yield* session.updatePartDelta({
|
|
|
sessionID: ctx.reasoningMap[value.id].sessionID,
|
|
|
messageID: ctx.reasoningMap[value.id].messageID,
|
|
|
@@ -432,36 +318,11 @@ export const layer = Layer.effect(
|
|
|
return
|
|
|
|
|
|
case "tool-input-delta":
|
|
|
- {
|
|
|
- const toolCall = yield* ensureToolCall(value)
|
|
|
- const assistantMessageID = mirrorAssistant ? yield* requireV2AssistantMessage(toolCall.call) : undefined
|
|
|
- if (assistantMessageID) {
|
|
|
- yield* events.publish(SessionEvent.Tool.Input.Delta, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- callID: value.id,
|
|
|
- delta: value.text,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
- ctx.toolcalls[value.id] = { ...toolCall.call, raw: toolCall.call.raw + value.text }
|
|
|
- }
|
|
|
+ yield* ensureToolCall(value)
|
|
|
return
|
|
|
|
|
|
case "tool-input-end": {
|
|
|
- const toolCall = yield* ensureToolCall(value)
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- const assistantMessageID = yield* requireV2AssistantMessage(toolCall.call)
|
|
|
- yield* events.publish(SessionEvent.Tool.Input.Ended, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- callID: value.id,
|
|
|
- text: toolCall.call.raw,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
- ctx.toolcalls[value.id] = { ...toolCall.call, inputEnded: true }
|
|
|
+ yield* ensureToolCall(value)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
@@ -469,37 +330,8 @@ export const layer = Layer.effect(
|
|
|
if (ctx.assistantMessage.summary) {
|
|
|
throw new Error(`Tool call not allowed while generating summary: ${value.name}`)
|
|
|
}
|
|
|
- const toolCall = yield* ensureToolCall(value)
|
|
|
+ yield* ensureToolCall(value)
|
|
|
const input = isRecord(value.input) ? value.input : { value: value.input }
|
|
|
- if (!toolCall.call.inputEnded) {
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- const assistantMessageID = yield* requireV2AssistantMessage(toolCall.call)
|
|
|
- yield* events.publish(SessionEvent.Tool.Input.Ended, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- callID: value.id,
|
|
|
- text: toolCall.call.raw,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
- }
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- const assistantMessageID = yield* requireV2AssistantMessage(toolCall.call)
|
|
|
- yield* events.publish(SessionEvent.Tool.Called, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- callID: value.id,
|
|
|
- tool: value.name,
|
|
|
- input,
|
|
|
- provider: {
|
|
|
- executed: toolCall.part.metadata?.providerExecuted === true,
|
|
|
- ...(value.providerMetadata ? { metadata: value.providerMetadata } : {}),
|
|
|
- },
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
yield* updateToolCall(value.id, (match) => ({
|
|
|
...match,
|
|
|
tool: value.name,
|
|
|
@@ -550,22 +382,6 @@ export const layer = Layer.effect(
|
|
|
const toolCall = yield* readToolCall(value.id)
|
|
|
if (!toolCall && value.result.type === "error") return
|
|
|
if (value.result.type === "error") {
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- const assistantMessageID = yield* requireV2AssistantMessage(toolCall?.call)
|
|
|
- yield* events.publish(SessionEvent.Tool.Failed, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- callID: value.id,
|
|
|
- error: { type: "unknown", message: errorMessage(value.result.value) },
|
|
|
- result: value.result,
|
|
|
- provider: {
|
|
|
- executed: value.providerExecuted === true || toolCall?.part.metadata?.providerExecuted === true,
|
|
|
- ...(value.providerMetadata ? { metadata: value.providerMetadata } : {}),
|
|
|
- },
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
yield* failToolCall(value.id, value.result.value)
|
|
|
return
|
|
|
}
|
|
|
@@ -591,81 +407,11 @@ export const layer = Layer.effect(
|
|
|
: `${rawOutput.output}\n\n[${omitted} image${omitted === 1 ? "" : "s"} omitted: could not be resized below the image size limit.]`,
|
|
|
attachments: attachments.length ? attachments : undefined,
|
|
|
}
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- const assistantMessageID = yield* requireV2AssistantMessage(toolCall?.call)
|
|
|
- const content = [
|
|
|
- { type: "text" as const, text: output.output },
|
|
|
- ...(output.attachments?.map(
|
|
|
- (item: SessionV1.FilePart) =>
|
|
|
- ({
|
|
|
- type: "file",
|
|
|
- uri: item.url,
|
|
|
- mime: item.mime,
|
|
|
- name: item.filename,
|
|
|
- }) as const,
|
|
|
- ) ?? []),
|
|
|
- ]
|
|
|
- const unsupported = content.find((item) => item.type === "file" && !item.uri.startsWith("data:"))
|
|
|
- if (unsupported?.type === "file") {
|
|
|
- const error = new Error(
|
|
|
- `Tool attachment URI "${unsupported.uri}" must be materialized before durable V2 settlement`,
|
|
|
- )
|
|
|
- yield* events.publish(SessionEvent.Tool.Failed, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- callID: value.id,
|
|
|
- error: {
|
|
|
- type: "unknown",
|
|
|
- message: error.message,
|
|
|
- },
|
|
|
- provider: {
|
|
|
- executed: value.providerExecuted === true || toolCall?.part.metadata?.providerExecuted === true,
|
|
|
- ...(value.providerMetadata ? { metadata: value.providerMetadata } : {}),
|
|
|
- },
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- yield* failToolCall(value.id, error)
|
|
|
- return
|
|
|
- } else
|
|
|
- yield* events.publish(SessionEvent.Tool.Success, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- callID: value.id,
|
|
|
- structured: output.metadata,
|
|
|
- content,
|
|
|
- result: value.result,
|
|
|
- provider: {
|
|
|
- executed: value.providerExecuted === true || toolCall?.part.metadata?.providerExecuted === true,
|
|
|
- ...(value.providerMetadata ? { metadata: value.providerMetadata } : {}),
|
|
|
- },
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
yield* completeToolCall(value.id, output)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
case "tool-error": {
|
|
|
- const toolCall = yield* readToolCall(value.id)
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- const assistantMessageID = yield* requireV2AssistantMessage(toolCall?.call)
|
|
|
- yield* events.publish(SessionEvent.Tool.Failed, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID,
|
|
|
- callID: value.id,
|
|
|
- error: {
|
|
|
- type: "unknown",
|
|
|
- message: value.message,
|
|
|
- },
|
|
|
- provider: {
|
|
|
- executed: toolCall?.part.metadata?.providerExecuted === true,
|
|
|
- ...(value.providerMetadata ? { metadata: value.providerMetadata } : {}),
|
|
|
- },
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
yield* failToolCall(value.id, value.error ?? new Error(value.message))
|
|
|
return
|
|
|
}
|
|
|
@@ -675,12 +421,6 @@ export const layer = Layer.effect(
|
|
|
|
|
|
case "step-start":
|
|
|
if (!ctx.snapshot) ctx.snapshot = yield* snapshot.track()
|
|
|
- if (!ctx.assistantMessage.summary) {
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- yield* ensureV2AssistantMessage()
|
|
|
- }
|
|
|
- }
|
|
|
yield* session.updatePart({
|
|
|
id: PartID.ascending(),
|
|
|
messageID: ctx.assistantMessage.id,
|
|
|
@@ -698,21 +438,6 @@ export const layer = Layer.effect(
|
|
|
usage: value.usage ?? new Usage({}),
|
|
|
metadata: value.providerMetadata,
|
|
|
})
|
|
|
- if (!ctx.assistantMessage.summary) {
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- yield* events.publish(SessionEvent.Step.Ended, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: yield* currentV2AssistantMessage(),
|
|
|
- finish: value.reason,
|
|
|
- cost: usage.cost,
|
|
|
- tokens: usage.tokens,
|
|
|
- snapshot: completedSnapshot,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- ctx.v2AssistantMessageID = undefined
|
|
|
- }
|
|
|
- }
|
|
|
ctx.assistantMessage.finish = value.reason
|
|
|
ctx.assistantMessage.cost += usage.cost
|
|
|
ctx.assistantMessage.tokens = usage.tokens
|
|
|
@@ -757,17 +482,6 @@ export const layer = Layer.effect(
|
|
|
}
|
|
|
|
|
|
case "text-start":
|
|
|
- if (!ctx.assistantMessage.summary) {
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- yield* events.publish(SessionEvent.Text.Started, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: yield* ensureV2AssistantMessage(),
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- textID: value.id,
|
|
|
- })
|
|
|
- }
|
|
|
- }
|
|
|
ctx.currentText = {
|
|
|
id: PartID.ascending(),
|
|
|
messageID: ctx.assistantMessage.id,
|
|
|
@@ -777,7 +491,6 @@ export const layer = Layer.effect(
|
|
|
time: { start: Date.now() },
|
|
|
metadata: value.providerMetadata,
|
|
|
}
|
|
|
- ctx.currentTextID = value.id
|
|
|
yield* session.updatePart(ctx.currentText)
|
|
|
return
|
|
|
|
|
|
@@ -785,15 +498,6 @@ export const layer = Layer.effect(
|
|
|
if (!ctx.currentText) return
|
|
|
ctx.currentText.text += value.text
|
|
|
if (value.providerMetadata) ctx.currentText.metadata = value.providerMetadata
|
|
|
- if (mirrorAssistant) {
|
|
|
- yield* events.publish(SessionEvent.Text.Delta, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: yield* currentV2AssistantMessage(),
|
|
|
- textID: value.id,
|
|
|
- delta: value.text,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
yield* session.updatePartDelta({
|
|
|
sessionID: ctx.currentText.sessionID,
|
|
|
messageID: ctx.currentText.messageID,
|
|
|
@@ -816,18 +520,6 @@ export const layer = Layer.effect(
|
|
|
},
|
|
|
{ text: ctx.currentText.text },
|
|
|
)).text
|
|
|
- if (!ctx.assistantMessage.summary) {
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- yield* events.publish(SessionEvent.Text.Ended, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: yield* currentV2AssistantMessage(),
|
|
|
- text: ctx.currentText.text,
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- textID: value.id,
|
|
|
- })
|
|
|
- }
|
|
|
- }
|
|
|
{
|
|
|
const end = Date.now()
|
|
|
ctx.currentText.time = { start: ctx.currentText.time?.start ?? end, end }
|
|
|
@@ -835,7 +527,6 @@ export const layer = Layer.effect(
|
|
|
if (value.providerMetadata) ctx.currentText.metadata = value.providerMetadata
|
|
|
yield* session.updatePart(ctx.currentText)
|
|
|
ctx.currentText = undefined
|
|
|
- ctx.currentTextID = undefined
|
|
|
return
|
|
|
|
|
|
case "finish":
|
|
|
@@ -864,7 +555,6 @@ export const layer = Layer.effect(
|
|
|
ctx.currentText.time = { start: ctx.currentText.time?.start ?? end, end }
|
|
|
yield* session.updatePart(ctx.currentText)
|
|
|
ctx.currentText = undefined
|
|
|
- ctx.currentTextID = undefined
|
|
|
}
|
|
|
|
|
|
for (const part of Object.values(ctx.reasoningMap)) {
|
|
|
@@ -886,16 +576,6 @@ export const layer = Layer.effect(
|
|
|
const match = yield* readToolCall(toolCallID)
|
|
|
if (!match) continue
|
|
|
const part = match.part
|
|
|
- if (mirrorAssistant && match.call.assistantMessageID) {
|
|
|
- yield* events.publish(SessionEvent.Tool.Failed, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: match.call.assistantMessageID,
|
|
|
- callID: toolCallID,
|
|
|
- error: { type: "unknown", message: "Tool execution aborted" },
|
|
|
- provider: { executed: part.metadata?.providerExecuted === true },
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
const end = Date.now()
|
|
|
const metadata = "metadata" in part.state && isRecord(part.state.metadata) ? part.state.metadata : {}
|
|
|
yield* session.updatePart({
|
|
|
@@ -922,7 +602,6 @@ export const layer = Layer.effect(
|
|
|
stack: e instanceof Error ? e.stack : undefined,
|
|
|
})
|
|
|
const error = parse(e)
|
|
|
- yield* flushV2Fragments()
|
|
|
if (SessionV1.ContextOverflowError.isInstance(error)) {
|
|
|
if ((yield* config.get()).compaction?.auto === false && !ctx.assistantMessage.summary) {
|
|
|
ctx.assistantMessage.error = error
|
|
|
@@ -935,20 +614,6 @@ export const layer = Layer.effect(
|
|
|
yield* events.publish(Session.Event.Error, { sessionID: ctx.sessionID, error })
|
|
|
return
|
|
|
}
|
|
|
- if (!ctx.assistantMessage.summary) {
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- if (mirrorAssistant) {
|
|
|
- yield* events.publish(SessionEvent.Step.Failed, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- assistantMessageID: yield* ensureV2AssistantMessage(),
|
|
|
- error: {
|
|
|
- type: "unknown",
|
|
|
- message: errorMessage(e),
|
|
|
- },
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- }
|
|
|
- }
|
|
|
ctx.assistantMessage.error = error
|
|
|
yield* events.publish(Session.Event.Error, {
|
|
|
sessionID: ctx.assistantMessage.sessionID,
|
|
|
@@ -968,7 +633,6 @@ export const layer = Layer.effect(
|
|
|
return yield* Effect.gen(function* () {
|
|
|
yield* Effect.gen(function* () {
|
|
|
ctx.currentText = undefined
|
|
|
- ctx.currentTextID = undefined
|
|
|
ctx.reasoningMap = {}
|
|
|
yield* status.set(ctx.sessionID, { type: "busy" })
|
|
|
const stream = llm.stream(streamInput)
|
|
|
@@ -996,30 +660,13 @@ export const layer = Layer.effect(
|
|
|
provider: input.model.providerID,
|
|
|
parse,
|
|
|
set: (info) => {
|
|
|
- // TODO(v2): Temporary dual-write while migrating session messages to v2 events.
|
|
|
- const event = mirrorAssistant
|
|
|
- ? events.publish(SessionEvent.Retried, {
|
|
|
- sessionID: ctx.sessionID,
|
|
|
- attempt: info.attempt,
|
|
|
- error: {
|
|
|
- message: info.message,
|
|
|
- isRetryable: true,
|
|
|
- },
|
|
|
- timestamp: DateTime.makeUnsafe(Date.now()),
|
|
|
- })
|
|
|
- : Effect.void
|
|
|
- return flushV2Fragments().pipe(
|
|
|
- Effect.andThen(event),
|
|
|
- Effect.andThen(
|
|
|
- status.set(ctx.sessionID, {
|
|
|
- type: "retry",
|
|
|
- attempt: info.attempt,
|
|
|
- message: info.message,
|
|
|
- action: info.action,
|
|
|
- next: info.next,
|
|
|
- }),
|
|
|
- ),
|
|
|
- )
|
|
|
+ return status.set(ctx.sessionID, {
|
|
|
+ type: "retry",
|
|
|
+ attempt: info.attempt,
|
|
|
+ message: info.message,
|
|
|
+ action: info.action,
|
|
|
+ next: info.next,
|
|
|
+ })
|
|
|
},
|
|
|
}),
|
|
|
),
|
|
|
@@ -1059,7 +706,6 @@ export const defaultLayer = Layer.suspend(() =>
|
|
|
Layer.provide(SessionStatus.defaultLayer),
|
|
|
Layer.provide(Image.defaultLayer),
|
|
|
Layer.provide(Config.defaultLayer),
|
|
|
- Layer.provide(RuntimeFlags.defaultLayer),
|
|
|
Layer.provide(Database.defaultLayer),
|
|
|
Layer.provide(EventV2Bridge.defaultLayer),
|
|
|
),
|
|
|
@@ -1080,7 +726,6 @@ export const node = LayerNode.make({
|
|
|
SessionStatus.node,
|
|
|
Image.node,
|
|
|
EventV2Bridge.node,
|
|
|
- RuntimeFlags.node,
|
|
|
Database.node,
|
|
|
],
|
|
|
})
|