mirror of https://github.com/langgenius/dify.git
65 lines
1.7 KiB
TypeScript
65 lines
1.7 KiB
TypeScript
import {
|
|
memo,
|
|
useCallback,
|
|
useMemo,
|
|
} from 'react'
|
|
import { useStore } from '../../store'
|
|
import UserInput from './user-input'
|
|
import { useChat } from './hooks'
|
|
import Chat from '@/app/components/base/chat/chat'
|
|
import type { OnSend } from '@/app/components/base/chat/types'
|
|
import { useFeaturesStore } from '@/app/components/base/features/hooks'
|
|
|
|
const ChatWrapper = () => {
|
|
const {
|
|
conversationId,
|
|
chatList,
|
|
handleStop,
|
|
isResponding,
|
|
suggestedQuestions,
|
|
handleSend,
|
|
} = useChat()
|
|
const featuresStore = useFeaturesStore()
|
|
const features = featuresStore!.getState().features
|
|
|
|
const config = useMemo(() => {
|
|
return {
|
|
opening_statement: features.opening.opening_statement,
|
|
suggested_questions: features.opening.suggested_questions,
|
|
suggested_questions_after_answer: features.suggested,
|
|
text_to_speech: features.text2speech,
|
|
speech_to_text: features.speech2text,
|
|
retriever_resource: features.citation,
|
|
sensitive_word_avoidance: features.moderation,
|
|
}
|
|
}, [features])
|
|
|
|
const doSend = useCallback<OnSend>((query, files) => {
|
|
handleSend({
|
|
query,
|
|
files,
|
|
inputs: useStore.getState().inputs,
|
|
conversationId,
|
|
})
|
|
}, [conversationId, handleSend])
|
|
|
|
return (
|
|
<Chat
|
|
config={config as any}
|
|
chatList={chatList}
|
|
isResponding={isResponding}
|
|
chatContainerclassName='px-4'
|
|
chatContainerInnerClassName='pt-6'
|
|
chatFooterClassName='px-4 rounded-bl-2xl'
|
|
chatFooterInnerClassName='pb-4'
|
|
onSend={doSend}
|
|
onStopResponding={handleStop}
|
|
chatNode={<UserInput />}
|
|
allToolIcons={{}}
|
|
suggestedQuestions={suggestedQuestions}
|
|
/>
|
|
)
|
|
}
|
|
|
|
export default memo(ChatWrapper)
|