Function waitForResponse

testing/panel/tests/helpers.ts:106–155

(
  page: Page,
  timeout: number = 60_000,
)

 * Wait for the assistant response to complete
 */
export async function waitForResponse(
  page: Page,
  timeout: number = 60_000,
): Promise<void> {
  // Wait for loading to start - the stop button should appear or we see loading indicator
  const stopButton = page.locator('button:has-text("Stop")')

  // First, wait a moment for loading to potentially start
  await page.waitForTimeout(1000)

  // Check if loading started by looking for the stop button
  const loadingStarted = await stopButton.isVisible().catch(() => false)

  if (loadingStarted) {
    // Wait for loading to complete (stop button to disappear)
    try {
      await stopButton.waitFor({ state: 'hidden', timeout: timeout - 1000 })
    } catch {
      // Stop button might still be visible if test times out
    }
  } else {
    // Loading might have been too fast or there's an error
    // Wait for either an assistant message or an error to appear
    const messagesJson = page
      .locator('pre')
      .filter({ hasText: '"role"' })
      .first()
    try {
      // Wait for the messages JSON to contain an assistant message
      await page.waitForFunction(
        () => {
          const preElements = document.querySelectorAll('pre')
          for (const pre of preElements) {
            const text = pre.textContent || ''
            if (text.includes('"assistant"')) {
              return true
            }
          }
          return false
        },
        { timeout: timeout - 1000 },
      )
    } catch {
      // Timeout waiting for response
    }
  }

  // Additional wait for message to fully render
  await page.waitForTimeout(500)
}

/**
 * Get the last assistant message text from the chat