Function yieldMissingToolResultBlocks

src/query.ts:128–158

(
  assistantMessages: AssistantMessage[],
  errorMessage: string,
  existingToolResultIds: ReadonlySet<string> = new Set(),
)

/* eslint-enable @typescript-eslint/no-require-imports */

function* yieldMissingToolResultBlocks(
  assistantMessages: AssistantMessage[],
  errorMessage: string,
  existingToolResultIds: ReadonlySet<string> = new Set(),
) {
  for (const assistantMessage of assistantMessages) {
    // Extract all tool use blocks from this assistant message
    const toolUseBlocks = assistantMessage.message.content.filter(
      content => content.type === 'tool_use',
    ) as ToolUseBlock[]

    // Emit an interruption message for each tool use
    for (const toolUse of toolUseBlocks) {
      if (existingToolResultIds.has(toolUse.id)) {
        continue
      }
      yield createUserMessage({
        content: [
          {
            type: 'tool_result',
            content: errorMessage,
            is_error: true,
            tool_use_id: toolUse.id,
          },
        ],
        toolUseResult: errorMessage,
        sourceToolAssistantUUID: assistantMessage.uuid,
      })
    }
  }
}

function collectToolResultIds(messages: Message[]): Set<string> {
  const ids = new Set<string>()