sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 15:38:55 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Enable regenerating Khoj messages on web app

Browse source 
- Replace Delete button with Regenerate button in chatMessage component
- Update chat API to delete previous conversation with turn id, if exists
This commit is contained in:
Debanjum (aider) 2024-11-07 15:14:02 -08:00 committed by Debanjum
parent adc8a764bb
commit 27137f1cdd
1 changed files with 40 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
src/interface/web/app/components/chatMessage/chatMessage.tsx
View file

@ -30,6 +30,7 @@ import {
Code,
Shapes,
Trash,
ArrowClockwise,
} from "@phosphor-icons/react";
import DOMPurify from "dompurify";
@ -682,6 +683,32 @@ const ChatMessage = forwardRef<HTMLDivElement, ChatMessageProps>((props, ref) =>
}
};
const regenerateMessage = async (message: SingleChatMessage) => {
const turnId = message.turnId || props.turnId;
if (!message.rawQuery) return;
const response = await fetch("/api/chat", {
method: "POST",
headers: {
"Content-Type": "application/json",
},
body: JSON.stringify({
q: message.rawQuery,
conversation_id: props.conversationId,
turn_id: turnId,
stream: false,
}),
});
if (response.ok) {
const result = await response.json();
// Update the UI after successful regeneration
props.onDeleteMessage(turnId);
} else {
console.error("Failed to regenerate message");
}
};
const allReferences = constructAllReferences(
props.chatMessage.context,
props.chatMessage.onlineContext,
@ -744,7 +771,7 @@ const ChatMessage = forwardRef<HTMLDivElement, ChatMessageProps>((props, ref) =>
/>
</button>
))}
{props.chatMessage.turnId && (
{props.chatMessage.by === "you" && props.chatMessage.turnId && (
<button
title="Delete"
className={`${styles.deleteButton}`}
@ -756,6 +783,18 @@ const ChatMessage = forwardRef<HTMLDivElement, ChatMessageProps>((props, ref) =>
/>
</button>
)}
{props.chatMessage.by === "khoj" && props.chatMessage.turnId && (
<button
title="Regenerate"
className={`${styles.regenerateButton}`}
onClick={() => regenerateMessage(props.chatMessage)}
>
<ArrowClockwise
alt="Regenerate Message"
className="hsl(var(--muted-foreground)) hover:text-blue-500"
/>
</button>
)}
<button
title="Copy"
className={`${styles.copyButton}`}