YeonGyu-Kim
81 lines
2.3 KiB
TypeScript
81 lines
2.3 KiB
TypeScript
import type { HookDeps } from "./types"
|
|
import type { SessionMessage, SessionMessagePart } from "./session-messages"
|
|
import { extractSessionMessages } from "./session-messages"
|
|
import { extractAutoRetrySignal } from "./error-classifier"
|
|
|
|
function getLastUserMessageIndex(messages: SessionMessage[]): number {
|
|
for (let index = messages.length - 1; index >= 0; index--) {
|
|
if (messages[index]?.info?.role === "user") {
|
|
return index
|
|
}
|
|
}
|
|
|
|
return -1
|
|
}
|
|
|
|
function getAssistantText(parts: SessionMessagePart[] | undefined): string {
|
|
return (parts ?? [])
|
|
.flatMap((part) => {
|
|
if (part.type !== "text") {
|
|
return []
|
|
}
|
|
|
|
const text = typeof part.text === "string" ? part.text.trim() : ""
|
|
return text.length > 0 ? [text] : []
|
|
})
|
|
.join("\n")
|
|
}
|
|
|
|
export function hasVisibleAssistantResponse(extractAutoRetrySignalFn: typeof extractAutoRetrySignal) {
|
|
return async (
|
|
ctx: HookDeps["ctx"],
|
|
sessionID: string,
|
|
_info: Record<string, unknown> | undefined,
|
|
): Promise<boolean> => {
|
|
try {
|
|
const messagesResponse = await ctx.client.session.messages({
|
|
path: { id: sessionID },
|
|
query: { directory: ctx.directory },
|
|
})
|
|
const messages = extractSessionMessages(messagesResponse)
|
|
if (!messages || messages.length === 0) return false
|
|
|
|
const lastUserMessageIndex = getLastUserMessageIndex(messages)
|
|
if (lastUserMessageIndex === -1) return false
|
|
|
|
for (let index = lastUserMessageIndex + 1; index < messages.length; index++) {
|
|
const message = messages[index]
|
|
if (message?.info?.role !== "assistant") {
|
|
continue
|
|
}
|
|
|
|
if (message.info?.error) {
|
|
continue
|
|
}
|
|
|
|
const infoParts = message.info?.parts
|
|
const infoMessageParts = Array.isArray(infoParts)
|
|
? infoParts.filter((part): part is SessionMessagePart => typeof part === "object" && part !== null)
|
|
: undefined
|
|
const parts = message.parts && message.parts.length > 0
|
|
? message.parts
|
|
: infoMessageParts
|
|
const assistantText = getAssistantText(parts)
|
|
if (!assistantText) {
|
|
continue
|
|
}
|
|
|
|
if (extractAutoRetrySignalFn({ message: assistantText })) {
|
|
continue
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
return false
|
|
} catch {
|
|
return false
|
|
}
|
|
}
|
|
}
|