can1357
89 lines
2.5 KiB
TypeScript
89 lines
2.5 KiB
TypeScript
/**
|
|
* Registry for handling tool events from subprocess agents.
|
|
*
|
|
* Tools can register handlers to:
|
|
* - Extract structured data from their execution results
|
|
* - Trigger subprocess termination on completion
|
|
* - Provide custom rendering for realtime/final display
|
|
*/
|
|
import type { Component } from "@oh-my-pi/pi-tui";
|
|
import type { Theme } from "../modes/theme/theme";
|
|
|
|
/** Event from subprocess tool execution (parsed from JSONL) */
|
|
export interface SubprocessToolEvent {
|
|
toolName: string;
|
|
toolCallId: string;
|
|
args?: Record<string, unknown>;
|
|
result?: {
|
|
content: Array<{ type: string; text?: string }>;
|
|
details?: unknown;
|
|
};
|
|
isError?: boolean;
|
|
}
|
|
|
|
/** Handler for subprocess tool events */
|
|
export interface SubprocessToolHandler<TData = unknown> {
|
|
/**
|
|
* Extract structured data from tool result.
|
|
* Extracted data is accumulated in progress.extractedToolData[toolName][].
|
|
*/
|
|
extractData?: (event: SubprocessToolEvent) => TData | undefined;
|
|
|
|
/**
|
|
* Whether this tool's completion should terminate the subprocess.
|
|
* Return true to send SIGTERM after the tool completes.
|
|
*/
|
|
shouldTerminate?: (event: SubprocessToolEvent) => boolean;
|
|
|
|
/**
|
|
* Render a single data item inline during streaming progress.
|
|
* Called for each tool execution end event.
|
|
*/
|
|
renderInline?: (data: TData, theme: Theme) => Component;
|
|
|
|
/**
|
|
* Render accumulated data in the final result view.
|
|
* Called once with all accumulated data for this tool.
|
|
*/
|
|
renderFinal?: (allData: TData[], theme: Theme, expanded: boolean) => Component;
|
|
}
|
|
|
|
/** Registry for subprocess tool handlers */
|
|
class SubprocessToolRegistryImpl {
|
|
#handlers = new Map<string, SubprocessToolHandler>();
|
|
|
|
/**
|
|
* Register a handler for a tool's subprocess events.
|
|
*/
|
|
register<T>(toolName: string, handler: SubprocessToolHandler<T>): void {
|
|
this.#handlers.set(toolName, handler as SubprocessToolHandler);
|
|
}
|
|
|
|
/**
|
|
* Get the handler for a tool, if registered.
|
|
*/
|
|
getHandler(toolName: string): SubprocessToolHandler | undefined {
|
|
return this.#handlers.get(toolName);
|
|
}
|
|
|
|
/**
|
|
* Check if a tool has a registered handler.
|
|
*/
|
|
hasHandler(toolName: string): boolean {
|
|
return this.#handlers.has(toolName);
|
|
}
|
|
|
|
/**
|
|
* Get all registered tool names.
|
|
*/
|
|
getRegisteredTools(): string[] {
|
|
return Array.from(this.#handlers.keys());
|
|
}
|
|
}
|
|
|
|
/** Singleton registry instance */
|
|
export const subprocessToolRegistry = new SubprocessToolRegistryImpl();
|
|
|
|
/** Type helper for extracted tool data in progress/result */
|
|
export type ExtractedToolData = Record<string, unknown[]>;
|