From 8fcc204f28830fdb390967d7df65454ba808881f Mon Sep 17 00:00:00 2001
From: Lily Delalande <119957291+lily-de@users.noreply.github.com>
Date: Tue, 6 May 2025 12:23:56 -0400
Subject: [PATCH] cxt-mgmt: only two retries for summarization and move to prod
 (#2440)

---
 ui/desktop/src/components/ChatView.tsx        |  26 ++---
 .../ContextLengthExceededHandler.tsx          | 104 ++++++++++++++----
 2 files changed, 92 insertions(+), 38 deletions(-)

diff --git a/ui/desktop/src/components/ChatView.tsx b/ui/desktop/src/components/ChatView.tsx
index 6ae0ffbe..4d7636f4 100644
--- a/ui/desktop/src/components/ChatView.tsx
+++ b/ui/desktop/src/components/ChatView.tsx
@@ -524,9 +524,8 @@ function ChatContent({
                     <UserMessage message={message} />
                   ) : (
                     <>
-                      {/* Only render GooseMessage if it's not a CLE message (and we are not in alpha mode) */}
-                      {process.env.NODE_ENV === 'development' &&
-                      hasContextLengthExceededContent(message) ? (
+                      {/* Only render GooseMessage if it's not a CLE message */}
+                      {hasContextLengthExceededContent(message) ? (
                         <ContextLengthExceededHandler
                           messages={messages}
                           messageId={message.id ?? message.created.toString()}
@@ -593,17 +592,16 @@ function ChatContent({
       </Card>
 
       {showGame && <FlappyGoose onClose={() => setShowGame(false)} />}
-      {process.env.NODE_ENV === 'development' && (
-        <SessionSummaryModal
-          isOpen={isSummaryModalOpen}
-          onClose={closeSummaryModal}
-          onSave={(editedContent) => {
-            updateSummary(editedContent);
-            closeSummaryModal();
-          }}
-          summaryContent={summaryContent}
-        />
-      )}
+
+      <SessionSummaryModal
+        isOpen={isSummaryModalOpen}
+        onClose={closeSummaryModal}
+        onSave={(editedContent) => {
+          updateSummary(editedContent);
+          closeSummaryModal();
+        }}
+        summaryContent={summaryContent}
+      />
     </div>
   );
 }
diff --git a/ui/desktop/src/components/context_management/ContextLengthExceededHandler.tsx b/ui/desktop/src/components/context_management/ContextLengthExceededHandler.tsx
index 1681c86c..484192a9 100644
--- a/ui/desktop/src/components/context_management/ContextLengthExceededHandler.tsx
+++ b/ui/desktop/src/components/context_management/ContextLengthExceededHandler.tsx
@@ -24,6 +24,7 @@ export const ContextLengthExceededHandler: React.FC<ContextLengthExceededHandler
   } = useChatContextManager();
 
   const [hasFetchStarted, setHasFetchStarted] = useState(false);
+  const [retryCount, setRetryCount] = useState(0);
 
   // Find the relevant message to check if it's the latest
   const isCurrentMessageLatest =
@@ -71,6 +72,11 @@ export const ContextLengthExceededHandler: React.FC<ContextLengthExceededHandler
   const handleRetry = () => {
     if (!shouldAllowSummaryInteraction) return;
 
+    // Only increment retry counter if there's an error
+    if (errorLoadingSummary) {
+      setRetryCount((prevCount) => prevCount + 1);
+    }
+
     // Reset states for retry
     setHasFetchStarted(false);
     fetchStartedRef.current = false;
@@ -79,7 +85,78 @@ export const ContextLengthExceededHandler: React.FC<ContextLengthExceededHandler
     triggerContextLengthExceeded();
   };
 
+  // Function to open a new goose window
+  const openNewSession = () => {
+    try {
+      // Use the workingDir from props directly without reassignment to avoid TypeScript error
+      const sessionWorkingDir = window.appConfig?.get('GOOSE_WORKING_DIR') || workingDir;
+      console.log(`Creating new chat window with working dir: ${sessionWorkingDir}`);
+      window.electron.createChatWindow(undefined, sessionWorkingDir as string);
+    } catch (error) {
+      console.error('Error creating new window:', error);
+      // Fallback to basic window.open if the electron API fails
+      window.open('/', '_blank');
+    }
+  };
+
   // Render the notification UI
+  const renderLoadingState = () => (
+    <div className="flex items-center text-xs text-gray-400">
+      <span className="mr-2">Preparing summary...</span>
+      <span className="animate-spin h-3 w-3 border-2 border-gray-400 rounded-full border-t-transparent"></span>
+    </div>
+  );
+
+  const renderFailedState = () => (
+    <>
+      <span className="text-xs text-gray-400">{`Your conversation has exceeded the model's context capacity`}</span>
+      <span className="text-xs text-gray-400">{`This conversation has too much information to continue. Extension data often takes up significant space.`}</span>
+      <button
+        onClick={openNewSession}
+        className="text-xs text-textStandard hover:text-textSubtle transition-colors mt-1 flex items-center"
+      >
+        Click here to start a new session
+      </button>
+    </>
+  );
+
+  const renderRetryState = () => (
+    <>
+      <span className="text-xs text-gray-400">{`Your conversation has exceeded the model's context capacity`}</span>
+      <button
+        onClick={handleRetry}
+        className="text-xs text-textStandard hover:text-textSubtle transition-colors mt-1 flex items-center"
+      >
+        Retry loading summary
+      </button>
+    </>
+  );
+
+  const renderSuccessState = () => (
+    <>
+      <span className="text-xs text-gray-400">{`Your conversation has exceeded the model's context capacity and a summary was prepared.`}</span>
+      <span className="text-xs text-gray-400">{`Messages above this line remain viewable but specific details are not included in active context.`}</span>
+      <button
+        onClick={openSummaryModal}
+        className="text-xs text-textStandard hover:text-textSubtle transition-colors mt-1 flex items-center"
+      >
+        View or edit summary (you may continue your conversation based on the summary)
+      </button>
+    </>
+  );
+
+  const renderContentState = () => {
+    if (!shouldAllowSummaryInteraction) {
+      return null;
+    }
+
+    if (errorLoadingSummary) {
+      return retryCount >= 2 ? renderFailedState() : renderRetryState();
+    }
+
+    return renderSuccessState();
+  };
+
   return (
     <div className="flex flex-col items-start mt-1 pl-4">
       {/* Horizontal line with text in the middle - shown regardless of loading state */}
@@ -88,30 +165,9 @@ export const ContextLengthExceededHandler: React.FC<ContextLengthExceededHandler
         <div className="flex-grow border-t border-gray-300"></div>
       </div>
 
-      {isLoadingSummary && shouldAllowSummaryInteraction ? (
-        // Show loading indicator during loading state
-        <div className="flex items-center text-xs text-gray-400">
-          <span className="mr-2">Preparing summary...</span>
-          <span className="animate-spin h-3 w-3 border-2 border-gray-400 rounded-full border-t-transparent"></span>
-        </div>
-      ) : (
-        // Show different UI based on whether it's already handled
-        <>
-          <span className="text-xs text-gray-400">{`Your conversation has exceeded the model's context capacity`}</span>
-          <span className="text-xs text-gray-400">{`Messages above this line remain viewable but are not included in the active context`}</span>
-          {/* Only show the button if its last message */}
-          {shouldAllowSummaryInteraction && (
-            <button
-              onClick={() => (errorLoadingSummary ? handleRetry() : openSummaryModal())}
-              className="text-xs text-textStandard hover:text-textSubtle transition-colors mt-1 flex items-center"
-            >
-              {errorLoadingSummary
-                ? 'Retry loading summary'
-                : 'View or edit summary (you may continue your conversation based on the summary)'}
-            </button>
-          )}
-        </>
-      )}
+      {isLoadingSummary && shouldAllowSummaryInteraction
+        ? renderLoadingState()
+        : renderContentState()}
     </div>
   );
 };