import { memo, useCallback, useImperativeHandle, useMemo } from 'react'
|
import {
|
useConfigFromDebugContext,
|
useFormattingChangedSubscription,
|
} from '../hooks'
|
import Chat from '@/app/components/base/chat/chat'
|
import { useChat } from '@/app/components/base/chat/chat/hooks'
|
import { useDebugConfigurationContext } from '@/context/debug-configuration'
|
import type { ChatConfig, ChatItem, ChatItemInTree, OnSend } from '@/app/components/base/chat/types'
|
import { useProviderContext } from '@/context/provider-context'
|
import {
|
fetchConversationMessages,
|
fetchSuggestedQuestions,
|
stopChatMessageResponding,
|
} from '@/service/debug'
|
import Avatar from '@/app/components/base/avatar'
|
import { useAppContext } from '@/context/app-context'
|
import { ModelFeatureEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
|
import { useStore as useAppStore } from '@/app/components/app/store'
|
import { useFeatures } from '@/app/components/base/features/hooks'
|
import { getLastAnswer, isValidGeneratedAnswer } from '@/app/components/base/chat/utils'
|
import type { InputForm } from '@/app/components/base/chat/chat/type'
|
import { canFindTool } from '@/utils'
|
import type { FileEntity } from '@/app/components/base/file-uploader/types'
|
|
type DebugWithSingleModelProps = {
|
checkCanSend?: () => boolean
|
}
|
export type DebugWithSingleModelRefType = {
|
handleRestart: () => void
|
}
|
const DebugWithSingleModel = (
|
{
|
ref,
|
checkCanSend,
|
}: DebugWithSingleModelProps & {
|
ref: React.RefObject<DebugWithSingleModelRefType>;
|
},
|
) => {
|
const { userProfile } = useAppContext()
|
const {
|
modelConfig,
|
appId,
|
inputs,
|
collectionList,
|
completionParams,
|
// isShowVisionConfig,
|
} = useDebugConfigurationContext()
|
const { textGenerationModelList } = useProviderContext()
|
const features = useFeatures(s => s.features)
|
const configTemplate = useConfigFromDebugContext()
|
const config = useMemo(() => {
|
return {
|
...configTemplate,
|
more_like_this: features.moreLikeThis,
|
opening_statement: features.opening?.enabled ? (features.opening?.opening_statement || '') : '',
|
suggested_questions: features.opening?.enabled ? (features.opening?.suggested_questions || []) : [],
|
sensitive_word_avoidance: features.moderation,
|
speech_to_text: features.speech2text,
|
text_to_speech: features.text2speech,
|
file_upload: features.file,
|
suggested_questions_after_answer: features.suggested,
|
retriever_resource: features.citation,
|
annotation_reply: features.annotationReply,
|
} as ChatConfig
|
}, [configTemplate, features])
|
const inputsForm = useMemo(() => {
|
return modelConfig.configs.prompt_variables.filter(item => item.type !== 'api').map(item => ({ ...item, label: item.name, variable: item.key })) as InputForm[]
|
}, [modelConfig.configs.prompt_variables])
|
const {
|
chatList,
|
setTargetMessageId,
|
isResponding,
|
handleSend,
|
suggestedQuestions,
|
handleStop,
|
handleRestart,
|
handleAnnotationAdded,
|
handleAnnotationEdited,
|
handleAnnotationRemoved,
|
} = useChat(
|
config,
|
{
|
inputs,
|
inputsForm,
|
},
|
[],
|
taskId => stopChatMessageResponding(appId, taskId),
|
)
|
useFormattingChangedSubscription(chatList)
|
|
const doSend: OnSend = useCallback((message, files, isRegenerate = false, parentAnswer: ChatItem | null = null) => {
|
if (checkCanSend && !checkCanSend())
|
return
|
const currentProvider = textGenerationModelList.find(item => item.provider === modelConfig.provider)
|
const currentModel = currentProvider?.models.find(model => model.model === modelConfig.model_id)
|
const supportVision = currentModel?.features?.includes(ModelFeatureEnum.vision)
|
|
const configData = {
|
...config,
|
model: {
|
provider: modelConfig.provider,
|
name: modelConfig.model_id,
|
mode: modelConfig.mode,
|
completion_params: completionParams,
|
},
|
}
|
|
const data: any = {
|
query: message,
|
inputs,
|
model_config: configData,
|
parent_message_id: (isRegenerate ? parentAnswer?.id : getLastAnswer(chatList)?.id) || null,
|
}
|
|
if ((config.file_upload as any)?.enabled && files?.length && supportVision)
|
data.files = files
|
|
handleSend(
|
`apps/${appId}/chat-messages`,
|
data,
|
{
|
onGetConversationMessages: (conversationId, getAbortController) => fetchConversationMessages(appId, conversationId, getAbortController),
|
onGetSuggestedQuestions: (responseItemId, getAbortController) => fetchSuggestedQuestions(appId, responseItemId, getAbortController),
|
},
|
)
|
}, [appId, chatList, checkCanSend, completionParams, config, handleSend, inputs, modelConfig.mode, modelConfig.model_id, modelConfig.provider, textGenerationModelList])
|
|
const doRegenerate = useCallback((chatItem: ChatItemInTree, editedQuestion?: { message: string, files?: FileEntity[] }) => {
|
const question = editedQuestion ? chatItem : chatList.find(item => item.id === chatItem.parentMessageId)!
|
const parentAnswer = chatList.find(item => item.id === question.parentMessageId)
|
doSend(editedQuestion ? editedQuestion.message : question.content,
|
editedQuestion ? editedQuestion.files : question.message_files,
|
true,
|
isValidGeneratedAnswer(parentAnswer) ? parentAnswer : null,
|
)
|
}, [chatList, doSend])
|
|
const allToolIcons = useMemo(() => {
|
const icons: Record<string, any> = {}
|
modelConfig.agentConfig.tools?.forEach((item: any) => {
|
icons[item.tool_name] = collectionList.find((collection: any) => canFindTool(collection.id, item.provider_id))?.icon
|
})
|
return icons
|
}, [collectionList, modelConfig.agentConfig.tools])
|
|
useImperativeHandle(ref, () => {
|
return {
|
handleRestart,
|
}
|
}, [handleRestart])
|
|
const setShowAppConfigureFeaturesModal = useAppStore(s => s.setShowAppConfigureFeaturesModal)
|
|
return (
|
<Chat
|
config={config}
|
chatList={chatList}
|
isResponding={isResponding}
|
chatContainerClassName='px-3 pt-6'
|
chatFooterClassName='px-3 pt-10 pb-0'
|
showFeatureBar
|
showFileUpload={false}
|
onFeatureBarClick={setShowAppConfigureFeaturesModal}
|
suggestedQuestions={suggestedQuestions}
|
onSend={doSend}
|
inputs={inputs}
|
inputsForm={inputsForm}
|
onRegenerate={doRegenerate}
|
switchSibling={siblingMessageId => setTargetMessageId(siblingMessageId)}
|
onStopResponding={handleStop}
|
showPromptLog
|
questionIcon={<Avatar avatar={userProfile.avatar_url} name={userProfile.name} size={40} />}
|
allToolIcons={allToolIcons}
|
onAnnotationEdited={handleAnnotationEdited}
|
onAnnotationAdded={handleAnnotationAdded}
|
onAnnotationRemoved={handleAnnotationRemoved}
|
noSpacing
|
/>
|
)
|
}
|
|
DebugWithSingleModel.displayName = 'DebugWithSingleModel'
|
|
export default memo(DebugWithSingleModel)
|
