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 57 58 59 60 61 62 63 | 79x 79x 79x 77x | import { Client } from '@modelcontextprotocol/sdk/client/index.js';
import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
import { version } from '../package.json';
import { USER_AGENT } from './consts';
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: {
// Version-bearing so /mcp requests are attributable to an exact SDK release.
// The transport sends no User-Agent of its own, and the plain USER_AGENT used on
// the other endpoints carries no version, which leaves tool listings anonymous.
'User-Agent': `${USER_AGENT}/${version}`,
...headers,
},
},
});
const client = new Client({
name: 'StackOne AI SDK',
version,
});
return {
client,
transport,
async [Symbol.asyncDispose]() {
await Promise.all([client.close(), transport.close()]);
},
};
}
|