From 26a4c549d7e630819d2c652dfc7328445b78b183 Mon Sep 17 00:00:00 2001
From: bracesproul <braceasproul@gmail.com>
Date: Tue, 8 Apr 2025 13:37:47 -0700
Subject: [PATCH 1/2] fix: Handle unknown interrupt objects

---
 src/components/thread/messages/ai.tsx         |  2 +-
 .../thread/messages/generic-interrupt.tsx     | 20 +++++++++++++------
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/src/components/thread/messages/ai.tsx b/src/components/thread/messages/ai.tsx
index 785978b..c761a6b 100644
--- a/src/components/thread/messages/ai.tsx
+++ b/src/components/thread/messages/ai.tsx
@@ -143,7 +143,7 @@ export function AssistantMessage({
             )}
           {threadInterrupt?.value &&
           !isAgentInboxInterruptSchema(threadInterrupt.value) &&
-          isLastMessage ? (
+          (isLastMessage || hasNoAIOrToolMessages) ? (
             <GenericInterruptView interrupt={threadInterrupt.value} />
           ) : null}
           <div
diff --git a/src/components/thread/messages/generic-interrupt.tsx b/src/components/thread/messages/generic-interrupt.tsx
index c4f069f..47b9b7c 100644
--- a/src/components/thread/messages/generic-interrupt.tsx
+++ b/src/components/thread/messages/generic-interrupt.tsx
@@ -9,7 +9,7 @@ function isComplexValue(value: any): boolean {
 export function GenericInterruptView({
   interrupt,
 }: {
-  interrupt: Record<string, any> | Record<string, any>[];
+  interrupt: unknown;
 }) {
   const [isExpanded, setIsExpanded] = useState(false);
 
@@ -39,23 +39,25 @@ export function GenericInterruptView({
   };
 
   // Process entries based on expanded state
-  const processEntries = () => {
+  const processEntries = (): unknown[] | string => {
     if (Array.isArray(interrupt)) {
       return isExpanded ? interrupt : interrupt.slice(0, 5);
-    } else {
+    } else if (typeof interrupt === "object" && interrupt !== null) {
       const entries = Object.entries(interrupt);
       if (!isExpanded && shouldTruncate) {
         // When collapsed, process each value to potentially truncate it
         return entries.map(([key, value]) => [key, truncateValue(value)]);
       }
       return entries;
+    } else {
+      return interrupt?.toString() ?? "";
     }
   };
 
   const displayEntries = processEntries();
 
   return (
-    <div className="border border-gray-200 rounded-lg overflow-hidden">
+    <div className="border border-gray-200 rounded-lg overflow-hidden max-w-sm sm:max-w-xl md:max-w-3xl">
       <div className="bg-gray-50 px-4 py-2 border-b border-gray-200">
         <div className="flex items-center justify-between gap-2 flex-wrap">
           <h3 className="font-medium text-gray-900">Human Interrupt</h3>
@@ -82,7 +84,7 @@ export function GenericInterruptView({
             >
               <table className="min-w-full divide-y divide-gray-200">
                 <tbody className="divide-y divide-gray-200">
-                  {displayEntries.map((item, argIdx) => {
+                  {Array.isArray(displayEntries) ? displayEntries.map((item, argIdx) => {
                     const [key, value] = Array.isArray(interrupt)
                       ? [argIdx.toString(), item]
                       : (item as [string, any]);
@@ -102,7 +104,13 @@ export function GenericInterruptView({
                         </td>
                       </tr>
                     );
-                  })}
+                  }) : (
+                    <tr>
+                      <td className="px-4 py-2 text-sm font-medium text-gray-900 whitespace-nowrap">
+                        {displayEntries}
+                      </td>
+                    </tr>
+                  )}
                 </tbody>
               </table>
             </motion.div>

From 3f25899bebaba8727949e9803a96012b0847bfbd Mon Sep 17 00:00:00 2001
From: bracesproul <braceasproul@gmail.com>
Date: Tue, 8 Apr 2025 14:12:09 -0700
Subject: [PATCH 2/2] cr

---
 src/components/thread/messages/ai.tsx         |  10 +-
 .../thread/messages/generic-interrupt.tsx     | 127 ++++++++++++++----
 2 files changed, 108 insertions(+), 29 deletions(-)

diff --git a/src/components/thread/messages/ai.tsx b/src/components/thread/messages/ai.tsx
index c761a6b..71402b1 100644
--- a/src/components/thread/messages/ai.tsx
+++ b/src/components/thread/messages/ai.tsx
@@ -12,7 +12,10 @@ import { Fragment } from "react/jsx-runtime";
 import { isAgentInboxInterruptSchema } from "@/lib/agent-inbox-interrupt";
 import { ThreadView } from "../agent-inbox";
 import { useQueryState, parseAsBoolean } from "nuqs";
-import { GenericInterruptView } from "./generic-interrupt";
+import {
+  GenericInterruptView,
+  ResumeGenericInterrupt,
+} from "./generic-interrupt";
 
 function CustomComponent({
   message,
@@ -144,7 +147,10 @@ export function AssistantMessage({
           {threadInterrupt?.value &&
           !isAgentInboxInterruptSchema(threadInterrupt.value) &&
           (isLastMessage || hasNoAIOrToolMessages) ? (
-            <GenericInterruptView interrupt={threadInterrupt.value} />
+            <div className="flex flex-col gap-2">
+              <GenericInterruptView interrupt={threadInterrupt.value} />
+              <ResumeGenericInterrupt />
+            </div>
           ) : null}
           <div
             className={cn(
diff --git a/src/components/thread/messages/generic-interrupt.tsx b/src/components/thread/messages/generic-interrupt.tsx
index 47b9b7c..360787a 100644
--- a/src/components/thread/messages/generic-interrupt.tsx
+++ b/src/components/thread/messages/generic-interrupt.tsx
@@ -1,16 +1,21 @@
-import { useState } from "react";
+import React, { useState } from "react";
 import { motion, AnimatePresence } from "framer-motion";
 import { ChevronDown, ChevronUp } from "lucide-react";
+import { Input } from "@/components/ui/input";
+import { Button } from "@/components/ui/button";
+import {
+  Tooltip,
+  TooltipContent,
+  TooltipProvider,
+  TooltipTrigger,
+} from "@/components/ui/tooltip";
+import { useStreamContext } from "@/providers/Stream";
 
 function isComplexValue(value: any): boolean {
   return Array.isArray(value) || (typeof value === "object" && value !== null);
 }
 
-export function GenericInterruptView({
-  interrupt,
-}: {
-  interrupt: unknown;
-}) {
+export function GenericInterruptView({ interrupt }: { interrupt: unknown }) {
   const [isExpanded, setIsExpanded] = useState(false);
 
   const contentStr = JSON.stringify(interrupt, null, 2);
@@ -84,27 +89,29 @@ export function GenericInterruptView({
             >
               <table className="min-w-full divide-y divide-gray-200">
                 <tbody className="divide-y divide-gray-200">
-                  {Array.isArray(displayEntries) ? displayEntries.map((item, argIdx) => {
-                    const [key, value] = Array.isArray(interrupt)
-                      ? [argIdx.toString(), item]
-                      : (item as [string, any]);
-                    return (
-                      <tr key={argIdx}>
-                        <td className="px-4 py-2 text-sm font-medium text-gray-900 whitespace-nowrap">
-                          {key}
-                        </td>
-                        <td className="px-4 py-2 text-sm text-gray-500">
-                          {isComplexValue(value) ? (
-                            <code className="bg-gray-50 rounded px-2 py-1 font-mono text-sm">
-                              {JSON.stringify(value, null, 2)}
-                            </code>
-                          ) : (
-                            String(value)
-                          )}
-                        </td>
-                      </tr>
-                    );
-                  }) : (
+                  {Array.isArray(displayEntries) ? (
+                    displayEntries.map((item, argIdx) => {
+                      const [key, value] = Array.isArray(interrupt)
+                        ? [argIdx.toString(), item]
+                        : (item as [string, any]);
+                      return (
+                        <tr key={argIdx}>
+                          <td className="px-4 py-2 text-sm font-medium text-gray-900 whitespace-nowrap">
+                            {key}
+                          </td>
+                          <td className="px-4 py-2 text-sm text-gray-500">
+                            {isComplexValue(value) ? (
+                              <code className="bg-gray-50 rounded px-2 py-1 font-mono text-sm">
+                                {JSON.stringify(value, null, 2)}
+                              </code>
+                            ) : (
+                              String(value)
+                            )}
+                          </td>
+                        </tr>
+                      );
+                    })
+                  ) : (
                     <tr>
                       <td className="px-4 py-2 text-sm font-medium text-gray-900 whitespace-nowrap">
                         {displayEntries}
@@ -132,3 +139,69 @@ export function GenericInterruptView({
     </div>
   );
 }
+
+const QuestionCircle = () => (
+  <div className="flex w-4 h-4 items-center justify-center rounded-full border-[1px] border-gray-300">
+    <p className="text-xs text-gray-500">?</p>
+  </div>
+);
+
+const Code = ({ children }: { children: React.ReactNode }) => (
+  <code className="text-orange-600 bg-gray-300 rounded px-1">{children}</code>
+);
+
+export function ResumeGenericInterrupt() {
+  const [resumeValue, setResumeValue] = useState("");
+  const stream = useStreamContext();
+
+  const handleResume = (
+    e:
+      | React.MouseEvent<HTMLButtonElement, MouseEvent>
+      | React.FormEvent<HTMLFormElement>,
+  ) => {
+    e.preventDefault();
+    stream.submit(null, {
+      command: {
+        resume: resumeValue,
+      },
+    });
+  };
+
+  return (
+    <div className="flex flex-col items-start justify-start gap-2 mt-3 max-w-sm sm:max-w-xl md:max-w-3xl">
+      <TooltipProvider>
+        <Tooltip delayDuration={200}>
+          <TooltipTrigger className="flex items-center justify-center gap-1">
+            <p className="text-xs text-gray-500">What&apos;s this</p>
+            <QuestionCircle />
+          </TooltipTrigger>
+          <TooltipContent className="flex flex-col gap-2 py-3">
+            <p>
+              This input offers a way to resume your graph from the point of the
+              interrupt.
+            </p>
+            <p>
+              The graph will be resumed using the <Code>Command</Code> API, with
+              the input you provide passed as a string value to the{" "}
+              <Code>resume</Code> key.
+            </p>
+          </TooltipContent>
+        </Tooltip>
+      </TooltipProvider>
+      <form
+        className="flex items-center gap-2 h-16 bg-muted rounded-2xl border w-full"
+        onSubmit={handleResume}
+      >
+        <Input
+          placeholder="Resume input"
+          value={resumeValue}
+          onChange={(e) => setResumeValue(e.target.value)}
+          className="bg-inherit border-none shadow-none"
+        />
+        <Button size="sm" variant="brand" className="mr-3" type="submit">
+          Resume
+        </Button>
+      </form>
+    </div>
+  );
+}