fix: infinite initialization (#793)

Author: dazy-ds

packages/ragbits-chat/CHANGELOG.md

@@ -1,6 +1,7 @@
 # CHANGELOG
 
 ## Unreleased
+- Fix bug causing infinite initialization screen (#793)
 - Fix bug that caused messages to be sent when changing chat settings; simplify and harden history logic (#791)
 - Add `clear_message` event type allowing to reset whole message (#789)
 - Add usage component to UI with backend support (#786)

typescript/ui/__tests__/integration/intergation.test.tsx

@@ -1,4 +1,4 @@
-import React, { ComponentProps } from "react";
+import React from "react";
 import {
   describe,
   it,
@@ -37,18 +37,6 @@ import { createHistoryStore } from "../../src/core/stores/HistoryStore/historySt
 import { createStore } from "zustand";
 import { useHistoryStore } from "../../src/core/stores/HistoryStore/useHistoryStore";
 import { HistoryStore } from "../../src/types/history";
-import HistoryStoreContextProvider from "../../src/core/stores/HistoryStore/HistoryStoreContextProvider";
-
-vi.mock(
-  "../../src/core/stores/HistoryStore/HistoryStoreContextProvider.tsx",
-  () => ({
-    default: ({
-      children,
-    }: ComponentProps<typeof HistoryStoreContextProvider>) => (
-      <div data-testid="history-store-context-provider">{children}</div>
-    ),
-  }),
-);
 
 vi.mock("../../src/core/stores/HistoryStore/useHistoryStore", () => {
   return {
@@ -246,8 +234,13 @@ describe("Integration tests", () => {
 
         render(<WrappedInput />);
         const user = userEvent.setup();
-        const chatOptionsButton =
-          await screen.findByTestId("open-chat-options");
+        const chatOptionsButton = await screen.findByTestId(
+          "open-chat-options",
+          undefined,
+          {
+            timeout: 5000,
+          },
+        );
         await user.click(chatOptionsButton);
         const selectTrigger = await screen.findByLabelText("Language", {
           selector: "select",

typescript/ui/src/core/stores/HistoryStore/HistoryStoreContextProvider.tsx

@@ -56,7 +56,11 @@ function initializeStore(shouldStoreHistory: boolean, storeKey: string) {
     );
   }
 
-  return createStore(createHistoryStore);
+  // Manually set _hasHydrated when we don't use any storage
+  const store = createStore(createHistoryStore);
+  store.getState()._internal._setHasHydrated(true);
+
+  return store;
 }
 
 export default function HistoryStoreContextProvider({
@@ -82,7 +86,7 @@ export default function HistoryStoreContextProvider({
     [store],
   );
 
-  if (!hasHydrated) {
+  if (shouldStoreHistory && !hasHydrated) {
     return <InitializationScreen />;
   }