klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-04 21:29:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add ability to retry a query from the web app

Browse Source
This commit is contained in:
Debanjum
2025-06-04 01:15:30 -07:00
parent 05d4e19cb8
commit d16f9f272b
4 changed files with 86 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/interface/web/app/components/chatHistory/chatHistory.tsx
View File

@@ -41,6 +41,7 @@ interface ChatHistoryProps {
customClassName?: string;
setIsChatSideBarOpen?: (isOpen: boolean) => void;
setIsOwner?: (isOwner: boolean) => void;
onRetryMessage?: (query: string, turnId?: string) => void;
}
interface TrainOfThoughtFrame {
@@ -508,6 +509,18 @@ export default function ChatHistory(props: ChatHistoryProps) {
}
};
const handleRetryMessage = (query: string, turnId?: string) => {
if (!query) return;
// Delete the message from local state first
if (turnId) {
handleDeleteMessage(turnId);
}
// Then trigger the retry
props.onRetryMessage?.(query, turnId);
};
if (!props.conversationId && !props.publicConversationSlug) {
return null;
}
@@ -561,6 +574,7 @@ export default function ChatHistory(props: ChatHistoryProps) {
borderLeftColor={`${data?.agent?.color}-500`}
isLastMessage={index === data.chat.length - 1}
onDeleteMessage={handleDeleteMessage}
onRetryMessage={handleRetryMessage}
conversationId={props.conversationId}
/>
</React.Fragment>
@@ -589,6 +603,7 @@ export default function ChatHistory(props: ChatHistoryProps) {
customClassName="fullHistory"
borderLeftColor={`${data?.agent?.color}-500`}
onDeleteMessage={handleDeleteMessage}
onRetryMessage={handleRetryMessage}
conversationId={props.conversationId}
turnId={messageTurnId}
/>
@@ -629,6 +644,7 @@ export default function ChatHistory(props: ChatHistoryProps) {
conversationId={props.conversationId}
turnId={messageTurnId}
onDeleteMessage={handleDeleteMessage}
onRetryMessage={handleRetryMessage}
customClassName="fullHistory"
borderLeftColor={`${data?.agent?.color}-500`}
isLastMessage={index === props.incomingMessages!.length - 1}
@@ -653,6 +669,7 @@ export default function ChatHistory(props: ChatHistoryProps) {
}}
conversationId={props.conversationId}
onDeleteMessage={handleDeleteMessage}
onRetryMessage={handleRetryMessage}
customClassName="fullHistory"
borderLeftColor={`${data?.agent?.color}-500`}
isLastMessage={true}