407 lines
14 KiB
TypeScript
407 lines
14 KiB
TypeScript
'use client'
|
|
import type { FC } from 'react'
|
|
import React, { useState } from 'react'
|
|
import { useTranslation } from 'react-i18next'
|
|
import { useBoolean } from 'ahooks'
|
|
import { useContext } from 'use-context-selector'
|
|
import produce from 'immer'
|
|
import {
|
|
RiDeleteBinLine,
|
|
} from '@remixicon/react'
|
|
import Panel from '../base/feature-panel'
|
|
import EditModal from './config-modal'
|
|
import IconTypeIcon from './input-type-icon'
|
|
import type { IInputTypeIconProps } from './input-type-icon'
|
|
import s from './style.module.css'
|
|
import SelectVarType from './select-var-type'
|
|
import { BracketsX as VarIcon } from '@/app/components/base/icons/src/vender/line/development'
|
|
import Tooltip from '@/app/components/base/tooltip'
|
|
import type { PromptVariable } from '@/models/debug'
|
|
import { DEFAULT_VALUE_MAX_LEN, getMaxVarNameLength } from '@/config'
|
|
import { checkKeys, getNewVar } from '@/utils/var'
|
|
import Switch from '@/app/components/base/switch'
|
|
import Toast from '@/app/components/base/toast'
|
|
import { Settings01 } from '@/app/components/base/icons/src/vender/line/general'
|
|
import Confirm from '@/app/components/base/confirm'
|
|
import ConfigContext from '@/context/debug-configuration'
|
|
import { AppType } from '@/types/app'
|
|
import type { ExternalDataTool } from '@/models/common'
|
|
import { useModalContext } from '@/context/modal-context'
|
|
import { useEventEmitterContextContext } from '@/context/event-emitter'
|
|
import type { InputVar } from '@/app/components/workflow/types'
|
|
import { InputVarType } from '@/app/components/workflow/types'
|
|
|
|
export const ADD_EXTERNAL_DATA_TOOL = 'ADD_EXTERNAL_DATA_TOOL'
|
|
|
|
interface ExternalDataToolParams {
|
|
key: string
|
|
type: string
|
|
index: number
|
|
name: string
|
|
config?: Record<string, any>
|
|
icon?: string
|
|
icon_background?: string
|
|
}
|
|
|
|
export interface IConfigVarProps {
|
|
promptVariables: PromptVariable[]
|
|
readonly?: boolean
|
|
onPromptVariablesChange?: (promptVariables: PromptVariable[]) => void
|
|
}
|
|
|
|
let conflictTimer: number
|
|
|
|
const ConfigVar: FC<IConfigVarProps> = ({ promptVariables, readonly, onPromptVariablesChange }) => {
|
|
const { t } = useTranslation()
|
|
const {
|
|
mode,
|
|
dataSets,
|
|
} = useContext(ConfigContext)
|
|
const { eventEmitter } = useEventEmitterContextContext()
|
|
|
|
const hasVar = promptVariables.length > 0
|
|
const updatePromptVariable = (key: string, updateKey: string, newValue: string | boolean) => {
|
|
const newPromptVariables = promptVariables.map((item) => {
|
|
if (item.key === key) {
|
|
return {
|
|
...item,
|
|
[updateKey]: newValue,
|
|
}
|
|
}
|
|
|
|
return item
|
|
})
|
|
onPromptVariablesChange?.(newPromptVariables)
|
|
}
|
|
const [currIndex, setCurrIndex] = useState<number>(-1)
|
|
const currItem = currIndex !== -1 ? promptVariables[currIndex] : null
|
|
const currItemToEdit: InputVar | null = (() => {
|
|
if (!currItem)
|
|
return null
|
|
|
|
return {
|
|
...currItem,
|
|
label: currItem.name,
|
|
variable: currItem.key,
|
|
type: currItem.type === 'string' ? InputVarType.textInput : currItem.type,
|
|
} as InputVar
|
|
})()
|
|
const updatePromptVariableItem = (payload: InputVar) => {
|
|
const newPromptVariables = produce(promptVariables, (draft) => {
|
|
const { variable, label, type, ...rest } = payload
|
|
draft[currIndex] = {
|
|
...rest,
|
|
type: type === InputVarType.textInput ? 'string' : type,
|
|
key: variable,
|
|
name: label as string,
|
|
}
|
|
|
|
if (payload.type === InputVarType.textInput)
|
|
draft[currIndex].max_length = draft[currIndex].max_length || DEFAULT_VALUE_MAX_LEN
|
|
|
|
if (payload.type !== InputVarType.select)
|
|
delete draft[currIndex].options
|
|
})
|
|
|
|
onPromptVariablesChange?.(newPromptVariables)
|
|
}
|
|
const updatePromptKey = (index: number, newKey: string) => {
|
|
window.clearTimeout(conflictTimer)
|
|
const { isValid, errorKey, errorMessageKey } = checkKeys([newKey], true)
|
|
if (!isValid) {
|
|
Toast.notify({
|
|
type: 'error',
|
|
message: t(`appDebug.varKeyError.${errorMessageKey}`, { key: errorKey }),
|
|
})
|
|
return
|
|
}
|
|
|
|
const newPromptVariables = promptVariables.map((item, i) => {
|
|
if (i === index) {
|
|
return {
|
|
...item,
|
|
key: newKey,
|
|
}
|
|
}
|
|
return item
|
|
})
|
|
|
|
conflictTimer = window.setTimeout(() => {
|
|
const isKeyExists = promptVariables.some(item => item.key?.trim() === newKey.trim())
|
|
if (isKeyExists) {
|
|
Toast.notify({
|
|
type: 'error',
|
|
message: t('appDebug.varKeyError.keyAlreadyExists', { key: newKey }),
|
|
})
|
|
}
|
|
}, 1000)
|
|
|
|
onPromptVariablesChange?.(newPromptVariables)
|
|
}
|
|
|
|
const updatePromptNameIfNameEmpty = (index: number, newKey: string) => {
|
|
if (!newKey)
|
|
return
|
|
const newPromptVariables = promptVariables.map((item, i) => {
|
|
if (i === index && !item.name) {
|
|
return {
|
|
...item,
|
|
name: newKey,
|
|
}
|
|
}
|
|
return item
|
|
})
|
|
|
|
onPromptVariablesChange?.(newPromptVariables)
|
|
}
|
|
|
|
const { setShowExternalDataToolModal } = useModalContext()
|
|
|
|
const handleOpenExternalDataToolModal = (
|
|
{ key, type, index, name, config, icon, icon_background }: ExternalDataToolParams,
|
|
oldPromptVariables: PromptVariable[],
|
|
) => {
|
|
setShowExternalDataToolModal({
|
|
payload: {
|
|
type,
|
|
variable: key,
|
|
label: name,
|
|
config,
|
|
icon,
|
|
icon_background,
|
|
},
|
|
onSaveCallback: (newExternalDataTool: ExternalDataTool) => {
|
|
const newPromptVariables = oldPromptVariables.map((item, i) => {
|
|
if (i === index) {
|
|
return {
|
|
key: newExternalDataTool.variable as string,
|
|
name: newExternalDataTool.label as string,
|
|
enabled: newExternalDataTool.enabled,
|
|
type: newExternalDataTool.type as string,
|
|
config: newExternalDataTool.config,
|
|
required: item.required,
|
|
icon: newExternalDataTool.icon,
|
|
icon_background: newExternalDataTool.icon_background,
|
|
}
|
|
}
|
|
return item
|
|
})
|
|
onPromptVariablesChange?.(newPromptVariables)
|
|
},
|
|
onCancelCallback: () => {
|
|
if (!key)
|
|
onPromptVariablesChange?.(promptVariables.filter((_, i) => i !== index))
|
|
},
|
|
onValidateBeforeSaveCallback: (newExternalDataTool: ExternalDataTool) => {
|
|
for (let i = 0; i < promptVariables.length; i++) {
|
|
if (promptVariables[i].key === newExternalDataTool.variable && i !== index) {
|
|
Toast.notify({ type: 'error', message: t('appDebug.varKeyError.keyAlreadyExists', { key: promptVariables[i].key }) })
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
},
|
|
})
|
|
}
|
|
|
|
const handleAddVar = (type: string) => {
|
|
const newVar = getNewVar('', type)
|
|
const newPromptVariables = [...promptVariables, newVar]
|
|
onPromptVariablesChange?.(newPromptVariables)
|
|
|
|
if (type === 'api') {
|
|
handleOpenExternalDataToolModal({
|
|
type,
|
|
key: newVar.key,
|
|
name: newVar.name,
|
|
index: promptVariables.length,
|
|
}, newPromptVariables)
|
|
}
|
|
}
|
|
|
|
eventEmitter?.useSubscription((v: any) => {
|
|
if (v.type === ADD_EXTERNAL_DATA_TOOL) {
|
|
const payload = v.payload
|
|
onPromptVariablesChange?.([
|
|
...promptVariables,
|
|
{
|
|
key: payload.variable as string,
|
|
name: payload.label as string,
|
|
enabled: payload.enabled,
|
|
type: payload.type as string,
|
|
config: payload.config,
|
|
required: true,
|
|
icon: payload.icon,
|
|
icon_background: payload.icon_background,
|
|
},
|
|
])
|
|
}
|
|
})
|
|
|
|
const [isShowDeleteContextVarModal, { setTrue: showDeleteContextVarModal, setFalse: hideDeleteContextVarModal }] = useBoolean(false)
|
|
const [removeIndex, setRemoveIndex] = useState<number | null>(null)
|
|
const didRemoveVar = (index: number) => {
|
|
onPromptVariablesChange?.(promptVariables.filter((_, i) => i !== index))
|
|
}
|
|
|
|
const handleRemoveVar = (index: number) => {
|
|
const removeVar = promptVariables[index]
|
|
|
|
if (mode === AppType.completion && dataSets.length > 0 && removeVar.is_context_var) {
|
|
showDeleteContextVarModal()
|
|
setRemoveIndex(index)
|
|
return
|
|
}
|
|
didRemoveVar(index)
|
|
}
|
|
|
|
// const [currKey, setCurrKey] = useState<string | null>(null)
|
|
const [isShowEditModal, { setTrue: showEditModal, setFalse: hideEditModal }] = useBoolean(false)
|
|
|
|
const handleConfig = ({ key, type, index, name, config, icon, icon_background }: ExternalDataToolParams) => {
|
|
// setCurrKey(key)
|
|
setCurrIndex(index)
|
|
if (type !== 'string' && type !== 'paragraph' && type !== 'select' && type !== 'number') {
|
|
handleOpenExternalDataToolModal({ key, type, index, name, config, icon, icon_background }, promptVariables)
|
|
return
|
|
}
|
|
|
|
showEditModal()
|
|
}
|
|
return (
|
|
<Panel
|
|
className="mt-2"
|
|
headerIcon={
|
|
<VarIcon className='w-4 h-4 text-primary-500' />
|
|
}
|
|
title={
|
|
<div className='flex items-center'>
|
|
<div className='mr-1'>{t('appDebug.variableTitle')}</div>
|
|
{!readonly && (
|
|
<Tooltip
|
|
popupContent={
|
|
<div className='w-[180px]'>
|
|
{t('appDebug.variableTip')}
|
|
</div>
|
|
}
|
|
/>
|
|
)}
|
|
</div>
|
|
}
|
|
headerRight={!readonly ? <SelectVarType onChange={handleAddVar} /> : null}
|
|
>
|
|
{!hasVar && (
|
|
<div className='pt-2 pb-1 text-xs text-gray-500'>{t('appDebug.notSetVar')}</div>
|
|
)}
|
|
{hasVar && (
|
|
<div className='rounded-lg border border-gray-200 bg-white overflow-x-auto'>
|
|
<table className={`${s.table} min-w-[440px] w-full max-w-full border-collapse border-0 rounded-lg text-sm`}>
|
|
<thead className="border-b border-gray-200 text-gray-500 text-xs font-medium">
|
|
<tr className='uppercase'>
|
|
<td>{t('appDebug.variableTable.key')}</td>
|
|
<td>{t('appDebug.variableTable.name')}</td>
|
|
{!readonly && (
|
|
<>
|
|
<td>{t('appDebug.variableTable.optional')}</td>
|
|
<td>{t('appDebug.variableTable.action')}</td>
|
|
</>
|
|
)}
|
|
|
|
</tr>
|
|
</thead>
|
|
<tbody className="text-gray-700">
|
|
{promptVariables.map(({ key, name, type, required, config, icon, icon_background }, index) => (
|
|
<tr key={index} className="h-9 leading-9">
|
|
<td className="w-[160px] border-b border-gray-100 pl-3">
|
|
<div className='flex items-center space-x-1'>
|
|
<IconTypeIcon type={type as IInputTypeIconProps['type']} className='text-gray-400' />
|
|
{!readonly
|
|
? (
|
|
<input
|
|
type="text"
|
|
placeholder="key"
|
|
value={key}
|
|
onChange={e => updatePromptKey(index, e.target.value)}
|
|
onBlur={e => updatePromptNameIfNameEmpty(index, e.target.value)}
|
|
maxLength={getMaxVarNameLength(name)}
|
|
className="h-6 leading-6 block w-full rounded-md border-0 py-1.5 text-gray-900 placeholder:text-gray-400 focus:outline-none focus:ring-1 focus:ring-inset focus:ring-gray-200"
|
|
/>
|
|
)
|
|
: (
|
|
<div className='h-6 leading-6 text-[13px] text-gray-700'>{key}</div>
|
|
)}
|
|
</div>
|
|
</td>
|
|
<td className="py-1 border-b border-gray-100">
|
|
{!readonly
|
|
? (
|
|
<input
|
|
type="text"
|
|
placeholder={key}
|
|
value={name}
|
|
onChange={e => updatePromptVariable(key, 'name', e.target.value)}
|
|
maxLength={getMaxVarNameLength(name)}
|
|
className="h-6 leading-6 block w-full rounded-md border-0 py-1.5 text-gray-900 placeholder:text-gray-400 focus:outline-none focus:ring-1 focus:ring-inset focus:ring-gray-200"
|
|
/>)
|
|
: (
|
|
<div className='h-6 leading-6 text-[13px] text-gray-700'>{name}</div>
|
|
)}
|
|
</td>
|
|
{!readonly && (
|
|
<>
|
|
<td className='w-[84px] border-b border-gray-100'>
|
|
<div className='flex items-center h-full'>
|
|
<Switch defaultValue={!required} size='md' onChange={value => updatePromptVariable(key, 'required', !value)} />
|
|
</div>
|
|
</td>
|
|
<td className='w-20 border-b border-gray-100'>
|
|
<div className='flex h-full items-center space-x-1'>
|
|
<div className=' p-1 rounded-md hover:bg-black/5 w-6 h-6 cursor-pointer' onClick={() => handleConfig({ type, key, index, name, config, icon, icon_background })}>
|
|
<Settings01 className='w-4 h-4 text-gray-500' />
|
|
</div>
|
|
<div className=' p-1 rounded-md hover:bg-black/5 w-6 h-6 cursor-pointer' onClick={() => handleRemoveVar(index)} >
|
|
<RiDeleteBinLine className='w-4 h-4 text-gray-500' />
|
|
</div>
|
|
</div>
|
|
</td>
|
|
</>
|
|
)}
|
|
</tr>
|
|
))}
|
|
</tbody>
|
|
</table>
|
|
</div>
|
|
)}
|
|
|
|
{isShowEditModal && (
|
|
<EditModal
|
|
payload={currItemToEdit!}
|
|
isShow={isShowEditModal}
|
|
onClose={hideEditModal}
|
|
onConfirm={(item) => {
|
|
updatePromptVariableItem(item)
|
|
hideEditModal()
|
|
}}
|
|
varKeys={promptVariables.map(v => v.key)}
|
|
/>
|
|
)}
|
|
|
|
{isShowDeleteContextVarModal && (
|
|
<Confirm
|
|
isShow={isShowDeleteContextVarModal}
|
|
title={t('appDebug.feature.dataSet.queryVariable.deleteContextVarTitle', { varName: promptVariables[removeIndex as number]?.name })}
|
|
content={t('appDebug.feature.dataSet.queryVariable.deleteContextVarTip')}
|
|
onConfirm={() => {
|
|
didRemoveVar(removeIndex as number)
|
|
hideDeleteContextVarModal()
|
|
}}
|
|
onCancel={hideDeleteContextVarModal}
|
|
/>
|
|
)}
|
|
|
|
</Panel>
|
|
)
|
|
}
|
|
export default React.memo(ConfigVar)
|