52 lines
1.2 KiB
TypeScript
52 lines
1.2 KiB
TypeScript
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
import type { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
|
|
import { McpResponse } from "./types.js";
|
|
|
|
export const validLogLevels = [
|
|
"trace",
|
|
"debug",
|
|
"info",
|
|
"warn",
|
|
"error",
|
|
] as const;
|
|
|
|
export type LogLevel = (typeof validLogLevels)[number];
|
|
|
|
export async function connect(
|
|
client: Client,
|
|
transport: Transport,
|
|
): Promise<void> {
|
|
try {
|
|
await client.connect(transport);
|
|
} catch (error) {
|
|
throw new Error(
|
|
`Failed to connect to MCP server: ${error instanceof Error ? error.message : String(error)}`,
|
|
);
|
|
}
|
|
}
|
|
|
|
export async function disconnect(transport: Transport): Promise<void> {
|
|
try {
|
|
await transport.close();
|
|
} catch (error) {
|
|
throw new Error(
|
|
`Failed to disconnect from MCP server: ${error instanceof Error ? error.message : String(error)}`,
|
|
);
|
|
}
|
|
}
|
|
|
|
// Set logging level
|
|
export async function setLoggingLevel(
|
|
client: Client,
|
|
level: LogLevel,
|
|
): Promise<McpResponse> {
|
|
try {
|
|
const response = await client.setLoggingLevel(level as any);
|
|
return response;
|
|
} catch (error) {
|
|
throw new Error(
|
|
`Failed to set logging level: ${error instanceof Error ? error.message : String(error)}`,
|
|
);
|
|
}
|
|
}
|