753be5d87f
- Added ToolSession interface and Bun Worker-based subagent execution replacing subprocess spawning. - Added BUILTIN_TOOLS registry and createTools factory function replacing individual tool exports. - Added toolNames parameter for filtering which built-in tools to include in agent sessions. - Removed deprecated tool exports (codingTools, readOnlyTools, createCodingTools, createReadOnlyTools). - Removed subagent custom tool example and question custom tool example.
2.3 KiB
2.3 KiB
Custom Tools Examples
Example custom tools for omp-coding-agent.
Examples
Each example uses the subdirectory/index.ts structure required for tool discovery.
hello/
Minimal example showing the basic structure of a custom tool.
todo/
Full-featured example demonstrating:
onSessionfor state reconstruction from session history- Custom
renderCallandrenderResult - Proper branching support via details storage
- State management without external files
Usage
# Test directly (can point to any .ts file)
omp --tool examples/custom-tools/todo/index.ts
# Or copy entire folder to tools directory for persistent use
cp -r todo ~/.omp/agent/tools/
Then in omp:
> add a todo "test custom tools"
> list todos
> toggle todo #1
> clear todos
Writing Custom Tools
See docs/custom-tools.md for full documentation.
Key Points
Factory pattern:
import { Type } from "@sinclair/typebox";
import { StringEnum } from "@mariozechner/pi-ai";
import { Text } from "@oh-my-pi/pi-tui";
import type { CustomToolFactory } from "@oh-my-pi/pi-coding-agent";
const factory: CustomToolFactory = (pi) => ({
name: "my_tool",
label: "My Tool",
description: "Tool description for LLM",
parameters: Type.Object({
action: StringEnum(["list", "add"] as const),
}),
// Called on session start/switch/branch/clear
onSession(event) {
// Reconstruct state from event.entries
},
async execute(toolCallId, params) {
return {
content: [{ type: "text", text: "Result" }],
details: {
/* for rendering and state reconstruction */
},
};
},
});
export default factory;
Custom rendering:
renderCall(args, theme) {
return new Text(
theme.fg("toolTitle", theme.bold("my_tool ")) + args.action,
0, 0 // No padding - Box handles it
);
},
renderResult(result, { expanded, isPartial }, theme) {
if (isPartial) {
return new Text(theme.fg("warning", "Working..."), 0, 0);
}
return new Text(theme.fg("success", "✓ Done"), 0, 0);
},
Use StringEnum for string parameters (required for Google API compatibility):
import { StringEnum } from "@mariozechner/pi-ai";
// Good
action: StringEnum(["list", "add"] as const);
// Bad - doesn't work with Google
action: Type.Union([Type.Literal("list"), Type.Literal("add")]);