fix: only give initialResponse unfinished true value when not a cancellable endpoint

This commit is contained in:
Daniel Avila 2023-04-10 18:27:06 -04:00
parent 0bd240939a
commit e0d5e75e73
3 changed files with 17 additions and 7 deletions

View file

@ -17,7 +17,7 @@ export default function MessageHandler() {
const messageHandler = (data, submission) => {
const { messages, message, initialResponse, isRegenerate = false } = submission;
if (isRegenerate)
if (isRegenerate) {
setMessages([
...messages,
{
@ -26,10 +26,10 @@ export default function MessageHandler() {
parentMessageId: message?.overrideParentMessageId,
messageId: message?.overrideParentMessageId + '_',
submitting: true,
unfinished: true
// unfinished: true
}
]);
else
} else {
setMessages([
...messages,
message,
@ -39,9 +39,10 @@ export default function MessageHandler() {
parentMessageId: message?.messageId,
messageId: message?.messageId + '_',
submitting: true,
unfinished: true
// unfinished: true
}
]);
}
};
const cancelHandler = (data, submission) => {
@ -50,8 +51,11 @@ export default function MessageHandler() {
const { requestMessage, responseMessage, conversation } = data;
// update the messages
if (isRegenerate) setMessages([...messages, responseMessage]);
else setMessages([...messages, requestMessage, responseMessage]);
if (isRegenerate) {
setMessages([...messages, responseMessage]);
} else {
setMessages([...messages, requestMessage, responseMessage]);
}
setIsSubmitting(false);
// refresh title

View file

@ -34,6 +34,12 @@ export default function Message({
const blinker = submitting && isSubmitting;
const getConversationQuery = useGetConversationByIdQuery(message.conversationId, { enabled: false });
// debugging
// useEffect(() => {
// console.log('isSubmitting:', isSubmitting);
// console.log('unfinished:', unfinished);
// }, [isSubmitting, unfinished]);
useEffect(() => {
if (blinker && !abortScroll) {
scrollToBottom();

View file

@ -106,7 +106,7 @@ const useMessageHandler = () => {
parentMessageId: isRegenerate ? messageId : fakeMessageId,
messageId: (isRegenerate ? messageId : fakeMessageId) + '_',
conversationId,
unfinished: true,
unfinished: (endpoint === 'azureOpenAI' || endpoint === 'openAI') ? false : true,
submitting: true
};