import type { CSSProperties, FC } from 'react'
|
import { useTranslation } from 'react-i18next'
|
import { memo } from 'react'
|
import type { ModelAndParameter } from '../types'
|
import ModelParameterTrigger from './model-parameter-trigger'
|
import ChatItem from './chat-item'
|
import TextGenerationItem from './text-generation-item'
|
import { useDebugWithMultipleModelContext } from './context'
|
import { useDebugConfigurationContext } from '@/context/debug-configuration'
|
import Dropdown from '@/app/components/base/dropdown'
|
import type { Item } from '@/app/components/base/dropdown'
|
import { useProviderContext } from '@/context/provider-context'
|
import { ModelStatusEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
|
|
type DebugItemProps = {
|
modelAndParameter: ModelAndParameter
|
className?: string
|
style?: CSSProperties
|
}
|
const DebugItem: FC<DebugItemProps> = ({
|
modelAndParameter,
|
className,
|
style,
|
}) => {
|
const { t } = useTranslation()
|
const { mode } = useDebugConfigurationContext()
|
const {
|
multipleModelConfigs,
|
onMultipleModelConfigsChange,
|
onDebugWithMultipleModelChange,
|
} = useDebugWithMultipleModelContext()
|
const { textGenerationModelList } = useProviderContext()
|
|
const index = multipleModelConfigs.findIndex(v => v.id === modelAndParameter.id)
|
const currentProvider = textGenerationModelList.find(item => item.provider === modelAndParameter.provider)
|
const currentModel = currentProvider?.models.find(item => item.model === modelAndParameter.model)
|
|
const handleSelect = (item: Item) => {
|
if (item.value === 'duplicate') {
|
if (multipleModelConfigs.length >= 4)
|
return
|
|
onMultipleModelConfigsChange(
|
true,
|
[
|
...multipleModelConfigs.slice(0, index + 1),
|
{
|
...modelAndParameter,
|
id: `${Date.now()}`,
|
},
|
...multipleModelConfigs.slice(index + 1),
|
],
|
)
|
}
|
if (item.value === 'debug-as-single-model')
|
onDebugWithMultipleModelChange(modelAndParameter)
|
if (item.value === 'remove') {
|
onMultipleModelConfigsChange(
|
true,
|
multipleModelConfigs.filter(item => item.id !== modelAndParameter.id),
|
)
|
}
|
}
|
|
return (
|
<div
|
className={`flex flex-col min-w-[320px] rounded-xl bg-white border-[0.5px] border-black/5 ${className}`}
|
style={style}
|
>
|
<div className='shrink-0 flex items-center justify-between h-10 px-3 border-b-[0.5px] border-b-black/5'>
|
<div className='flex items-center justify-center w-6 h-5 font-medium italic text-gray-500'>
|
#{index + 1}
|
</div>
|
<ModelParameterTrigger
|
modelAndParameter={modelAndParameter}
|
/>
|
<Dropdown
|
onSelect={handleSelect}
|
items={[
|
...(
|
multipleModelConfigs.length <= 3
|
? [
|
{
|
value: 'duplicate',
|
text: t('appDebug.duplicateModel'),
|
},
|
]
|
: []
|
),
|
...(
|
(modelAndParameter.provider && modelAndParameter.model)
|
? [
|
{
|
value: 'debug-as-single-model',
|
text: t('appDebug.debugAsSingleModel'),
|
},
|
]
|
: []
|
),
|
]}
|
secondItems={
|
multipleModelConfigs.length > 2
|
? [
|
{
|
value: 'remove',
|
text: t('common.operation.remove') as string,
|
},
|
]
|
: undefined
|
}
|
/>
|
</div>
|
<div style={{ height: 'calc(100% - 40px)' }}>
|
{
|
(mode === 'chat' || mode === 'agent-chat') && currentProvider && currentModel && currentModel.status === ModelStatusEnum.active && (
|
<ChatItem modelAndParameter={modelAndParameter} />
|
)
|
}
|
{
|
mode === 'completion' && currentProvider && currentModel && currentModel.status === ModelStatusEnum.active && (
|
<TextGenerationItem modelAndParameter={modelAndParameter}/>
|
)
|
}
|
</div>
|
</div>
|
)
|
}
|
|
export default memo(DebugItem)
|
