model-trigger.tsx 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. import type { FC } from 'react'
  2. import { RiArrowDownSLine } from '@remixicon/react'
  3. import type {
  4. Model,
  5. ModelItem,
  6. } from '../declarations'
  7. import {
  8. MODEL_STATUS_TEXT,
  9. ModelStatusEnum,
  10. } from '../declarations'
  11. import { useLanguage } from '../hooks'
  12. import ModelIcon from '../model-icon'
  13. import ModelName from '../model-name'
  14. import { AlertTriangle } from '@/app/components/base/icons/src/vender/line/alertsAndFeedback'
  15. import Tooltip from '@/app/components/base/tooltip'
  16. import classNames from '@/utils/classnames'
  17. type ModelTriggerProps = {
  18. open: boolean
  19. provider: Model
  20. model: ModelItem
  21. className?: string
  22. readonly?: boolean
  23. }
  24. const ModelTrigger: FC<ModelTriggerProps> = ({
  25. open,
  26. provider,
  27. model,
  28. className,
  29. readonly,
  30. }) => {
  31. const language = useLanguage()
  32. return (
  33. <div
  34. className={classNames(
  35. 'group flex items-center px-2 h-8 rounded-lg bg-components-input-bg-normal',
  36. !readonly && 'hover:bg-components-input-bg-hover cursor-pointer',
  37. !!readonly && 'opacity-50',
  38. className,
  39. open && '!bg-components-input-bg-hover',
  40. model.status !== ModelStatusEnum.active && '!bg-[#FFFAEB]',
  41. )}
  42. >
  43. <ModelIcon
  44. className='shrink-0 mr-1.5'
  45. provider={provider}
  46. modelName={model.model}
  47. />
  48. <ModelName
  49. className='grow'
  50. modelItem={model}
  51. showMode
  52. showFeatures
  53. />
  54. {!readonly && (
  55. <div className='shrink-0 flex items-center justify-center w-4 h-4'>
  56. {
  57. model.status !== ModelStatusEnum.active
  58. ? (
  59. <Tooltip popupContent={MODEL_STATUS_TEXT[model.status][language]}>
  60. <AlertTriangle className='w-4 h-4 text-[#F79009]' />
  61. </Tooltip>
  62. )
  63. : (
  64. <RiArrowDownSLine
  65. className='w-3.5 h-3.5 text-gray-500'
  66. />
  67. )
  68. }
  69. </div>
  70. )}
  71. </div>
  72. )
  73. }
  74. export default ModelTrigger