3c1e71f256
Doctor checks: - model-resolution-cache.ts, model-resolution-config.ts - model-resolution-details.ts, model-resolution-effective-model.ts - model-resolution-types.ts, model-resolution-variant.ts Run events: - event-formatting.ts, event-handlers.ts - event-state.ts, event-stream-processor.ts
44 lines
1.2 KiB
TypeScript
44 lines
1.2 KiB
TypeScript
import pc from "picocolors"
|
|
import type { RunContext, EventPayload } from "./types"
|
|
import type { EventState } from "./event-state"
|
|
import { logEventVerbose } from "./event-formatting"
|
|
import {
|
|
handleSessionError,
|
|
handleSessionIdle,
|
|
handleSessionStatus,
|
|
handleMessagePartUpdated,
|
|
handleMessageUpdated,
|
|
handleToolExecute,
|
|
handleToolResult,
|
|
} from "./event-handlers"
|
|
|
|
export async function processEvents(
|
|
ctx: RunContext,
|
|
stream: AsyncIterable<unknown>,
|
|
state: EventState
|
|
): Promise<void> {
|
|
for await (const event of stream) {
|
|
if (ctx.abortController.signal.aborted) break
|
|
|
|
try {
|
|
const payload = event as EventPayload
|
|
if (!payload?.type) {
|
|
console.error(pc.dim(`[event] no type: ${JSON.stringify(event)}`))
|
|
continue
|
|
}
|
|
|
|
logEventVerbose(ctx, payload)
|
|
|
|
handleSessionError(ctx, payload, state)
|
|
handleSessionIdle(ctx, payload, state)
|
|
handleSessionStatus(ctx, payload, state)
|
|
handleMessagePartUpdated(ctx, payload, state)
|
|
handleMessageUpdated(ctx, payload, state)
|
|
handleToolExecute(ctx, payload, state)
|
|
handleToolResult(ctx, payload, state)
|
|
} catch (err) {
|
|
console.error(pc.red(`[event error] ${err}`))
|
|
}
|
|
}
|
|
}
|