feat: support config top max value by env (#12375)

This commit is contained in:
Joel 2025-01-06 10:38:14 +08:00 • committed by GitHub
commit 3b8f6233b0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 47 additions and 54 deletions

View file

@ -11,11 +11,17 @@ type Props = {
enable: boolean
}
const maxTopK = (() => {
const configValue = 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: 10,
max: maxTopK,
}
const key = 'top_k'

View file

@ -44,6 +44,7 @@ const LocaleLayout = ({
data-public-maintenance-notice={process.env.NEXT_PUBLIC_MAINTENANCE_NOTICE}
data-public-site-about={process.env.NEXT_PUBLIC_SITE_ABOUT}
data-public-text-generation-timeout-ms={process.env.NEXT_PUBLIC_TEXT_GENERATION_TIMEOUT_MS}
data-public-top-k-max-value={process.env.NEXT_PUBLIC_TOP_K_MAX_VALUE}
>
<BrowserInitor>
<SentryInitor>