132 lines
3.4 KiB
TypeScript
132 lines
3.4 KiB
TypeScript
import { redactToolDetail } from "../logging/redact.js";
|
|
import { shortenHomeInString } from "../utils.js";
|
|
import {
|
|
defaultTitle,
|
|
formatDetailKey,
|
|
normalizeToolName,
|
|
normalizeVerb,
|
|
resolveActionSpec,
|
|
resolveDetailFromKeys,
|
|
resolveReadDetail,
|
|
resolveWriteDetail,
|
|
type ToolDisplaySpec as ToolDisplaySpecBase,
|
|
} from "./tool-display-common.js";
|
|
import TOOL_DISPLAY_JSON from "./tool-display.json" with { type: "json" };
|
|
|
|
type ToolDisplaySpec = ToolDisplaySpecBase & {
|
|
emoji?: string;
|
|
};
|
|
|
|
type ToolDisplayConfig = {
|
|
version?: number;
|
|
fallback?: ToolDisplaySpec;
|
|
tools?: Record<string, ToolDisplaySpec>;
|
|
};
|
|
|
|
export type ToolDisplay = {
|
|
name: string;
|
|
emoji: string;
|
|
title: string;
|
|
label: string;
|
|
verb?: string;
|
|
detail?: string;
|
|
};
|
|
|
|
const TOOL_DISPLAY_CONFIG = TOOL_DISPLAY_JSON as ToolDisplayConfig;
|
|
const FALLBACK = TOOL_DISPLAY_CONFIG.fallback ?? { emoji: "🧩" };
|
|
const TOOL_MAP = TOOL_DISPLAY_CONFIG.tools ?? {};
|
|
const DETAIL_LABEL_OVERRIDES: Record<string, string> = {
|
|
agentId: "agent",
|
|
sessionKey: "session",
|
|
targetId: "target",
|
|
targetUrl: "url",
|
|
nodeId: "node",
|
|
requestId: "request",
|
|
messageId: "message",
|
|
threadId: "thread",
|
|
channelId: "channel",
|
|
guildId: "guild",
|
|
userId: "user",
|
|
runTimeoutSeconds: "timeout",
|
|
timeoutSeconds: "timeout",
|
|
includeTools: "tools",
|
|
pollQuestion: "poll",
|
|
maxChars: "max chars",
|
|
};
|
|
const MAX_DETAIL_ENTRIES = 8;
|
|
|
|
export function resolveToolDisplay(params: {
|
|
name?: string;
|
|
args?: unknown;
|
|
meta?: string;
|
|
}): ToolDisplay {
|
|
const name = normalizeToolName(params.name);
|
|
const key = name.toLowerCase();
|
|
const spec = TOOL_MAP[key];
|
|
const emoji = spec?.emoji ?? FALLBACK.emoji ?? "🧩";
|
|
const title = spec?.title ?? defaultTitle(name);
|
|
const label = spec?.label ?? title;
|
|
const actionRaw =
|
|
params.args && typeof params.args === "object"
|
|
? ((params.args as Record<string, unknown>).action as string | undefined)
|
|
: undefined;
|
|
const action = typeof actionRaw === "string" ? actionRaw.trim() : undefined;
|
|
const actionSpec = resolveActionSpec(spec, action);
|
|
const verb = normalizeVerb(actionSpec?.label ?? action);
|
|
|
|
let detail: string | undefined;
|
|
if (key === "read") {
|
|
detail = resolveReadDetail(params.args);
|
|
}
|
|
if (!detail && (key === "write" || key === "edit" || key === "attach")) {
|
|
detail = resolveWriteDetail(params.args);
|
|
}
|
|
|
|
const detailKeys = actionSpec?.detailKeys ?? spec?.detailKeys ?? FALLBACK.detailKeys ?? [];
|
|
if (!detail && detailKeys.length > 0) {
|
|
detail = resolveDetailFromKeys(params.args, detailKeys, {
|
|
mode: "summary",
|
|
maxEntries: MAX_DETAIL_ENTRIES,
|
|
formatKey: (raw) => formatDetailKey(raw, DETAIL_LABEL_OVERRIDES),
|
|
});
|
|
}
|
|
|
|
if (!detail && params.meta) {
|
|
detail = params.meta;
|
|
}
|
|
|
|
if (detail) {
|
|
detail = shortenHomeInString(detail);
|
|
}
|
|
|
|
return {
|
|
name,
|
|
emoji,
|
|
title,
|
|
label,
|
|
verb,
|
|
detail,
|
|
};
|
|
}
|
|
|
|
export function formatToolDetail(display: ToolDisplay): string | undefined {
|
|
const parts: string[] = [];
|
|
if (display.verb) {
|
|
parts.push(display.verb);
|
|
}
|
|
if (display.detail) {
|
|
parts.push(redactToolDetail(display.detail));
|
|
}
|
|
if (parts.length === 0) {
|
|
return undefined;
|
|
}
|
|
return parts.join(" · ");
|
|
}
|
|
|
|
export function formatToolSummary(display: ToolDisplay): string {
|
|
const detail = formatToolDetail(display);
|
|
return detail
|
|
? `${display.emoji} ${display.label}: ${detail}`
|
|
: `${display.emoji} ${display.label}`;
|
|
}
|