Fix sort messages (jupyterlab#975)

* fix sort messages * prettier
Marchlak · Oct 28, 2024 · 08a6c17 · 08a6c17
1 parent 3b8ff46
commit 08a6c17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/jupyter-ai/src/components/chat-messages.tsx b/packages/jupyter-ai/src/components/chat-messages.tsx
@@ -49,11 +49,11 @@ function sortMessages(
      */
 
     const aOriginTimestamp =
-      a.type === 'agent' && a.reply_to in timestampsById
+      'reply_to' in a && a.reply_to in timestampsById
         ? timestampsById[a.reply_to]
         : a.time;
     const bOriginTimestamp =
-      b.type === 'agent' && b.reply_to in timestampsById
+      'reply_to' in b && b.reply_to in timestampsById
         ? timestampsById[b.reply_to]
         : b.time;