dify

dify/web/app/components/base/param-item/top-k-item.tsx

@@ -0,0 +1,53 @@
+'use client'
+import type { FC } from 'react'
+import React from 'react'
+import { useTranslation } from 'react-i18next'
+import ParamItem from '.'
+
+type Props = {
+  className?: string
+  value: number
+  onChange: (key: string, value: number) => void
+  enable: boolean
+}
+
+const maxTopK = (() => {
+  const configValue = Number.parseInt(globalThis.document?.body?.getAttribute('data-public-top-k-max-value') || '', 10)
+  if (configValue && !isNaN(configValue))
+    return configValue
+  return 10
+})()
+const VALUE_LIMIT = {
+  default: 2,
+  step: 1,
+  min: 1,
+  max: maxTopK,
+}
+
+const TopKItem: FC<Props> = ({
+  className,
+  value,
+  enable,
+  onChange,
+}) => {
+  const { t } = useTranslation()
+  const handleParamChange = (key: string, value: number) => {
+    let notOutRangeValue = Number.parseInt(value.toFixed(0))
+    notOutRangeValue = Math.max(VALUE_LIMIT.min, notOutRangeValue)
+    notOutRangeValue = Math.min(VALUE_LIMIT.max, notOutRangeValue)
+    onChange(key, notOutRangeValue)
+  }
+  return (
+    <ParamItem
+      className={className}
+      id='top_k'
+      name={t('appDebug.datasetConfig.top_k')}
+      tip={t('appDebug.datasetConfig.top_kTip') as string}
+      {...VALUE_LIMIT}
+      value={value}
+      enable={enable}
+      onChange={handleParamChange}
+    />
+  )
+}
+export default React.memo(TopKItem)