update

TabbyML · Oct 18, 2024 · f875e09 · f875e09
1 parent 3fa0100
commit f875e09
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 12 deletions.
diff --git a/ee/tabby-ui/components/chat/chat.tsx b/ee/tabby-ui/components/chat/chat.tsx
@@ -198,6 +198,11 @@ function ChatRenderer(
     const nextQaPairs = qaPairs.filter(o => o.user.id !== userMessageId)
     setQaPairs(nextQaPairs)
     setInput(userMessage.message)
+    if (userMessage.activeContext) {
+      onNavigateToContext(userMessage.activeContext, {
+        openInEditor: true
+      })
+    }
 
     deleteThreadMessagePair(threadId, qaPair?.user.id, qaPair?.assistant?.id)
   }
@@ -385,10 +390,11 @@ function ChatRenderer(
         message: userMessage.message + selectCodeSnippet,
         // If no id is provided, set a fallback id.
         id: userMessage.id ?? nanoid(),
-        // If a temporary user message draft exists, use its selectContext and activeContext
+        // FIXME: perhaps should use the current active selection
         selectContext: userMessageForEditTemp.current
           ? userMessageForEditTemp.current.selectContext
           : userMessage.selectContext,
+        // todo: use current active selection
         activeContext: userMessageForEditTemp.current
           ? userMessageForEditTemp.current.activeContext
           : userMessage.activeContext

diff --git a/ee/tabby-ui/components/chat/question-answer.tsx b/ee/tabby-ui/components/chat/question-answer.tsx
@@ -208,7 +208,17 @@ function UserMessageCardActions(props: { message: UserMessage }) {
         <Button
           variant="ghost"
           size="icon"
-          onClick={e => handleMessageAction?.(message.id, 'delete')}
+          onClick={e => handleMessageAction(message.id, 'edit')}
+        >
+          <IconEdit />
+          <span className="sr-only">Edit message</span>
+        </Button>
+      )}
+      {!isLoading && (
+        <Button
+          variant="ghost"
+          size="icon"
+          onClick={e => handleMessageAction(message.id, 'delete')}
         >
           <IconTrash />
           <span className="sr-only">Delete message</span>
@@ -444,16 +454,6 @@ function AssistantMessageCardActions(props: AssistantMessageActionProps) {
           <span className="sr-only">Regenerate message</span>
         </Button>
       )}
-      {!isGenerating && (
-        <Button
-          variant="ghost"
-          size="icon"
-          onClick={e => handleMessageAction(userMessageId, 'edit')}
-        >
-          <IconEdit />
-          <span className="sr-only">Edit message</span>
-        </Button>
-      )}
       <CopyButton value={copyContent} onCopyContent={onCopyContent} />
     </ChatMessageActionsWrapper>
   )