mirror of
https://github.com/supabase/supabase.git
synced 2026-05-06 22:18:00 +08:00
e23175f00b
* updated commands and expose ai key locally * added tests for AI assistant * added OPEN_API_KEY for e2e test suite * updated log drain options * updated README
47 lines
1.6 KiB
TypeScript
47 lines
1.6 KiB
TypeScript
import { expect } from '@playwright/test'
|
|
import { test } from '../utils/test.js'
|
|
import { toUrl } from '../utils/to-url.js'
|
|
|
|
test.describe('AI Assistant', async () => {
|
|
test('Can send a message to the assistant and receive a response', async ({ page, ref }) => {
|
|
// Skip the test if the OPENAI_API_KEY is not set
|
|
test.skip(!process.env.OPENAI_API_KEY, 'OPENAI_API_KEY is not set')
|
|
|
|
await page.goto(toUrl(`/project/${ref}`))
|
|
|
|
// Wait for the page to load
|
|
await expect(page.getByRole('heading', { level: 1 })).toBeVisible()
|
|
|
|
// Click the assistant button to open the assistant panel
|
|
await page.locator('#assistant-trigger').click()
|
|
|
|
// Wait for the assistant panel to be visible
|
|
await expect(page.getByRole('heading', { name: 'How can I assist you?' })).toBeVisible()
|
|
|
|
// Type "hello" in the chat input
|
|
const chatInput = page.getByRole('textbox', { name: 'Chat to Postgres...' })
|
|
await chatInput.fill('hello')
|
|
|
|
const responsePromise = page.waitForResponse(
|
|
(response) =>
|
|
response.url().includes('/api/ai/sql/generate-v4') &&
|
|
response.request().method() === 'POST',
|
|
{ timeout: 60000 }
|
|
)
|
|
|
|
// Click the send message button
|
|
const sendButton = page.getByRole('button', { name: 'Send message' })
|
|
await sendButton.click()
|
|
|
|
// Wait for the API request to complete
|
|
const response = await responsePromise
|
|
|
|
// Verify the response was successful
|
|
expect(response.status()).toBe(200)
|
|
|
|
// AI response has values
|
|
const body = await response.text()
|
|
expect(body).toContain('data')
|
|
})
|
|
})
|