pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2026-01-29 22:56:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Add a think button to the chat box. #12742 (#12743)

Browse Source 
### What problem does this PR solve?

Feat: Add a think button to the chat box. #12742
### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2026-01-21 15:39:18 +08:00
committed by GitHub
parent f98abf14a8
commit e1143d40bc
12 changed files with 190 additions and 149 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
web/src/pages/agent/chat/use-send-agent-message.ts
View File

@ -1,3 +1,4 @@
import { NextMessageInputOnPressEnterParameter } from '@/components/message-input/next';
import sonnerMessage from '@/components/ui/message';
import { MessageType } from '@/constants/chat';
import {
@ -289,6 +290,7 @@ export const useSendAgentMessage = ({
params.files = uploadResponseList;
params.session_id = sessionId;
params.reasoning = message.reasoning;
}
try {
@ -355,28 +357,31 @@ export const useSendAgentMessage = ({
removeAllMessagesExceptFirst,
]);
const handlePressEnter = useCallback(() => {
if (trim(value) === '') return;
const msgBody = buildRequestBody(value);
if (done) {
setValue('');
sendMessage({
message: msgBody,
});
}
addNewestOneQuestion({ ...msgBody, files: fileList });
setTimeout(() => {
scrollToBottom();
}, 100);
}, [
value,
done,
addNewestOneQuestion,
fileList,
setValue,
sendMessage,
scrollToBottom,
]);
const handlePressEnter = useCallback(
(...[{ enableThinking }]: NextMessageInputOnPressEnterParameter) => {
if (trim(value) === '') return;
const msgBody = buildRequestBody(value);
if (done) {
setValue('');
sendMessage({
message: { ...msgBody, reasoning: enableThinking },
});
}
addNewestOneQuestion({ ...msgBody, files: fileList });
setTimeout(() => {
scrollToBottom();
}, 100);
},
[
value,
done,
addNewestOneQuestion,
fileList,
setValue,
sendMessage,
scrollToBottom,
],
);
const sendedTaskMessage = useRef<boolean>(false);

4
web/src/pages/agent/debug-content/index.tsx
View File

@ -164,7 +164,7 @@ const DebugContent = ({
render={({ field }) => (
<div className="space-y-6">
<FormItem className="w-full">
<FormLabel>{t('assistantAvatar')}</FormLabel>
<FormLabel>{props.label}</FormLabel>
<FormControl>
<FileUploadDirectUpload
value={field.value}
@ -218,7 +218,7 @@ const DebugContent = ({
BeginQueryTypeMap[BeginQueryType.Paragraph]
);
},
[form, t],
[form],
);
const onSubmit = useCallback(

5
web/src/pages/next-chats/chat/app-settings/chat-prompt-engine.tsx
View File

@ -50,11 +50,6 @@ export function ChatPromptEngine() {
tooltip={t('multiTurnTip')}
></SwitchFormField>
<UseKnowledgeGraphFormField name="prompt_config.use_kg"></UseKnowledgeGraphFormField>
<SwitchFormField
name={'prompt_config.reasoning'}
label={t('reasoning')}
tooltip={t('reasoningTip')}
></SwitchFormField>
<RerankFormFields></RerankFormFields>
<CrossLanguageFormField></CrossLanguageFormField>
<DynamicVariableForm></DynamicVariableForm>

1
web/src/pages/next-chats/chat/chat-box/multiple-chat-box.tsx
View File

@ -260,6 +260,7 @@ export function MultipleChatBox({
}
stopOutputMessage={stopOutputMessage}
onUpload={handleUploadFile}
showReasoning
/>
</div>
{visible && (

1
web/src/pages/next-chats/chat/chat-box/single-chat-box.tsx
View File

@ -119,6 +119,7 @@ export function SingleChatBox({
onUpload={handleUploadFile}
isUploading={isUploading}
removeFile={removeFile}
showReasoning
/>
{visible && (
<PdfSheet

85
web/src/pages/next-chats/hooks/use-send-chat-message.ts
View File

@ -1,3 +1,4 @@
import { NextMessageInputOnPressEnterParameter } from '@/components/message-input/next';
import { MessageType } from '@/constants/chat';
import {
useHandleMessageInputChange,
@ -133,52 +134,56 @@ export const useSendMessage = (controller: AbortController) => {
const { createConversationBeforeSendMessage } =
useCreateConversationBeforeSendMessage();
const handlePressEnter = useCallback(async () => {
if (trim(value) === '') return;
const handlePressEnter = useCallback(
async (...[{ enableThinking }]: NextMessageInputOnPressEnterParameter) => {
if (trim(value) === '') return;
const data = await createConversationBeforeSendMessage(value);
const data = await createConversationBeforeSendMessage(value);
if (data === undefined) {
return;
}
if (data === undefined) {
return;
}
const { targetConversationId, currentMessages } = data;
const { targetConversationId, currentMessages } = data;
const id = uuid();
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,
},
addNewestQuestion({
content: value,
files: files,
id,
role: MessageType.User,
conversationId: targetConversationId,
});
}
clearFiles();
}, [
value,
createConversationBeforeSendMessage,
addNewestQuestion,
files,
done,
clearFiles,
setValue,
sendMessage,
]);
if (done) {
setValue('');
sendMessage({
currentConversationId: targetConversationId,
messages: currentMessages,
message: {
id,
content: value.trim(),
role: MessageType.User,
files: files,
conversationId: targetConversationId,
reasoning: enableThinking,
},
});
}
clearFiles();
},
[
value,
createConversationBeforeSendMessage,
addNewestQuestion,
files,
done,
clearFiles,
setValue,
sendMessage,
],
);
useEffect(() => {
// #1289

99
web/src/pages/next-chats/hooks/use-send-multiple-message.ts
View File

@ -1,3 +1,4 @@
import { NextMessageInputOnPressEnterParameter } from '@/components/message-input/next';
import showMessage from '@/components/ui/message';
import { MessageType } from '@/constants/chat';
import {
@ -175,63 +176,67 @@ export function useSendMultipleChatMessage(
],
);
const handlePressEnter = useCallback(async () => {
if (trim(value) === '') return;
const id = uuid();
const handlePressEnter = useCallback(
async (...[{ enableThinking }]: NextMessageInputOnPressEnterParameter) => {
if (trim(value) === '') return;
const id = uuid();
const data = await createConversationBeforeSendMessage(value);
const data = await createConversationBeforeSendMessage(value);
if (data === undefined) {
return;
}
const { targetConversationId, currentMessages } = data;
chatBoxIds.forEach((chatBoxId) => {
if (!isLLMConfigEmpty(chatBoxId)) {
addNewestQuestion({
content: value,
id,
role: MessageType.User,
chatBoxId,
files,
conversationId: targetConversationId,
});
if (data === undefined) {
return;
}
});
if (allDone) {
setValue('');
const { targetConversationId, currentMessages } = data;
chatBoxIds.forEach((chatBoxId) => {
if (!isLLMConfigEmpty(chatBoxId)) {
sendMessage({
message: {
id,
content: value.trim(),
role: MessageType.User,
files,
conversationId: targetConversationId,
},
addNewestQuestion({
content: value,
id,
role: MessageType.User,
chatBoxId,
currentConversationId: targetConversationId,
messages: currentMessages,
files,
conversationId: targetConversationId,
});
}
});
}
clearFiles();
}, [
value,
createConversationBeforeSendMessage,
chatBoxIds,
allDone,
clearFiles,
isLLMConfigEmpty,
addNewestQuestion,
files,
setValue,
sendMessage,
]);
if (allDone) {
setValue('');
chatBoxIds.forEach((chatBoxId) => {
if (!isLLMConfigEmpty(chatBoxId)) {
sendMessage({
message: {
id,
content: value.trim(),
role: MessageType.User,
files,
conversationId: targetConversationId,
reasoning: enableThinking,
},
chatBoxId,
currentConversationId: targetConversationId,
messages: currentMessages,
});
}
});
}
clearFiles();
},
[
value,
createConversationBeforeSendMessage,
chatBoxIds,
allDone,
clearFiles,
isLLMConfigEmpty,
addNewestQuestion,
files,
setValue,
sendMessage,
],
);
useEffect(() => {
if (answer.answer && conversationId) {

10
web/src/pages/next-chats/hooks/use-send-shared-message.ts
View File

@ -1,3 +1,4 @@
import { NextMessageInputOnPressEnterParameter } from '@/components/message-input/next';
import { MessageType, SharedFrom } from '@/constants/chat';
import {
useHandleMessageInputChange,
@ -24,8 +25,7 @@ export const useGetSharedChatSearchParams = () => {
const [searchParams] = useSearchParams();
const data_prefix = 'data_';
const data = Object.fromEntries(
searchParams
.entries()
Array.from(searchParams.entries())
.filter(([key]) => key.startsWith(data_prefix))
.map(([key, value]) => [key.replace(data_prefix, ''), value]),
);
@ -72,6 +72,7 @@ export const useSendSharedMessage = () => {
quote: true,
question: message.content,
session_id: get(derivedMessages, '0.session_id'),
reasoning: message.reasoning,
});
if (isCompletionError(res)) {
@ -118,14 +119,14 @@ export const useSendSharedMessage = () => {
}, [answer, addNewestAnswer]);
const handlePressEnter = useCallback(
(documentIds: string[]) => {
(...[{ enableThinking }]: NextMessageInputOnPressEnterParameter) => {
if (trim(value) === '') return;
const id = uuid();
if (done) {
setValue('');
addNewestQuestion({
content: value,
doc_ids: documentIds,
doc_ids: [],
id,
role: MessageType.User,
});
@ -133,6 +134,7 @@ export const useSendSharedMessage = () => {
content: value.trim(),
id,
role: MessageType.User,
reasoning: enableThinking,
});
}
},

1
web/src/pages/next-chats/share/index.tsx
View File

@ -123,6 +123,7 @@ const ChatContainer = () => {
uploadMethod="external_upload_and_parse"
showUploadIcon={false}
stopOutputMessage={stopOutputMessage}
showReasoning
></NextMessageInput>
</div>
</div>