feat: handle questioin classify

This commit is contained in:
Joel 2024-03-18 11:26:26 +08:00
parent b8ecfd859b
commit 13dbc7f0ce
4 changed files with 18 additions and 9 deletions

View File

@ -133,7 +133,7 @@ const MemoryConfig: FC<Props> = ({
<div className='flex items-center h-8 space-x-2'>
<Slider
className='w-[144px]'
value={payload.window?.size as number}
value={(payload.window?.size || WINDOW_SIZE_DEFAULT) as number}
min={WINDOW_SIZE_MIN}
max={WINDOW_SIZE_MAX}
step={1}
@ -141,7 +141,7 @@ const MemoryConfig: FC<Props> = ({
disabled={readonly}
/>
<input
value={(payload.window?.size || '') as number}
value={(payload.window?.size || WINDOW_SIZE_DEFAULT) as number}
className='shrink-0 block ml-4 pl-3 w-12 h-8 appearance-none outline-none rounded-lg bg-gray-100 text-[13px] text-gra-900'
type='number'
min={WINDOW_SIZE_MIN}

View File

@ -10,6 +10,7 @@ const i18nPrefix = 'workflow.nodes.questionClassifiers'
type Props = {
instruction: string
onInstructionChange: (instruction: string) => void
hideMemorySetting: boolean
memory: Memory
onMemoryChange: (memory: Memory) => void
}
@ -17,6 +18,7 @@ type Props = {
const AdvancedSetting: FC<Props> = ({
instruction,
onInstructionChange,
hideMemorySetting,
memory,
onMemoryChange,
}) => {
@ -37,13 +39,15 @@ const AdvancedSetting: FC<Props> = ({
</div>
)}
/>
<MemoryConfig
className='mt-4'
readonly={false}
payload={memory}
onChange={onMemoryChange}
canSetRoleName={false}
/>
{!hideMemorySetting && (
<MemoryConfig
className='mt-4'
readonly={false}
payload={memory}
onChange={onMemoryChange}
canSetRoleName={false}
/>
)}
</>
)
}

View File

@ -24,6 +24,7 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
const {
inputs,
handleModelChanged,
isChatModel,
handleCompletionParamsChange,
handleQueryVarChange,
handleTopicsChange,
@ -86,6 +87,7 @@ const Panel: FC<NodePanelProps<QuestionClassifierNodeType>> = ({
supportFold
>
<AdvancedSetting
hideMemorySetting={!isChatModel}
instruction={inputs.instruction}
onInstructionChange={handleInstructionChange}
memory={inputs.memory}

View File

@ -28,6 +28,8 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
} = useModelListAndDefaultModelAndCurrentProviderAndModel(1)
const model = inputs.model
const modelMode = inputs.model?.mode
const isChatModel = modelMode === 'chat'
const handleModelChanged = useCallback((model: { provider: string; modelId: string; mode?: string }) => {
const newInputs = produce(inputRef.current, (draft) => {
@ -133,6 +135,7 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
return {
inputs,
handleModelChanged,
isChatModel,
handleCompletionParamsChange,
handleQueryVarChange,
filterVar,