Function extractOutputText

packages/blocks/src/output-text.ts:14–53

(output: unknown, options?: ExtractOutputTextOptions)

 * Returns null if the output type is unrecognized or has no textual representation.
 */
export function extractOutputText(output: unknown, options?: ExtractOutputTextOptions): string | null {
  const outResult = unknownObjectSchema.safeParse(output)
  if (!outResult.success) return null
  const out = outResult.data

  if (out.output_type === 'stream' && typeof out.text === 'string') {
    return out.text
  }

  if (out.output_type === 'execute_result' || out.output_type === 'display_data') {
    const dataResult = unknownObjectSchema.safeParse(out.data)
    const data = dataResult.success ? dataResult.data : undefined
    if (data?.['text/plain']) {
      return String(data['text/plain'])
    }
    if (data?.['text/html']) {
      return '[HTML output]'
    }
    if (data?.['image/png'] || data?.['image/jpeg']) {
      return '[Image output]'
    }
  }

  if (out.output_type === 'error') {
    const ename = String(out.ename || 'Error')
    const evalue = String(out.evalue || '')
    let text = `Error: ${ename}: ${evalue}`
    if (options?.includeTraceback && Array.isArray(out.traceback)) {
      text +=
        '\n' +
        (out.traceback as string[])
          // biome-ignore lint/suspicious/noControlCharactersInRegex: strip ANSI escape sequences from traceback
          .map(line => line.replace(/\x1b\[[0-9;]*m/g, ''))
          .join('\n')
    }
    return text
  }

  return null
}

/**
 * Extract human-readable text from an array of Jupyter-style output objects.