fix: handle ESC properly on policy generate selection screen

src/handlers/gateway/policy/generate.screen.test.tsx

@@ -47,6 +47,22 @@ describe("gateway policy generate screen", () => {
     expect(frame).toContain("[enter] generate");
   });
 
+  test("escape returns the form through the picker to the Gateway menu", async () => {
+    const screen = renderScreen("/agentcore/gateway/policy/generate", {
+      core: coreWith(ENGINE_ARN),
+    });
+
+    await waitForText(screen.lastFrame, "checkout-gateway");
+    await screen.press("return");
+    await waitForText(screen.lastFrame, PLACEHOLDER);
+
+    await screen.press("escape");
+    await waitForText(screen.lastFrame, "choose a Gateway to generate a policy for");
+    await screen.press("escape");
+    await waitForText(screen.lastFrame, "manage AgentCore Gateways");
+    expect(screen.lastFrame()).toContain("policy");
+  });
+
   test("explains when the gateway has no engine and offers no prompt", async () => {
     const screen = renderScreen(`/agentcore/gateway/policy/generate/${GATEWAY_ID}`, {
       core: coreWith(undefined),

src/handlers/gateway/policy/screen.tsx

@@ -38,6 +38,7 @@ export function GatewayPolicyGenerateScreen(props: ScreenProps) {
         breadcrumb={["agentcore", "gateway", "policy", "generate"]}
         description="choose a Gateway to generate a policy for"
         onSelect={(id) => navigate(`${PICKER_PATH}/${encodeURIComponent(id)}`)}
+        onEscape={() => navigate("/agentcore/gateway")}
       />
     );
   }