Feat: Users can chat directly without first creating a conversation. #11768 (#11769)

### What problem does this PR solve?

Feat: Users can chat directly without first creating a conversation.
#11768
### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2025-12-05 17:34:41 +08:00
committed by GitHub
parent 885eb2eab9
commit ea38e12d42
20 changed files with 410 additions and 308 deletions

View File

@ -1,4 +1,3 @@
import { FileUploadProps } from '@/components/file-upload';
import { MessageType } from '@/constants/chat';
import {
useHandleMessageInputChange,
@ -6,19 +5,15 @@ import {
useSelectDerivedMessages,
useSendMessageWithSse,
} from '@/hooks/logic-hooks';
import {
useFetchConversation,
useGetChatSearchParams,
} from '@/hooks/use-chat-request';
import { IMessage, Message } from '@/interfaces/database/chat';
import { useGetChatSearchParams } from '@/hooks/use-chat-request';
import { IMessage } from '@/interfaces/database/chat';
import api from '@/utils/api';
import { trim } from 'lodash';
import { useCallback, useEffect } from 'react';
import { useParams } from 'umi';
import { v4 as uuid } from 'uuid';
import { useCreateConversationBeforeSendMessage } from './use-chat-url';
import { useFindPrologueFromDialogList } from './use-select-conversation-list';
import { useSetChatRouteParams } from './use-set-chat-route';
import { useSetConversation } from './use-set-conversation';
import { useUploadFile } from './use-upload-file';
export const useSelectNextMessages = () => {
@ -33,8 +28,7 @@ export const useSelectNextMessages = () => {
removeMessageById,
removeMessagesAfterCurrentMessage,
} = useSelectDerivedMessages();
const { data: conversation, loading } = useFetchConversation();
const { conversationId, isNew } = useGetChatSearchParams();
const { isNew, conversationId } = useGetChatSearchParams();
const { id: dialogId } = useParams();
const prologue = useFindPrologueFromDialogList();
@ -44,45 +38,31 @@ export const useSelectNextMessages = () => {
role: MessageType.Assistant,
content: prologue,
id: uuid(),
conversationId: conversationId,
} as IMessage;
setDerivedMessages([nextMessage]);
}
}, [dialogId, isNew, prologue, setDerivedMessages]);
}, [conversationId, dialogId, isNew, prologue, setDerivedMessages]);
useEffect(() => {
addPrologue();
}, [addPrologue]);
useEffect(() => {
if (
conversationId &&
isNew !== 'true' &&
conversation.message?.length > 0
) {
setDerivedMessages(conversation.message);
}
if (!conversationId) {
setDerivedMessages([]);
}
}, [conversation.message, conversationId, setDerivedMessages, isNew]);
return {
scrollRef,
messageContainerRef,
derivedMessages,
loading,
addNewestAnswer,
addNewestQuestion,
removeLatestMessage,
removeMessageById,
removeMessagesAfterCurrentMessage,
setDerivedMessages,
};
};
export const useSendMessage = (controller: AbortController) => {
const { setConversation } = useSetConversation();
const { conversationId, isNew } = useGetChatSearchParams();
const { handleInputChange, value, setValue } = useHandleMessageInputChange();
@ -96,31 +76,13 @@ export const useSendMessage = (controller: AbortController) => {
scrollRef,
messageContainerRef,
derivedMessages,
loading,
addNewestAnswer,
addNewestQuestion,
removeLatestMessage,
removeMessageById,
removeMessagesAfterCurrentMessage,
setDerivedMessages,
} = useSelectNextMessages();
const { setConversationIsNew, getConversationIsNew } =
useSetChatRouteParams();
const onUploadFile: NonNullable<FileUploadProps['onUpload']> = useCallback(
async (files, options) => {
const isNew = getConversationIsNew();
if (isNew === 'true' && Array.isArray(files) && files.length) {
const data = await setConversation(files[0].name, true);
if (data.code === 0) {
handleUploadFile(files, options, data.data?.id);
}
} else {
handleUploadFile(files, options);
}
},
[getConversationIsNew, handleUploadFile, setConversation],
);
const sendMessage = useCallback(
async ({
@ -128,14 +90,19 @@ export const useSendMessage = (controller: AbortController) => {
currentConversationId,
messages,
}: {
message: Message;
message: IMessage;
currentConversationId?: string;
messages?: Message[];
messages?: IMessage[];
}) => {
const res = await send(
{
conversation_id: currentConversationId ?? conversationId,
messages: [...(messages ?? derivedMessages ?? []), message],
messages: [
...(Array.isArray(messages) && messages?.length > 0
? messages
: derivedMessages ?? []),
message,
],
},
controller,
);
@ -157,44 +124,62 @@ export const useSendMessage = (controller: AbortController) => {
],
);
const handleSendMessage = useCallback(
async (message: Message) => {
const isNew = getConversationIsNew();
if (isNew !== 'true') {
sendMessage({ message });
} else {
const data = await setConversation(
message.content,
true,
conversationId,
);
if (data.code === 0) {
setConversationIsNew('');
const id = data.data.id;
// currentConversationIdRef.current = id;
sendMessage({
message,
currentConversationId: id,
messages: data.data.message,
});
}
}
},
[
setConversation,
sendMessage,
setConversationIsNew,
getConversationIsNew,
conversationId,
],
);
const { regenerateMessage } = useRegenerateMessage({
removeMessagesAfterCurrentMessage,
sendMessage,
messages: derivedMessages,
});
const { createConversationBeforeSendMessage } =
useCreateConversationBeforeSendMessage();
const handlePressEnter = useCallback(async () => {
if (trim(value) === '') return;
const data = await createConversationBeforeSendMessage(value);
if (data === undefined) {
return;
}
const { targetConversationId, currentMessages } = data;
const id = uuid();
addNewestQuestion({
content: value,
files: files,
id,
role: MessageType.User,
conversationId: targetConversationId,
});
if (done) {
setValue('');
sendMessage({
currentConversationId: targetConversationId,
messages: currentMessages,
message: {
id,
content: value.trim(),
role: MessageType.User,
files: files,
conversationId: targetConversationId,
},
});
}
clearFiles();
}, [
value,
createConversationBeforeSendMessage,
addNewestQuestion,
files,
done,
clearFiles,
setValue,
sendMessage,
]);
useEffect(() => {
// #1289
if (answer.answer && conversationId && isNew !== 'true') {
@ -202,36 +187,6 @@ export const useSendMessage = (controller: AbortController) => {
}
}, [answer, addNewestAnswer, conversationId, isNew]);
const handlePressEnter = useCallback(() => {
if (trim(value) === '') return;
const id = uuid();
addNewestQuestion({
content: value,
files: files,
id,
role: MessageType.User,
});
if (done) {
setValue('');
handleSendMessage({
id,
content: value.trim(),
role: MessageType.User,
files: files,
});
}
clearFiles();
}, [
value,
addNewestQuestion,
files,
done,
clearFiles,
setValue,
handleSendMessage,
]);
return {
handlePressEnter,
handleInputChange,
@ -239,13 +194,13 @@ export const useSendMessage = (controller: AbortController) => {
setValue,
regenerateMessage,
sendLoading: !done,
loading,
scrollRef,
messageContainerRef,
derivedMessages,
removeMessageById,
handleUploadFile: onUploadFile,
handleUploadFile,
isUploading,
removeFile,
setDerivedMessages,
};
};