58 lines
1.4 KiB
TypeScript
58 lines
1.4 KiB
TypeScript
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
import { McpResponse } from "./types.js";
|
|
|
|
// JSON value type matching the client utils
|
|
type JsonValue =
|
|
| string
|
|
| number
|
|
| boolean
|
|
| null
|
|
| undefined
|
|
| JsonValue[]
|
|
| { [key: string]: JsonValue };
|
|
|
|
// List available prompts
|
|
export async function listPrompts(client: Client): Promise<McpResponse> {
|
|
try {
|
|
const response = await client.listPrompts();
|
|
return response;
|
|
} catch (error) {
|
|
throw new Error(
|
|
`Failed to list prompts: ${error instanceof Error ? error.message : String(error)}`,
|
|
);
|
|
}
|
|
}
|
|
|
|
// Get a prompt
|
|
export async function getPrompt(
|
|
client: Client,
|
|
name: string,
|
|
args?: Record<string, JsonValue>,
|
|
): Promise<McpResponse> {
|
|
try {
|
|
// Convert all arguments to strings for prompt arguments
|
|
const stringArgs: Record<string, string> = {};
|
|
if (args) {
|
|
for (const [key, value] of Object.entries(args)) {
|
|
if (typeof value === "string") {
|
|
stringArgs[key] = value;
|
|
} else if (value === null || value === undefined) {
|
|
stringArgs[key] = String(value);
|
|
} else {
|
|
stringArgs[key] = JSON.stringify(value);
|
|
}
|
|
}
|
|
}
|
|
|
|
const response = await client.getPrompt({
|
|
name,
|
|
arguments: stringArgs,
|
|
});
|
|
|
|
return response;
|
|
} catch (error) {
|
|
throw new Error(
|
|
`Failed to get prompt: ${error instanceof Error ? error.message : String(error)}`,
|
|
);
|
|
}
|
|
}
|