Function buildPrompt

packages/agent-runtime/src/runtime.ts:258–294

(
  message: string,
  images?: Array<{ base64: string; mediaType: string }>,
)

type ContentBlock = TextBlock | ImageBlock

function buildPrompt(
  message: string,
  images?: Array<{ base64: string; mediaType: string }>,
): string | AsyncIterable<SDKUserMessage> {
  if (!images || images.length === 0) {
    return message
  }

  const content: ContentBlock[] = []

  for (const img of images) {
    content.push({
      type: "image",
      source: {
        type: "base64",
        media_type: img.mediaType,
        data: img.base64,
      },
    })
  }

  content.push({ type: "text", text: message })

  async function* messageGenerator(): AsyncGenerator<SDKUserMessage> {
    yield {
      type: "user",
      message: {
        role: "user",
        content,
      },
      parent_tool_use_id: null,
      session_id: "",
    } as SDKUserMessage
  }

  return messageGenerator()
}

// --- SDK options ---