Simon
68 lines
1.5 KiB
TypeScript
68 lines
1.5 KiB
TypeScript
import type { LLMConfig } from '@page-agent/llms'
|
|
|
|
export default defineUnlistedScript(() => {
|
|
const w = window as any
|
|
|
|
let _lastId = 0
|
|
function getId() {
|
|
_lastId += 1
|
|
return _lastId
|
|
}
|
|
|
|
w.execute = async (task: string, llmConfig: LLMConfig) => {
|
|
if (typeof task !== 'string') throw new Error('Task must be a string')
|
|
if (task.trim().length === 0) throw new Error('Task cannot be empty')
|
|
if (!llmConfig) throw new Error('LLM config is required')
|
|
if (!llmConfig.baseURL) throw new Error('LLM config must have a baseURL')
|
|
if (!llmConfig.apiKey) throw new Error('LLM config must have an apiKey')
|
|
if (!llmConfig.model) throw new Error('LLM config must have a model')
|
|
|
|
const id = getId()
|
|
|
|
const promise = new Promise((resolve, reject) => {
|
|
function handleMessage(e: MessageEvent) {
|
|
const data = e.data
|
|
if (typeof data !== 'object' || data === null) return
|
|
if (data.channel !== 'PAGE_AGENT_EXT_RESPONSE') return
|
|
if (data.action !== 'execute_result') return
|
|
if (data.id !== id) return
|
|
|
|
window.removeEventListener('message', handleMessage)
|
|
|
|
if (data.error) {
|
|
reject(new Error(data.error))
|
|
} else {
|
|
resolve(data.payload)
|
|
}
|
|
}
|
|
|
|
window.addEventListener('message', handleMessage)
|
|
})
|
|
|
|
window.postMessage(
|
|
{
|
|
channel: 'PAGE_AGENT_EXT_REQUEST',
|
|
id,
|
|
action: 'execute',
|
|
payload: { task, llmConfig },
|
|
},
|
|
'*'
|
|
)
|
|
|
|
return promise
|
|
}
|
|
|
|
w.dispose = () => {
|
|
const id = getId()
|
|
|
|
window.postMessage(
|
|
{
|
|
channel: 'PAGE_AGENT_EXT_REQUEST',
|
|
id,
|
|
action: 'dispose',
|
|
},
|
|
'*'
|
|
)
|
|
}
|
|
})
|