fix: drain pending stream output after interrupt in ClaudeCodeSdk (#11)

- keep reading ReceiveResponseAsync during cancellation to flush buffered messages
  - trigger InterruptAsync from cancellation callback instead of aborting the reader token
  - stop disposing the client inside ClaudeSdkClient.InterruptAsync so buffered output is not lost to the next query

(cherry picked from commit 2f1d4ec9a0cc27ce5741a10d6ff6afd3ae277485)

Signed-off-by: zxyao145 <[email protected]>

Author: zxyao145

src/ClaudeCodeSdk.MAF/ClaudeCodeAIAgent.cs

@@ -100,11 +100,6 @@ protected override async Task<AgentResponse> RunCoreAsync(
 
             await foreach (var claudeMessage in asyncEnumMsgs)
             {
-                if (client != null && cancellationToken.IsCancellationRequested)
-                {
-                    await InterruptAsync(client);
-                }
-
                 if (claudeMessage is ResultMessage resultMessage)
                 {
                     usageDetails = resultMessage.ToUsageDetails();
@@ -150,7 +145,7 @@ protected override async IAsyncEnumerable<AgentResponseUpdate> RunCoreStreamingA
         if (!string.IsNullOrWhiteSpace(content))
         {
             var (asyncEnumMsgs, client) = await SendUserInput(claudeThread, content, cancellationToken);
-
+           
             if (client != null && cancellationToken.IsCancellationRequested)
             {
                 await InterruptAsync(client);
@@ -160,11 +155,6 @@ protected override async IAsyncEnumerable<AgentResponseUpdate> RunCoreStreamingA
             // Receive and yield responses
             await foreach (var claudeMessage in asyncEnumMsgs)
             {
-                if (client != null && cancellationToken.IsCancellationRequested)
-                {
-                    await InterruptAsync(client);
-                }
-
                 var update = claudeMessage.ToAgentRunResponseUpdate();
                 if (update != null)
                 {
@@ -203,15 +193,37 @@ private async Task InterruptAsync(ClaudeSdkClient client)
         }
         else
         {
-            client = await _clientManager.GetClientAsync(claudeThread, cancellationToken);
+            client = await _clientManager.GetClientAsync(claudeThread, CancellationToken.None);
 
             await client.QueryAsync(content,
                  sessionId: claudeThread.SessionId.ToString(),
-                 cancellationToken: cancellationToken);
+                 cancellationToken: CancellationToken.None);
 
-            asyncEnumMsgs = client.ReceiveResponseAsync(cancellationToken);
+            asyncEnumMsgs = client.ReceiveResponseAsync(CancellationToken.None);
         }
 
+        // 
+        var interruptRequested = 0;
+        using var cancellationRegistration = cancellationToken.Register(() =>
+        {
+            if (client == null) return;
+            if (Interlocked.Exchange(ref interruptRequested, 1) != 0)
+                return;
+
+            _ = Task.Run(async () =>
+            {
+                try
+                {
+                    await InterruptAsync(client);
+                }
+                catch (Exception ex)
+                {
+                    _logger?.LogDebug(ex, "Failed to interrupt Claude SDK client during streaming cancellation");
+                }
+            });
+        });
+
+
         return (asyncEnumMsgs, client);
     }
 

src/ClaudeCodeSdk/ClaudeSDKClient.cs

@@ -120,7 +120,7 @@ public async Task InterruptAsync(CancellationToken cancellationToken = default)
 
         ConnectStatus = ConnectStatus.DisConnecting;
         await _process.InterruptAsync();
-        await this.DisposeAsync();
+        ConnectStatus = ConnectStatus.DisConnected;
     }
 
     /// <summary>
@@ -171,4 +171,4 @@ public async ValueTask DisposeAsync()
         }
         GC.SuppressFinalize(this);
     }
-}
+}