mirror of
https://github.com/modelcontextprotocol/servers.git
synced 2026-04-02 03:00:15 -04:00
97c6408f04
- add .idea/ for Jetbrains IDEs
* in everything.ts
- remove import of SetLevelRequestSchema
- remove logLevel var
- add sessionId var
- in startNotificationIntervals function
- add optional sid argument
- set sessionId to sid
- define messages to be sent, adding sessionId if present
- remove setRequestHandler call for SetLevelRequestSchema
- replace server.notification calls that sent "notifications/message" objects with calls to server.sendLoggingMessage, passing just the parameters and sessionId.
* In package.json & package-lock.json
- bump TS SDK version to 1.17.5
* In sse.ts, pass transport.sessionId to startNotificationIntervals call
* In stdio.ts
- destructure startNotificationIntervals from createServer call
- implement custom logging request handler and server.sendLoggingMessage implementation, as a
workaround for the fact that the SDK's automatic log level handling currently only tracks requested log level by session id. This will be fixed in a followup PR for the SDK
- call the startNotificationIntervals function after connecting the transport to the server
* In streamableHttp.ts
- destructure startNotificationIntervals from createServer call
- call startNotificationIntervals passing the transport.sessionId after connecting the transport to the server
57 lines
1.8 KiB
TypeScript
57 lines
1.8 KiB
TypeScript
import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
|
|
import express from "express";
|
|
import { createServer } from "./everything.js";
|
|
|
|
console.error('Starting SSE server...');
|
|
|
|
const app = express();
|
|
|
|
const transports: Map<string, SSEServerTransport> = new Map<string, SSEServerTransport>();
|
|
|
|
app.get("/sse", async (req, res) => {
|
|
let transport: SSEServerTransport;
|
|
const { server, cleanup, startNotificationIntervals } = createServer();
|
|
|
|
if (req?.query?.sessionId) {
|
|
const sessionId = (req?.query?.sessionId as string);
|
|
transport = transports.get(sessionId) as SSEServerTransport;
|
|
console.error("Client Reconnecting? This shouldn't happen; when client has a sessionId, GET /sse should not be called again.", transport.sessionId);
|
|
} else {
|
|
// Create and store transport for new session
|
|
transport = new SSEServerTransport("/message", res);
|
|
transports.set(transport.sessionId, transport);
|
|
|
|
// Connect server to transport
|
|
await server.connect(transport);
|
|
console.error("Client Connected: ", transport.sessionId);
|
|
|
|
// Start notification intervals after client connects
|
|
startNotificationIntervals(transport.sessionId);
|
|
|
|
// Handle close of connection
|
|
server.onclose = async () => {
|
|
console.error("Client Disconnected: ", transport.sessionId);
|
|
transports.delete(transport.sessionId);
|
|
await cleanup();
|
|
};
|
|
|
|
}
|
|
|
|
});
|
|
|
|
app.post("/message", async (req, res) => {
|
|
const sessionId = (req?.query?.sessionId as string);
|
|
const transport = transports.get(sessionId);
|
|
if (transport) {
|
|
console.error("Client Message from", sessionId);
|
|
await transport.handlePostMessage(req, res);
|
|
} else {
|
|
console.error(`No transport found for sessionId ${sessionId}`)
|
|
}
|
|
});
|
|
|
|
const PORT = process.env.PORT || 3001;
|
|
app.listen(PORT, () => {
|
|
console.error(`Server is running on port ${PORT}`);
|
|
});
|