1823aae891Co-authored-by: Matt <77928207+mattzcarey@users.noreply.github.com> Co-authored-by: Konstantin Konstantinov <KKonstantinov@users.noreply.github.com> Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
33 lines
1.4 KiB
TypeScript
33 lines
1.4 KiB
TypeScript
/**
|
|
* Drives the prompts example: list, complete an argument, get a prompt.
|
|
*/
|
|
import { check, parseExampleArgs, siblingPath } from '@mcp-examples/shared';
|
|
import { Client, StreamableHTTPClientTransport } from '@modelcontextprotocol/client';
|
|
import { StdioClientTransport } from '@modelcontextprotocol/client/stdio';
|
|
|
|
const { transport, url, era } = parseExampleArgs();
|
|
|
|
const client = new Client(
|
|
{ name: 'prompts-example-client', version: '1.0.0' },
|
|
{ versionNegotiation: { mode: era === 'modern' ? 'auto' : 'legacy' } }
|
|
);
|
|
|
|
await (transport === 'stdio'
|
|
? client.connect(new StdioClientTransport({ command: 'npx', args: ['-y', 'tsx', siblingPath(import.meta.url, 'server.ts')] }))
|
|
: client.connect(new StreamableHTTPClientTransport(new URL(url))));
|
|
|
|
const list = await client.listPrompts();
|
|
check.ok(list.prompts.some(p => p.name === 'review-code'));
|
|
|
|
const completion = await client.complete({
|
|
ref: { type: 'ref/prompt', name: 'review-code' },
|
|
argument: { name: 'language', value: 'ty' }
|
|
});
|
|
check.ok(completion.completion.values.includes('typescript'));
|
|
|
|
const got = await client.getPrompt({ name: 'review-code', arguments: { language: 'rust', code: 'fn main() {}' } });
|
|
check.equal(got.messages.length, 1);
|
|
const text = got.messages[0]?.content.type === 'text' ? got.messages[0].content.text : '';
|
|
check.match(text, /Review this rust code/);
|
|
|
|
await client.close();
|