chore: update elicitation/sampling streaming for v2

Author: LucaButBoring

examples/client/src/simpleStreamableHttp.ts

@@ -16,6 +16,7 @@ import {
     ErrorCode,
     getDisplayName,
     GetPromptResultSchema,
+    InMemoryTaskStore,
     ListPromptsResultSchema,
     ListResourcesResultSchema,
     ListToolsResultSchema,
@@ -27,7 +28,6 @@ import {
     StreamableHTTPClientTransport
 } from '@modelcontextprotocol/client';
 import { Ajv } from 'ajv';
-import { InMemoryTaskStore } from '../../experimental/tasks/stores/in-memory.js';
 
 // Create readline interface for user input
 const readline = createInterface({

packages/server/src/experimental/tasks/server.ts

@@ -140,7 +140,7 @@ export class ExperimentalServerTasks<
         // These may appear even without tools/toolChoice in the current request when
         // a previous sampling request returned tool_use and this is a follow-up with results.
         if (params.messages.length > 0) {
-            const lastMessage = params.messages[params.messages.length - 1];
+            const lastMessage = params.messages[params.messages.length - 1]!;
             const lastContent = Array.isArray(lastMessage.content) ? lastMessage.content : [lastMessage.content];
             const hasToolResults = lastContent.some(c => c.type === 'tool_result');