File size: 2,937 Bytes
867b17d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 |
import { expect, Page } from '@playwright/test';
export class ArtifactPage {
constructor(private page: Page) {}
public get artifact() {
return this.page.getByTestId('artifact');
}
public get sendButton() {
return this.artifact.getByTestId('send-button');
}
public get stopButton() {
return this.page.getByTestId('stop-button');
}
public get multimodalInput() {
return this.page.getByTestId('multimodal-input');
}
async isGenerationComplete() {
const response = await this.page.waitForResponse((response) =>
response.url().includes('/api/chat'),
);
await response.finished();
}
async sendUserMessage(message: string) {
await this.artifact.getByTestId('multimodal-input').click();
await this.artifact.getByTestId('multimodal-input').fill(message);
await this.artifact.getByTestId('send-button').click();
}
async getRecentAssistantMessage() {
const messageElements = await this.artifact
.getByTestId('message-assistant')
.all();
const lastMessageElement = messageElements[messageElements.length - 1];
const content = await lastMessageElement
.getByTestId('message-content')
.innerText()
.catch(() => null);
const reasoningElement = await lastMessageElement
.getByTestId('message-reasoning')
.isVisible()
.then(async (visible) =>
visible
? await lastMessageElement
.getByTestId('message-reasoning')
.innerText()
: null,
)
.catch(() => null);
return {
element: lastMessageElement,
content,
reasoning: reasoningElement,
async toggleReasoningVisibility() {
await lastMessageElement
.getByTestId('message-reasoning-toggle')
.click();
},
};
}
async getRecentUserMessage() {
const messageElements = await this.artifact
.getByTestId('message-user')
.all();
const lastMessageElement = messageElements[messageElements.length - 1];
const content = await lastMessageElement.innerText();
const hasAttachments = await lastMessageElement
.getByTestId('message-attachments')
.isVisible()
.catch(() => false);
const attachments = hasAttachments
? await lastMessageElement.getByTestId('message-attachments').all()
: [];
const page = this.artifact;
return {
element: lastMessageElement,
content,
attachments,
async edit(newMessage: string) {
await page.getByTestId('message-edit-button').click();
await page.getByTestId('message-editor').fill(newMessage);
await page.getByTestId('message-editor-send-button').click();
await expect(
page.getByTestId('message-editor-send-button'),
).not.toBeVisible();
},
};
}
async closeArtifact() {
return this.page.getByTestId('artifact-close-button').click();
}
}
|