fix(ai-help): handle invalid chat ids correctly

client/src/plus/ai-help/use-ai.ts

@@ -2,7 +2,7 @@
 // License: Apache 2.0 - https://github.com/supabase/supabase/blob/0f1254252f6b066e088a40617f239744e3a1e22b/LICENSE
 import type { OpenAI } from "openai";
 import { useCallback, useEffect, useReducer, useRef, useState } from "react";
-import { useSearchParams } from "react-router-dom";
+import { useNavigate, useSearchParams } from "react-router-dom";
 
 import { SSE } from "sse.js";
 import useSWR, { mutate } from "swr";
@@ -341,6 +341,7 @@
 export function useAiChat({
   messageTemplate = (message) => message,
 }: UseAiChatOptions = {}) {
+  const navigate = useNavigate();
   const gleanClick = useGleanClick();
   const eventSourceRef = useRef<SSE>();
 
@@ -407,8 +408,23 @@
     });
   }, [setSearchParams, chatId]);
 
+  const removeChatIdFromUrl = () => {
+    const searchParams = new URLSearchParams(window.location.search);
+
+    searchParams.delete("c");
+
+    navigate(
+      {
+        search: searchParams.toString(),
+      },
+      { replace: true }
+    );
+  };
+
   useEffect(() => {
     if (!isHistoryEnabled) {
+      // If we got a chat id passed in without history enabled, clear parameters from URL
+      removeChatIdFromUrl();
       return;
     }
     let timeoutID;
@@ -451,12 +467,19 @@
           }
         } catch (e) {
           setPreviousChatId(undefined);
-          setChatId(convId);
           setPath([]);
           dispatchState({
             type: "reset",
           });
-          handleError(e);
+          // If we got a 404 from the API, reset and remove the parameter from the URL,
+          // do not show an error.
+          if (e instanceof Error && e.message.startsWith("404")) {
+            setChatId(undefined);
+            removeChatIdFromUrl();
+          } else {
+            setChatId(convId);
+            handleError(e);
+          }
         }
         setIsHistoryLoading(false);
       };
@@ -466,7 +489,7 @@
     if (r) {
       reset();
     }
   }, [isHistoryEnabled, searchParams, chatId, reset, handleError]);
 
   useEffect(() => {
     if (remoteQuota !== undefined) {