mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-15 13:03:42 +00:00
b129154cc8
- Changed ToolRenderer return type from TemplateResult to ToolRenderResult
- ToolRenderResult = { content: TemplateResult, isCustom: boolean }
- isCustom: true = no card wrapper, false = wrap in card
- Updated all existing tool renderers to return new format
- Updated Messages.ts to handle custom rendering
This enables tools to render without default card chrome when needed.
29 lines
1,006 B
TypeScript
29 lines
1,006 B
TypeScript
import type { ToolResultMessage } from "@mariozechner/pi-ai";
|
|
import "./javascript-repl.js"; // Auto-registers the renderer
|
|
import { getToolRenderer, registerToolRenderer } from "./renderer-registry.js";
|
|
import { BashRenderer } from "./renderers/BashRenderer.js";
|
|
import { DefaultRenderer } from "./renderers/DefaultRenderer.js";
|
|
import type { ToolRenderResult } from "./types.js";
|
|
|
|
// Register all built-in tool renderers
|
|
registerToolRenderer("bash", new BashRenderer());
|
|
|
|
const defaultRenderer = new DefaultRenderer();
|
|
|
|
/**
|
|
* Render tool - unified function that handles params, result, and streaming state
|
|
*/
|
|
export function renderTool(
|
|
toolName: string,
|
|
params: any | undefined,
|
|
result: ToolResultMessage | undefined,
|
|
isStreaming?: boolean,
|
|
): ToolRenderResult {
|
|
const renderer = getToolRenderer(toolName);
|
|
if (renderer) {
|
|
return renderer.render(params, result, isStreaming);
|
|
}
|
|
return defaultRenderer.render(params, result, isStreaming);
|
|
}
|
|
|
|
export { getToolRenderer, registerToolRenderer };
|