Merge branch 'master' into fs2

Author: williamstein

src/packages/frontend/chat/actions.ts

@@ -574,7 +574,29 @@ export class ChatActions extends Actions<ChatState> {
     return true;
   };
 
-  markThreadRead = (threadKey: string, count: number): boolean => {
+  setThreadPin = (threadKey: string, pinned: boolean): boolean => {
+    if (this.syncdb == null) {
+      return false;
+    }
+    const entry = this.getThreadRootDoc(threadKey);
+    if (entry == null) {
+      return false;
+    }
+    if (pinned) {
+      entry.doc.pin = true;
+    } else {
+      entry.doc.pin = false;
+    }
+    this.syncdb.set(entry.doc);
+    this.syncdb.commit();
+    return true;
+  };
+
+  markThreadRead = (
+    threadKey: string,
+    count: number,
+    commit = true,
+  ): boolean => {
     if (this.syncdb == null) {
       return false;
     }
@@ -588,7 +610,9 @@ export class ChatActions extends Actions<ChatState> {
     }
     entry.doc[`read-${account_id}`] = count;
     this.syncdb.set(entry.doc);
-    this.syncdb.commit();
+    if (commit) {
+      this.syncdb.commit();
+    }
     return true;
   };
 
@@ -752,30 +776,44 @@ export class ChatActions extends Actions<ChatState> {
    * This checks a thread of messages to see if it is a language model thread and if so, returns it.
    */
   isLanguageModelThread = (date?: Date): false | LanguageModel => {
-    if (date == null) {
+    if (date == null || this.store == null) {
       return false;
     }
-    const thread = this.getMessagesInThread(date.toISOString());
-    if (thread == null) {
+    const messages = this.store.get("messages");
+    if (messages == null) {
+      return false;
+    }
+    const rootMs =
+      getThreadRootDate({ date: date.valueOf(), messages }) || date.valueOf();
+    const entry = this.getThreadRootDoc(`${rootMs}`);
+    const rootMessage = entry?.message;
+    if (rootMessage == null) {
       return false;
     }
 
-    // We deliberately start at the last most recent message.
-    // Why? If we use the LLM regenerate dropdown button to change the LLM, we want to keep it.
-    for (const message of thread.reverse()) {
-      const lastHistory = message.get("history")?.first();
-      // this must be an invalid message, because there is no history
-      if (lastHistory == null) continue;
-      const sender_id = lastHistory.get("author_id");
-      if (isLanguageModelService(sender_id)) {
-        return service2model(sender_id);
-      }
-      const input = lastHistory.get("content")?.toLowerCase();
-      if (mentionsLanguageModel(input)) {
-        return getLanguageModel(input);
-      }
+    const thread = this.getMessagesInThread(
+      rootMessage.get("date")?.toISOString?.() ?? `${rootMs}`,
+    );
+    if (thread == null) {
+      return false;
     }
 
+    const firstMessage = thread.first();
+    if (firstMessage == null) {
+      return false;
+    }
+    const firstHistory = firstMessage.get("history")?.first();
+    if (firstHistory == null) {
+      return false;
+    }
+    const sender_id = firstHistory.get("author_id");
+    if (isLanguageModelService(sender_id)) {
+      return service2model(sender_id);
+    }
+    const input = firstHistory.get("content")?.toLowerCase();
+    if (mentionsLanguageModel(input)) {
+      return getLanguageModel(input);
+    }
     return false;
   };
 
@@ -1270,7 +1308,6 @@ export class ChatActions extends Actions<ChatState> {
   };
 
   setSelectedThread = (threadKey: string | null) => {
-    console.log("setSelectedThread", { threadKey });
     this.frameTreeActions?.set_frame_data({
       id: this.frameId,
       selectedThreadKey: threadKey,

src/packages/frontend/chat/chat-log.tsx

@@ -546,7 +546,7 @@ export function MessageList({
   return (
     <Virtuoso
       ref={virtuosoRef}
-      totalCount={sortedDates.length}
+      totalCount={sortedDates.length + 1}
       itemSize={(el) => {
         // see comment in jupyter/cell-list.tsx
         const h = el.getBoundingClientRect().height;
@@ -558,6 +558,9 @@ export function MessageList({
         return h;
       }}
       itemContent={(index) => {
+        if (sortedDates.length == index) {
+          return <div style={{ height: "25vh" }} />;
+        }
         const date = sortedDates[index];
         const message: ChatMessageTyped | undefined = messages.get(date);
         if (message == null) {