ummaya 0.2.4 → 0.2.6

package/tui/src/query/toolRunner.ts

@@ -0,0 +1,687 @@
+import { findToolByName, type Tool, type ToolUseContext } from '../Tool.js'
+import type { AssistantMessage, Message, UserMessage } from '../types/message.js'
+import { getAdapterToolByName } from '../tools/AdapterTool/AdapterTool.js'
+import {
+  repairUmmayaDocumentToolInputForDispatch,
+} from '../tools/_shared/toolChoiceRepair.js'
+import { deriveLocationQueryFromUserText } from '../tools/_shared/locationInputRepair.js'
+import { createUserMessage } from '../utils/messages.js'
+import { isUnregisteredRawJsonToolUseId } from '../utils/rawJsonToolCall.js'
+import { formatZodValidationError } from '../utils/toolErrors.js'
+import {
+  latestTextUserMessageIndex,
+  messageText,
+  type ToolUseBlock,
+} from './messageGuards.js'
+import { createToolUnavailableErrorPayload } from './toolResultErrors.js'
+
+function serializeToolResult(data: unknown): string {
+  if (typeof data === 'string') return data
+  if (data === null || data === undefined) return ''
+  try {
+    return JSON.stringify(data)
+  } catch {
+    return String(data)
+  }
+}
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+  return typeof value === 'object' && value !== null && !Array.isArray(value)
+}
+
+function parseJsonRecord(value: unknown): Record<string, unknown> | undefined {
+  if (isRecord(value)) return value
+  if (typeof value !== 'string') return undefined
+  try {
+    const parsed: unknown = JSON.parse(value)
+    return isRecord(parsed) ? parsed : undefined
+  } catch {
+    return undefined
+  }
+}
+
+function isStructuredFailureResult(value: unknown): boolean {
+  return parseJsonRecord(value)?.ok === false
+}
+
+function normalizeToolResultDataForModel(
+  toolName: string,
+  data: unknown,
+): unknown {
+  if (toolName !== 'document' || !isRecord(data) || data.ok !== false) {
+    return data
+  }
+  if (isRecord(data.result) && data.result.tool_id === 'document') {
+    return data
+  }
+  const error = isRecord(data.error) ? data.error : undefined
+  const message = typeof error?.message === 'string'
+    ? error.message
+    : 'Document tool failed before producing a document result.'
+  return {
+    ...data,
+    result: {
+      tool_id: 'document',
+      status: 'failed',
+      text_summary: message,
+    },
+  }
+}
+
+function permissionErrorMessage(message: string | undefined): string {
+  const trimmed = message?.trim()
+  return trimmed
+    ? `Permission denied: ${trimmed}`
+    : 'Permission denied: tool execution was not approved.'
+}
+
+type ToolInput = { readonly [key: string]: unknown }
+
+type SuccessfulToolCall = {
+  readonly toolName: string
+  readonly input: ToolInput
+}
+
+const MAX_SUCCESSFUL_CALLS_PER_EFFECTIVE_TOOL = 2
+const HOMETAX_LOOKUP_TOOL_NAME = 'mock_lookup_module_hometax_simplified'
+const KMA_METAR_TOOL_NAME = 'kma_apihub_url_air_metar_decoded'
+const ROOT_FIND_TOOL_NAME = 'find'
+const KAKAO_LOCATION_TOOL_NAMES = new Set([
+  'kakao_address_search',
+  'kakao_keyword_search',
+])
+const KMA_ORDINARY_WEATHER_TOOL_NAMES = new Set([
+  'kma_apihub_upp_mtly_info_service_get_max_wind',
+  'kma_current_observation',
+  'kma_forecast_fetch',
+  'kma_short_term_forecast',
+  'kma_ultra_short_term_forecast',
+])
+
+function stableJson(value: unknown): string {
+  if (Array.isArray(value)) {
+    return `[${value.map(stableJson).join(',')}]`
+  }
+  if (isRecord(value)) {
+    return `{${Object.keys(value)
+      .sort()
+      .map(key => `${JSON.stringify(key)}:${stableJson(value[key])}`)
+      .join(',')}}`
+  }
+  const encoded = JSON.stringify(value)
+  return encoded === undefined ? String(value) : encoded
+}
+
+function toolInputsEquivalent(
+  tool: Tool,
+  left: ToolInput,
+  right: ToolInput,
+): boolean {
+  return tool.inputsEquivalent?.(left, right) ?? stableJson(left) === stableJson(right)
+}
+
+function effectiveToolKey(toolName: string, input: ToolInput): string {
+  const concreteToolId = input.tool_id
+  if (typeof concreteToolId === 'string' && concreteToolId.trim().length > 0) {
+    return `${toolName}:${concreteToolId}`
+  }
+  return toolName
+}
+
+function isCitizenPromptMessage(message: Message): boolean {
+  if (message.type !== 'user') return false
+  if (message.isMeta === true) return false
+  const content = message.message.content
+  if (!Array.isArray(content)) return true
+  return !content.every(block => block.type === 'tool_result')
+}
+
+function latestCitizenPromptIndex(messages: readonly Message[]): number {
+  for (let index = messages.length - 1; index >= 0; index -= 1) {
+    const message = messages[index]
+    if (message !== undefined && isCitizenPromptMessage(message)) {
+      return index
+    }
+  }
+  return -1
+}
+
+function latestCitizenPromptText(messages: readonly Message[]): string {
+  const index = latestTextUserMessageIndex(messages)
+  const message = index >= 0 ? messages[index] : undefined
+  return message ? messageText(message) : ''
+}
+
+function successfulToolResultIdsSinceLatestPrompt(
+  messages: readonly Message[],
+): ReadonlySet<string> {
+  const startIndex = latestCitizenPromptIndex(messages)
+  const ids = new Set<string>()
+  for (const message of messages.slice(startIndex + 1)) {
+    if (message.type !== 'user' || !Array.isArray(message.message.content)) {
+      continue
+    }
+    for (const block of message.message.content) {
+      if (
+        block.type === 'tool_result' &&
+        block.is_error !== true &&
+        !isStructuredFailureResult(block.content)
+      ) {
+        ids.add(block.tool_use_id)
+      }
+    }
+  }
+  return ids
+}
+
+function successfulToolCallsSinceLatestPrompt(
+  messages: readonly Message[],
+): readonly SuccessfulToolCall[] {
+  const startIndex = latestCitizenPromptIndex(messages)
+  const successfulResultIds = successfulToolResultIdsSinceLatestPrompt(messages)
+  const calls: SuccessfulToolCall[] = []
+  for (const message of messages.slice(startIndex + 1)) {
+    if (message.type !== 'assistant') continue
+    for (const block of message.message.content) {
+      if (
+        block.type === 'tool_use' &&
+        successfulResultIds.has(block.id) &&
+        isRecord(block.input)
+      ) {
+        calls.push({
+          toolName: block.name,
+          input: repairDirectLocationToolInputForValidation({
+            toolName: block.name,
+            input: block.input,
+            messages,
+          }),
+        })
+      }
+    }
+  }
+  return calls
+}
+
+type PriorToolResult = {
+  readonly toolName: string
+  readonly content: unknown
+  readonly isError: boolean
+  readonly isStructuredFailure: boolean
+}
+
+function toolResultsSinceLatestPrompt(
+  messages: readonly Message[],
+): readonly PriorToolResult[] {
+  const startIndex = latestCitizenPromptIndex(messages)
+  const toolNamesByUseId = new Map<string, string>()
+  const results: PriorToolResult[] = []
+  for (const message of messages.slice(startIndex + 1)) {
+    if (message.type === 'assistant') {
+      for (const block of message.message.content) {
+        if (block.type === 'tool_use') {
+          toolNamesByUseId.set(block.id, block.name)
+        }
+      }
+      continue
+    }
+    if (message.type !== 'user' || !Array.isArray(message.message.content)) {
+      continue
+    }
+    for (const block of message.message.content) {
+      if (block.type !== 'tool_result') continue
+      const toolName = toolNamesByUseId.get(block.tool_use_id)
+      if (toolName === undefined) continue
+      results.push({
+        toolName,
+        content: block.content,
+        isError: block.is_error === true,
+        isStructuredFailure: isStructuredFailureResult(block.content),
+      })
+    }
+  }
+  return results
+}
+
+function repeatedToolUseMessage(toolName: string): string {
+  return `<tool_use_error>RepeatedToolUseError: Tool '${toolName}' already returned a successful result for the same effective inputs in this query loop. Use the prior tool result, or call with different parameters only if the citizen goal requires a broader follow-up.</tool_use_error>`
+}
+
+function hometaxProgressionMessage(): string {
+  return [
+    'Hometax lookup already returned a usable simplified-tax lookup result for this citizen request.',
+    'Do not repeat the same lookup before the required 본인확인, 위임, and 동의 progression.',
+    'Next valid step: call the verify tool for 본인확인/위임 동의, or write a Korean needs-input answer asking for that approval.',
+  ].join(' ')
+}
+
+function kmaAviationRepeatMessage(kind: 'usable' | 'blocked'): string {
+  return kind === 'usable'
+    ? [
+        'KMA aviation lookup already returned usable METAR evidence for this airport request.',
+        'Do not repeat the decoded METAR tool in the same query loop.',
+        'Use the prior aviation tool_result to answer the delay-risk question and state any remaining manual notification limitation.',
+      ].join(' ')
+    : [
+        'KMA aviation lookup already returned a blocked result for this airport request.',
+        'Do not retry the decoded METAR tool without new credential/configuration evidence.',
+        'Write a Korean handoff answer that names the aviation weather credential limitation.',
+      ].join(' ')
+}
+
+function kmaOrdinaryFallbackMessage(toolName: string): string {
+  return [
+    `Do not call ${toolName} as a fallback for airport aviation weather after aviation METAR evidence exists.`,
+    'Do not ask the user for KMA nx/ny grid coordinates for an airport-flight risk question.',
+    'Use the aviation weather evidence already returned, or give a Korean handoff answer for official airline/airport status checks.',
+  ].join(' ')
+}
+
+function kmaGridRepeatMessage(toolName: string): string {
+  return [
+    'Do not repeat ordinary KMA current or forecast tools after a missing-grid or invalid-grid result.',
+    `Tool ${toolName} already failed for this route/weather turn without required grid parameters.`,
+    'Answer from the existing location evidence and ask for a precise district/date-time only if more weather detail is required.',
+  ].join(' ')
+}
+
+function locationRepeatMessage(toolName: string, query: string): string {
+  return [
+    `Location lookup ${toolName} already returned usable Kakao coordinates for "${query}" in this query loop.`,
+    'Do not repeat the same location lookup.',
+    'Use the prior location evidence, or call the same adapter with a different origin, destination, radius, or refined query when the citizen goal requires it.',
+  ].join(' ')
+}
+
+function shouldUseGenericRepeatGuard(tool: Tool, toolName: string): boolean {
+  return tool.inputsEquivalent !== undefined || toolName === ROOT_FIND_TOOL_NAME
+}
+
+function priorKmaAviationResultKind(
+  priorResults: readonly PriorToolResult[],
+): 'usable' | 'blocked' | undefined {
+  const aviationResult = priorResults.find(
+    result => result.toolName === KMA_METAR_TOOL_NAME,
+  )
+  if (aviationResult === undefined) return undefined
+  return aviationResult.isStructuredFailure || aviationResult.isError
+    ? 'blocked'
+    : 'usable'
+}
+
+function isKmaGridFailureResult(result: PriorToolResult): boolean {
+  if (!KMA_ORDINARY_WEATHER_TOOL_NAMES.has(result.toolName)) return false
+  const text = serializeToolResult(result.content)
+  return /Missing or invalid fields|LOCATE FIRST|nx\/ny|base_date|base_time/iu.test(text)
+}
+
+function createGuardToolResult(
+  block: ToolUseBlock,
+  assistantMessage: AssistantMessage,
+  message: string,
+): UserMessage {
+  return createUserMessage({
+    content: [
+      {
+        type: 'tool_result',
+        tool_use_id: block.id,
+        content: message,
+        is_error: true,
+      },
+    ],
+    toolUseResult: `Error: ${message}`,
+    sourceToolAssistantUUID: assistantMessage.uuid,
+  })
+}
+
+function domainRepeatGuardMessage(params: {
+  readonly block: ToolUseBlock
+  readonly input: ToolInput
+  readonly priorResults: readonly PriorToolResult[]
+  readonly successfulCalls: readonly SuccessfulToolCall[]
+}): string | undefined {
+  if (KAKAO_LOCATION_TOOL_NAMES.has(params.block.name)) {
+    const query = typeof params.input.query === 'string'
+      ? params.input.query.trim()
+      : ''
+    if (
+      query.length > 0 &&
+      params.successfulCalls.some(call =>
+        KAKAO_LOCATION_TOOL_NAMES.has(call.toolName) &&
+        typeof call.input.query === 'string' &&
+        call.input.query.trim() === query
+      )
+    ) {
+      return locationRepeatMessage(params.block.name, query)
+    }
+  }
+
+  if (
+    params.block.name === HOMETAX_LOOKUP_TOOL_NAME &&
+    params.successfulCalls.some(call => call.toolName === HOMETAX_LOOKUP_TOOL_NAME)
+  ) {
+    return hometaxProgressionMessage()
+  }
+
+  if (params.block.name === KMA_METAR_TOOL_NAME) {
+    const kind = priorKmaAviationResultKind(params.priorResults)
+    return kind ? kmaAviationRepeatMessage(kind) : undefined
+  }
+
+  const aviationKind = priorKmaAviationResultKind(params.priorResults)
+  if (aviationKind !== undefined) {
+    if (KMA_ORDINARY_WEATHER_TOOL_NAMES.has(params.block.name)) {
+      return kmaOrdinaryFallbackMessage(params.block.name)
+    }
+    if (
+      params.block.name === ROOT_FIND_TOOL_NAME &&
+      typeof params.block.input.tool_id === 'string' &&
+      KMA_ORDINARY_WEATHER_TOOL_NAMES.has(params.block.input.tool_id)
+    ) {
+      return kmaOrdinaryFallbackMessage(params.block.input.tool_id)
+    }
+  }
+
+  if (
+    KMA_ORDINARY_WEATHER_TOOL_NAMES.has(params.block.name) &&
+    params.priorResults.some(
+      result =>
+        result.toolName === params.block.name && isKmaGridFailureResult(result),
+    )
+  ) {
+    return kmaGridRepeatMessage(params.block.name)
+  }
+
+  return undefined
+}
+
+function repairDirectLocationToolInputForValidation(params: {
+  readonly toolName: string
+  readonly input: ToolInput
+  readonly messages: readonly Message[]
+}): ToolInput {
+  if (
+    !KAKAO_LOCATION_TOOL_NAMES.has(params.toolName) ||
+    (typeof params.input.query === 'string' &&
+      params.input.query.trim().length > 0)
+  ) {
+    return params.input
+  }
+  const query = deriveLocationQueryFromUserText(
+    latestCitizenPromptText(params.messages),
+  )
+  return query === undefined ? params.input : { ...params.input, query }
+}
+
+function unexplainedRepeatedToolMessage(toolName: string): string {
+  return `<tool_use_error>RepeatedToolUseError: Tool '${toolName}' already returned a successful result in this query loop. Before calling the same tool again with different inputs, write a visible progress sentence explaining which parameter, radius, source, or scope is changing; otherwise answer from the prior tool_result.</tool_use_error>`
+}
+
+function excessiveRepeatedToolMessage(toolName: string): string {
+  return `<tool_use_error>RepeatedToolUseError: Tool '${toolName}' already returned multiple successful results in this query loop. Stop calling the same effective tool again; answer from the prior tool_result set or state the remaining limitation.</tool_use_error>`
+}
+
+function hasVisibleAssistantText(message: AssistantMessage): boolean {
+  return message.message.content.some(
+    block => block.type === 'text' && block.text.trim().length > 0,
+  )
+}
+
+export async function runToolUseBlocks(params: {
+  readonly blocks: readonly ToolUseBlock[]
+  readonly assistantMessage: AssistantMessage
+  readonly messages: readonly Message[]
+  readonly toolUseContext: ToolUseContext
+  readonly canUseTool: Parameters<
+    ToolUseContext['options']['tools'][number]['call']
+  >[2]
+}): Promise<readonly UserMessage[]> {
+  const results: UserMessage[] = []
+  const successfulCalls: SuccessfulToolCall[] = [
+    ...successfulToolCallsSinceLatestPrompt(params.messages),
+  ]
+  for (const block of params.blocks) {
+    const localTool = findToolByName(params.toolUseContext.options.tools, block.name)
+    const tool = isUnregisteredRawJsonToolUseId(block.id)
+      ? undefined
+      : localTool ?? getAdapterToolByName(block.name)
+    if (!tool) {
+      const errorPayload = createToolUnavailableErrorPayload(block.name)
+      const errorContent = serializeToolResult(errorPayload)
+      results.push(
+        createUserMessage({
+          content: [
+            {
+              type: 'tool_result',
+              tool_use_id: block.id,
+              content: errorContent,
+              is_error: true,
+            },
+          ],
+          toolUseResult: errorPayload,
+          sourceToolAssistantUUID: params.assistantMessage.uuid,
+        }),
+      )
+      continue
+    }
+
+    const documentRepair = repairUmmayaDocumentToolInputForDispatch({
+      toolName: block.name,
+      input: block.input,
+      messages: params.messages,
+    })
+    if (documentRepair.kind === 'blocked') {
+      results.push(
+        createUserMessage({
+          content: [
+            {
+              type: 'tool_result',
+              tool_use_id: block.id,
+              content: documentRepair.message,
+              is_error: true,
+            },
+          ],
+          toolUseResult: documentRepair.message,
+          sourceToolAssistantUUID: params.assistantMessage.uuid,
+        }),
+      )
+      continue
+    }
+    const toolUseContext = {
+      ...params.toolUseContext,
+      messages: params.messages,
+    }
+    const inputForValidation = localTool === undefined
+      ? documentRepair.input
+      : repairDirectLocationToolInputForValidation({
+          toolName: block.name,
+          input: documentRepair.input,
+          messages: params.messages,
+        })
+    const parsedInput = tool.inputSchema.safeParse(inputForValidation)
+    if (!parsedInput.success) {
+      const errorContent = formatZodValidationError(
+        block.name,
+        parsedInput.error,
+      )
+      results.push(
+        createUserMessage({
+          content: [
+            {
+              type: 'tool_result',
+              tool_use_id: block.id,
+              content: `<tool_use_error>InputValidationError: ${errorContent}</tool_use_error>`,
+              is_error: true,
+            },
+          ],
+          toolUseResult: `InputValidationError: ${parsedInput.error.message}`,
+          sourceToolAssistantUUID: params.assistantMessage.uuid,
+        }),
+      )
+      continue
+    }
+    const priorResults = toolResultsSinceLatestPrompt(params.messages)
+    const domainGuardMessage = domainRepeatGuardMessage({
+      block,
+      input: parsedInput.data,
+      priorResults,
+      successfulCalls,
+    })
+    if (domainGuardMessage !== undefined) {
+      results.push(
+        createGuardToolResult(block, params.assistantMessage, domainGuardMessage),
+      )
+      continue
+    }
+    const repeatedSuccessfulCall = successfulCalls.some(
+      call =>
+        effectiveToolKey(call.toolName, call.input) ===
+          effectiveToolKey(block.name, parsedInput.data) &&
+        toolInputsEquivalent(tool, parsedInput.data, call.input),
+    )
+    if (repeatedSuccessfulCall && shouldUseGenericRepeatGuard(tool, block.name)) {
+      const errorContent = repeatedToolUseMessage(block.name)
+      results.push(
+        createUserMessage({
+          content: [
+            {
+              type: 'tool_result',
+              tool_use_id: block.id,
+              content: errorContent,
+              is_error: true,
+            },
+          ],
+          toolUseResult: `Error: ${errorContent}`,
+          sourceToolAssistantUUID: params.assistantMessage.uuid,
+        }),
+      )
+      continue
+    }
+    const successfulCallCountForEffectiveTool = successfulCalls.filter(
+      call =>
+        effectiveToolKey(call.toolName, call.input) ===
+          effectiveToolKey(block.name, parsedInput.data),
+    ).length
+    if (
+      successfulCallCountForEffectiveTool >=
+      MAX_SUCCESSFUL_CALLS_PER_EFFECTIVE_TOOL &&
+      shouldUseGenericRepeatGuard(tool, block.name)
+    ) {
+      const errorContent = excessiveRepeatedToolMessage(block.name)
+      results.push(
+        createUserMessage({
+          content: [
+            {
+              type: 'tool_result',
+              tool_use_id: block.id,
+              content: errorContent,
+              is_error: true,
+            },
+          ],
+          toolUseResult: `Error: ${errorContent}`,
+          sourceToolAssistantUUID: params.assistantMessage.uuid,
+        }),
+      )
+      continue
+    }
+    const sameToolAlreadySucceeded = successfulCallCountForEffectiveTool > 0
+    if (
+      sameToolAlreadySucceeded &&
+      shouldUseGenericRepeatGuard(tool, block.name) &&
+      !hasVisibleAssistantText(params.assistantMessage)
+    ) {
+      const errorContent = unexplainedRepeatedToolMessage(block.name)
+      results.push(
+        createUserMessage({
+          content: [
+            {
+              type: 'tool_result',
+              tool_use_id: block.id,
+              content: errorContent,
+              is_error: true,
+            },
+          ],
+          toolUseResult: `Error: ${errorContent}`,
+          sourceToolAssistantUUID: params.assistantMessage.uuid,
+        }),
+      )
+      continue
+    }
+    const isValidCall = await tool.validateInput?.(
+      parsedInput.data,
+      toolUseContext,
+    )
+    if (isValidCall?.result === false) {
+      results.push(
+        createUserMessage({
+          content: [
+            {
+              type: 'tool_result',
+              tool_use_id: block.id,
+              content: `<tool_use_error>${isValidCall.message}</tool_use_error>`,
+              is_error: true,
+            },
+          ],
+          toolUseResult: `Error: ${isValidCall.message}`,
+          sourceToolAssistantUUID: params.assistantMessage.uuid,
+        }),
+      )
+      continue
+    }
+    const permissionDecision = await params.canUseTool(
+      tool,
+      parsedInput.data,
+      toolUseContext,
+      params.assistantMessage,
+      block.id,
+    )
+    if (permissionDecision.behavior !== 'allow') {
+      const errorContent = permissionErrorMessage(permissionDecision.message)
+      results.push(
+        createUserMessage({
+          content: [
+            {
+              type: 'tool_result',
+              tool_use_id: block.id,
+              content: errorContent,
+              is_error: true,
+            },
+          ],
+          toolUseResult: `Error: ${errorContent}`,
+          sourceToolAssistantUUID: params.assistantMessage.uuid,
+        }),
+      )
+      continue
+    }
+    const callInput = permissionDecision.updatedInput ?? parsedInput.data
+    const result = await tool.call(
+      callInput,
+      {
+        ...toolUseContext,
+        toolUseId: block.id,
+        userModified: permissionDecision.userModified ?? false,
+      },
+      params.canUseTool,
+      params.assistantMessage,
+    )
+    const modelData = normalizeToolResultDataForModel(block.name, result.data)
+    const isStructuredFailure = isStructuredFailureResult(modelData)
+    results.push(
+      createUserMessage({
+        content: [
+          {
+            type: 'tool_result',
+            tool_use_id: block.id,
+            content: serializeToolResult(modelData),
+            ...(isStructuredFailure ? { is_error: true as const } : {}),
+          },
+        ],
+        toolUseResult: modelData,
+        sourceToolAssistantUUID: params.assistantMessage.uuid,
+      }),
+    )
+    if (!isStructuredFailure) {
+      successfulCalls.push({ toolName: block.name, input: parsedInput.data })
+    }
+  }
+  return results
+}