add examples with notifications

Author: ihrpr

src/examples/client/simpleStreamableHttp.ts

@@ -10,7 +10,8 @@ import {
   GetPromptRequest,
   GetPromptResultSchema,
   ListResourcesRequest,
-  ListResourcesResultSchema
+  ListResourcesResultSchema,
+  LoggingMessageNotificationSchema
 } from '../../types.js';
 
 async function main(): Promise<void> {
@@ -19,12 +20,17 @@ async function main(): Promise<void> {
     name: 'example-client',
     version: '1.0.0'
   });
+
   const transport = new StreamableHTTPClientTransport(
     new URL('http://localhost:3000/mcp')
   );
 
   // Connect the client using the transport and initialize the server
   await client.connect(transport);
+  client.setNotificationHandler(LoggingMessageNotificationSchema, (notification) => {
+    console.log(`Notification received: ${notification.params.level} - ${notification.params.data}`);
+  });
+
 
   console.log('Connected to MCP server');
   // List available tools
@@ -46,6 +52,23 @@ async function main(): Promise<void> {
   const greetResult = await client.request(greetRequest, CallToolResultSchema);
   console.log('Greeting result:', greetResult.content[0].text);
 
+  // Call the new 'multi-greet' tool
+  console.log('\nCalling multi-greet tool (with notifications)...');
+  const multiGreetRequest: CallToolRequest = {
+    method: 'tools/call',
+    params: {
+      name: 'multi-greet',
+      arguments: { name: 'MCP User' }
+    }
+  };
+  const multiGreetResult = await client.request(multiGreetRequest, CallToolResultSchema);
+  console.log('Multi-greet results:');
+  multiGreetResult.content.forEach(item => {
+    if (item.type === 'text') {
+      console.log(`- ${item.text}`);
+    }
+  });
+
   // List available prompts
   try {
     const promptsRequest: ListPromptsRequest = {

src/examples/server/jsonResponseStreamableHttp.ts

@@ -9,6 +9,10 @@ import { CallToolResult } from '../../types.js';
 const server = new McpServer({
   name: 'json-response-streamable-http-server',
   version: '1.0.0',
+}, {
+  capabilities: {
+    logging: {},
+  }
 });
 
 // Register a simple tool that returns a greeting
@@ -30,6 +34,46 @@ server.tool(
   }
 );
 
+// Register a tool that sends multiple greetings with notifications
+server.tool(
+  'multi-greet',
+  'A tool that sends different greetings with delays between them',
+  {
+    name: z.string().describe('Name to greet'),
+  },
+  async ({ name }, { sendNotification }): Promise<CallToolResult> => {
+    const sleep = (ms: number) => new Promise(resolve => setTimeout(resolve, ms));
+
+    await sendNotification({
+      method: "notifications/message",
+      params: { level: "debug", data: `Starting multi-greet for ${name}` }
+    });
+
+    await sleep(1000); // Wait 1 second before first greeting
+
+    await sendNotification({
+      method: "notifications/message",
+      params: { level: "info", data: `Sending first greeting to ${name}` }
+    });
+
+    await sleep(1000); // Wait another second before second greeting
+
+    await sendNotification({
+      method: "notifications/message",
+      params: { level: "info", data: `Sending second greeting to ${name}` }
+    });
+
+    return {
+      content: [
+        {
+          type: 'text',
+          text: `Good morning, ${name}!`,
+        }
+      ],
+    };
+  }
+);
+
 const app = express();
 app.use(express.json());
 

src/examples/server/simpleStreamableHttp.ts

@@ -9,7 +9,7 @@ import { CallToolResult, GetPromptResult, ReadResourceResult } from '../../types
 const server = new McpServer({
   name: 'simple-streamable-http-server',
   version: '1.0.0',
-});
+}, { capabilities: { logging: {} } });
 
 // Register a simple tool that returns a greeting
 server.tool(
@@ -30,6 +30,46 @@ server.tool(
   }
 );
 
+// Register a tool that sends multiple greetings with notifications
+server.tool(
+  'multi-greet',
+  'A tool that sends different greetings with delays between them',
+  {
+    name: z.string().describe('Name to greet'),
+  },
+  async ({ name }, { sendNotification }): Promise<CallToolResult> => {
+    const sleep = (ms: number) => new Promise(resolve => setTimeout(resolve, ms));
+
+    await sendNotification({
+      method: "notifications/message",
+      params: { level: "debug", data: `Starting multi-greet for ${name}` }
+    });
+
+    await sleep(1000); // Wait 1 second before first greeting
+
+    await sendNotification({
+      method: "notifications/message",
+      params: { level: "info", data: `Sending first greeting to ${name}` }
+    });
+
+    await sleep(1000); // Wait another second before second greeting
+
+    await sendNotification({
+      method: "notifications/message",
+      params: { level: "info", data: `Sending second greeting to ${name}` }
+    });
+
+    return {
+      content: [
+        {
+          type: 'text',
+          text: `Good morning, ${name}!`,
+        }
+      ],
+    };
+  }
+);
+
 // Register a simple prompt
 server.prompt(
   'greeting-template',
@@ -81,7 +121,7 @@ app.post('/mcp', async (req: Request, res: Response) => {
     // Check for existing session ID
     const sessionId = req.headers['mcp-session-id'] as string | undefined;
     let transport: StreamableHTTPServerTransport;
-    
+
     if (sessionId && transports[sessionId]) {
       // Reuse existing transport
       transport = transports[sessionId];
@@ -90,14 +130,14 @@ app.post('/mcp', async (req: Request, res: Response) => {
       transport = new StreamableHTTPServerTransport({
         sessionIdGenerator: () => randomUUID(),
       });
-      
+
       // Connect the transport to the MCP server BEFORE handling the request
       // so responses can flow back through the same transport
       await server.connect(transport);
-      
+
       // After handling the request, if we get a session ID back, store the transport
       await transport.handleRequest(req, res, req.body);
-      
+
       // Store the transport by session ID for future requests
       if (transport.sessionId) {
         transports[transport.sessionId] = transport;