fix: pass custom instructions to agent when rejecting tool permissions

Author: CaioDGallo

apps/code/src/renderer/components/action-selector/constants.ts

@@ -1,8 +1,20 @@
+import type { PermissionOption } from "@agentclientprotocol/sdk";
+
 export const OTHER_OPTION_ID = "_other";
 export const OTHER_OPTION_ID_ALT = "other";
 export const SUBMIT_OPTION_ID = "_submit";
 export const CANCEL_OPTION_ID = "cancel";
 export const OPTION_ID_PREFIX = "option_";
+export const REJECT_OPTION_ID = "reject";
+
+export function isRejectWithInstructionOption(
+  permissionOption: PermissionOption,
+): boolean {
+  return (
+    permissionOption.optionId === REJECT_OPTION_ID &&
+    permissionOption._meta?.customInput === true
+  );
+}
 
 export function isOtherOption(optionId: string): boolean {
   return optionId === OTHER_OPTION_ID || optionId === OTHER_OPTION_ID_ALT;

apps/code/src/renderer/features/sessions/components/SessionView.tsx

@@ -1,4 +1,7 @@
-import { isOtherOption } from "@components/action-selector/constants";
+import {
+  isOtherOption,
+  isRejectWithInstructionOption,
+} from "@components/action-selector/constants";
 import { PermissionSelector } from "@components/permissions/PermissionSelector";
 import {
   MessageEditor,
@@ -252,7 +255,10 @@ export function SessionView({
       }
 
       if (customInput) {
-        if (isOtherOption(optionId)) {
+        if (
+          isOtherOption(optionId) ||
+          (selectedOption && isRejectWithInstructionOption(selectedOption))
+        ) {
           await getSessionService().respondToPermission(
             taskId,
             firstPendingPermission.toolCallId,

packages/agent/src/adapters/claude/permissions/permission-handlers.ts

@@ -388,12 +388,14 @@ async function handleDefaultPermissionFlow(
       updatedInput: toolInput as Record<string, unknown>,
     };
   } else {
-    const message = "User refused permission to run tool";
+    const feedback = (
+      response._meta?.customInput as string | undefined
+    )?.trim();
+    const message = feedback
+      ? `User refused permission to run tool with feedback: ${feedback}`
+      : "User refused permission to run tool";
     await emitToolDenial(context, message);
-    return {
-      behavior: "deny",
-      message,
-    };
+    return { behavior: "deny", message, interrupt: !feedback };
   }
 }