fix(ai-help): send correct context when editing question

client/src/plus/ai-help/rust-types.ts

@@ -13,9 +13,9 @@ export interface AIHelpMeta {
   type: MetaType;
   chat_id: string;
   message_id: string;
-  parent_id?: string;
+  parent_id?: string | null;
   sources: RefDoc[];
-  quota?: AIHelpLimit;
+  quota?: AIHelpLimit | null;
   created_at: string;
 }
 

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

@@ -0,0 +1,203 @@
+import { MessageRole, apiDataToStorage, stateToMessagePath } from "./use-ai";
+import { AIHelpLog, MetaType } from "./rust-types";
+
+const data: AIHelpLog = {
+  chat_id: "a2befad4-be1e-4b01-8d09-e45bebfee490",
+  messages: [
+    {
+      metadata: {
+        type: MetaType.Metadata,
+        chat_id: "a2befad4-be1e-4b01-8d09-e45bebfee490",
+        message_id: "5d49b710-20ad-4371-a92f-a283c26e2a7f",
+        parent_id: null,
+        sources: [
+          {
+            url: "/en-US/docs/Learn/CSS/Howto/Center_an_item",
+            title: "How to center an item",
+          },
+          {
+            url: "/en-US/docs/Web/CSS/Layout_cookbook/Center_an_element",
+            title: "Center an element",
+          },
+          { url: "/en-US/docs/Learn/CSS/Howto/CSS_FAQ", title: "CSS FAQ" },
+          {
+            url: "/en-US/docs/Learn/CSS/Howto/Create_fancy_boxes",
+            title: "Create fancy boxes",
+          },
+          { url: "/en-US/docs/Web/CSS/place-content", title: "place-content" },
+        ],
+        quota: null,
+        created_at: "2024-02-13T17:05:51.171127Z",
+      },
+      user: { role: "user", content: "How to center a div with CSS?" },
+      assistant: {
+        role: "assistant",
+        content: "To center a `div` element ...",
+      },
+    },
+    {
+      metadata: {
+        type: MetaType.Metadata,
+        chat_id: "a2befad4-be1e-4b01-8d09-e45bebfee490",
+        message_id: "ba70e34a-9fc2-4e57-88f7-f4bb8c5a46ef",
+        parent_id: "5d49b710-20ad-4371-a92f-a283c26e2a7f",
+        sources: [
+          {
+            url: "/en-US/docs/Learn/CSS/Howto/Center_an_item",
+            title: "How to center an item",
+          },
+          {
+            url: "/en-US/docs/Web/CSS/Layout_cookbook/Center_an_element",
+            title: "Center an element",
+          },
+          {
+            url: "/en-US/docs/Web/CSS/vertical-align",
+            title: "vertical-align",
+          },
+          {
+            url: "/en-US/docs/Web/CSS/CSS_flexible_box_layout/Aligning_items_in_a_flex_container",
+            title: "Aligning items in a flex container",
+          },
+          { url: "/en-US/docs/Web/CSS/justify-self", title: "justify-self" },
+        ],
+        quota: null,
+        created_at: "2024-02-13T17:06:52.596985Z",
+      },
+      user: { role: "user", content: "How to center it vertically only?" },
+      assistant: {
+        role: "assistant",
+        content: "To center a `div` ...",
+      },
+    },
+    {
+      metadata: {
+        type: MetaType.Metadata,
+        chat_id: "a2befad4-be1e-4b01-8d09-e45bebfee490",
+        message_id: "e853350f-4c83-4cfd-ad7b-a33a13e8f53f",
+        parent_id: "5d49b710-20ad-4371-a92f-a283c26e2a7f",
+        sources: [
+          {
+            url: "/en-US/docs/Learn/CSS/Howto/Center_an_item",
+            title: "How to center an item",
+          },
+          {
+            url: "/en-US/docs/Web/CSS/Layout_cookbook/Center_an_element",
+            title: "Center an element",
+          },
+          {
+            url: "/en-US/docs/Web/CSS/CSS_flexible_box_layout/Aligning_items_in_a_flex_container",
+            title: "Aligning items in a flex container",
+          },
+          { url: "/en-US/docs/Web/CSS/text-align", title: "text-align" },
+          {
+            url: "/en-US/docs/Learn/CSS/Building_blocks/Writing_Modes_Tasks",
+            title: "Test your skills: Writing modes and logical properties",
+          },
+        ],
+        quota: null,
+        created_at: "2024-02-13T17:07:53.234230Z",
+      },
+      user: { role: "user", content: "How to center it horizontally only?" },
+      assistant: {
+        role: "assistant",
+        content: "To center a `div` vertically ...",
+      },
+    },
+    {
+      metadata: {
+        type: MetaType.Metadata,
+        chat_id: "a2befad4-be1e-4b01-8d09-e45bebfee490",
+        message_id: "e844b41e-1648-463a-875b-e7bfc2479c77",
+        parent_id: "5d49b710-20ad-4371-a92f-a283c26e2a7f",
+        sources: [
+          {
+            url: "/en-US/docs/Learn/Common_questions/Tools_and_setup/How_much_does_it_cost",
+            title: "How much does it cost to do something on the Web?",
+          },
+          { url: "/en-US/docs/Glossary/FPS", title: "Frame rate (FPS)" },
+          { url: "/en-US/docs/Glossary/FTU", title: "FTU" },
+          {
+            url: "/en-US/docs/Games/Tutorials/2D_breakout_game_Phaser/Player_paddle_and_controls",
+            title: "Player paddle and controls",
+          },
+          {
+            url: "/en-US/docs/Web/SVG/Namespaces_Crash_Course/Example",
+            title: "Example",
+          },
+        ],
+        quota: null,
+        created_at: "2024-02-14T08:52:42.178701Z",
+      },
+      user: { role: "user", content: "How much is the fish?" },
+      assistant: {
+        role: "assistant",
+        content:
+          "I'm sorry, but I can only provide assistance with web development topics...",
+      },
+    },
+    {
+      metadata: {
+        type: MetaType.Metadata,
+        chat_id: "a2befad4-be1e-4b01-8d09-e45bebfee490",
+        message_id: "c863991c-d44b-4c33-9725-06a5f0d9add9",
+        parent_id: "5d49b710-20ad-4371-a92f-a283c26e2a7f",
+        sources: [
+          {
+            url: "/en-US/docs/Learn/Common_questions/Tools_and_setup/How_much_does_it_cost",
+            title: "How much does it cost to do something on the Web?",
+          },
+          {
+            url: "/en-US/docs/Games/Publishing_games/Game_monetization",
+            title: "Game monetization",
+          },
+          { url: "/en-US/docs/Web/CSS/actual_value", title: "Actual value" },
+          {
+            url: "/en-US/docs/Web/Performance/Fundamentals",
+            title: "Performance fundamentals",
+          },
+          { url: "/en-US/docs/Glossary/CSS_pixel", title: "CSS pixel" },
+        ],
+        quota: null,
+        created_at: "2024-02-14T09:07:26.459750Z",
+      },
+      user: { role: "user", content: "How much really?" },
+      assistant: {
+        role: "assistant",
+        content: "To center a `div` horizontally and vertically with CSS...",
+      },
+    },
+  ],
+};
+
+const state = apiDataToStorage(data, data.chat_id);
+
+describe("Testing ai-help utility code", () => {
+  it("test stateToMessagePath with traverseWithDefault", async () => {
+    expect(stateToMessagePath(state.treeState!, [])).toEqual([]);
+    expect(stateToMessagePath(state.treeState!, [], true)).toHaveLength(4);
+
+    expect(stateToMessagePath(state.treeState!, [0, 0])).toHaveLength(4);
+    expect(stateToMessagePath(state.treeState!, [0, 1])).toHaveLength(4);
+    expect(stateToMessagePath(state.treeState!, [0, 2], true)).toHaveLength(4);
+    expect(stateToMessagePath(state.treeState!, [0], true)).toHaveLength(4);
+    expect(stateToMessagePath(state.treeState!, [0])).toHaveLength(2);
+
+    expect(
+      stateToMessagePath(state.treeState!, [0, 0])
+        .filter((x) => x?.role === MessageRole.User)
+        .map((x) => x.messageId)
+    ).toEqual([
+      "5d49b710-20ad-4371-a92f-a283c26e2a7f",
+      "ba70e34a-9fc2-4e57-88f7-f4bb8c5a46ef",
+    ]);
+
+    expect(
+      stateToMessagePath(state.treeState!, [0, 2])
+        .filter((x) => x?.role === MessageRole.User)
+        .map((x) => x.messageId)
+    ).toEqual([
+      "5d49b710-20ad-4371-a92f-a283c26e2a7f",
+      "e844b41e-1648-463a-875b-e7bfc2479c77",
+    ]);
+  });
+});