tool.ts•1.38 kB
import { McpError, ErrorCode } from "@modelcontextprotocol/sdk/types.js";
export interface ToolResponse {
content: {
type: "text";
text: string;
}[];
isError: boolean;
_meta?: Record<string, unknown>;
}
export type ToolParams = {
[key: string]: unknown;
};
export abstract class BaseTool<T extends ToolParams = ToolParams> {
abstract name: string;
abstract description: string;
abstract inputSchema: {
type: "object";
properties: Record<string, unknown>;
required?: string[];
};
abstract execute(params: T): Promise<ToolResponse>;
protected validateCollection(collection: unknown): string {
if (typeof collection !== "string") {
throw new McpError(
ErrorCode.InvalidRequest,
`Collection name must be a string, got ${typeof collection}`
);
}
return collection;
}
protected validateObject(
value: unknown,
name: string
): Record<string, unknown> {
if (!value || typeof value !== "object") {
throw new McpError(ErrorCode.InvalidRequest, `${name} must be an object`);
}
return value as Record<string, unknown>;
}
protected handleError(error: unknown): ToolResponse {
return {
content: [
{
type: "text",
text: error instanceof Error ? error.message : String(error),
},
],
isError: true,
};
}
}