All files / src mcp-client.ts

100% Statements 4/4
100% Branches 0/0
100% Functions 2/2
100% Lines 4/4

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56                                                                        16x           16x         16x       14x        
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
import { version } from '../package.json';
 
interface MCPClientOptions {
	baseUrl: string;
	headers?: Record<string, string>;
}
 
interface MCPClient {
	/** underlying MCP client */
	client: Client;
 
	/** underlying transport */
	transport: StreamableHTTPClientTransport;
 
	/** cleanup client and transport */
	[Symbol.asyncDispose](): Promise<void>;
}
 
/**
 * Create a Model Context Protocol (MCP) client.
 *
 * @example
 * ```ts
 * import { createMCPClient } from '@stackone/ai';
 *
 * await using clients = await createMCPClient({
 *   baseUrl: 'https://api.modelcontextprotocol.org',
 *   headers: {
 *     'Authorization': 'Bearer YOUR_API_KEY',
 *   },
 * });
 * ```
 */
export async function createMCPClient({ baseUrl, headers }: MCPClientOptions): Promise<MCPClient> {
	const transport = new StreamableHTTPClientTransport(new URL(baseUrl), {
		requestInit: {
			headers,
		},
	});
 
	const client = new Client({
		name: 'StackOne AI SDK',
		version,
	});
 
	return {
		client,
		transport,
		async [Symbol.asyncDispose]() {
			await Promise.all([client.close(), transport.close()]);
		},
	};
}