mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-28 18:19:28 +00:00
7dec2c9640
probe-model-picker.mjs times pill-click → menu painted over N rounds; profile-model-picker.mjs wraps one open in a CPU profile with a top-self-time table. Both attach to the perf:serve instance.
60 lines
2.2 KiB
JavaScript
60 lines
2.2 KiB
JavaScript
// CPU-profile one model-picker open.
|
|
// node scripts/profile-model-picker.mjs [--port 9222]
|
|
import { writeFileSync } from 'node:fs'
|
|
|
|
import { CDP } from './perf/lib/cdp.mjs'
|
|
import { cpuProfileTopSelf } from './perf/lib/stats.mjs'
|
|
|
|
const port = Number(process.argv.includes('--port') ? process.argv[process.argv.indexOf('--port') + 1] : 9222)
|
|
const cdp = await CDP.connect({ port })
|
|
await cdp.send('Runtime.enable')
|
|
await cdp.send('Profiler.enable')
|
|
await cdp.send('Profiler.setSamplingInterval', { interval: 100 })
|
|
|
|
const OPEN = `
|
|
(async () => {
|
|
// Reset: close any open menu first.
|
|
document.dispatchEvent(new KeyboardEvent('keydown', { key: 'Escape', bubbles: true }))
|
|
await new Promise(r => setTimeout(r, 250))
|
|
|
|
const btns = [...document.querySelectorAll('button[aria-label]')]
|
|
const pill = btns.find(b => /model/i.test(b.getAttribute('aria-label') || ''))
|
|
if (!pill) return -1
|
|
const t0 = performance.now()
|
|
pill.dispatchEvent(new PointerEvent('pointerdown', { bubbles: true }))
|
|
pill.dispatchEvent(new PointerEvent('pointerup', { bubbles: true }))
|
|
pill.click()
|
|
const found = await new Promise(resolve => {
|
|
const deadline = performance.now() + 8000
|
|
const check = () => {
|
|
const menu = document.querySelector('[role="menu"]')
|
|
if (menu && menu.childElementCount > 0) {
|
|
requestAnimationFrame(() => requestAnimationFrame(() => resolve(performance.now())))
|
|
return
|
|
}
|
|
if (performance.now() > deadline) { resolve(-1); return }
|
|
requestAnimationFrame(check)
|
|
}
|
|
check()
|
|
})
|
|
return found < 0 ? -1 : found - t0
|
|
})()
|
|
`
|
|
|
|
await cdp.send('Profiler.start')
|
|
const openMs = await cdp.eval(OPEN)
|
|
const { profile } = await cdp.send('Profiler.stop')
|
|
|
|
console.log('openMs:', Math.round(openMs))
|
|
const out = `/tmp/model-picker-open.cpuprofile`
|
|
writeFileSync(out, JSON.stringify(profile))
|
|
console.log('wrote', out)
|
|
console.log('top self-time (ms):')
|
|
|
|
for (const r of cpuProfileTopSelf(profile, 20)) {
|
|
console.log(` ${r.ms.toFixed(1).padStart(7)} ${r.name.padEnd(44)} ${r.url.split('/').slice(-2).join('/')}:${r.line}`)
|
|
}
|
|
|
|
// Close the menu again.
|
|
await cdp.eval(`document.dispatchEvent(new KeyboardEvent('keydown', { key: 'Escape', bubbles: true }))`)
|
|
cdp.close()
|