test(e2e): stabilize MCP tool message flow

Author: yiliang114

integration-tests/sdk-typescript/mcp-server.test.ts

@@ -477,7 +477,8 @@ describe('MCP Server Integration (E2E)', () => {
   describe('MCP Tool Message Flow', () => {
     it('should receive proper message sequence for MCP tool usage', async () => {
       const q = query({
-        prompt: 'Use add to calculate 2 + 3',
+        prompt:
+          'Use the add tool to calculate 2 + 3. You must call the tool. Just give me the result.',
         options: {
           ...SHARED_TEST_OPTIONS,
           cwd: testDir,
@@ -500,10 +501,9 @@ describe('MCP Server Integration (E2E)', () => {
           messageTypes.push(message.type);
 
           if (isSDKAssistantMessage(message)) {
-            const toolUseBlocks = findToolUseBlocks(message);
+            const toolUseBlocks = findToolUseBlocks(message, MCP_ADD_TOOL);
             if (toolUseBlocks.length > 0) {
               foundToolUse = true;
-              expect(toolUseBlocks[0].name).toBe(MCP_ADD_TOOL);
               expect(toolUseBlocks[0].input).toBeDefined();
             }
           }