75 lines
2.4 KiB
TypeScript
75 lines
2.4 KiB
TypeScript
/**
|
|
* Tab control tools for browser extension
|
|
*
|
|
* These tools allow the agent to manage multiple browser tabs:
|
|
* - open_new_tab: Open a new tab and set it as current
|
|
* - switch_to_tab: Switch to an existing tab
|
|
* - close_tab: Close a tab (optionally switch to another)
|
|
*/
|
|
import * as z from 'zod/v4'
|
|
|
|
import type { TabsController } from './TabsController'
|
|
|
|
/** Tool definition compatible with PageAgentCore customTools */
|
|
interface TabTool {
|
|
description: string
|
|
inputSchema: z.ZodType
|
|
execute: (input: unknown) => Promise<string>
|
|
}
|
|
|
|
/**
|
|
* Create tab control tools bound to a TabsManager instance.
|
|
* These tools are injected into PageAgentCore via customTools config.
|
|
*/
|
|
export function createTabTools(tabsController: TabsController): Record<string, TabTool> {
|
|
return {
|
|
open_new_tab: {
|
|
description:
|
|
'Open a new browser tab with the specified URL. The new tab becomes the current tab for all subsequent page operations.',
|
|
inputSchema: z.object({
|
|
url: z.string().describe('The URL to open in the new tab'),
|
|
}),
|
|
execute: async (input: unknown) => {
|
|
const { url } = input as { url: string }
|
|
try {
|
|
return await tabsController.openNewTab(url)
|
|
} catch (error) {
|
|
return `❌ Failed: ${error instanceof Error ? error.message : String(error)}`
|
|
}
|
|
},
|
|
},
|
|
|
|
switch_to_tab: {
|
|
description:
|
|
'Switch to an existing tab by its ID. After switching, all page operations will target the new current tab. You can only switch to tabs in the tab list shown in browser state.',
|
|
inputSchema: z.object({
|
|
tab_id: z.number().int().describe('The tab ID to switch to'),
|
|
}),
|
|
execute: async (input: unknown) => {
|
|
const { tab_id } = input as { tab_id: number }
|
|
try {
|
|
return await tabsController.switchToTab(tab_id)
|
|
} catch (error) {
|
|
return `❌ Failed: ${error instanceof Error ? error.message : String(error)}`
|
|
}
|
|
},
|
|
},
|
|
|
|
close_tab: {
|
|
description:
|
|
'Close a tab by its ID. Cannot close the initial tab. Optionally specify which tab to switch to after closing.',
|
|
inputSchema: z.object({
|
|
tab_id: z.number().int().describe('The tab ID to close'),
|
|
}),
|
|
execute: async (input: unknown) => {
|
|
const { tab_id } = input as { tab_id: number }
|
|
try {
|
|
return await tabsController.closeTab(tab_id)
|
|
} catch (error) {
|
|
return `❌ Failed: ${error instanceof Error ? error.message : String(error)}`
|
|
}
|
|
},
|
|
},
|
|
}
|
|
}
|