Function createToolWrapper

src/tools/REPLTool/toolWrappers.ts:151–254

(
  toolName: string,
  runtime: ReplWrapperRuntime,
)

}

export function createToolWrapper(
  toolName: string,
  runtime: ReplWrapperRuntime,
): (args: Record<string, unknown>) => Promise<unknown> {
  return async (rawArgs: Record<string, unknown>) => {
    const toolInput = ensureRecord(rawArgs)
    const innerToolUseID = randomUUID()

    const toolUse: ToolUseBlock = {
      type: 'tool_use',
      id: innerToolUseID,
      name: toolName,
      input: toolInput,
    }

    const assistantMessage = createAssistantMessage({
      content: [toolUse],
      isVirtual: true,
    }) as AssistantMessage

    runtime.pushMessage(assistantMessage)
    runtime.onProgress?.({
      toolUseID: runtime.outerToolUseID,
      data: {
        type: 'repl_tool_call',
        phase: 'start',
        toolName,
        toolInput,
      },
    })

    const innerContext: ToolUseContext = {
      ...runtime.toolUseContext,
      getAppState: createNestedGetAppState(
        runtime.toolUseContext.getAppState,
        runtime.availableTools,
      ),
      options: {
        ...runtime.toolUseContext.options,
        tools: runtime.availableTools,
      },
    }

    let sawToolResult = false
    let toolResult: unknown
    let toolError: string | undefined

    for await (const update of runToolUse(
      toolUse,
      assistantMessage,
      runtime.canUseTool,
      innerContext,
    )) {
      if (update.contextModifier) {
        runtime.pushContextModifier(update.contextModifier.modifyContext)
      }
      if (update.message.type === 'progress') {
        continue