mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 00:40:14 +01:00
🔧 style: Improve UI and UX with Style Fixes and Code Refactors (#2002)
* refactor(useSSE): add useCallback to all event handlers * chore: remove modelName in defaultAssistantFormValues * fix(SidePanel): fix layout shift on chrome my removing sidenav scrollbar * style(ChatForm): match ChatGPT textarea effect styling * style: fix flickering of old background color on refresh
This commit is contained in:
parent
b2ab6fd19d
commit
ab66747e97
7 changed files with 412 additions and 375 deletions
|
|
@ -43,7 +43,7 @@ export default function ChatForm({ index = 0 }) {
|
||||||
>
|
>
|
||||||
<div className="relative flex h-full flex-1 items-stretch md:flex-col">
|
<div className="relative flex h-full flex-1 items-stretch md:flex-col">
|
||||||
<div className="flex w-full items-center">
|
<div className="flex w-full items-center">
|
||||||
<div className="[&:has(textarea:focus)]:border-token-border-xheavy border-token-border-heavy shadow-xs dark:shadow-xs relative flex w-full flex-grow flex-col overflow-hidden rounded-2xl border border-black/10 bg-white shadow-[0_0_0_2px_rgba(255,255,255,0.95)] dark:border-gray-600 dark:bg-gray-800 dark:text-white dark:shadow-[0_0_0_2px_rgba(33,33,33,1)] [&:has(textarea:focus)]:shadow-[0_2px_6px_rgba(0,0,0,0.3)]">
|
<div className="[&:has(textarea:focus)]:border-token-border-xheavy dark:border-token-border-medium border-token-border-medium bg-token-main-surface-primary relative flex w-full flex-grow flex-col overflow-hidden rounded-2xl border dark:text-white [&:has(textarea:focus)]:shadow-[0_2px_6px_rgba(0,0,0,.05)]">
|
||||||
<FileRow
|
<FileRow
|
||||||
files={files}
|
files={files}
|
||||||
setFiles={setFiles}
|
setFiles={setFiles}
|
||||||
|
|
|
||||||
|
|
@ -20,7 +20,7 @@ export default function Nav({ links, isCollapsed, resize, defaultActive }: NavPr
|
||||||
return (
|
return (
|
||||||
<div
|
<div
|
||||||
data-collapsed={isCollapsed}
|
data-collapsed={isCollapsed}
|
||||||
className="bg-token-sidebar-surface-primary group flex-shrink-0 overflow-x-hidden py-2 data-[collapsed=true]:py-2"
|
className="bg-token-sidebar-surface-primary hide-scrollbar group flex-shrink-0 overflow-x-hidden py-2 data-[collapsed=true]:py-2"
|
||||||
>
|
>
|
||||||
<div className="h-full">
|
<div className="h-full">
|
||||||
<div className="flex h-full min-h-0 flex-col">
|
<div className="flex h-full min-h-0 flex-col">
|
||||||
|
|
|
||||||
|
|
@ -172,7 +172,7 @@ export default function SidePanel({
|
||||||
localStorage.setItem('react-resizable-panels:collapsed', 'true');
|
localStorage.setItem('react-resizable-panels:collapsed', 'true');
|
||||||
}}
|
}}
|
||||||
className={cn(
|
className={cn(
|
||||||
'sidenav border-l border-gray-200 bg-white dark:border-gray-800/50 dark:bg-gray-900',
|
'sidenav hide-scrollbar border-l border-gray-200 bg-white dark:border-gray-800/50 dark:bg-gray-900',
|
||||||
isCollapsed ? 'min-w-[50px]' : 'min-w-[340px] sm:min-w-[352px]',
|
isCollapsed ? 'min-w-[50px]' : 'min-w-[340px] sm:min-w-[352px]',
|
||||||
minSize === 0 ? 'min-w-0' : '',
|
minSize === 0 ? 'min-w-0' : '',
|
||||||
)}
|
)}
|
||||||
|
|
|
||||||
|
|
@ -6,6 +6,7 @@ import type {
|
||||||
ContentPart,
|
ContentPart,
|
||||||
TMessageContentParts,
|
TMessageContentParts,
|
||||||
} from 'librechat-data-provider';
|
} from 'librechat-data-provider';
|
||||||
|
import { useCallback, useMemo } from 'react';
|
||||||
|
|
||||||
type TUseContentHandler = {
|
type TUseContentHandler = {
|
||||||
setMessages: (messages: TMessage[]) => void;
|
setMessages: (messages: TMessage[]) => void;
|
||||||
|
|
@ -18,49 +19,53 @@ type TContentHandler = {
|
||||||
};
|
};
|
||||||
|
|
||||||
export default function useContentHandler({ setMessages, getMessages }: TUseContentHandler) {
|
export default function useContentHandler({ setMessages, getMessages }: TUseContentHandler) {
|
||||||
const messageMap = new Map<string, TMessage>();
|
const messageMap = useMemo(() => new Map<string, TMessage>(), []);
|
||||||
return ({ data, submission }: TContentHandler) => {
|
return useCallback(
|
||||||
const { type, messageId, thread_id, conversationId, index, stream } = data;
|
({ data, submission }: TContentHandler) => {
|
||||||
|
const { type, messageId, thread_id, conversationId, index, stream } = data;
|
||||||
|
|
||||||
const _messages = getMessages();
|
const _messages = getMessages();
|
||||||
const messages =
|
const messages =
|
||||||
_messages?.filter((m) => m.messageId !== messageId)?.map((msg) => ({ ...msg, thread_id })) ??
|
_messages
|
||||||
[];
|
?.filter((m) => m.messageId !== messageId)
|
||||||
const userMessage = messages[messages.length - 1];
|
?.map((msg) => ({ ...msg, thread_id })) ?? [];
|
||||||
|
const userMessage = messages[messages.length - 1];
|
||||||
|
|
||||||
const { initialResponse } = submission;
|
const { initialResponse } = submission;
|
||||||
|
|
||||||
let response = messageMap.get(messageId);
|
let response = messageMap.get(messageId);
|
||||||
if (!response) {
|
if (!response) {
|
||||||
response = {
|
response = {
|
||||||
...initialResponse,
|
...initialResponse,
|
||||||
parentMessageId: userMessage?.messageId,
|
parentMessageId: userMessage?.messageId,
|
||||||
conversationId,
|
conversationId,
|
||||||
messageId,
|
messageId,
|
||||||
thread_id,
|
thread_id,
|
||||||
};
|
};
|
||||||
messageMap.set(messageId, response);
|
messageMap.set(messageId, response);
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: handle streaming for non-text
|
// TODO: handle streaming for non-text
|
||||||
const part: ContentPart =
|
const part: ContentPart =
|
||||||
stream && data[ContentTypes.TEXT] ? { value: data[ContentTypes.TEXT] } : data[type];
|
stream && data[ContentTypes.TEXT] ? { value: data[ContentTypes.TEXT] } : data[type];
|
||||||
|
|
||||||
/* spreading the content array to avoid mutation */
|
/* spreading the content array to avoid mutation */
|
||||||
response.content = [...(response.content ?? [])];
|
response.content = [...(response.content ?? [])];
|
||||||
|
|
||||||
response.content[index] = { type, [type]: part } as TMessageContentParts;
|
response.content[index] = { type, [type]: part } as TMessageContentParts;
|
||||||
|
|
||||||
if (
|
if (
|
||||||
type !== ContentTypes.TEXT &&
|
type !== ContentTypes.TEXT &&
|
||||||
initialResponse.content &&
|
initialResponse.content &&
|
||||||
((response.content[response.content.length - 1].type === ContentTypes.TOOL_CALL &&
|
((response.content[response.content.length - 1].type === ContentTypes.TOOL_CALL &&
|
||||||
response.content[response.content.length - 1][ContentTypes.TOOL_CALL].progress === 1) ||
|
response.content[response.content.length - 1][ContentTypes.TOOL_CALL].progress === 1) ||
|
||||||
response.content[response.content.length - 1].type === ContentTypes.IMAGE_FILE)
|
response.content[response.content.length - 1].type === ContentTypes.IMAGE_FILE)
|
||||||
) {
|
) {
|
||||||
response.content.push(initialResponse.content[0]);
|
response.content.push(initialResponse.content[0]);
|
||||||
}
|
}
|
||||||
|
|
||||||
setMessages([...messages, response]);
|
setMessages([...messages, response]);
|
||||||
};
|
},
|
||||||
|
[getMessages, messageMap, setMessages],
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,8 +1,8 @@
|
||||||
import { v4 } from 'uuid';
|
import { v4 } from 'uuid';
|
||||||
import { useSetRecoilState } from 'recoil';
|
import { useSetRecoilState } from 'recoil';
|
||||||
import { useEffect, useState } from 'react';
|
|
||||||
import { useParams } from 'react-router-dom';
|
import { useParams } from 'react-router-dom';
|
||||||
import { useQueryClient } from '@tanstack/react-query';
|
import { useQueryClient } from '@tanstack/react-query';
|
||||||
|
import { useEffect, useState, useCallback } from 'react';
|
||||||
import {
|
import {
|
||||||
/* @ts-ignore */
|
/* @ts-ignore */
|
||||||
SSE,
|
SSE,
|
||||||
|
|
@ -83,392 +83,424 @@ export default function useSSE(submission: TSubmission | null, index = 0) {
|
||||||
enabled: !!isAuthenticated && startupConfig?.checkBalance,
|
enabled: !!isAuthenticated && startupConfig?.checkBalance,
|
||||||
});
|
});
|
||||||
|
|
||||||
const messageHandler = (data: string, submission: TSubmission) => {
|
const messageHandler = useCallback(
|
||||||
const {
|
(data: string, submission: TSubmission) => {
|
||||||
messages,
|
const {
|
||||||
message,
|
messages,
|
||||||
plugin,
|
|
||||||
plugins,
|
|
||||||
initialResponse,
|
|
||||||
isRegenerate = false,
|
|
||||||
} = submission;
|
|
||||||
|
|
||||||
if (isRegenerate) {
|
|
||||||
setMessages([
|
|
||||||
...messages,
|
|
||||||
{
|
|
||||||
...initialResponse,
|
|
||||||
text: data,
|
|
||||||
parentMessageId: message?.overrideParentMessageId ?? null,
|
|
||||||
messageId: message?.overrideParentMessageId + '_',
|
|
||||||
plugin: plugin ?? null,
|
|
||||||
plugins: plugins ?? [],
|
|
||||||
// unfinished: true
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
} else {
|
|
||||||
setMessages([
|
|
||||||
...messages,
|
|
||||||
message,
|
message,
|
||||||
|
plugin,
|
||||||
|
plugins,
|
||||||
|
initialResponse,
|
||||||
|
isRegenerate = false,
|
||||||
|
} = submission;
|
||||||
|
|
||||||
|
if (isRegenerate) {
|
||||||
|
setMessages([
|
||||||
|
...messages,
|
||||||
|
{
|
||||||
|
...initialResponse,
|
||||||
|
text: data,
|
||||||
|
parentMessageId: message?.overrideParentMessageId ?? null,
|
||||||
|
messageId: message?.overrideParentMessageId + '_',
|
||||||
|
plugin: plugin ?? null,
|
||||||
|
plugins: plugins ?? [],
|
||||||
|
// unfinished: true
|
||||||
|
},
|
||||||
|
]);
|
||||||
|
} else {
|
||||||
|
setMessages([
|
||||||
|
...messages,
|
||||||
|
message,
|
||||||
|
{
|
||||||
|
...initialResponse,
|
||||||
|
text: data,
|
||||||
|
parentMessageId: message?.messageId,
|
||||||
|
messageId: message?.messageId + '_',
|
||||||
|
plugin: plugin ?? null,
|
||||||
|
plugins: plugins ?? [],
|
||||||
|
// unfinished: true
|
||||||
|
},
|
||||||
|
]);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[setMessages],
|
||||||
|
);
|
||||||
|
|
||||||
|
const cancelHandler = useCallback(
|
||||||
|
(data: TResData, submission: TSubmission) => {
|
||||||
|
const { requestMessage, responseMessage, conversation } = data;
|
||||||
|
const { messages, isRegenerate = false } = submission;
|
||||||
|
|
||||||
|
const convoUpdate = conversation ?? submission.conversation;
|
||||||
|
|
||||||
|
// update the messages
|
||||||
|
if (isRegenerate) {
|
||||||
|
const messagesUpdate = [...messages, responseMessage].filter((msg) => msg);
|
||||||
|
setMessages(messagesUpdate);
|
||||||
|
} else {
|
||||||
|
const messagesUpdate = [...messages, requestMessage, responseMessage].filter((msg) => msg);
|
||||||
|
setMessages(messagesUpdate);
|
||||||
|
}
|
||||||
|
|
||||||
|
const isNewConvo = conversation.conversationId !== submission.conversation.conversationId;
|
||||||
|
if (isNewConvo) {
|
||||||
|
queryClient.setQueryData<ConversationData>([QueryKeys.allConversations], (convoData) => {
|
||||||
|
if (!convoData) {
|
||||||
|
return convoData;
|
||||||
|
}
|
||||||
|
return deleteConversation(convoData, submission.conversation.conversationId as string);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// refresh title
|
||||||
|
if (isNewConvo && requestMessage?.parentMessageId === Constants.NO_PARENT) {
|
||||||
|
setTimeout(() => {
|
||||||
|
genTitle.mutate({ conversationId: convoUpdate.conversationId as string });
|
||||||
|
}, 2500);
|
||||||
|
}
|
||||||
|
|
||||||
|
setConversation((prevState) => {
|
||||||
|
const update = {
|
||||||
|
...prevState,
|
||||||
|
...convoUpdate,
|
||||||
|
};
|
||||||
|
|
||||||
|
setStorage(update);
|
||||||
|
return update;
|
||||||
|
});
|
||||||
|
|
||||||
|
setIsSubmitting(false);
|
||||||
|
},
|
||||||
|
[setMessages, setConversation, setStorage, genTitle, queryClient, setIsSubmitting],
|
||||||
|
);
|
||||||
|
|
||||||
|
const syncHandler = useCallback(
|
||||||
|
(data: TSyncData, submission: TSubmission) => {
|
||||||
|
const { conversationId, thread_id, responseMessage, requestMessage } = data;
|
||||||
|
const { initialResponse, messages: _messages, message } = submission;
|
||||||
|
|
||||||
|
const messages = _messages.filter((msg) => msg.messageId !== message.messageId);
|
||||||
|
|
||||||
|
setMessages([
|
||||||
|
...messages,
|
||||||
|
requestMessage,
|
||||||
{
|
{
|
||||||
...initialResponse,
|
...initialResponse,
|
||||||
text: data,
|
...responseMessage,
|
||||||
parentMessageId: message?.messageId,
|
|
||||||
messageId: message?.messageId + '_',
|
|
||||||
plugin: plugin ?? null,
|
|
||||||
plugins: plugins ?? [],
|
|
||||||
// unfinished: true
|
|
||||||
},
|
},
|
||||||
]);
|
]);
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const cancelHandler = (data: TResData, submission: TSubmission) => {
|
let update = {} as TConversation;
|
||||||
const { requestMessage, responseMessage, conversation } = data;
|
setConversation((prevState) => {
|
||||||
const { messages, isRegenerate = false } = submission;
|
update = tConvoUpdateSchema.parse({
|
||||||
|
...prevState,
|
||||||
|
conversationId,
|
||||||
|
thread_id,
|
||||||
|
messages: [requestMessage.messageId, responseMessage.messageId],
|
||||||
|
}) as TConversation;
|
||||||
|
|
||||||
const convoUpdate = conversation ?? submission.conversation;
|
setStorage(update);
|
||||||
|
return update;
|
||||||
|
});
|
||||||
|
|
||||||
// update the messages
|
|
||||||
if (isRegenerate) {
|
|
||||||
const messagesUpdate = [...messages, responseMessage].filter((msg) => msg);
|
|
||||||
setMessages(messagesUpdate);
|
|
||||||
} else {
|
|
||||||
const messagesUpdate = [...messages, requestMessage, responseMessage].filter((msg) => msg);
|
|
||||||
setMessages(messagesUpdate);
|
|
||||||
}
|
|
||||||
|
|
||||||
const isNewConvo = conversation.conversationId !== submission.conversation.conversationId;
|
|
||||||
if (isNewConvo) {
|
|
||||||
queryClient.setQueryData<ConversationData>([QueryKeys.allConversations], (convoData) => {
|
queryClient.setQueryData<ConversationData>([QueryKeys.allConversations], (convoData) => {
|
||||||
if (!convoData) {
|
if (!convoData) {
|
||||||
return convoData;
|
return convoData;
|
||||||
}
|
}
|
||||||
return deleteConversation(convoData, submission.conversation.conversationId as string);
|
if (requestMessage.parentMessageId === Constants.NO_PARENT) {
|
||||||
|
return addConversation(convoData, update);
|
||||||
|
} else {
|
||||||
|
return updateConversation(convoData, update);
|
||||||
|
}
|
||||||
});
|
});
|
||||||
}
|
|
||||||
|
|
||||||
// refresh title
|
setShowStopButton(true);
|
||||||
if (isNewConvo && requestMessage?.parentMessageId === Constants.NO_PARENT) {
|
|
||||||
setTimeout(() => {
|
|
||||||
genTitle.mutate({ conversationId: convoUpdate.conversationId as string });
|
|
||||||
}, 2500);
|
|
||||||
}
|
|
||||||
|
|
||||||
setConversation((prevState) => {
|
resetLatestMessage();
|
||||||
const update = {
|
},
|
||||||
...prevState,
|
[setMessages, setConversation, setStorage, queryClient, setShowStopButton, resetLatestMessage],
|
||||||
...convoUpdate,
|
);
|
||||||
};
|
|
||||||
|
|
||||||
setStorage(update);
|
const createdHandler = useCallback(
|
||||||
return update;
|
(data: TResData, submission: TSubmission) => {
|
||||||
});
|
const { messages, message, initialResponse, isRegenerate = false } = submission;
|
||||||
|
|
||||||
setIsSubmitting(false);
|
if (isRegenerate) {
|
||||||
};
|
setMessages([
|
||||||
|
...messages,
|
||||||
const syncHandler = (data: TSyncData, submission: TSubmission) => {
|
{
|
||||||
const { conversationId, thread_id, responseMessage, requestMessage } = data;
|
...initialResponse,
|
||||||
const { initialResponse, messages: _messages, message } = submission;
|
parentMessageId: message?.overrideParentMessageId ?? null,
|
||||||
|
messageId: message?.overrideParentMessageId + '_',
|
||||||
const messages = _messages.filter((msg) => msg.messageId !== message.messageId);
|
},
|
||||||
|
]);
|
||||||
setMessages([
|
|
||||||
...messages,
|
|
||||||
requestMessage,
|
|
||||||
{
|
|
||||||
...initialResponse,
|
|
||||||
...responseMessage,
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
|
|
||||||
let update = {} as TConversation;
|
|
||||||
setConversation((prevState) => {
|
|
||||||
update = tConvoUpdateSchema.parse({
|
|
||||||
...prevState,
|
|
||||||
conversationId,
|
|
||||||
thread_id,
|
|
||||||
messages: [requestMessage.messageId, responseMessage.messageId],
|
|
||||||
}) as TConversation;
|
|
||||||
|
|
||||||
setStorage(update);
|
|
||||||
return update;
|
|
||||||
});
|
|
||||||
|
|
||||||
queryClient.setQueryData<ConversationData>([QueryKeys.allConversations], (convoData) => {
|
|
||||||
if (!convoData) {
|
|
||||||
return convoData;
|
|
||||||
}
|
|
||||||
if (requestMessage.parentMessageId === Constants.NO_PARENT) {
|
|
||||||
return addConversation(convoData, update);
|
|
||||||
} else {
|
} else {
|
||||||
return updateConversation(convoData, update);
|
setMessages([
|
||||||
}
|
...messages,
|
||||||
});
|
message,
|
||||||
|
{
|
||||||
setShowStopButton(true);
|
...initialResponse,
|
||||||
|
parentMessageId: message?.messageId,
|
||||||
resetLatestMessage();
|
messageId: message?.messageId + '_',
|
||||||
};
|
},
|
||||||
|
]);
|
||||||
const createdHandler = (data: TResData, submission: TSubmission) => {
|
|
||||||
const { messages, message, initialResponse, isRegenerate = false } = submission;
|
|
||||||
|
|
||||||
if (isRegenerate) {
|
|
||||||
setMessages([
|
|
||||||
...messages,
|
|
||||||
{
|
|
||||||
...initialResponse,
|
|
||||||
parentMessageId: message?.overrideParentMessageId ?? null,
|
|
||||||
messageId: message?.overrideParentMessageId + '_',
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
} else {
|
|
||||||
setMessages([
|
|
||||||
...messages,
|
|
||||||
message,
|
|
||||||
{
|
|
||||||
...initialResponse,
|
|
||||||
parentMessageId: message?.messageId,
|
|
||||||
messageId: message?.messageId + '_',
|
|
||||||
},
|
|
||||||
]);
|
|
||||||
}
|
|
||||||
|
|
||||||
const { conversationId, parentMessageId } = message;
|
|
||||||
|
|
||||||
let update = {} as TConversation;
|
|
||||||
setConversation((prevState) => {
|
|
||||||
let title = prevState?.title;
|
|
||||||
if (parentMessageId !== Constants.NO_PARENT && title?.toLowerCase()?.includes('new chat')) {
|
|
||||||
const convos = queryClient.getQueryData<ConversationData>([QueryKeys.allConversations]);
|
|
||||||
const cachedConvo = getConversationById(convos, conversationId);
|
|
||||||
title = cachedConvo?.title;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
update = tConvoUpdateSchema.parse({
|
const { conversationId, parentMessageId } = message;
|
||||||
...prevState,
|
|
||||||
conversationId,
|
|
||||||
title,
|
|
||||||
}) as TConversation;
|
|
||||||
|
|
||||||
setStorage(update);
|
let update = {} as TConversation;
|
||||||
return update;
|
setConversation((prevState) => {
|
||||||
});
|
let title = prevState?.title;
|
||||||
|
if (parentMessageId !== Constants.NO_PARENT && title?.toLowerCase()?.includes('new chat')) {
|
||||||
|
const convos = queryClient.getQueryData<ConversationData>([QueryKeys.allConversations]);
|
||||||
|
const cachedConvo = getConversationById(convos, conversationId);
|
||||||
|
title = cachedConvo?.title;
|
||||||
|
}
|
||||||
|
|
||||||
queryClient.setQueryData<ConversationData>([QueryKeys.allConversations], (convoData) => {
|
update = tConvoUpdateSchema.parse({
|
||||||
if (!convoData) {
|
...prevState,
|
||||||
return convoData;
|
conversationId,
|
||||||
}
|
title,
|
||||||
if (parentMessageId === Constants.NO_PARENT) {
|
}) as TConversation;
|
||||||
return addConversation(convoData, update);
|
|
||||||
} else {
|
|
||||||
return updateConversation(convoData, update);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
resetLatestMessage();
|
|
||||||
};
|
|
||||||
|
|
||||||
const finalHandler = (data: TResData, submission: TSubmission) => {
|
setStorage(update);
|
||||||
const { requestMessage, responseMessage, conversation, runMessages } = data;
|
return update;
|
||||||
const { messages, conversation: submissionConvo, isRegenerate = false } = submission;
|
});
|
||||||
|
|
||||||
setShowStopButton(false);
|
|
||||||
setCompleted((prev) => new Set(prev.add(submission?.initialResponse?.messageId)));
|
|
||||||
|
|
||||||
// update the messages; if assistants endpoint, client doesn't receive responseMessage
|
|
||||||
if (runMessages) {
|
|
||||||
setMessages([...runMessages]);
|
|
||||||
} else if (isRegenerate && responseMessage) {
|
|
||||||
setMessages([...messages, responseMessage]);
|
|
||||||
} else if (responseMessage) {
|
|
||||||
setMessages([...messages, requestMessage, responseMessage]);
|
|
||||||
}
|
|
||||||
|
|
||||||
const isNewConvo = conversation.conversationId !== submissionConvo.conversationId;
|
|
||||||
if (isNewConvo) {
|
|
||||||
queryClient.setQueryData<ConversationData>([QueryKeys.allConversations], (convoData) => {
|
queryClient.setQueryData<ConversationData>([QueryKeys.allConversations], (convoData) => {
|
||||||
if (!convoData) {
|
if (!convoData) {
|
||||||
return convoData;
|
return convoData;
|
||||||
}
|
}
|
||||||
return deleteConversation(convoData, submissionConvo.conversationId as string);
|
if (parentMessageId === Constants.NO_PARENT) {
|
||||||
|
return addConversation(convoData, update);
|
||||||
|
} else {
|
||||||
|
return updateConversation(convoData, update);
|
||||||
|
}
|
||||||
});
|
});
|
||||||
}
|
resetLatestMessage();
|
||||||
|
},
|
||||||
|
[setMessages, setConversation, setStorage, queryClient, resetLatestMessage],
|
||||||
|
);
|
||||||
|
|
||||||
// refresh title
|
const finalHandler = useCallback(
|
||||||
if (isNewConvo && requestMessage && requestMessage.parentMessageId === Constants.NO_PARENT) {
|
(data: TResData, submission: TSubmission) => {
|
||||||
setTimeout(() => {
|
const { requestMessage, responseMessage, conversation, runMessages } = data;
|
||||||
genTitle.mutate({ conversationId: conversation.conversationId as string });
|
const { messages, conversation: submissionConvo, isRegenerate = false } = submission;
|
||||||
}, 2500);
|
|
||||||
}
|
|
||||||
|
|
||||||
setConversation((prevState) => {
|
setShowStopButton(false);
|
||||||
const update = {
|
setCompleted((prev) => new Set(prev.add(submission?.initialResponse?.messageId)));
|
||||||
...prevState,
|
|
||||||
...conversation,
|
|
||||||
};
|
|
||||||
|
|
||||||
if (prevState?.model && prevState.model !== submissionConvo.model) {
|
// update the messages; if assistants endpoint, client doesn't receive responseMessage
|
||||||
update.model = prevState.model;
|
if (runMessages) {
|
||||||
|
setMessages([...runMessages]);
|
||||||
|
} else if (isRegenerate && responseMessage) {
|
||||||
|
setMessages([...messages, responseMessage]);
|
||||||
|
} else if (responseMessage) {
|
||||||
|
setMessages([...messages, requestMessage, responseMessage]);
|
||||||
}
|
}
|
||||||
|
|
||||||
setStorage(update);
|
const isNewConvo = conversation.conversationId !== submissionConvo.conversationId;
|
||||||
return update;
|
if (isNewConvo) {
|
||||||
});
|
queryClient.setQueryData<ConversationData>([QueryKeys.allConversations], (convoData) => {
|
||||||
|
if (!convoData) {
|
||||||
|
return convoData;
|
||||||
|
}
|
||||||
|
return deleteConversation(convoData, submissionConvo.conversationId as string);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
setIsSubmitting(false);
|
// refresh title
|
||||||
};
|
if (isNewConvo && requestMessage && requestMessage.parentMessageId === Constants.NO_PARENT) {
|
||||||
|
setTimeout(() => {
|
||||||
|
genTitle.mutate({ conversationId: conversation.conversationId as string });
|
||||||
|
}, 2500);
|
||||||
|
}
|
||||||
|
|
||||||
const errorHandler = ({ data, submission }: { data?: TResData; submission: TSubmission }) => {
|
setConversation((prevState) => {
|
||||||
const { messages, message, initialResponse } = submission;
|
const update = {
|
||||||
|
...prevState,
|
||||||
|
...conversation,
|
||||||
|
};
|
||||||
|
|
||||||
setCompleted((prev) => new Set(prev.add(initialResponse.messageId)));
|
if (prevState?.model && prevState.model !== submissionConvo.model) {
|
||||||
|
update.model = prevState.model;
|
||||||
|
}
|
||||||
|
|
||||||
const conversationId = message?.conversationId ?? submission?.conversationId;
|
setStorage(update);
|
||||||
|
return update;
|
||||||
|
});
|
||||||
|
|
||||||
const parseErrorResponse = (data: TResData | Partial<TMessage>) => {
|
setIsSubmitting(false);
|
||||||
const metadata = data['responseMessage'] ?? data;
|
},
|
||||||
const errorMessage = {
|
[
|
||||||
...initialResponse,
|
setMessages,
|
||||||
...metadata,
|
setConversation,
|
||||||
|
setStorage,
|
||||||
|
genTitle,
|
||||||
|
queryClient,
|
||||||
|
setIsSubmitting,
|
||||||
|
setShowStopButton,
|
||||||
|
],
|
||||||
|
);
|
||||||
|
|
||||||
|
const errorHandler = useCallback(
|
||||||
|
({ data, submission }: { data?: TResData; submission: TSubmission }) => {
|
||||||
|
const { messages, message, initialResponse } = submission;
|
||||||
|
|
||||||
|
setCompleted((prev) => new Set(prev.add(initialResponse.messageId)));
|
||||||
|
|
||||||
|
const conversationId = message?.conversationId ?? submission?.conversationId;
|
||||||
|
|
||||||
|
const parseErrorResponse = (data: TResData | Partial<TMessage>) => {
|
||||||
|
const metadata = data['responseMessage'] ?? data;
|
||||||
|
const errorMessage = {
|
||||||
|
...initialResponse,
|
||||||
|
...metadata,
|
||||||
|
error: true,
|
||||||
|
parentMessageId: message?.messageId,
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!errorMessage.messageId) {
|
||||||
|
errorMessage.messageId = v4();
|
||||||
|
}
|
||||||
|
|
||||||
|
return tMessageSchema.parse(errorMessage);
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!data) {
|
||||||
|
const convoId = conversationId ?? v4();
|
||||||
|
const errorResponse = parseErrorResponse({
|
||||||
|
text: 'Error connecting to server',
|
||||||
|
...submission,
|
||||||
|
conversationId: convoId,
|
||||||
|
});
|
||||||
|
setMessages([...messages, message, errorResponse]);
|
||||||
|
newConversation({
|
||||||
|
template: { conversationId: convoId },
|
||||||
|
preset: tPresetSchema.parse(submission?.conversation),
|
||||||
|
});
|
||||||
|
setIsSubmitting(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!conversationId && !data.conversationId) {
|
||||||
|
const convoId = v4();
|
||||||
|
const errorResponse = parseErrorResponse(data);
|
||||||
|
setMessages([...messages, message, errorResponse]);
|
||||||
|
newConversation({
|
||||||
|
template: { conversationId: convoId },
|
||||||
|
preset: tPresetSchema.parse(submission?.conversation),
|
||||||
|
});
|
||||||
|
setIsSubmitting(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('Error:', data);
|
||||||
|
const errorResponse = tMessageSchema.parse({
|
||||||
|
...data,
|
||||||
error: true,
|
error: true,
|
||||||
parentMessageId: message?.messageId,
|
parentMessageId: message?.messageId,
|
||||||
};
|
});
|
||||||
|
|
||||||
if (!errorMessage.messageId) {
|
setMessages([...messages, message, errorResponse]);
|
||||||
errorMessage.messageId = v4();
|
if (data.conversationId && paramId === 'new') {
|
||||||
|
newConversation({
|
||||||
|
template: { conversationId: data.conversationId },
|
||||||
|
preset: tPresetSchema.parse(submission?.conversation),
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
return tMessageSchema.parse(errorMessage);
|
|
||||||
};
|
|
||||||
|
|
||||||
if (!data) {
|
|
||||||
const convoId = conversationId ?? v4();
|
|
||||||
const errorResponse = parseErrorResponse({
|
|
||||||
text: 'Error connecting to server',
|
|
||||||
...submission,
|
|
||||||
conversationId: convoId,
|
|
||||||
});
|
|
||||||
setMessages([...messages, message, errorResponse]);
|
|
||||||
newConversation({
|
|
||||||
template: { conversationId: convoId },
|
|
||||||
preset: tPresetSchema.parse(submission?.conversation),
|
|
||||||
});
|
|
||||||
setIsSubmitting(false);
|
setIsSubmitting(false);
|
||||||
return;
|
return;
|
||||||
}
|
},
|
||||||
|
[setMessages, paramId, setIsSubmitting, newConversation],
|
||||||
|
);
|
||||||
|
|
||||||
if (!conversationId && !data.conversationId) {
|
const abortConversation = useCallback(
|
||||||
const convoId = v4();
|
async (conversationId = '', submission: TSubmission) => {
|
||||||
const errorResponse = parseErrorResponse(data);
|
console.log(submission);
|
||||||
setMessages([...messages, message, errorResponse]);
|
let runAbortKey = '';
|
||||||
newConversation({
|
try {
|
||||||
template: { conversationId: convoId },
|
const conversation = (JSON.parse(localStorage.getItem('lastConversationSetup') ?? '') ??
|
||||||
preset: tPresetSchema.parse(submission?.conversation),
|
{}) as TConversation;
|
||||||
});
|
const { conversationId, messages } = conversation;
|
||||||
setIsSubmitting(false);
|
runAbortKey = `${conversationId}:${messages?.[messages.length - 1]}`;
|
||||||
return;
|
} catch (error) {
|
||||||
}
|
console.error('Error getting last conversation setup');
|
||||||
|
console.error(error);
|
||||||
|
}
|
||||||
|
const { endpoint: _endpoint, endpointType } = submission?.conversation || {};
|
||||||
|
const endpoint = endpointType ?? _endpoint;
|
||||||
|
try {
|
||||||
|
const response = await fetch(`${EndpointURLs[endpoint ?? '']}/abort`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json',
|
||||||
|
Authorization: `Bearer ${token}`,
|
||||||
|
},
|
||||||
|
body: JSON.stringify({
|
||||||
|
abortKey: _endpoint === EModelEndpoint.assistants ? runAbortKey : conversationId,
|
||||||
|
endpoint,
|
||||||
|
}),
|
||||||
|
});
|
||||||
|
|
||||||
console.log('Error:', data);
|
// Check if the response is JSON
|
||||||
const errorResponse = tMessageSchema.parse({
|
const contentType = response.headers.get('content-type');
|
||||||
...data,
|
if (contentType && contentType.includes('application/json')) {
|
||||||
error: true,
|
const data = await response.json();
|
||||||
parentMessageId: message?.messageId,
|
console.log('aborted', data);
|
||||||
});
|
if (response.status === 404) {
|
||||||
|
setIsSubmitting(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (data.final) {
|
||||||
|
finalHandler(data, submission);
|
||||||
|
} else {
|
||||||
|
cancelHandler(data, submission);
|
||||||
|
}
|
||||||
|
} else if (response.status === 204) {
|
||||||
|
const responseMessage = {
|
||||||
|
...submission.initialResponse,
|
||||||
|
};
|
||||||
|
|
||||||
setMessages([...messages, message, errorResponse]);
|
const data = {
|
||||||
if (data.conversationId && paramId === 'new') {
|
requestMessage: submission.message,
|
||||||
newConversation({
|
responseMessage: responseMessage,
|
||||||
template: { conversationId: data.conversationId },
|
conversation: submission.conversation,
|
||||||
preset: tPresetSchema.parse(submission?.conversation),
|
};
|
||||||
});
|
console.log('aborted', data);
|
||||||
}
|
|
||||||
|
|
||||||
setIsSubmitting(false);
|
|
||||||
return;
|
|
||||||
};
|
|
||||||
|
|
||||||
const abortConversation = async (conversationId = '', submission: TSubmission) => {
|
|
||||||
console.log(submission);
|
|
||||||
let runAbortKey = '';
|
|
||||||
try {
|
|
||||||
const conversation = (JSON.parse(localStorage.getItem('lastConversationSetup') ?? '') ??
|
|
||||||
{}) as TConversation;
|
|
||||||
const { conversationId, messages } = conversation;
|
|
||||||
runAbortKey = `${conversationId}:${messages?.[messages.length - 1]}`;
|
|
||||||
} catch (error) {
|
|
||||||
console.error('Error getting last conversation setup');
|
|
||||||
console.error(error);
|
|
||||||
}
|
|
||||||
const { endpoint: _endpoint, endpointType } = submission?.conversation || {};
|
|
||||||
const endpoint = endpointType ?? _endpoint;
|
|
||||||
try {
|
|
||||||
const response = await fetch(`${EndpointURLs[endpoint ?? '']}/abort`, {
|
|
||||||
method: 'POST',
|
|
||||||
headers: {
|
|
||||||
'Content-Type': 'application/json',
|
|
||||||
Authorization: `Bearer ${token}`,
|
|
||||||
},
|
|
||||||
body: JSON.stringify({
|
|
||||||
abortKey: _endpoint === EModelEndpoint.assistants ? runAbortKey : conversationId,
|
|
||||||
endpoint,
|
|
||||||
}),
|
|
||||||
});
|
|
||||||
|
|
||||||
// Check if the response is JSON
|
|
||||||
const contentType = response.headers.get('content-type');
|
|
||||||
if (contentType && contentType.includes('application/json')) {
|
|
||||||
const data = await response.json();
|
|
||||||
console.log('aborted', data);
|
|
||||||
if (response.status === 404) {
|
|
||||||
setIsSubmitting(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (data.final) {
|
|
||||||
finalHandler(data, submission);
|
|
||||||
} else {
|
} else {
|
||||||
cancelHandler(data, submission);
|
throw new Error(
|
||||||
|
'Unexpected response from server; Status: ' +
|
||||||
|
response.status +
|
||||||
|
' ' +
|
||||||
|
response.statusText,
|
||||||
|
);
|
||||||
}
|
}
|
||||||
} else if (response.status === 204) {
|
} catch (error) {
|
||||||
const responseMessage = {
|
console.error('Error cancelling request');
|
||||||
|
console.error(error);
|
||||||
|
const convoId = conversationId ?? v4();
|
||||||
|
const text =
|
||||||
|
submission.initialResponse?.text?.length > 45 ? submission.initialResponse?.text : '';
|
||||||
|
const errorMessage = {
|
||||||
|
...submission,
|
||||||
...submission.initialResponse,
|
...submission.initialResponse,
|
||||||
|
text: text ?? (error as Error).message ?? 'Error cancelling request',
|
||||||
|
unfinished: !!text.length,
|
||||||
|
error: true,
|
||||||
};
|
};
|
||||||
|
const errorResponse = tMessageSchema.parse(errorMessage);
|
||||||
const data = {
|
setMessages([...submission.messages, submission.message, errorResponse]);
|
||||||
requestMessage: submission.message,
|
newConversation({
|
||||||
responseMessage: responseMessage,
|
template: { conversationId: convoId },
|
||||||
conversation: submission.conversation,
|
preset: tPresetSchema.parse(submission?.conversation),
|
||||||
};
|
});
|
||||||
console.log('aborted', data);
|
setIsSubmitting(false);
|
||||||
} else {
|
|
||||||
throw new Error(
|
|
||||||
'Unexpected response from server; Status: ' + response.status + ' ' + response.statusText,
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
} catch (error) {
|
},
|
||||||
console.error('Error cancelling request');
|
[token, setIsSubmitting, finalHandler, cancelHandler, setMessages, newConversation],
|
||||||
console.error(error);
|
);
|
||||||
const convoId = conversationId ?? v4();
|
|
||||||
const text =
|
|
||||||
submission.initialResponse?.text?.length > 45 ? submission.initialResponse?.text : '';
|
|
||||||
const errorMessage = {
|
|
||||||
...submission,
|
|
||||||
...submission.initialResponse,
|
|
||||||
text: text ?? (error as Error).message ?? 'Error cancelling request',
|
|
||||||
unfinished: !!text.length,
|
|
||||||
error: true,
|
|
||||||
};
|
|
||||||
const errorResponse = tMessageSchema.parse(errorMessage);
|
|
||||||
setMessages([...submission.messages, submission.message, errorResponse]);
|
|
||||||
newConversation({
|
|
||||||
template: { conversationId: convoId },
|
|
||||||
preset: tPresetSchema.parse(submission?.conversation),
|
|
||||||
});
|
|
||||||
setIsSubmitting(false);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (submission === null) {
|
if (submission === null) {
|
||||||
|
|
|
||||||
|
|
@ -1410,7 +1410,7 @@ html {
|
||||||
.dark body,
|
.dark body,
|
||||||
.dark html {
|
.dark html {
|
||||||
--tw-bg-opacity:1;
|
--tw-bg-opacity:1;
|
||||||
background-color:rgba(52,53,65,var(--tw-bg-opacity))
|
background-color:rgba(33,33,33,var(--tw-bg-opacity))
|
||||||
}
|
}
|
||||||
#__next,
|
#__next,
|
||||||
#root {
|
#root {
|
||||||
|
|
|
||||||
|
|
@ -22,7 +22,7 @@ export const defaultAssistantFormValues = {
|
||||||
name: '',
|
name: '',
|
||||||
description: '',
|
description: '',
|
||||||
instructions: '',
|
instructions: '',
|
||||||
model: 'gpt-3.5-turbo-1106',
|
model: '',
|
||||||
functions: [],
|
functions: [],
|
||||||
code_interpreter: false,
|
code_interpreter: false,
|
||||||
retrieval: false,
|
retrieval: false,
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue