mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-07-17 00:31:45 +08:00

Signed-off-by: yihong0618 <zouzou0208@gmail.com> Signed-off-by: -LAN- <laipz8200@outlook.com> Signed-off-by: xhe <xw897002528@gmail.com> Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: takatost <takatost@gmail.com> Co-authored-by: kurokobo <kuro664@gmail.com> Co-authored-by: Novice Lee <novicelee@NoviPro.local> Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: AkaraChen <akarachen@outlook.com> Co-authored-by: Yi <yxiaoisme@gmail.com> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: JzoNg <jzongcode@gmail.com> Co-authored-by: twwu <twwu@dify.ai> Co-authored-by: Hiroshi Fujita <fujita-h@users.noreply.github.com> Co-authored-by: AkaraChen <85140972+AkaraChen@users.noreply.github.com> Co-authored-by: NFish <douxc512@gmail.com> Co-authored-by: Wu Tianwei <30284043+WTW0313@users.noreply.github.com> Co-authored-by: 非法操作 <hjlarry@163.com> Co-authored-by: Novice <857526207@qq.com> Co-authored-by: Hiroki Nagai <82458324+nagaihiroki-git@users.noreply.github.com> Co-authored-by: Gen Sato <52241300+halogen22@users.noreply.github.com> Co-authored-by: eux <euxuuu@gmail.com> Co-authored-by: huangzhuo1949 <167434202+huangzhuo1949@users.noreply.github.com> Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com> Co-authored-by: lotsik <lotsik@mail.ru> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> Co-authored-by: nite-knite <nkCoding@gmail.com> Co-authored-by: Jyong <76649700+JohnJyong@users.noreply.github.com> Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: gakkiyomi <gakkiyomi@aliyun.com> Co-authored-by: CN-P5 <heibai2006@gmail.com> Co-authored-by: CN-P5 <heibai2006@qq.com> Co-authored-by: Chuehnone <1897025+chuehnone@users.noreply.github.com> Co-authored-by: yihong <zouzou0208@gmail.com> Co-authored-by: Kevin9703 <51311316+Kevin9703@users.noreply.github.com> Co-authored-by: -LAN- <laipz8200@outlook.com> Co-authored-by: Boris Feld <lothiraldan@gmail.com> Co-authored-by: mbo <himabo@gmail.com> Co-authored-by: mabo <mabo@aeyes.ai> Co-authored-by: Warren Chen <warren.chen830@gmail.com> Co-authored-by: JzoNgKVO <27049666+JzoNgKVO@users.noreply.github.com> Co-authored-by: jiandanfeng <chenjh3@wangsu.com> Co-authored-by: zhu-an <70234959+xhdd123321@users.noreply.github.com> Co-authored-by: zhaoqingyu.1075 <zhaoqingyu.1075@bytedance.com> Co-authored-by: 海狸大師 <86974027+yenslife@users.noreply.github.com> Co-authored-by: Xu Song <xusong.vip@gmail.com> Co-authored-by: rayshaw001 <396301947@163.com> Co-authored-by: Ding Jiatong <dingjiatong@gmail.com> Co-authored-by: Bowen Liang <liangbowen@gf.com.cn> Co-authored-by: JasonVV <jasonwangiii@outlook.com> Co-authored-by: le0zh <newlight@qq.com> Co-authored-by: zhuxinliang <zhuxinliang@didiglobal.com> Co-authored-by: k-zaku <zaku99@outlook.jp> Co-authored-by: luckylhb90 <luckylhb90@gmail.com> Co-authored-by: hobo.l <hobo.l@binance.com> Co-authored-by: jiangbo721 <365065261@qq.com> Co-authored-by: 刘江波 <jiangbo721@163.com> Co-authored-by: Shun Miyazawa <34241526+miya@users.noreply.github.com> Co-authored-by: EricPan <30651140+Egfly@users.noreply.github.com> Co-authored-by: crazywoola <427733928@qq.com> Co-authored-by: sino <sino2322@gmail.com> Co-authored-by: Jhvcc <37662342+Jhvcc@users.noreply.github.com> Co-authored-by: lowell <lowell.hu@zkteco.in> Co-authored-by: Boris Polonsky <BorisPolonsky@users.noreply.github.com> Co-authored-by: Ademílson Tonato <ademilsonft@outlook.com> Co-authored-by: Ademílson Tonato <ademilson.tonato@refurbed.com> Co-authored-by: IWAI, Masaharu <iwaim.sub@gmail.com> Co-authored-by: Yueh-Po Peng (Yabi) <94939112+y10ab1@users.noreply.github.com> Co-authored-by: Jason <ggbbddjm@gmail.com> Co-authored-by: Xin Zhang <sjhpzx@gmail.com> Co-authored-by: yjc980121 <3898524+yjc980121@users.noreply.github.com> Co-authored-by: heyszt <36215648+hieheihei@users.noreply.github.com> Co-authored-by: Abdullah AlOsaimi <osaimiacc@gmail.com> Co-authored-by: Abdullah AlOsaimi <189027247+osaimi@users.noreply.github.com> Co-authored-by: Yingchun Lai <laiyingchun@apache.org> Co-authored-by: Hash Brown <hi@xzd.me> Co-authored-by: zuodongxu <192560071+zuodongxu@users.noreply.github.com> Co-authored-by: Masashi Tomooka <tmokmss@users.noreply.github.com> Co-authored-by: aplio <ryo.091219@gmail.com> Co-authored-by: Obada Khalili <54270856+obadakhalili@users.noreply.github.com> Co-authored-by: Nam Vu <zuzoovn@gmail.com> Co-authored-by: Kei YAMAZAKI <1715090+kei-yamazaki@users.noreply.github.com> Co-authored-by: TechnoHouse <13776377+deephbz@users.noreply.github.com> Co-authored-by: Riddhimaan-Senapati <114703025+Riddhimaan-Senapati@users.noreply.github.com> Co-authored-by: MaFee921 <31881301+2284730142@users.noreply.github.com> Co-authored-by: te-chan <t-nakanome@sakura-is.co.jp> Co-authored-by: HQidea <HQidea@users.noreply.github.com> Co-authored-by: Joshbly <36315710+Joshbly@users.noreply.github.com> Co-authored-by: xhe <xw897002528@gmail.com> Co-authored-by: weiwenyan-dev <154779315+weiwenyan-dev@users.noreply.github.com> Co-authored-by: ex_wenyan.wei <ex_wenyan.wei@tcl.com> Co-authored-by: engchina <12236799+engchina@users.noreply.github.com> Co-authored-by: engchina <atjapan2015@gmail.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: 呆萌闷油瓶 <253605712@qq.com> Co-authored-by: Kemal <kemalmeler@outlook.com> Co-authored-by: Lazy_Frog <4590648+lazyFrogLOL@users.noreply.github.com> Co-authored-by: Yi Xiao <54782454+YIXIAO0@users.noreply.github.com> Co-authored-by: Steven sun <98230804+Tuyohai@users.noreply.github.com> Co-authored-by: steven <sunzwj@digitalchina.com> Co-authored-by: Kalo Chin <91766386+fdb02983rhy@users.noreply.github.com> Co-authored-by: Katy Tao <34019945+KatyTao@users.noreply.github.com> Co-authored-by: depy <42985524+h4ckdepy@users.noreply.github.com> Co-authored-by: 胡春东 <gycm520@gmail.com> Co-authored-by: Junjie.M <118170653@qq.com> Co-authored-by: MuYu <mr.muzea@gmail.com> Co-authored-by: Naoki Takashima <39912547+takatea@users.noreply.github.com> Co-authored-by: Summer-Gu <37869445+gubinjie@users.noreply.github.com> Co-authored-by: Fei He <droxer.he@gmail.com> Co-authored-by: ybalbert001 <120714773+ybalbert001@users.noreply.github.com> Co-authored-by: Yuanbo Li <ybalbert@amazon.com> Co-authored-by: douxc <7553076+douxc@users.noreply.github.com> Co-authored-by: liuzhenghua <1090179900@qq.com> Co-authored-by: Wu Jiayang <62842862+Wu-Jiayang@users.noreply.github.com> Co-authored-by: Your Name <you@example.com> Co-authored-by: kimjion <45935338+kimjion@users.noreply.github.com> Co-authored-by: AugNSo <song.tiankai@icloud.com> Co-authored-by: llinvokerl <38915183+llinvokerl@users.noreply.github.com> Co-authored-by: liusurong.lsr <liusurong.lsr@alibaba-inc.com> Co-authored-by: Vasu Negi <vasu-negi@users.noreply.github.com> Co-authored-by: Hundredwz <1808096180@qq.com> Co-authored-by: Xiyuan Chen <52963600+GareArc@users.noreply.github.com>
269 lines
9.2 KiB
TypeScript
269 lines
9.2 KiB
TypeScript
import type {
|
||
FC,
|
||
ReactNode,
|
||
} from 'react'
|
||
import { useMemo, useState } from 'react'
|
||
import useSWR from 'swr'
|
||
import { useTranslation } from 'react-i18next'
|
||
import type {
|
||
DefaultModel,
|
||
FormValue,
|
||
ModelParameterRule,
|
||
} from '../declarations'
|
||
import { ModelStatusEnum } from '../declarations'
|
||
import ModelSelector from '../model-selector'
|
||
import {
|
||
useTextGenerationCurrentProviderAndModelAndModelList,
|
||
} from '../hooks'
|
||
import ParameterItem from './parameter-item'
|
||
import type { ParameterValue } from './parameter-item'
|
||
import Trigger from './trigger'
|
||
import type { TriggerProps } from './trigger'
|
||
import PresetsParameter from './presets-parameter'
|
||
import cn from '@/utils/classnames'
|
||
import {
|
||
PortalToFollowElem,
|
||
PortalToFollowElemContent,
|
||
PortalToFollowElemTrigger,
|
||
} from '@/app/components/base/portal-to-follow-elem'
|
||
import { fetchModelParameterRules } from '@/service/common'
|
||
import Loading from '@/app/components/base/loading'
|
||
import { useProviderContext } from '@/context/provider-context'
|
||
import { TONE_LIST } from '@/config'
|
||
import { ArrowNarrowLeft } from '@/app/components/base/icons/src/vender/line/arrows'
|
||
|
||
export type ModelParameterModalProps = {
|
||
popupClassName?: string
|
||
portalToFollowElemContentClassName?: string
|
||
isAdvancedMode: boolean
|
||
mode: string
|
||
modelId: string
|
||
provider: string
|
||
setModel: (model: { modelId: string; provider: string; mode?: string; features?: string[] }) => void
|
||
completionParams: FormValue
|
||
onCompletionParamsChange: (newParams: FormValue) => void
|
||
hideDebugWithMultipleModel?: boolean
|
||
debugWithMultipleModel?: boolean
|
||
onDebugWithMultipleModelChange?: () => void
|
||
renderTrigger?: (v: TriggerProps) => ReactNode
|
||
readonly?: boolean
|
||
isInWorkflow?: boolean
|
||
scope?: string
|
||
}
|
||
const stopParameterRule: ModelParameterRule = {
|
||
default: [],
|
||
help: {
|
||
en_US: 'Up to four sequences where the API will stop generating further tokens. The returned text will not contain the stop sequence.',
|
||
zh_Hans: '最多四个序列,API 将停止生成更多的 token。返回的文本将不包含停止序列。',
|
||
},
|
||
label: {
|
||
en_US: 'Stop sequences',
|
||
zh_Hans: '停止序列',
|
||
},
|
||
name: 'stop',
|
||
required: false,
|
||
type: 'tag',
|
||
tagPlaceholder: {
|
||
en_US: 'Enter sequence and press Tab',
|
||
zh_Hans: '输入序列并按 Tab 键',
|
||
},
|
||
}
|
||
|
||
const PROVIDER_WITH_PRESET_TONE = ['langgenius/openai/openai', 'langgenius/azure_openai/azure_openai']
|
||
const ModelParameterModal: FC<ModelParameterModalProps> = ({
|
||
popupClassName,
|
||
portalToFollowElemContentClassName,
|
||
isAdvancedMode,
|
||
modelId,
|
||
provider,
|
||
setModel,
|
||
completionParams,
|
||
onCompletionParamsChange,
|
||
hideDebugWithMultipleModel,
|
||
debugWithMultipleModel,
|
||
onDebugWithMultipleModelChange,
|
||
renderTrigger,
|
||
readonly,
|
||
isInWorkflow,
|
||
scope = 'text-generation',
|
||
}) => {
|
||
const { t } = useTranslation()
|
||
const { isAPIKeySet } = useProviderContext()
|
||
const [open, setOpen] = useState(false)
|
||
const { data: parameterRulesData, isLoading } = useSWR((provider && modelId) ? `/workspaces/current/model-providers/${provider}/models/parameter-rules?model=${modelId}` : null, fetchModelParameterRules)
|
||
const {
|
||
currentProvider,
|
||
currentModel,
|
||
activeTextGenerationModelList,
|
||
} = useTextGenerationCurrentProviderAndModelAndModelList(
|
||
{ provider, model: modelId },
|
||
)
|
||
|
||
const hasDeprecated = !currentProvider || !currentModel
|
||
const modelDisabled = currentModel?.status !== ModelStatusEnum.active
|
||
const disabled = !isAPIKeySet || hasDeprecated || modelDisabled
|
||
|
||
const parameterRules: ModelParameterRule[] = useMemo(() => {
|
||
return parameterRulesData?.data || []
|
||
}, [parameterRulesData])
|
||
|
||
const handleParamChange = (key: string, value: ParameterValue) => {
|
||
onCompletionParamsChange({
|
||
...completionParams,
|
||
[key]: value,
|
||
})
|
||
}
|
||
|
||
const handleChangeModel = ({ provider, model }: DefaultModel) => {
|
||
const targetProvider = activeTextGenerationModelList.find(modelItem => modelItem.provider === provider)
|
||
const targetModelItem = targetProvider?.models.find(modelItem => modelItem.model === model)
|
||
setModel({
|
||
modelId: model,
|
||
provider,
|
||
mode: targetModelItem?.model_properties.mode as string,
|
||
features: targetModelItem?.features || [],
|
||
})
|
||
}
|
||
|
||
const handleSwitch = (key: string, value: boolean, assignValue: ParameterValue) => {
|
||
if (!value) {
|
||
const newCompletionParams = { ...completionParams }
|
||
delete newCompletionParams[key]
|
||
|
||
onCompletionParamsChange(newCompletionParams)
|
||
}
|
||
if (value) {
|
||
onCompletionParamsChange({
|
||
...completionParams,
|
||
[key]: assignValue,
|
||
})
|
||
}
|
||
}
|
||
|
||
const handleSelectPresetParameter = (toneId: number) => {
|
||
const tone = TONE_LIST.find(tone => tone.id === toneId)
|
||
if (tone) {
|
||
onCompletionParamsChange({
|
||
...completionParams,
|
||
...tone.config,
|
||
})
|
||
}
|
||
}
|
||
|
||
return (
|
||
<PortalToFollowElem
|
||
open={open}
|
||
onOpenChange={setOpen}
|
||
placement={isInWorkflow ? 'left' : 'bottom-end'}
|
||
offset={4}
|
||
>
|
||
<div className='relative'>
|
||
<PortalToFollowElemTrigger
|
||
onClick={() => {
|
||
if (readonly)
|
||
return
|
||
setOpen(v => !v)
|
||
}}
|
||
className='block'
|
||
>
|
||
{
|
||
renderTrigger
|
||
? renderTrigger({
|
||
open,
|
||
disabled,
|
||
modelDisabled,
|
||
hasDeprecated,
|
||
currentProvider,
|
||
currentModel,
|
||
providerName: provider,
|
||
modelId,
|
||
})
|
||
: (
|
||
<Trigger
|
||
disabled={disabled}
|
||
isInWorkflow={isInWorkflow}
|
||
modelDisabled={modelDisabled}
|
||
hasDeprecated={hasDeprecated}
|
||
currentProvider={currentProvider}
|
||
currentModel={currentModel}
|
||
providerName={provider}
|
||
modelId={modelId}
|
||
/>
|
||
)
|
||
}
|
||
</PortalToFollowElemTrigger>
|
||
<PortalToFollowElemContent className={cn('z-[60]', portalToFollowElemContentClassName)}>
|
||
<div className={cn(popupClassName, 'w-[389px] rounded-2xl border-[0.5px] border-components-panel-border bg-components-panel-bg shadow-lg')}>
|
||
<div className={cn('max-h-[420px] p-4 pt-3 overflow-y-auto')}>
|
||
<div className='relative'>
|
||
<div className={cn('mb-1 h-6 flex items-center text-text-secondary system-sm-semibold')}>
|
||
{t('common.modelProvider.model').toLocaleUpperCase()}
|
||
</div>
|
||
<ModelSelector
|
||
defaultModel={(provider || modelId) ? { provider, model: modelId } : undefined}
|
||
modelList={activeTextGenerationModelList}
|
||
onSelect={handleChangeModel}
|
||
/>
|
||
</div>
|
||
{
|
||
!!parameterRules.length && (
|
||
<div className='my-3 h-[1px] bg-divider-subtle' />
|
||
)
|
||
}
|
||
{
|
||
isLoading && (
|
||
<div className='mt-5'><Loading /></div>
|
||
)
|
||
}
|
||
{
|
||
!isLoading && !!parameterRules.length && (
|
||
<div className='flex items-center justify-between mb-2'>
|
||
<div className={cn('h-6 flex items-center text-text-secondary system-sm-semibold')}>{t('common.modelProvider.parameters')}</div>
|
||
{
|
||
PROVIDER_WITH_PRESET_TONE.includes(provider) && (
|
||
<PresetsParameter onSelect={handleSelectPresetParameter} />
|
||
)
|
||
}
|
||
</div>
|
||
)
|
||
}
|
||
{
|
||
!isLoading && !!parameterRules.length && (
|
||
[
|
||
...parameterRules,
|
||
...(isAdvancedMode ? [stopParameterRule] : []),
|
||
].map(parameter => (
|
||
<ParameterItem
|
||
key={`${modelId}-${parameter.name}`}
|
||
parameterRule={parameter}
|
||
value={completionParams?.[parameter.name]}
|
||
onChange={v => handleParamChange(parameter.name, v)}
|
||
onSwitch={(checked, assignValue) => handleSwitch(parameter.name, checked, assignValue)}
|
||
isInWorkflow={isInWorkflow}
|
||
/>
|
||
))
|
||
)
|
||
}
|
||
</div>
|
||
{!hideDebugWithMultipleModel && (
|
||
<div
|
||
className='flex items-center justify-between px-4 h-[50px] bg-components-section-burn border-t border-t-divider-subtle system-sm-regular text-text-accent cursor-pointer rounded-b-xl'
|
||
onClick={() => onDebugWithMultipleModelChange?.()}
|
||
>
|
||
{
|
||
debugWithMultipleModel
|
||
? t('appDebug.debugAsSingleModel')
|
||
: t('appDebug.debugAsMultipleModel')
|
||
}
|
||
<ArrowNarrowLeft className='w-3 h-3 rotate-180' />
|
||
</div>
|
||
)}
|
||
</div>
|
||
</PortalToFollowElemContent>
|
||
</div>
|
||
</PortalToFollowElem>
|
||
)
|
||
}
|
||
|
||
export default ModelParameterModal
|