reasoning-mode-picker.tsx 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. 'use client'
  2. import type { FC } from 'react'
  3. import React, { useCallback } from 'react'
  4. import { useTranslation } from 'react-i18next'
  5. import { ReasoningModeType } from '../types'
  6. import Field from '../../_base/components/field'
  7. import OptionCard from '../../_base/components/option-card'
  8. const i18nPrefix = 'workflow.nodes.parameterExtractor'
  9. interface Props {
  10. type: ReasoningModeType
  11. onChange: (type: ReasoningModeType) => void
  12. }
  13. const ReasoningModePicker: FC<Props> = ({
  14. type,
  15. onChange,
  16. }) => {
  17. const { t } = useTranslation()
  18. const handleChange = useCallback((type: ReasoningModeType) => {
  19. return () => {
  20. onChange(type)
  21. }
  22. }, [onChange])
  23. return (
  24. <Field
  25. title={t(`${i18nPrefix}.reasoningMode`)}
  26. tooltip={t(`${i18nPrefix}.reasoningModeTip`)!}
  27. >
  28. <div className='grid grid-cols-2 gap-x-1'>
  29. <OptionCard
  30. title='Function/Tool Calling'
  31. onSelect={handleChange(ReasoningModeType.functionCall)}
  32. selected={type === ReasoningModeType.functionCall}
  33. />
  34. <OptionCard
  35. title='Prompt'
  36. selected={type === ReasoningModeType.prompt}
  37. onSelect={handleChange(ReasoningModeType.prompt)}
  38. />
  39. </div>
  40. </Field>
  41. )
  42. }
  43. export default React.memo(ReasoningModePicker)