Update tests and CLI
This commit is contained in:
parent
2b0599e163
commit
e9c64fbc53
1 changed files with 12 additions and 12 deletions
16
src/cli.ts
16
src/cli.ts
|
|
@ -31,17 +31,17 @@ async function runClient(url_or_command: string, args: string[]) {
|
|||
}
|
||||
|
||||
if (url?.protocol === "http:" || url?.protocol === "https:") {
|
||||
clientTransport = new SSEClientTransport();
|
||||
await clientTransport.connect(new URL(url_or_command));
|
||||
clientTransport = new SSEClientTransport(new URL(url_or_command));
|
||||
await clientTransport.start();
|
||||
} else if (url?.protocol === "ws:" || url?.protocol === "wss:") {
|
||||
clientTransport = new WebSocketClientTransport();
|
||||
await clientTransport.connect(new URL(url_or_command));
|
||||
clientTransport = new WebSocketClientTransport(new URL(url_or_command));
|
||||
await clientTransport.start();
|
||||
} else {
|
||||
clientTransport = new StdioClientTransport();
|
||||
await clientTransport.spawn({
|
||||
clientTransport = new StdioClientTransport({
|
||||
command: url_or_command,
|
||||
args,
|
||||
});
|
||||
await clientTransport.start();
|
||||
}
|
||||
|
||||
console.log("Connected to server.");
|
||||
|
|
@ -62,7 +62,7 @@ async function runServer(port: number | null) {
|
|||
app.get("/sse", async (req, res) => {
|
||||
console.log("Got new SSE connection");
|
||||
|
||||
const transport = new SSEServerTransport("/message");
|
||||
const transport = new SSEServerTransport("/message", res);
|
||||
const server = new Server({
|
||||
name: "mcp-typescript test server",
|
||||
version: "0.1.0",
|
||||
|
|
@ -75,7 +75,7 @@ async function runServer(port: number | null) {
|
|||
servers = servers.filter((s) => s !== server);
|
||||
};
|
||||
|
||||
await transport.connectSSE(req, res);
|
||||
await transport.start();
|
||||
await server.connect(transport);
|
||||
});
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue