fix(ai): retry DNS transport failures (#6946)

Fixes #6904
2026-07-22 12:26:19 +03:00
parent a5afc3f171
commit 33e40c3e15
3 changed files with 42 additions and 0 deletions
@@ -0,0 +1,28 @@
+import { fauxAssistantMessage } from "@earendil-works/pi-ai";
+import { describe, expect, it } from "vitest";
+import { createHarness } from "../harness.ts";
+
+const wrappedDnsLookupError =
+	"The pending stream has been canceled (caused by: getaddrinfo ENOTFOUND bedrock-runtime.us-east-1.amazonaws.com)";
+
+describe("issue #6904 DNS transport failure retry", () => {
+	it("retries a transient DNS lookup failure", async () => {
+		const harness = await createHarness({ settings: { retry: { enabled: true, maxRetries: 3, baseDelayMs: 1 } } });
+		try {
+			harness.setResponses([
+				fauxAssistantMessage("", { stopReason: "error", errorMessage: wrappedDnsLookupError }),
+				fauxAssistantMessage("recovered after DNS retry"),
+			]);
+
+			await harness.session.prompt("test");
+
+			expect(harness.faux.state.callCount).toBe(2);
+			expect(harness.eventsOfType("auto_retry_start").map((event) => event.errorMessage)).toEqual([
+				wrappedDnsLookupError,
+			]);
+			expect(harness.eventsOfType("auto_retry_end").map((event) => event.success)).toEqual([true]);
+		} finally {
+			harness.cleanup();
+		}
+	});
+});