moderation-setting-modal.tsx 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386
  1. import type { ChangeEvent, FC } from 'react'
  2. import { useState } from 'react'
  3. import useSWR from 'swr'
  4. import { useContext } from 'use-context-selector'
  5. import { useTranslation } from 'react-i18next'
  6. import { RiCloseLine } from '@remixicon/react'
  7. import ModerationContent from './moderation-content'
  8. import FormGeneration from './form-generation'
  9. import ApiBasedExtensionSelector from '@/app/components/header/account-setting/api-based-extension-page/selector'
  10. import Modal from '@/app/components/base/modal'
  11. import Button from '@/app/components/base/button'
  12. import Divider from '@/app/components/base/divider'
  13. // zlwu12 2025-09-18 注释
  14. // import { BookOpen01 } from '@/app/components/base/icons/src/vender/line/education'
  15. import type { ModerationConfig, ModerationContentConfig } from '@/models/debug'
  16. import { useToastContext } from '@/app/components/base/toast'
  17. import {
  18. fetchCodeBasedExtensionList,
  19. fetchModelProviders,
  20. } from '@/service/common'
  21. import type { CodeBasedExtensionItem } from '@/models/common'
  22. import I18n from '@/context/i18n'
  23. import { LanguagesSupported } from '@/i18n-config/language'
  24. import { InfoCircle } from '@/app/components/base/icons/src/vender/line/general'
  25. import { useModalContext } from '@/context/modal-context'
  26. import { CustomConfigurationStatusEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
  27. import cn from '@/utils/classnames'
  28. import { noop } from 'lodash-es'
  29. // zlwu12 2025-09-18 注释
  30. // import { useDocLink } from '@/context/i18n'
  31. const systemTypes = ['openai_moderation', 'keywords', 'api']
  32. type Provider = {
  33. key: string
  34. name: string
  35. form_schema?: CodeBasedExtensionItem['form_schema']
  36. }
  37. type ModerationSettingModalProps = {
  38. data: ModerationConfig
  39. onCancel: () => void
  40. onSave: (moderationConfig: ModerationConfig) => void
  41. }
  42. const ModerationSettingModal: FC<ModerationSettingModalProps> = ({
  43. data,
  44. onCancel,
  45. onSave,
  46. }) => {
  47. const { t } = useTranslation()
  48. // zlwu12 2025-09-18 注释
  49. // const docLink = useDocLink()
  50. const { notify } = useToastContext()
  51. const { locale } = useContext(I18n)
  52. const { data: modelProviders, isLoading, mutate } = useSWR('/workspaces/current/model-providers', fetchModelProviders)
  53. const [localeData, setLocaleData] = useState<ModerationConfig>(data)
  54. const { setShowAccountSettingModal } = useModalContext()
  55. const handleOpenSettingsModal = () => {
  56. setShowAccountSettingModal({
  57. payload: 'provider',
  58. onCancelCallback: () => {
  59. mutate()
  60. },
  61. })
  62. }
  63. const { data: codeBasedExtensionList } = useSWR(
  64. '/code-based-extension?module=moderation',
  65. fetchCodeBasedExtensionList,
  66. )
  67. const openaiProvider = modelProviders?.data.find(item => item.provider === 'langgenius/openai/openai')
  68. const systemOpenaiProviderEnabled = openaiProvider?.system_configuration.enabled
  69. const systemOpenaiProviderQuota = systemOpenaiProviderEnabled ? openaiProvider?.system_configuration.quota_configurations.find(item => item.quota_type === openaiProvider.system_configuration.current_quota_type) : undefined
  70. const systemOpenaiProviderCanUse = systemOpenaiProviderQuota?.is_valid
  71. const customOpenaiProvidersCanUse = openaiProvider?.custom_configuration.status === CustomConfigurationStatusEnum.active
  72. const isOpenAIProviderConfigured = customOpenaiProvidersCanUse || systemOpenaiProviderCanUse
  73. const providers: Provider[] = [
  74. {
  75. key: 'openai_moderation',
  76. name: t('appDebug.feature.moderation.modal.provider.openai'),
  77. },
  78. {
  79. key: 'keywords',
  80. name: t('appDebug.feature.moderation.modal.provider.keywords'),
  81. },
  82. {
  83. key: 'api',
  84. name: t('common.apiBasedExtension.selector.title'),
  85. },
  86. ...(
  87. codeBasedExtensionList
  88. ? codeBasedExtensionList.data.map((item) => {
  89. return {
  90. key: item.name,
  91. name: locale === 'zh-Hans' ? item.label['zh-Hans'] : item.label['en-US'],
  92. form_schema: item.form_schema,
  93. }
  94. })
  95. : []
  96. ),
  97. ]
  98. const currentProvider = providers.find(provider => provider.key === localeData.type)
  99. const handleDataTypeChange = (type: string) => {
  100. let config: undefined | Record<string, any>
  101. const currProvider = providers.find(provider => provider.key === type)
  102. if (systemTypes.findIndex(t => t === type) < 0 && currProvider?.form_schema) {
  103. config = currProvider?.form_schema.reduce((prev, next) => {
  104. prev[next.variable] = next.default
  105. return prev
  106. }, {} as Record<string, any>)
  107. }
  108. setLocaleData({
  109. ...localeData,
  110. type,
  111. config,
  112. })
  113. }
  114. const handleDataKeywordsChange = (e: ChangeEvent<HTMLTextAreaElement>) => {
  115. const value = e.target.value
  116. const arr = value.split('\n').reduce((prev: string[], next: string) => {
  117. if (next !== '')
  118. prev.push(next.slice(0, 100))
  119. if (next === '' && prev[prev.length - 1] !== '')
  120. prev.push(next)
  121. return prev
  122. }, [])
  123. setLocaleData({
  124. ...localeData,
  125. config: {
  126. ...localeData.config,
  127. keywords: arr.slice(0, 100).join('\n'),
  128. },
  129. })
  130. }
  131. const handleDataContentChange = (contentType: string, contentConfig: ModerationContentConfig) => {
  132. setLocaleData({
  133. ...localeData,
  134. config: {
  135. ...localeData.config,
  136. [contentType]: contentConfig,
  137. },
  138. })
  139. }
  140. const handleDataApiBasedChange = (apiBasedExtensionId: string) => {
  141. setLocaleData({
  142. ...localeData,
  143. config: {
  144. ...localeData.config,
  145. api_based_extension_id: apiBasedExtensionId,
  146. },
  147. })
  148. }
  149. const handleDataExtraChange = (extraValue: Record<string, string>) => {
  150. setLocaleData({
  151. ...localeData,
  152. config: {
  153. ...localeData.config,
  154. ...extraValue,
  155. },
  156. })
  157. }
  158. const formatData = (originData: ModerationConfig) => {
  159. const { enabled, type, config } = originData
  160. const { inputs_config, outputs_config } = config!
  161. const params: Record<string, string | undefined> = {}
  162. if (type === 'keywords')
  163. params.keywords = config?.keywords
  164. if (type === 'api')
  165. params.api_based_extension_id = config?.api_based_extension_id
  166. if (systemTypes.findIndex(t => t === type) < 0 && currentProvider?.form_schema) {
  167. currentProvider.form_schema.forEach((form) => {
  168. params[form.variable] = config?.[form.variable]
  169. })
  170. }
  171. return {
  172. type,
  173. enabled,
  174. config: {
  175. inputs_config: inputs_config || { enabled: false },
  176. outputs_config: outputs_config || { enabled: false },
  177. ...params,
  178. },
  179. }
  180. }
  181. const handleSave = () => {
  182. if (localeData.type === 'openai_moderation' && !isOpenAIProviderConfigured)
  183. return
  184. if (!localeData.config?.inputs_config?.enabled && !localeData.config?.outputs_config?.enabled) {
  185. notify({ type: 'error', message: t('appDebug.feature.moderation.modal.content.condition') })
  186. return
  187. }
  188. if (localeData.type === 'keywords' && !localeData.config.keywords) {
  189. notify({ type: 'error', message: t('appDebug.errorMessage.valueOfVarRequired', { key: locale !== LanguagesSupported[1] ? 'keywords' : '关键词' }) })
  190. return
  191. }
  192. if (localeData.type === 'api' && !localeData.config.api_based_extension_id) {
  193. notify({ type: 'error', message: t('appDebug.errorMessage.valueOfVarRequired', { key: locale !== LanguagesSupported[1] ? 'API Extension' : 'API 扩展' }) })
  194. return
  195. }
  196. if (systemTypes.findIndex(t => t === localeData.type) < 0 && currentProvider?.form_schema) {
  197. for (let i = 0; i < currentProvider.form_schema.length; i++) {
  198. if (!localeData.config?.[currentProvider.form_schema[i].variable] && currentProvider.form_schema[i].required) {
  199. notify({
  200. type: 'error',
  201. message: t('appDebug.errorMessage.valueOfVarRequired', { key: locale !== LanguagesSupported[1] ? currentProvider.form_schema[i].label['en-US'] : currentProvider.form_schema[i].label['zh-Hans'] }),
  202. })
  203. return
  204. }
  205. }
  206. }
  207. if (localeData.config.inputs_config?.enabled && !localeData.config.inputs_config.preset_response && localeData.type !== 'api') {
  208. notify({ type: 'error', message: t('appDebug.feature.moderation.modal.content.errorMessage') })
  209. return
  210. }
  211. if (localeData.config.outputs_config?.enabled && !localeData.config.outputs_config.preset_response && localeData.type !== 'api') {
  212. notify({ type: 'error', message: t('appDebug.feature.moderation.modal.content.errorMessage') })
  213. return
  214. }
  215. onSave(formatData(localeData))
  216. }
  217. return (
  218. <Modal
  219. isShow
  220. onClose={noop}
  221. className='!mt-14 !w-[600px] !max-w-none !p-6'
  222. >
  223. <div className='flex items-center justify-between'>
  224. <div className='title-2xl-semi-bold text-text-primary'>{t('appDebug.feature.moderation.modal.title')}</div>
  225. <div className='cursor-pointer p-1' onClick={onCancel}><RiCloseLine className='h-4 w-4 text-text-tertiary' /></div>
  226. </div>
  227. <div className='py-2'>
  228. <div className='text-sm font-medium leading-9 text-text-primary'>
  229. {t('appDebug.feature.moderation.modal.provider.title')}
  230. </div>
  231. <div className='grid grid-cols-3 gap-2.5'>
  232. {
  233. providers.map(provider => (
  234. <div
  235. key={provider.key}
  236. className={cn(
  237. 'system-sm-regular flex h-8 cursor-default items-center rounded-md border border-components-option-card-option-border bg-components-option-card-option-bg px-2 text-text-secondary',
  238. localeData.type !== provider.key && 'cursor-pointer hover:border-components-option-card-option-border-hover hover:bg-components-option-card-option-bg-hover hover:shadow-xs',
  239. localeData.type === provider.key && 'system-sm-medium border-[1.5px] border-components-option-card-option-selected-border bg-components-option-card-option-selected-bg shadow-xs',
  240. localeData.type === 'openai_moderation' && provider.key === 'openai_moderation' && !isOpenAIProviderConfigured && 'text-text-disabled',
  241. )}
  242. onClick={() => handleDataTypeChange(provider.key)}
  243. >
  244. <div className={cn(
  245. 'mr-2 h-4 w-4 rounded-full border border-components-radio-border bg-components-radio-bg shadow-xs',
  246. localeData.type === provider.key && 'border-[5px] border-components-radio-border-checked',
  247. )}></div>
  248. {provider.name}
  249. </div>
  250. ))
  251. }
  252. </div>
  253. {
  254. !isLoading && !isOpenAIProviderConfigured && localeData.type === 'openai_moderation' && (
  255. <div className='mt-2 flex items-center rounded-lg border border-[#FEF0C7] bg-[#FFFAEB] px-3 py-2'>
  256. <InfoCircle className='mr-1 h-4 w-4 text-[#F79009]' />
  257. <div className='flex items-center text-xs font-medium text-gray-700'>
  258. {t('appDebug.feature.moderation.modal.openaiNotConfig.before')}
  259. <span
  260. className='cursor-pointer text-primary-600'
  261. onClick={handleOpenSettingsModal}
  262. >
  263. &nbsp;{t('common.settings.provider')}&nbsp;
  264. </span>
  265. {t('appDebug.feature.moderation.modal.openaiNotConfig.after')}
  266. </div>
  267. </div>
  268. )
  269. }
  270. </div>
  271. {
  272. localeData.type === 'keywords' && (
  273. <div className='py-2'>
  274. <div className='mb-1 text-sm font-medium text-text-primary'>{t('appDebug.feature.moderation.modal.provider.keywords')}</div>
  275. <div className='mb-2 text-xs text-text-tertiary'>{t('appDebug.feature.moderation.modal.keywords.tip')}</div>
  276. <div className='relative h-[88px] rounded-lg bg-components-input-bg-normal px-3 py-2'>
  277. <textarea
  278. value={localeData.config?.keywords || ''}
  279. onChange={handleDataKeywordsChange}
  280. className='block h-full w-full resize-none appearance-none bg-transparent text-sm text-text-secondary outline-none'
  281. placeholder={t('appDebug.feature.moderation.modal.keywords.placeholder') || ''}
  282. />
  283. <div className='absolute bottom-2 right-2 flex h-5 items-center rounded-md bg-background-section px-1 text-xs font-medium text-text-quaternary'>
  284. <span>{(localeData.config?.keywords || '').split('\n').filter(Boolean).length}</span>/<span className='text-text-tertiary'>100 {t('appDebug.feature.moderation.modal.keywords.line')}</span>
  285. </div>
  286. </div>
  287. </div>
  288. )
  289. }
  290. {
  291. localeData.type === 'api' && (
  292. <div className='py-2'>
  293. <div className='flex h-9 items-center justify-between'>
  294. <div className='text-sm font-medium text-text-primary'>{t('common.apiBasedExtension.selector.title')}</div>
  295. {/* zlwu12 2025-09-18 注释 */}
  296. {/* <a
  297. href={docLink('/guides/extension/api-based-extension/README')}
  298. target='_blank' rel='noopener noreferrer'
  299. className='group flex items-center text-xs text-text-tertiary hover:text-primary-600'
  300. >
  301. <BookOpen01 className='mr-1 h-3 w-3 text-text-tertiary group-hover:text-primary-600' />
  302. {t('common.apiBasedExtension.link')}
  303. </a> */}
  304. </div>
  305. <ApiBasedExtensionSelector
  306. value={localeData.config?.api_based_extension_id || ''}
  307. onChange={handleDataApiBasedChange}
  308. />
  309. </div>
  310. )
  311. }
  312. {
  313. systemTypes.findIndex(t => t === localeData.type) < 0
  314. && currentProvider?.form_schema
  315. && (
  316. <FormGeneration
  317. forms={currentProvider?.form_schema}
  318. value={localeData.config}
  319. onChange={handleDataExtraChange}
  320. />
  321. )
  322. }
  323. <Divider bgStyle='gradient' className='my-3 h-px' />
  324. <ModerationContent
  325. title={t('appDebug.feature.moderation.modal.content.input') || ''}
  326. config={localeData.config?.inputs_config || { enabled: false, preset_response: '' }}
  327. onConfigChange={config => handleDataContentChange('inputs_config', config)}
  328. info={(localeData.type === 'api' && t('appDebug.feature.moderation.modal.content.fromApi')) || ''}
  329. showPreset={localeData.type !== 'api'}
  330. />
  331. <ModerationContent
  332. title={t('appDebug.feature.moderation.modal.content.output') || ''}
  333. config={localeData.config?.outputs_config || { enabled: false, preset_response: '' }}
  334. onConfigChange={config => handleDataContentChange('outputs_config', config)}
  335. info={(localeData.type === 'api' && t('appDebug.feature.moderation.modal.content.fromApi')) || ''}
  336. showPreset={localeData.type !== 'api'}
  337. />
  338. <div className='mb-8 mt-1 text-xs font-medium text-text-tertiary'>{t('appDebug.feature.moderation.modal.content.condition')}</div>
  339. <div className='flex items-center justify-end'>
  340. <Button
  341. onClick={onCancel}
  342. className='mr-2'
  343. >
  344. {t('common.operation.cancel')}
  345. </Button>
  346. <Button
  347. variant='primary'
  348. onClick={handleSave}
  349. disabled={localeData.type === 'openai_moderation' && !isOpenAIProviderConfigured}
  350. >
  351. {t('common.operation.save')}
  352. </Button>
  353. </div>
  354. </Modal>
  355. )
  356. }
  357. export default ModerationSettingModal