Update @github/copilot to 1.0.55-5

dotnet/test/E2E/SessionE2ETests.cs

@@ -110,16 +110,19 @@ public async Task Should_Create_A_Session_With_Customized_SystemMessage_Config()
             }
         });
 
-        await session.SendAsync(new MessageOptions { Prompt = "Who are you?" });
-        var assistantMessage = await TestHelper.GetFinalAssistantMessageAsync(session);
-        Assert.NotNull(assistantMessage);
-
-        var traffic = await Ctx.GetExchangesAsync();
-        Assert.NotEmpty(traffic);
-        var systemMessage = GetSystemMessage(traffic[0]);
-        Assert.Contains(customTone, systemMessage);
-        Assert.Contains(appendedContent, systemMessage);
-        Assert.DoesNotContain("<code_change_instructions>", systemMessage);
+        try
+        {
+            await session.SendAsync(new MessageOptions { Prompt = "Who are you?" });
+            var traffic = await WaitForExchangesAsync();
+            var systemMessage = GetSystemMessage(traffic[0]);
+            Assert.Contains(customTone, systemMessage);
+            Assert.Contains(appendedContent, systemMessage);
+            Assert.DoesNotContain("<code_change_instructions>", systemMessage);
+        }
+        finally
+        {
+            await session.DisposeAsync();
+        }
     }
 
     [Fact]

go/internal/e2e/session_e2e_test.go

@@ -209,20 +209,15 @@ func TestSessionE2E(t *testing.T) {
 		if err != nil {
 			t.Fatalf("Failed to create session: %v", err)
 		}
+		t.Cleanup(func() { _ = session.Disconnect() })
 
-		_, err = session.SendAndWait(t.Context(), copilot.MessageOptions{Prompt: "Who are you?"})
+		_, err = session.Send(t.Context(), copilot.MessageOptions{Prompt: "Who are you?"})
 		if err != nil {
 			t.Fatalf("Failed to send message: %v", err)
 		}
 
 		// Validate the system message sent to the model
-		traffic, err := ctx.GetExchanges()
-		if err != nil {
-			t.Fatalf("Failed to get exchanges: %v", err)
-		}
-		if len(traffic) == 0 {
-			t.Fatal("Expected at least one exchange")
-		}
+		traffic := ctx.WaitForExchanges(t, 1)
 		systemMessage := getSystemMessage(traffic[0])
 		if !strings.Contains(systemMessage, customTone) {
 			t.Errorf("Expected system message to contain custom tone, got %q", systemMessage)