can1357
779ca4872b
- Removed Prettier configuration files (.prettierignore and .prettierrc) and migrated formatting to Biome. - Updated Biome configuration from version 2.3.11 to 2.3.12 and changed arrowParentheses rule from 'always' to 'asNeeded'. - Pinned @biomejs/biome dependency to exact version 2.3.12 in package.json and bun.lock. - Applied consistent arrow function formatting across 489 files by removing unnecessary parentheses around single parameters. - Removed blank lines after comment blocks and reorganized imports for consistency across the codebase.
90 lines
2.6 KiB
TypeScript
90 lines
2.6 KiB
TypeScript
/**
|
|
* API Demo Extension
|
|
*
|
|
* Demonstrates using ExtensionAPI's logger, typebox, and pi module access.
|
|
* These features are now exposed directly on the ExtensionAPI, matching
|
|
* the CustomToolAPI interface.
|
|
*/
|
|
import type { ExtensionAPI } from "@oh-my-pi/pi-coding-agent";
|
|
|
|
export default function (pi: ExtensionAPI) {
|
|
// 1. Access TypeBox directly from pi.typebox (no separate import needed)
|
|
const { Type } = pi.typebox;
|
|
|
|
// 2. Access the logger for debugging
|
|
pi.logger.debug("API demo extension loaded");
|
|
|
|
// 3. Register a tool that uses all three API features
|
|
// Import StringEnum from typebox helpers
|
|
const { StringEnum } = pi.pi;
|
|
|
|
pi.registerTool({
|
|
name: "api_demo",
|
|
label: "API Demo",
|
|
description: "Demonstrates ExtensionAPI capabilities: logger, typebox, and pi module access",
|
|
parameters: Type.Object({
|
|
message: Type.String({ description: "Test message" }),
|
|
logLevel: Type.Optional(
|
|
StringEnum(["error", "warn", "debug"], {
|
|
description: "Log level to use",
|
|
default: "debug",
|
|
}),
|
|
),
|
|
}),
|
|
|
|
async execute(_toolCallId, params, _onUpdate, ctx, _signal) {
|
|
const { message, logLevel = "debug" } = params as { message: string; logLevel?: "error" | "warn" | "debug" };
|
|
|
|
// Use logger at specified level
|
|
pi.logger[logLevel]("API demo tool executed", { message, logLevel });
|
|
|
|
// Access pi module utilities
|
|
const { logger: piLogger } = pi.pi;
|
|
piLogger.debug("Accessed pi module from extension", { sessionFile: ctx.sessionManager.getSessionFile() });
|
|
|
|
// Get session information
|
|
const sessionInfo = `Session: ${ctx.sessionManager.getSessionFile()}`;
|
|
const modelInfo = ctx.model ? `Model: ${ctx.model.id}` : "Model: none";
|
|
|
|
return {
|
|
content: [
|
|
{
|
|
type: "text",
|
|
text: [
|
|
`API Demo Tool executed successfully!`,
|
|
``,
|
|
`Message: ${message}`,
|
|
`Log Level: ${logLevel}`,
|
|
``,
|
|
`Features demonstrated:`,
|
|
`1. ✓ Logger access via pi.logger`,
|
|
`2. ✓ TypeBox access via pi.typebox`,
|
|
`3. ✓ Pi module access via pi.pi`,
|
|
``,
|
|
`Context:`,
|
|
`- ${sessionInfo}`,
|
|
`- ${modelInfo}`,
|
|
`- CWD: ${ctx.cwd}`,
|
|
].join("\n"),
|
|
},
|
|
],
|
|
details: {
|
|
message,
|
|
logLevel,
|
|
sessionFile: ctx.sessionManager.getSessionFile(),
|
|
modelId: ctx.model?.id,
|
|
},
|
|
};
|
|
},
|
|
});
|
|
|
|
// Demonstrate event handling with logger
|
|
pi.on("session_start", async () => {
|
|
pi.logger.debug("Session started", { extension: "api-demo" });
|
|
});
|
|
|
|
pi.on("agent_start", async () => {
|
|
pi.logger.debug("Agent started", { extension: "api-demo" });
|
|
});
|
|
}
|