can1357
827841d19f
- Fixed Google provider schema sanitization to collapse anyOf/oneOf with const values into enum arrays. - Fixed const-to-enum conversion to infer type from const value when type is not specified. - Changed review finding priority format from numeric (0-3) to string labels (P0-P3). - Replaced Type.Union with Type.Literal patterns with StringEnum helper across tool schemas. - Added const modifier to StringEnum generic type parameter for improved TypeScript type inference. - Consolidated priority metadata into single PRIORITY_INFO record with exported getPriorityInfo function.
91 lines
2.6 KiB
TypeScript
91 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" });
|
|
});
|
|
}
|