mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-15 19:05:11 +00:00
030788140a
- Remove setApiKey, resolveApiKey, and global apiKeys Map from stream.ts - Rename getApiKey to getApiKeyFromEnv (only checks env vars) - Remove OAuth storage layer (storage.ts deleted) - OAuth login/refresh functions now return credentials instead of saving - getOAuthApiKey/refreshOAuthToken now take credentials as params - Add test/oauth.ts helper for ai package tests - Simplify root npm run check (single biome + tsgo pass) - Remove redundant check scripts from most packages - Add web-ui and coding-agent examples to biome/tsgo includes coding-agent still has compile errors - needs refactoring for new API
83 lines
2.4 KiB
TypeScript
83 lines
2.4 KiB
TypeScript
/**
|
|
* Question Tool - Let the LLM ask the user a question with options
|
|
*/
|
|
|
|
import type { CustomAgentTool, CustomToolFactory } from "@mariozechner/pi-coding-agent";
|
|
import { Text } from "@mariozechner/pi-tui";
|
|
import { Type } from "@sinclair/typebox";
|
|
|
|
interface QuestionDetails {
|
|
question: string;
|
|
options: string[];
|
|
answer: string | null;
|
|
}
|
|
|
|
const QuestionParams = Type.Object({
|
|
question: Type.String({ description: "The question to ask the user" }),
|
|
options: Type.Array(Type.String(), { description: "Options for the user to choose from" }),
|
|
});
|
|
|
|
const factory: CustomToolFactory = (pi) => {
|
|
const tool: CustomAgentTool<typeof QuestionParams, QuestionDetails> = {
|
|
name: "question",
|
|
label: "Question",
|
|
description: "Ask the user a question and let them pick from options. Use when you need user input to proceed.",
|
|
parameters: QuestionParams,
|
|
|
|
async execute(_toolCallId, params) {
|
|
if (!pi.hasUI) {
|
|
return {
|
|
content: [{ type: "text", text: "Error: UI not available (running in non-interactive mode)" }],
|
|
details: { question: params.question, options: params.options, answer: null },
|
|
};
|
|
}
|
|
|
|
if (params.options.length === 0) {
|
|
return {
|
|
content: [{ type: "text", text: "Error: No options provided" }],
|
|
details: { question: params.question, options: [], answer: null },
|
|
};
|
|
}
|
|
|
|
const answer = await pi.ui.select(params.question, params.options);
|
|
|
|
if (answer === null) {
|
|
return {
|
|
content: [{ type: "text", text: "User cancelled the selection" }],
|
|
details: { question: params.question, options: params.options, answer: null },
|
|
};
|
|
}
|
|
|
|
return {
|
|
content: [{ type: "text", text: `User selected: ${answer}` }],
|
|
details: { question: params.question, options: params.options, answer },
|
|
};
|
|
},
|
|
|
|
renderCall(args, theme) {
|
|
let text = theme.fg("toolTitle", theme.bold("question ")) + theme.fg("muted", args.question);
|
|
if (args.options?.length) {
|
|
text += `\n${theme.fg("dim", ` Options: ${args.options.join(", ")}`)}`;
|
|
}
|
|
return new Text(text, 0, 0);
|
|
},
|
|
|
|
renderResult(result, _options, theme) {
|
|
const { details } = result;
|
|
if (!details) {
|
|
const text = result.content[0];
|
|
return new Text(text?.type === "text" ? text.text : "", 0, 0);
|
|
}
|
|
|
|
if (details.answer === null) {
|
|
return new Text(theme.fg("warning", "Cancelled"), 0, 0);
|
|
}
|
|
|
|
return new Text(theme.fg("success", "✓ ") + theme.fg("accent", details.answer), 0, 0);
|
|
},
|
|
};
|
|
|
|
return tool;
|
|
};
|
|
|
|
export default factory;
|