diff --git a/webapp/packages/chat-sdk/src/Chat/ChatFooter/index.tsx b/webapp/packages/chat-sdk/src/Chat/ChatFooter/index.tsx index 54a313c61..c81f66536 100644 --- a/webapp/packages/chat-sdk/src/Chat/ChatFooter/index.tsx +++ b/webapp/packages/chat-sdk/src/Chat/ChatFooter/index.tsx @@ -18,7 +18,7 @@ type Props = { onToggleHistoryVisible: () => void; onOpenAgents: () => void; onInputMsgChange: (value: string) => void; - onSendMsg: (msg: string, modelId?: number) => void; + onSendMsg: (msg: string, dataSetId?: number) => void; onAddConversation: (agent?: AgentType) => void; onSelectAgent: (agent: AgentType) => void; onOpenShowcase: () => void; @@ -94,7 +94,7 @@ const ChatFooter: ForwardRefRenderFunction = ( }, []); const getStepOptions = (recommends: any[]) => { - const data = groupByColumn(recommends, 'modelName'); + const data = groupByColumn(recommends, 'dataSetName'); return isMobile && recommends.length > 6 ? Object.keys(data) .slice(0, 4) @@ -110,12 +110,12 @@ const ChatFooter: ForwardRefRenderFunction = ( const processMsg = (msg: string) => { let msgValue = msg; - let modelId: number | undefined; + let dataSetId: number | undefined; if (msg?.[0] === '/') { const agent = agentList.find(item => msg.includes(`/${item.name}`)); msgValue = agent ? msg.replace(`/${agent.name}`, '') : msg; } - return { msgValue, modelId }; + return { msgValue, dataSetId }; }; const debounceGetWordsFunc = useCallback(() => { @@ -128,14 +128,13 @@ const ChatFooter: ForwardRefRenderFunction = ( } fetchRef.current += 1; const fetchId = fetchRef.current; - const { msgValue, modelId } = processMsg(msg); - const res = await searchRecommend(msgValue.trim(), chatId, modelId, currentAgent?.id); + const { msgValue, dataSetId } = processMsg(msg); + const res = await searchRecommend(msgValue.trim(), chatId, dataSetId, currentAgent?.id); if (fetchId !== fetchRef.current) { return; } const recommends = msgValue ? res.data || [] : []; const stepOptionList = recommends.map((item: any) => item.subRecommend); - if (stepOptionList.length > 0 && stepOptionList.every((item: any) => item !== null)) { setStepOptions(getStepOptions(recommends)); } else { @@ -203,12 +202,13 @@ const ChatFooter: ForwardRefRenderFunction = ( .find(item => Object.keys(stepOptions).length === 1 ? item.recommend === value - : `${item.modelName || ''}${item.recommend}` === value + : `${item.dataSetName || ''}${item.recommend}` === value ); + if (option && isSelect) { - onSendMsg(option.recommend, Object.keys(stepOptions).length > 1 ? option.modelId : undefined); + onSendMsg(option.recommend, option.dataSetIds); } else { - onSendMsg(value.trim()); + onSendMsg(value.trim(), option?.dataSetId); } }; @@ -255,14 +255,14 @@ const ChatFooter: ForwardRefRenderFunction = ( let optionValue = Object.keys(stepOptions).length === 1 ? option.recommend - : `${option.modelName || ''}${option.recommend}`; + : `${option.dataSetName || ''}${option.recommend}`; if (inputMsg[0] === '/') { const agent = agentList.find(item => inputMsg.includes(item.name)); optionValue = agent ? `/${agent.name} ${option.recommend}` : optionValue; } return (