workspace-chat.spec.ts 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516
  1. import { test, expect, type Page } from '@playwright/test'
  2. async function login(page: Page) {
  3. await page.goto('/conversations')
  4. await page.getByRole('textbox', { name: '账号', exact: true }).fill('fixture')
  5. await page.getByRole('textbox', { name: '密码', exact: true }).fill('fixture-pass')
  6. await page.getByRole('button', { name: '登录', exact: true }).click()
  7. await expect(page.getByRole('heading', { name: '还没有聊天' })).toBeVisible()
  8. }
  9. async function openCreate(page: Page, name: '新建 Bot' | '新建群聊') {
  10. await page.locator('.desktop-sidebar .sidebar-create-trigger').click()
  11. await page.getByRole('menuitem', { name, exact: true }).click()
  12. }
  13. async function createAgent(page: Page, name: string) {
  14. await openCreate(page, '新建 Bot')
  15. const dialog = page.getByRole('dialog')
  16. await dialog.getByRole('textbox', { name: '名称', exact: true }).fill(name)
  17. await dialog
  18. .getByRole('textbox', { name: '角色提示词与规则', exact: true })
  19. .fill(`你是${name},每次回复给出验收证据。`)
  20. await dialog.getByRole('button', { name: `使用颜色 ${name === '产品经理' ? '#ff6b00' : '#00b9ac'}`, exact: true }).click()
  21. await dialog.locator('.mascot-grid--shape button').filter({ hasText: name === '产品经理' ? '圆形' : '小三角' }).first().click()
  22. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  23. await expect(page.getByRole('heading', { name, exact: true })).toBeVisible()
  24. }
  25. test('an open Web chat list receives cross-client sessions and real title events without reload', async ({
  26. page,
  27. context,
  28. }) => {
  29. await login(page)
  30. const pageAChannelOpened = page.waitForResponse(response =>
  31. response.request().method() === 'POST'
  32. && response.url().endsWith('/api/realtime/channels'))
  33. await page.goto('/chat')
  34. await pageAChannelOpened
  35. await expect(page.getByRole('button', { name: '新建聊天', exact: true })).toBeVisible()
  36. const marker = `${Date.now()}-${Math.random().toString(16).slice(2)}`
  37. const initialTitle = `跨端新会话 ${marker}`
  38. const generatedTitle = `自动标题 ${marker}`
  39. const sentinel = `page-a-${marker}`
  40. await page.evaluate(value => {
  41. ;(window as Window & { __crossClientProjectionSentinel?: string })
  42. .__crossClientProjectionSentinel = value
  43. }, sentinel)
  44. const pageB = await context.newPage()
  45. const pageBChannelOpened = pageB.waitForResponse(response =>
  46. response.request().method() === 'POST'
  47. && response.url().endsWith('/api/realtime/channels'))
  48. await pageB.goto('/chat')
  49. await pageBChannelOpened
  50. await pageB.getByRole('button', { name: '新建聊天', exact: true }).click()
  51. const pageBComposer = pageB.getByRole('textbox', {
  52. name: '输入消息,Enter 发送,Shift + Enter 换行',
  53. })
  54. await pageBComposer.fill(`[cross-client-title:${initialTitle}] 创建跨端会话`)
  55. const pageBSend = pageB.getByRole('button', { name: '发送消息', exact: true })
  56. await expect(pageBSend).toBeEnabled()
  57. await pageBSend.click()
  58. await expect(pageB).toHaveURL(/\/chat\/(?!draft-)[^/?]+(?:\?.*)?$/)
  59. const sessionId = decodeURIComponent(new URL(pageB.url()).pathname.split('/').at(-1) || '')
  60. expect(sessionId).not.toMatch(/^draft-/)
  61. await expect(pageB.locator('.message--assistant')).toContainText('会话独立保存')
  62. const sessionRow = page.locator(`.desktop-sidebar .sidebar-item[data-sidebar-id="${sessionId}"]`)
  63. await expect(sessionRow).toContainText(initialTitle)
  64. expect(await page.evaluate(() =>
  65. (window as Window & { __crossClientProjectionSentinel?: string })
  66. .__crossClientProjectionSentinel)).toBe(sentinel)
  67. await pageB.close()
  68. const titled = await page.request.post(
  69. `http://127.0.0.1:19120/__test/sessions/title?id=${encodeURIComponent(sessionId)}`
  70. + `&title=${encodeURIComponent(generatedTitle)}`,
  71. )
  72. expect(titled.ok(), await titled.text()).toBe(true)
  73. await expect(sessionRow).toContainText(generatedTitle)
  74. await expect(sessionRow).not.toContainText(initialTitle)
  75. expect(await page.evaluate(() =>
  76. (window as Window & { __crossClientProjectionSentinel?: string })
  77. .__crossClientProjectionSentinel)).toBe(sentinel)
  78. })
  79. test('an idle Web viewer follows an iOS-origin run and restores thinking after reload', async ({ page }, testInfo) => {
  80. await login(page)
  81. const capabilities = await (await page.request.get('/api/realtime/capabilities')).json()
  82. const headers = { 'X-CSRF-Token': capabilities.csrfToken, Origin: new URL(page.url()).origin }
  83. const opened = await page.request.post('/api/realtime/channels', { headers, data: { channel: 'chat' } })
  84. expect(opened.ok(), await opened.text()).toBe(true)
  85. const channel = (await opened.json()).id
  86. let sequence = 0
  87. async function command(method: string, params: Record<string, unknown>) {
  88. const response = await page.request.post(`/api/realtime/channels/${channel}/commands`, {
  89. headers: { ...headers, 'Idempotency-Key': `ios-live-${Date.now()}-${++sequence}` },
  90. data: { jsonrpc: '2.0', method, params },
  91. })
  92. expect(response.ok(), await response.text()).toBe(true)
  93. const receipt = await response.json()
  94. expect(receipt.state).toBe('confirmed')
  95. return receipt.response.result
  96. }
  97. const session = await command('session.create', { profile: 'default', source: 'ios' })
  98. const subscribed = page.waitForResponse(response => response.url().endsWith('/commands')
  99. && response.request().postDataJSON()?.method === 'session.resume')
  100. await page.goto(`/chat/${session.stored_session_id}?profile=default`)
  101. await subscribed
  102. await command('prompt.submit', { session_id: session.session_id, text: '[cross-client-live] 检查文件' })
  103. await expect(page.getByRole('status', { name: '正在思考', exact: true })).toBeVisible()
  104. await page.getByText('思考与工具', { exact: true }).click()
  105. await expect(page.getByText('read_file', { exact: true })).toBeVisible()
  106. await page.screenshot({ path: testInfo.outputPath('ios-run-web-thinking-tools.png') })
  107. // This Web page has never submitted a prompt or written an in-flight marker.
  108. await page.evaluate(() => sessionStorage.clear())
  109. await page.reload()
  110. await expect(page.getByRole('status', { name: '正在思考', exact: true })).toBeVisible()
  111. await page.request.post('http://127.0.0.1:19120/__release')
  112. await expect(page.locator('.message--assistant')).toContainText('会话独立保存')
  113. await expect(page.getByRole('status', { name: '正在思考', exact: true })).toHaveCount(0)
  114. await page.screenshot({ path: testInfo.outputPath('ios-run-web-completed.png') })
  115. })
  116. test('created roles and editable teams share a durable chat list without plugin calls', async ({
  117. page,
  118. context,
  119. }, testInfo) => {
  120. const pluginCalls: string[] = []
  121. page.on('request', (r) => {
  122. if (r.url().includes('/plugins/yaoyao')) pluginCalls.push(r.url())
  123. })
  124. await page.setViewportSize({ width: 1638, height: 874 })
  125. await login(page)
  126. const rail = page.locator('.desktop-sidebar')
  127. await expect(rail.locator('.sidebar-feature-nav')).toHaveCount(0)
  128. await expect(rail.locator('.sidebar-primary-action')).toHaveCount(0)
  129. await expect(rail).toHaveCSS('width', '360px')
  130. await expect(rail.locator('.sidebar-context__heading')).toHaveCount(0)
  131. await expect(rail.locator('.bot-logo-trigger .brand-mark')).toBeVisible()
  132. await expect(rail.locator('.sidebar-footer')).toBeVisible()
  133. await expect(rail.locator('.sidebar-account-switcher__main')).toContainText('fixture')
  134. await expect(rail.locator('.sidebar-account-switcher__main .account-initial-avatar')).toHaveText('F')
  135. await rail.getByRole('button', { name: '设置与模式', exact: true }).click()
  136. await expect(page.getByRole('menu', { name: '设置与模式' }).getByRole('menuitem')).toHaveText(['进入设置', '进入聊天模式'])
  137. await page.getByRole('menuitem', { name: '进入设置', exact: true }).click()
  138. const settings = page.getByRole('dialog', { name: '设置中心', exact: true })
  139. await expect(settings.locator('.settings-account-summary')).toContainText('fixture')
  140. await expect(settings.locator('.settings-account-summary .account-initial-avatar')).toHaveText('F')
  141. await settings.locator('.account-avatar-card input[type=file]').setInputFiles({name:'account.png',mimeType:'image/png',buffer:Buffer.from('iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNk+A8AAQUBAScY42YAAAAASUVORK5CYII=','base64')})
  142. await expect(settings.locator('.settings-account-summary .account-initial-avatar img')).toBeVisible()
  143. expect((await (await page.request.get('/api/app/bootstrap')).json()).user.avatar).toMatch(/^data:image\/png;base64,/)
  144. await settings.getByRole('button', {name:'使用首字母',exact:true}).click()
  145. await expect(settings.locator('.settings-account-summary .account-initial-avatar')).toHaveText('F')
  146. await settings.getByRole('button', { name: '身份与头像', exact: true }).click()
  147. await settings.getByRole('button', { name: '造型:星星', exact: true }).click()
  148. await settings.getByRole('button', { name: '使用颜色 #1488ff', exact: true }).click()
  149. await settings.locator('.mascot-grid--expression button').filter({ hasText: '开心' }).click()
  150. await settings.getByRole('button', { name: '保存更改', exact: true }).click()
  151. await expect(settings.getByRole('button', { name: '保存更改', exact: true })).toBeDisabled()
  152. await settings.getByRole('button', { name: '关闭设置中心', exact: true }).click()
  153. await rail.locator('.sidebar-account-switcher__main').click()
  154. await settings.getByRole('button', { name: '身份与头像', exact: true }).click()
  155. await expect(settings.getByRole('button', { name: '造型:星星', exact: true })).toHaveAttribute('aria-pressed', 'true')
  156. await expect(settings.getByRole('button', { name: '使用颜色 #1488ff', exact: true })).toHaveAttribute('aria-pressed', 'true')
  157. await expect(settings.locator('.mascot-grid--expression button').filter({ hasText: '开心' })).toHaveAttribute('aria-pressed', 'true')
  158. await settings.getByRole('button', { name: '关闭设置中心', exact: true }).click()
  159. await expect(rail.getByRole('checkbox', { name: '显示已归档' })).toHaveCount(0)
  160. const plus = rail.getByRole('button', { name: '新建', exact: true })
  161. await expect(rail.locator('.sidebar-collapse')).toHaveCount(0)
  162. await plus.click()
  163. await expect(page.getByRole('menu', { name: '新建聊天' }).getByRole('menuitem')).toHaveText(['新建 Bot', '添加远程 Agent', '新建群聊'])
  164. await page.keyboard.press('Escape')
  165. await rail.getByRole('button', { name: '搜索', exact: true }).click()
  166. await expect(page.getByRole('dialog', { name: '搜索聊天' })).toBeVisible()
  167. await page.keyboard.press('Escape')
  168. await page.goto('/chat')
  169. await expect(rail).toHaveCSS('width', '264px')
  170. await expect(rail.getByRole('button', { name: '聊天', exact: true })).toHaveCount(0)
  171. await rail.getByRole('button', { name: '设置与模式', exact: true }).click()
  172. const settingsMenu = page.getByRole('menu', { name: '设置与模式', exact: true })
  173. await expect(settingsMenu.getByRole('menuitem')).toHaveText(['进入设置', '进入 Bot 模式'])
  174. await page.screenshot({ path: testInfo.outputPath('chat-settings-mode-menu.png') })
  175. await settingsMenu.getByRole('menuitem', { name: '进入设置', exact: true }).click()
  176. await expect(page.getByRole('dialog', { name: '设置中心', exact: true })).toBeVisible()
  177. await page.getByRole('button', { name: '关闭设置中心', exact: true }).click()
  178. await expect(rail.getByRole('button', { name: '设置与模式', exact: true })).toBeFocused()
  179. await rail.getByRole('button', { name: '设置与模式', exact: true }).click()
  180. await settingsMenu.getByRole('menuitem', { name: '进入 Bot 模式', exact: true }).click()
  181. await expect(page).toHaveURL(/\/conversations$/)
  182. await expect(rail).toHaveCSS('width', '360px')
  183. await page.addInitScript(() => { Object.defineProperty(crypto, 'randomUUID', { value: undefined, configurable: true }) })
  184. await page.reload()
  185. await createAgent(page, '产品经理')
  186. await page.getByRole('button', {name:'聊天设置',exact:true}).click()
  187. const avatarEditor = page.getByRole('dialog')
  188. await expect(avatarEditor.locator('.color-grid button')).toHaveCount(11)
  189. await expect(avatarEditor.locator('.mascot-grid--expression button')).toHaveCount(10)
  190. await avatarEditor.getByRole('button', {name:'造型:星星',exact:true}).click()
  191. await avatarEditor.locator('.mascot-grid--expression button').filter({hasText:'自豪'}).click()
  192. await avatarEditor.getByRole('button', {name:'使用颜色 #000000',exact:true}).click()
  193. await avatarEditor.getByRole('button', {name:'保存',exact:true}).click()
  194. const identities = await (await page.request.get('/api/app/agents')).json()
  195. const edited = identities.agents.find((a: {name:string}) => a.name === '产品经理')
  196. expect(JSON.parse(edited.avatar.slice('yaoyao-avatar:v2:'.length))).toMatchObject({version:2,bodyId:'star',expression:'proud',color:'#000000'})
  197. await page.reload()
  198. await page.getByRole('button', {name:'聊天设置',exact:true}).click()
  199. await expect(avatarEditor.getByRole('button', {name:'造型:星星',exact:true})).toHaveAttribute('aria-pressed','true')
  200. await avatarEditor.locator('input[type=file]').setInputFiles({name:'avatar.png',mimeType:'image/png',buffer:Buffer.from('iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNk+A8AAQUBAScY42YAAAAASUVORK5CYII=','base64')})
  201. await expect(avatarEditor.getByText('图片裁剪',{exact:true})).toBeVisible()
  202. let photoBytes=''
  203. for(const [label,crop] of [['圆形','circle'],['方形','square'],['圆角方形','rounded']]){
  204. await avatarEditor.getByRole('button',{name:label,exact:true}).click()
  205. await avatarEditor.getByRole('button',{name:'保存',exact:true}).click()
  206. const payload=await (await page.request.get('/api/app/agents')).json()
  207. const avatar=JSON.parse(payload.agents.find((a:{name:string})=>a.name==='产品经理').avatar.slice('yaoyao-avatar:v2:'.length))
  208. expect(avatar).toMatchObject({avatarMode:'image',imageCrop:crop})
  209. if(photoBytes)expect(avatar.imageDataURL).toBe(photoBytes)
  210. else photoBytes=avatar.imageDataURL
  211. await page.getByRole('button',{name:'聊天设置',exact:true}).click()
  212. }
  213. await avatarEditor.getByRole('button',{name:'动态吉祥物',exact:true}).click()
  214. await avatarEditor.locator('.mascot-grid--shape button').filter({hasText:'圆形'}).first().click()
  215. await avatarEditor.getByRole('button', {name:'使用颜色 #ff6b00',exact:true}).click()
  216. await avatarEditor.locator('.mascot-grid--expression button').filter({hasText:'默认'}).click()
  217. await avatarEditor.getByRole('button', {name:'保存',exact:true}).click()
  218. await expect(avatarEditor).toBeHidden()
  219. const composerInput = page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行')
  220. const sendButton = page.getByRole('button', { name: '发送消息', exact: true })
  221. await expect(composerInput).toBeEnabled()
  222. await expect(rail.getByText('聊天', { exact: true })).toHaveCount(0)
  223. await expect(rail.getByText('聊天列表', { exact: true })).toHaveCount(0)
  224. await expect(rail.locator('.sidebar-item__row strong').first()).toHaveCSS('font-size', '15px')
  225. await expect(rail.locator('.sidebar-item__row--secondary').first()).toHaveCSS('font-size', '13px')
  226. await expect(rail.locator('.sidebar-item__row small').first()).toHaveCSS('font-size', '10px')
  227. await composerInput.fill('请说明角色职责')
  228. await expect(sendButton).toBeEnabled()
  229. await sendButton.click()
  230. await expect(page.locator('.message--assistant')).toContainText('我是产品经理')
  231. for (const colorScheme of ['light', 'dark'] as const) {
  232. await page.emulateMedia({ colorScheme })
  233. for (const selector of ['.message__avatar.agent-avatar', '.desktop-sidebar .sidebar-item__icon--avatar', '.desktop-sidebar .agent-avatar']) {
  234. await expect(page.locator(selector).first()).toHaveCSS('background-color', 'rgba(0, 0, 0, 0)')
  235. await expect(page.locator(selector).first()).toHaveCSS('background-image', 'none')
  236. }
  237. }
  238. await page.emulateMedia({ colorScheme: 'light' })
  239. await expect(page.locator('.message--assistant a').first()).toHaveAttribute(
  240. 'href',
  241. /\/api\/app\/files\/.+\/download/,
  242. )
  243. await page.locator('.message--assistant a').first().click()
  244. const textPreview = page.getByRole('dialog', { name: '预览 workspace-report.txt', exact: true })
  245. await expect(textPreview).toBeVisible()
  246. await expect(textPreview).toContainText('Fixture report')
  247. await textPreview.getByRole('button', { name: '关闭预览' }).click()
  248. await page.getByRole('button', {name:'置顶聊天',exact:true}).click()
  249. await expect(page.getByRole('button', {name:'取消置顶',exact:true})).toBeVisible()
  250. const directURL = page.url()
  251. await createAgent(page, '开发工程师')
  252. await expect(page.locator('.message')).toHaveCount(0)
  253. await openCreate(page, '新建群聊')
  254. const dialog = page.getByRole('dialog')
  255. await dialog.getByRole('textbox', { name: '名称', exact: true }).fill('产品开发团队')
  256. await dialog.getByRole('checkbox', { name: '产品经理', exact: true }).check()
  257. await dialog.getByRole('checkbox', { name: '开发工程师', exact: true }).check()
  258. await dialog
  259. .getByRole('combobox', { name: '管理员', exact: true })
  260. .selectOption({ label: '产品经理' })
  261. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  262. await expect(page.getByRole('heading', { name: '产品开发团队', exact: true })).toBeVisible()
  263. await expect(page.locator('.desktop-sidebar .sidebar-item')).toHaveCount(3)
  264. // Every cluster portrait must be the same portrait shown in that member's direct chat.
  265. const groupRow = page.locator('.desktop-sidebar .sidebar-item').filter({ hasText: '产品开发团队' })
  266. await expect(groupRow.locator('.team-avatar__member')).toHaveCount(2)
  267. for (const name of ['产品经理', '开发工程师']) {
  268. const portrait = `.agent-avatar[aria-label="${name} 的头像"]`
  269. const direct = page.locator('.desktop-sidebar .sidebar-item').filter({ hasText: name }).filter({ hasNot: page.locator('.team-avatar') }).locator(portrait)
  270. const member = groupRow.locator(portrait)
  271. await expect(member.locator('[data-part=outline] > :first-child')).toHaveAttribute('fill', await direct.locator('[data-part=outline] > :first-child').getAttribute('fill') || '')
  272. expect(await member.locator('[data-part=outline] > :first-child').evaluate(el => el.tagName))
  273. .toBe(await direct.locator('[data-part=outline] > :first-child').evaluate(el => el.tagName))
  274. }
  275. await expect(page.locator('.desktop-sidebar')).not.toContainText('话题')
  276. await expect(page.locator('.desktop-sidebar .sidebar-account-switcher__main')).not.toContainText(
  277. '通用助手',
  278. )
  279. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  280. await expect(dialog.getByRole('checkbox', { name: '产品经理', exact: true })).toBeDisabled()
  281. await expect(dialog.getByRole('checkbox', { name: '开发工程师', exact: true })).toBeEnabled()
  282. await dialog.getByRole('button', { name: '取消', exact: true }).click()
  283. await page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行').fill('请@开发工程师给出实现结果')
  284. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  285. await expect(page.locator('.message--assistant')).toHaveCount(3)
  286. await expect(page.locator('.message--assistant').nth(1)).toContainText('我是开发工程师')
  287. await expect(page.locator('.run-status')).toHaveCount(0)
  288. const second = await context.newPage()
  289. await second.goto(page.url())
  290. await expect(second.locator('.message--assistant')).toHaveCount(3)
  291. await page.screenshot({ path: testInfo.outputPath('mixed-chats.png'), fullPage: true })
  292. const groupURL=page.url()
  293. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  294. await dialog.getByRole('checkbox', { name: '开发工程师', exact: true }).uncheck()
  295. await expect(dialog.getByRole('checkbox', { name: '产品经理', exact: true })).toBeDisabled()
  296. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  297. await expect(groupRow.locator('.team-avatar__member')).toHaveCount(1)
  298. await expect(page.locator('.message--assistant')).toHaveCount(3)
  299. await expect(second.locator('.desktop-sidebar .sidebar-item').filter({ hasText: '产品开发团队' }).locator('.team-avatar__member')).toHaveCount(1)
  300. await page.reload()
  301. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  302. await expect(dialog.getByRole('checkbox', { name: '开发工程师', exact: true })).not.toBeChecked()
  303. await dialog.getByRole('button', { name: '取消', exact: true }).click()
  304. await createAgent(page, '新成员')
  305. await page.goto(groupURL)
  306. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  307. await dialog.getByRole('checkbox', { name: '新成员', exact: true }).check()
  308. await dialog.getByRole('checkbox', { name: '产品经理', exact: true }).scrollIntoViewIfNeeded()
  309. await page.screenshot({ path: testInfo.outputPath('member-settings.png'), fullPage: true })
  310. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  311. await expect(groupRow.locator('.team-avatar__member')).toHaveCount(2)
  312. await expect(groupRow.locator('[aria-label="开发工程师 的头像"]')).toHaveCount(0)
  313. await expect(groupRow.locator('[aria-label="新成员 的头像"]')).toHaveCount(1)
  314. await page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行').fill('请@新成员继续')
  315. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  316. await expect(page.locator('.message--assistant')).toHaveCount(6)
  317. await expect(page.locator('.message--assistant').nth(4)).toContainText('我是新成员')
  318. await page.goto('/files')
  319. await page.getByRole('button',{name:'查看来源',exact:true}).first().click()
  320. await expect(page).toHaveURL(groupURL)
  321. await page.getByRole('button', { name: '归档聊天', exact: true }).click()
  322. await expect(page.locator('.desktop-sidebar .sidebar-item')).toHaveCount(3)
  323. await page.locator('.desktop-sidebar .sidebar-search-trigger').click()
  324. const search = page.getByRole('dialog', { name: '搜索聊天', exact: true })
  325. await expect(search).toBeVisible()
  326. await search.getByRole('searchbox', { name: '搜索聊天', exact: true }).fill('产品开发')
  327. await expect(search.getByRole('option')).toHaveCount(0)
  328. await search.getByRole('tab', { name: '已归档', exact: true }).click()
  329. await expect(search.getByRole('searchbox')).toHaveValue('产品开发')
  330. await expect(search.getByRole('option')).toHaveCount(1)
  331. await page.screenshot({ path: testInfo.outputPath('archived-chat-search.png'), fullPage: true })
  332. await search.getByRole('option').click()
  333. await expect(search).not.toBeVisible()
  334. await page.getByRole('button', { name: '恢复聊天', exact: true }).click()
  335. await page.goto(directURL)
  336. await expect(page.locator('.message--assistant')).toHaveCount(1)
  337. await expect(page.locator('.message--assistant')).toContainText('我是产品经理')
  338. await page.goto('/chat')
  339. await expect(page).toHaveURL(/\/chat/)
  340. expect(pluginCalls).toEqual([])
  341. await second.close()
  342. await page.goto('/conversations')
  343. await openCreate(page, '新建群聊')
  344. await expect(dialog.getByRole('button', { name: '信息收集团队', exact: true })).toBeDisabled()
  345. await dialog.getByRole('button', { name: '取消', exact: true }).click()
  346. await createAgent(page, '模板成员四')
  347. await createAgent(page, '模板成员五')
  348. await openCreate(page, '新建群聊')
  349. for (const name of ['信息收集团队', '产品设计团队', '软件开发团队', '文案书写团队', '项目管理团队', '数据分析团队', '会议协作团队', '运维保障团队']) {
  350. await expect(dialog.getByRole('button', { name, exact: true })).toBeEnabled()
  351. }
  352. await page.screenshot({ path: testInfo.outputPath('restored-team-presets.png'), fullPage: true })
  353. await dialog.getByRole('button', { name: '产品设计团队', exact: true }).click()
  354. await expect(dialog.getByRole('textbox', { name: '名称', exact: true })).toHaveValue('产品设计团队')
  355. await dialog.getByRole('combobox', { name: '产品负责人对应的 Agent', exact: true }).selectOption({ label: '产品经理' })
  356. await dialog.getByRole('combobox', { name: '交互设计对应的 Agent', exact: true }).selectOption({ label: '开发工程师' })
  357. const [submitted] = await Promise.all([
  358. page.waitForRequest(r => r.method() === 'POST' && r.url().endsWith('/api/app/conversations')),
  359. dialog.getByRole('button', { name: '保存', exact: true }).click(),
  360. ])
  361. const payload = submitted.postDataJSON()
  362. expect(new Set(payload.memberIds).size).toBe(5)
  363. expect(Object.keys(payload.memberRoles)).toHaveLength(5)
  364. expect(payload.memberRoles[payload.administratorId].name).toBe('产品负责人')
  365. await expect(page.getByRole('heading', { name: '产品设计团队', exact: true })).toBeVisible()
  366. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  367. await expect(dialog.getByText('产品经理 · 产品负责人', { exact: false })).toBeVisible()
  368. await expect(dialog.getByText('开发工程师 · 交互设计', { exact: false })).toBeVisible()
  369. await dialog.getByRole('button', { name: '取消', exact: true }).click()
  370. await page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行').fill('请汇总设计方案')
  371. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  372. await expect(page.locator('.message--assistant')).toContainText('我是产品经理')
  373. const composer = page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行')
  374. await composer.fill('[hold-workspace] 正在执行的任务')
  375. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  376. await expect(page.getByRole('button', { name: '停止生成', exact: true })).toBeVisible()
  377. await composer.fill('后续排队任务')
  378. await expect(page.getByRole('button', { name: '发送消息', exact: true })).toBeEnabled()
  379. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  380. await expect(page.locator('.message--user').last()).toContainText('后续排队任务')
  381. const detail = await (await page.request.get(new URL(page.url()).pathname.replace('/conversations/', '/api/app/conversations/'))).json()
  382. expect(detail.conversation.queuedMessageCount).toBe(1)
  383. await page.request.post('http://127.0.0.1:19120/__release')
  384. await expect.poll(async () => (await (await page.request.get(`/api/app/conversations/${detail.conversation.id}`)).json()).conversation.activeRunId).toBeUndefined()
  385. await page.setViewportSize({ width: 390, height: 844 })
  386. await page.getByRole('button', { name: '返回 Bot 列表', exact: true }).click()
  387. await expect(rail).toBeVisible()
  388. await expect(rail).toHaveCSS('width', '390px')
  389. await expect(page.locator('.mobile-header')).toBeHidden()
  390. await expect(rail.locator('.sidebar-feature-nav')).toHaveCount(0)
  391. await page.screenshot({ path: testInfo.outputPath('mobile-bot-list.png') })
  392. await rail.getByRole('button', { name: '新建', exact: true }).click()
  393. await expect(page.getByRole('menuitem', { name: '新建 Bot', exact: true })).toBeVisible()
  394. await page.keyboard.press('Escape')
  395. await rail.getByRole('button', { name: '搜索', exact: true }).click()
  396. await expect(page.getByRole('dialog', { name: '搜索聊天' })).toBeVisible()
  397. await expect(page.getByRole('tab', { name: '未归档', exact: true })).toBeVisible()
  398. await page.getByRole('button', { name: '关闭搜索', exact: true }).click()
  399. await rail.locator('.sidebar-account-switcher__main').click()
  400. await page.getByRole('button', { name: '进入聊天模式', exact: true }).click()
  401. await expect(page).toHaveURL(/\/chat$/)
  402. await page.goto('/')
  403. await expect(page).toHaveURL(/\/chat$/)
  404. await page.setViewportSize({ width: 1280, height: 800 })
  405. await page.getByRole('button', { name: '设置与模式', exact: true }).click()
  406. await page.getByRole('menuitem', { name: '进入 Bot 模式', exact: true }).click()
  407. await page.goto('/')
  408. await expect(page).toHaveURL(/\/conversations$/)
  409. await page.reload()
  410. await expect(rail.locator('.bot-list-header')).toBeVisible()
  411. expect(pluginCalls).toEqual([])
  412. })
  413. test('subaccount allocation and Bot-only navigation work through the browser', async ({ page, browser }, testInfo) => {
  414. await login(page)
  415. await page.locator('.desktop-sidebar').getByRole('button', { name: '设置与模式', exact: true }).click()
  416. await page.getByRole('menuitem', { name: '进入设置', exact: true }).click()
  417. const settings = page.getByRole('dialog', { name: '设置中心', exact: true })
  418. await settings.getByRole('button', { name: '用户与权限', exact: true }).click()
  419. const username = `child-${Date.now()}`
  420. await settings.locator('input[name=managed-username]').fill(username)
  421. await settings.locator('input[name=managed-temporary-password]').fill('temporary-pass')
  422. await settings.getByRole('listbox', { name: '新子账号的基础 Agent' }).selectOption('default')
  423. await settings.getByRole('button', { name: '创建用户', exact: true }).click()
  424. await expect(settings.getByRole('listbox', { name: `分配给 ${username} 的基础 Agent`, exact: true })).toHaveValues(['default'])
  425. await page.screenshot({ path: testInfo.outputPath('admin-agent-allocation.png'), fullPage: true })
  426. const childContext = await browser.newContext({ viewport: { width: 1280, height: 800 } })
  427. const child = await childContext.newPage()
  428. await child.goto(new URL('/chat', page.url()).href)
  429. await child.getByRole('textbox', { name: '账号', exact: true }).fill(username)
  430. await child.getByRole('textbox', { name: '密码', exact: true }).fill('temporary-pass')
  431. await child.getByRole('button', { name: '登录', exact: true }).click()
  432. await child.getByLabel('当前密码', { exact: true }).fill('temporary-pass')
  433. await child.getByLabel('新密码', { exact: true }).fill('permanent-pass')
  434. await child.getByLabel('确认新密码', { exact: true }).fill('permanent-pass')
  435. await child.getByRole('button', { name: '保存并进入夭夭', exact: true }).click()
  436. await expect(child).toHaveURL(/\/conversations$/)
  437. await expect(child.getByRole('heading', { name: '还没有聊天' })).toBeVisible()
  438. await child.locator('.desktop-sidebar').getByRole('button', { name: '设置与模式', exact: true }).click()
  439. await expect(child.getByRole('menu', { name: '设置与模式' }).getByRole('menuitem')).toHaveText(['进入设置'])
  440. await child.keyboard.press('Escape')
  441. await openCreate(child, '新建 Bot')
  442. const dialog = child.getByRole('dialog')
  443. await expect(dialog.getByLabel('基础 Agent').locator('option')).toHaveCount(1)
  444. await expect(dialog.getByLabel('基础 Agent').locator('option')).toContainText('通用助手')
  445. await dialog.getByLabel('基础 Agent').scrollIntoViewIfNeeded()
  446. await child.screenshot({ path: testInfo.outputPath('subaccount-assigned-source.png'), fullPage: true })
  447. await dialog.getByRole('textbox', { name: '名称', exact: true }).fill('子账号助手')
  448. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  449. await expect(child.getByRole('heading', { name: '子账号助手', exact: true })).toBeVisible()
  450. for (const path of ['/history', '/kanban', '/files', '/chat']) {
  451. await child.goto(new URL(path, page.url()).href)
  452. await expect(child).toHaveURL(/\/conversations$/)
  453. }
  454. await childContext.close()
  455. })
  456. for (const mode of ['普通聊天', 'Bot 单聊', '群聊', '普通长回复'] as const) {
  457. test(`${mode} 在完成前格式化流式 Markdown 并保留最终内容`, async ({ page }, testInfo) => {
  458. await login(page)
  459. if (mode.startsWith('普通')) {
  460. await page.goto('/chat')
  461. await page.getByRole('button', { name: '新建聊天', exact: true }).click()
  462. } else {
  463. await createAgent(page, `${mode}流式验证助手`)
  464. if (mode === '群聊') {
  465. await createAgent(page, '群聊流式协作成员')
  466. await openCreate(page, '新建群聊')
  467. const dialog = page.getByRole('dialog')
  468. await dialog.getByRole('textbox', { name: '名称', exact: true }).fill('流式验证群')
  469. await dialog.getByRole('checkbox', { name: `${mode}流式验证助手`, exact: true }).check()
  470. await dialog.getByRole('checkbox', { name: '群聊流式协作成员', exact: true }).check()
  471. await dialog.getByRole('combobox', { name: '管理员', exact: true }).selectOption({ label: `${mode}流式验证助手` })
  472. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  473. await expect(page.getByRole('heading', { name: '流式验证群', exact: true })).toBeVisible()
  474. }
  475. }
  476. await page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行').fill(`[streaming-markdown] ${mode === '普通长回复' ? '[long]' : ''} 演示排版`)
  477. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  478. const message = page.locator('.message--assistant').last()
  479. await expect(message.locator('.markdown--streaming h1')).toHaveText('流式标题')
  480. await expect(message.locator('strong').filter({ hasText: '即时格式化' })).toBeVisible()
  481. await expect(message.locator('li')).toHaveText(['第一项', '第二项'])
  482. await expect(message.locator('pre code').first()).toContainText('const answer = 42')
  483. await expect(message).not.toContainText('最终完整标记')
  484. await page.screenshot({ path: testInfo.outputPath('streaming-markdown.png'), fullPage: true })
  485. if (mode === '普通长回复') {
  486. await expect(message.getByText('流式末尾标记', { exact: true })).toBeInViewport()
  487. await page.locator('.timeline').evaluate(el => { el.scrollTop = 0; el.dispatchEvent(new Event('scroll')) })
  488. await expect(message.getByRole('heading', { name: '流式标题', exact: true })).toBeInViewport()
  489. }
  490. const released = await page.request.post('http://127.0.0.1:19120/__release')
  491. expect(released.ok()).toBe(true)
  492. await expect(message).toContainText('最终完整标记')
  493. await expect(message.locator('table')).toContainText('项目')
  494. await expect(message.locator('.markdown--streaming')).toHaveCount(0)
  495. if (mode === '普通长回复') {
  496. await expect(message.getByRole('heading', { name: '流式标题', exact: true })).toBeInViewport()
  497. await expect(message.getByText('最终完整标记', { exact: true })).not.toBeInViewport()
  498. }
  499. })
  500. }