From a0e9d7f5f316e715681bf97ee161afabd4591b02 Mon Sep 17 00:00:00 2001
From: bracesproul <braceasproul@gmail.com>
Date: Mon, 17 Mar 2025 14:09:04 -0700
Subject: [PATCH] fix: Allow for human interrupt to be passed without array

---
 .../components/thread-actions-view.tsx        | 19 ++++++++++-------
 src/components/thread/agent-inbox/index.tsx   |  7 ++++---
 src/components/thread/messages/ai.tsx         |  2 +-
 src/lib/agent-inbox-interrupt.ts              | 21 ++++++++++---------
 4 files changed, 27 insertions(+), 22 deletions(-)

diff --git a/src/components/thread/agent-inbox/components/thread-actions-view.tsx b/src/components/thread/agent-inbox/components/thread-actions-view.tsx
index a0fe53a..096f5e7 100644
--- a/src/components/thread/agent-inbox/components/thread-actions-view.tsx
+++ b/src/components/thread/agent-inbox/components/thread-actions-view.tsx
@@ -100,6 +100,7 @@ export function ThreadActionsView({
 
   const threadTitle = interrupt.action_request.action || "Unknown";
   const actionsDisabled = loading || streaming;
+  const ignoreAllowed = interrupt.config.allow_ignore;
 
   return (
     <div className="flex flex-col min-h-full w-full gap-9">
@@ -138,14 +139,16 @@ export function ThreadActionsView({
         >
           Mark as Resolved
         </Button>
-        <Button
-          variant="outline"
-          className="text-gray-800 border-gray-500 font-normal bg-white"
-          onClick={handleIgnore}
-          disabled={actionsDisabled}
-        >
-          Ignore
-        </Button>
+        {ignoreAllowed && (
+          <Button
+            variant="outline"
+            className="text-gray-800 border-gray-500 font-normal bg-white"
+            onClick={handleIgnore}
+            disabled={actionsDisabled}
+          >
+            Ignore
+          </Button>
+        )}
       </div>
 
       {/* Actions */}
diff --git a/src/components/thread/agent-inbox/index.tsx b/src/components/thread/agent-inbox/index.tsx
index 141bba1..2bbd614 100644
--- a/src/components/thread/agent-inbox/index.tsx
+++ b/src/components/thread/agent-inbox/index.tsx
@@ -5,10 +5,11 @@ import { HumanInterrupt } from "@langchain/langgraph/prebuilt";
 import { useStreamContext } from "@/providers/Stream";
 
 interface ThreadViewProps {
-  interrupt: HumanInterrupt;
+  interrupt: HumanInterrupt | HumanInterrupt[];
 }
 
 export function ThreadView({ interrupt }: ThreadViewProps) {
+  const interruptObj = Array.isArray(interrupt) ? interrupt[0] : interrupt;
   const thread = useStreamContext();
   const [showDescription, setShowDescription] = useState(false);
   const [showState, setShowState] = useState(false);
@@ -39,13 +40,13 @@ export function ThreadView({ interrupt }: ThreadViewProps) {
       {showSidePanel ? (
         <StateView
           handleShowSidePanel={handleShowSidePanel}
-          description={interrupt.description}
+          description={interruptObj.description}
           values={thread.values}
           view={showState ? "state" : "description"}
         />
       ) : (
         <ThreadActionsView
-          interrupt={interrupt}
+          interrupt={interruptObj}
           handleShowSidePanel={handleShowSidePanel}
           showState={showState}
           showDescription={showDescription}
diff --git a/src/components/thread/messages/ai.tsx b/src/components/thread/messages/ai.tsx
index 4f90d87..6799d9e 100644
--- a/src/components/thread/messages/ai.tsx
+++ b/src/components/thread/messages/ai.tsx
@@ -131,7 +131,7 @@ export function AssistantMessage({
 
           <CustomComponent message={message} thread={thread} />
           {isAgentInboxInterruptSchema(interrupt?.value) && isLastMessage && (
-            <ThreadView interrupt={interrupt.value[0]} />
+            <ThreadView interrupt={interrupt.value} />
           )}
           <div
             className={cn(
diff --git a/src/lib/agent-inbox-interrupt.ts b/src/lib/agent-inbox-interrupt.ts
index 6b1792a..01c7925 100644
--- a/src/lib/agent-inbox-interrupt.ts
+++ b/src/lib/agent-inbox-interrupt.ts
@@ -2,16 +2,17 @@ import { HumanInterrupt } from "@langchain/langgraph/prebuilt";
 
 export function isAgentInboxInterruptSchema(
   value: unknown,
-): value is HumanInterrupt[] {
+): value is HumanInterrupt | HumanInterrupt[] {
+  const valueAsObject = Array.isArray(value) ? value[0] : value;
   return (
-    Array.isArray(value) &&
-    "action_request" in value[0] &&
-    typeof value[0].action_request === "object" &&
-    "config" in value[0] &&
-    typeof value[0].config === "object" &&
-    "allow_respond" in value[0].config &&
-    "allow_accept" in value[0].config &&
-    "allow_edit" in value[0].config &&
-    "allow_ignore" in value[0].config
+    valueAsObject &&
+    "action_request" in valueAsObject &&
+    typeof valueAsObject.action_request === "object" &&
+    "config" in valueAsObject &&
+    typeof valueAsObject.config === "object" &&
+    "allow_respond" in valueAsObject.config &&
+    "allow_accept" in valueAsObject.config &&
+    "allow_edit" in valueAsObject.config &&
+    "allow_ignore" in valueAsObject.config
   );
 }