31 lines
1.2 KiB
TypeScript
31 lines
1.2 KiB
TypeScript
/**
|
|
* Tool wrappers for extension-registered tools.
|
|
*
|
|
* These wrappers only adapt tool execution so extension tools receive the runner context.
|
|
* Tool call and tool result interception is handled by AgentSession via agent-core hooks.
|
|
*/
|
|
|
|
import type { AgentTool } from "@mariozechner/pi-agent-core";
|
|
import { wrapToolDefinition, wrapToolDefinitions } from "../tools/tool-definition-wrapper.js";
|
|
import type { ExtensionRunner } from "./runner.js";
|
|
import type { RegisteredTool } from "./types.js";
|
|
|
|
/**
|
|
* Wrap a RegisteredTool into an AgentTool.
|
|
* Uses the runner's createContext() for consistent context across tools and event handlers.
|
|
*/
|
|
export function wrapRegisteredTool(registeredTool: RegisteredTool, runner: ExtensionRunner): AgentTool {
|
|
return wrapToolDefinition(registeredTool.definition, () => runner.createContext());
|
|
}
|
|
|
|
/**
|
|
* Wrap all registered tools into AgentTools.
|
|
* Uses the runner's createContext() for consistent context across tools and event handlers.
|
|
*/
|
|
export function wrapRegisteredTools(registeredTools: RegisteredTool[], runner: ExtensionRunner): AgentTool[] {
|
|
return wrapToolDefinitions(
|
|
registeredTools.map((registeredTool) => registeredTool.definition),
|
|
() => runner.createContext(),
|
|
);
|
|
}
|