Feat: Add a web search button to the chat box on the chat page. (#12786)

### What problem does this PR solve?

Feat: Add a web search button to the chat box on the chat page.

### Type of change


- [x] New Feature (non-breaking change which adds functionality)

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

@@ -291,6 +291,7 @@ export const useSendAgentMessage = ({
 
         params.session_id = sessionId;
         params.reasoning = message.reasoning;
+        params.internet = message.internet;
       }
 
       try {
@@ -358,13 +359,21 @@ export const useSendAgentMessage = ({
   ]);
 
   const handlePressEnter = useCallback(
-    (...[{ enableThinking }]: NextMessageInputOnPressEnterParameter) => {
+    (
+      ...[
+        { enableThinking, enableInternet },
+      ]: NextMessageInputOnPressEnterParameter
+    ) => {
       if (trim(value) === '') return;
       const msgBody = buildRequestBody(value);
       if (done) {
         setValue('');
         sendMessage({
-          message: { ...msgBody, reasoning: enableThinking },
+          message: {
+            ...msgBody,
+            reasoning: enableThinking,
+            internet: enableInternet,
+          },
         });
       }
       addNewestOneQuestion({ ...msgBody, files: fileList });

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

@@ -39,6 +39,7 @@ import { useSendMessage } from '../../hooks/use-send-chat-message';
 import { useSendMultipleChatMessage } from '../../hooks/use-send-multiple-message';
 import { buildMessageItemReference } from '../../utils';
 import { useAddChatBox } from '../use-add-box';
+import { useShowInternet } from '../use-show-internet';
 import { useSetDefaultModel } from './use-set-default-model';
 
 type MultipleChatBoxProps = {
@@ -226,6 +227,8 @@ export function MultipleChatBox({
   const { visible, hideModal, documentId, selectedChunk, clickDocumentButton } =
     useClickDrawer();
 
+  const showInternet = useShowInternet();
+
   return (
     <section className="h-full flex flex-col px-5">
       <div className="flex gap-4 flex-1 px-5 pb-14 min-h-0">
@@ -261,6 +264,7 @@ export function MultipleChatBox({
           stopOutputMessage={stopOutputMessage}
           onUpload={handleUploadFile}
           showReasoning
+          showInternet={showInternet}
         />
       </div>
       {visible && (

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

@@ -18,6 +18,7 @@ import {
 import { useCreateConversationBeforeUploadDocument } from '../../hooks/use-create-conversation';
 import { useSendMessage } from '../../hooks/use-send-chat-message';
 import { buildMessageItemReference } from '../../utils';
+import { useShowInternet } from '../use-show-internet';
 
 interface IProps {
   controller: AbortController;
@@ -55,6 +56,8 @@ export function SingleChatBox({
   const { visible, hideModal, documentId, selectedChunk, clickDocumentButton } =
     useClickDrawer();
 
+  const showInternet = useShowInternet();
+
   useEffect(() => {
     const messages = conversation?.message;
     if (Array.isArray(messages)) {
@@ -120,6 +123,7 @@ export function SingleChatBox({
         isUploading={isUploading}
         removeFile={removeFile}
         showReasoning
+        showInternet={showInternet}
       />
       {visible && (
         <PdfSheet

web/src/pages/next-chats/chat/use-show-internet.ts

@@ -0,0 +1,8 @@
+import { useFetchDialog } from '@/hooks/use-chat-request';
+import { isEmpty } from 'lodash';
+
+export function useShowInternet() {
+  const { data: currentDialog } = useFetchDialog();
+
+  return !isEmpty(currentDialog?.prompt_config?.tavily_api_key);
+}

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

@@ -135,7 +135,11 @@ export const useSendMessage = (controller: AbortController) => {
     useCreateConversationBeforeSendMessage();
 
   const handlePressEnter = useCallback(
-    async (...[{ enableThinking }]: NextMessageInputOnPressEnterParameter) => {
+    async (
+      ...[
+        { enableThinking, enableInternet },
+      ]: NextMessageInputOnPressEnterParameter
+    ) => {
       if (trim(value) === '') return;
 
       const data = await createConversationBeforeSendMessage(value);
@@ -168,6 +172,7 @@ export const useSendMessage = (controller: AbortController) => {
             files: files,
             conversationId: targetConversationId,
             reasoning: enableThinking,
+            internet: enableInternet,
           },
         });
       }

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

@@ -177,7 +177,11 @@ export function useSendMultipleChatMessage(
   );
 
   const handlePressEnter = useCallback(
-    async (...[{ enableThinking }]: NextMessageInputOnPressEnterParameter) => {
+    async (
+      ...[
+        { enableThinking, enableInternet },
+      ]: NextMessageInputOnPressEnterParameter
+    ) => {
       if (trim(value) === '') return;
       const id = uuid();
 
@@ -214,6 +218,7 @@ export function useSendMultipleChatMessage(
                 files,
                 conversationId: targetConversationId,
                 reasoning: enableThinking,
+                internet: enableInternet,
               },
               chatBoxId,
               currentConversationId: targetConversationId,

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

@@ -73,6 +73,7 @@ export const useSendSharedMessage = () => {
         question: message.content,
         session_id: get(derivedMessages, '0.session_id'),
         reasoning: message.reasoning,
+        internet: message.internet,
       });
 
       if (isCompletionError(res)) {
@@ -119,7 +120,11 @@ export const useSendSharedMessage = () => {
   }, [answer, addNewestAnswer]);
 
   const handlePressEnter = useCallback(
-    (...[{ enableThinking }]: NextMessageInputOnPressEnterParameter) => {
+    (
+      ...[
+        { enableThinking, enableInternet },
+      ]: NextMessageInputOnPressEnterParameter
+    ) => {
       if (trim(value) === '') return;
       const id = uuid();
       if (done) {
@@ -135,6 +140,7 @@ export const useSendSharedMessage = () => {
           id,
           role: MessageType.User,
           reasoning: enableThinking,
+          internet: enableInternet,
         });
       }
     },

web/src/pages/next-chats/share/index.tsx

@@ -124,6 +124,7 @@ const ChatContainer = () => {
                 showUploadIcon={false}
                 stopOutputMessage={stopOutputMessage}
                 showReasoning
+                showInternet={chatInfo?.has_tavily_key}
               ></NextMessageInput>
             </div>
           </div>