index.tsx 3.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. import { useTranslation } from 'react-i18next'
  2. import useSWR from 'swr'
  3. import { useContext } from 'use-context-selector'
  4. import { FileSearch02 } from '@/app/components/base/icons/src/vender/solid/files'
  5. import { Settings01 } from '@/app/components/base/icons/src/vender/line/general'
  6. import { useModalContext } from '@/context/modal-context'
  7. import ConfigContext from '@/context/debug-configuration'
  8. import { fetchCodeBasedExtensionList } from '@/service/common'
  9. import I18n from '@/context/i18n'
  10. import { getModelRuntimeSupported } from '@/utils/language'
  11. const Moderation = () => {
  12. const { t } = useTranslation()
  13. const { setShowModerationSettingModal } = useModalContext()
  14. const { locale } = useContext(I18n)
  15. const language = getModelRuntimeSupported(locale)
  16. const {
  17. moderationConfig,
  18. setModerationConfig,
  19. } = useContext(ConfigContext)
  20. const { data: codeBasedExtensionList } = useSWR(
  21. '/code-based-extension?module=moderation',
  22. fetchCodeBasedExtensionList,
  23. )
  24. const handleOpenModerationSettingModal = () => {
  25. setShowModerationSettingModal({
  26. payload: moderationConfig,
  27. onSaveCallback: setModerationConfig,
  28. })
  29. }
  30. const renderInfo = () => {
  31. let prefix = ''
  32. let suffix = ''
  33. if (moderationConfig.type === 'openai_moderation')
  34. prefix = t('appDebug.feature.moderation.modal.provider.openai')
  35. else if (moderationConfig.type === 'keywords')
  36. prefix = t('appDebug.feature.moderation.modal.provider.keywords')
  37. else if (moderationConfig.type === 'api')
  38. prefix = t('common.apiBasedExtension.selector.title')
  39. else
  40. prefix = codeBasedExtensionList?.data.find(item => item.name === moderationConfig.type)?.label[language] || ''
  41. if (moderationConfig.config?.inputs_config?.enabled && moderationConfig.config?.outputs_config?.enabled)
  42. suffix = t('appDebug.feature.moderation.allEnabled')
  43. else if (moderationConfig.config?.inputs_config?.enabled)
  44. suffix = t('appDebug.feature.moderation.inputEnabled')
  45. else if (moderationConfig.config?.outputs_config?.enabled)
  46. suffix = t('appDebug.feature.moderation.outputEnabled')
  47. return `${prefix} · ${suffix}`
  48. }
  49. return (
  50. <div className='flex items-center px-3 h-12 bg-gray-50 rounded-xl overflow-hidden'>
  51. <div className='shrink-0 flex items-center justify-center mr-1 w-6 h-6'>
  52. <FileSearch02 className='shrink-0 w-4 h-4 text-[#039855]' />
  53. </div>
  54. <div className='shrink-0 mr-2 whitespace-nowrap text-sm text-gray-800 font-semibold'>
  55. {t('appDebug.feature.moderation.title')}
  56. </div>
  57. <div
  58. className='grow block w-0 text-right text-xs text-gray-500 truncate'
  59. title={renderInfo()}>
  60. {renderInfo()}
  61. </div>
  62. <div className='shrink-0 ml-4 mr-1 w-[1px] h-3.5 bg-gray-200'></div>
  63. <div
  64. className={`
  65. shrink-0 flex items-center px-3 h-7 cursor-pointer rounded-md
  66. text-xs text-gray-700 font-medium hover:bg-gray-200
  67. `}
  68. onClick={handleOpenModerationSettingModal}
  69. >
  70. <Settings01 className='mr-[5px] w-3.5 h-3.5' />
  71. {t('common.operation.settings')}
  72. </div>
  73. </div>
  74. )
  75. }
  76. export default Moderation