Dynamic title in problem set initial message

Author: jonkafton

src/components/AiChat/AiChat.stories.tsx

@@ -132,7 +132,7 @@ export const AssignmentSelection: Story = {
     problemSetInitialMessages: [
       {
         role: "assistant",
-        content: "Which question are you working on?",
+        content: "Can I help you with any of the problems in <title>?",
       },
     ],
   },

src/components/AiChat/AiChat.tsx

@@ -306,7 +306,11 @@ const AiChatDisplay: FC<AiChatDisplayProps> = ({
     if (problemSetInitialMessages) {
       setMessages(
         problemSetInitialMessages.map((message, i) => ({
-          ...message,
+          content: message.content?.replace(
+            "<title>",
+            event.target.value as string,
+          ),
+          role: message.role,
           id: `initial-${i}`,
         })),
       )

src/components/AiChat/types.ts

@@ -132,6 +132,7 @@ type AiChatDisplayProps = {
 
   /**
    * Initial messages to display on problem set selection.
+   * Occurrences of "<title>" in the content will be replaced with the problem set title.
    */
   problemSetInitialMessages?: Omit<AiChatMessage, "id">[]