Преглед изворни кода

feat: user message support generate prompt (#25689)

tags/1.9.0
Joel пре 1 месец
родитељ
комит
9bb7bcf52e
No account linked to committer's email address

+ 5
- 2
web/app/components/app/configuration/config/automatic/get-automatic-res.tsx Прегледај датотеку

@@ -50,6 +50,7 @@ export type IGetAutomaticResProps = {
onFinished: (res: GenRes) => void
flowId?: string
nodeId?: string
editorId?: string
currentPrompt?: string
isBasicMode?: boolean
}
@@ -76,6 +77,7 @@ const GetAutomaticRes: FC<IGetAutomaticResProps> = ({
onClose,
flowId,
nodeId,
editorId,
currentPrompt,
isBasicMode,
onFinished,
@@ -132,7 +134,8 @@ const GetAutomaticRes: FC<IGetAutomaticResProps> = ({
},
]

const [instructionFromSessionStorage, setInstruction] = useSessionStorageState<string>(`improve-instruction-${flowId}${isBasicMode ? '' : `-${nodeId}`}`)
// eslint-disable-next-line sonarjs/no-nested-template-literals, sonarjs/no-nested-conditional
const [instructionFromSessionStorage, setInstruction] = useSessionStorageState<string>(`improve-instruction-${flowId}${isBasicMode ? '' : `-${nodeId}${editorId ? `-${editorId}` : ''}`}`)
const instruction = instructionFromSessionStorage || ''
const [ideaOutput, setIdeaOutput] = useState<string>('')

@@ -166,7 +169,7 @@ const GetAutomaticRes: FC<IGetAutomaticResProps> = ({
return true
}
const [isLoading, { setTrue: setLoadingTrue, setFalse: setLoadingFalse }] = useBoolean(false)
const storageKey = `${flowId}${isBasicMode ? '' : `-${nodeId}`}`
const storageKey = `${flowId}${isBasicMode ? '' : `-${nodeId}${editorId ? `-${editorId}` : ''}`}`
const { addVersion, current, currentVersionIndex, setCurrentVersionIndex, versions } = useGenData({
storageKey,
})

+ 3
- 0
web/app/components/workflow/nodes/_base/components/prompt/editor.tsx Прегледај датотеку

@@ -42,6 +42,7 @@ type Props = {
headerClassName?: string
instanceId?: string
nodeId?: string
editorId?: string
title: string | React.JSX.Element
value: string
onChange: (value: string) => void
@@ -85,6 +86,7 @@ const Editor: FC<Props> = ({
headerClassName,
instanceId,
nodeId,
editorId,
title,
value,
onChange,
@@ -163,6 +165,7 @@ const Editor: FC<Props> = ({
{isSupportPromptGenerator && (
<PromptGeneratorBtn
nodeId={nodeId!}
editorId={editorId}
className='ml-[5px]'
onGenerated={onGenerated}
modelConfig={modelConfig}

+ 2
- 1
web/app/components/workflow/nodes/llm/components/config-prompt-item.tsx Прегледај датотеку

@@ -136,7 +136,8 @@ const ConfigPromptItem: FC<Props> = ({
nodesOutputVars={availableVars}
availableNodes={availableNodes}
nodeId={nodeId}
isSupportPromptGenerator={payload.role === PromptRole.system}
editorId={id}
isSupportPromptGenerator
onGenerated={handleGenerated}
modelConfig={modelConfig}
isSupportJinja

+ 3
- 0
web/app/components/workflow/nodes/llm/components/prompt-generator-btn.tsx Прегледај датотеку

@@ -16,6 +16,7 @@ type Props = {
onGenerated?: (prompt: string) => void
modelConfig?: ModelConfig
nodeId: string
editorId?: string
currentPrompt?: string
}

@@ -23,6 +24,7 @@ const PromptGeneratorBtn: FC<Props> = ({
className,
onGenerated,
nodeId,
editorId,
currentPrompt,
}) => {
const [showAutomatic, { setTrue: showAutomaticTrue, setFalse: showAutomaticFalse }] = useBoolean(false)
@@ -46,6 +48,7 @@ const PromptGeneratorBtn: FC<Props> = ({
onFinished={handleAutomaticRes}
flowId={configsMap?.flowId || ''}
nodeId={nodeId}
editorId={editorId}
currentPrompt={currentPrompt}
/>
)}

Loading…
Откажи
Сачувај