model-trigger.tsx 1.9 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. className,
  38. open && '!bg-components-input-bg-hover',
  39. model.status !== ModelStatusEnum.active && '!bg-[#FFFAEB]',
  40. )}
  41. >
  42. <ModelIcon
  43. className='shrink-0 mr-1.5'
  44. provider={provider}
  45. modelName={model.model}
  46. />
  47. <ModelName
  48. className='grow'
  49. modelItem={model}
  50. showMode
  51. showFeatures
  52. />
  53. {!readonly && (
  54. <div className='shrink-0 flex items-center justify-center w-4 h-4'>
  55. {
  56. model.status !== ModelStatusEnum.active
  57. ? (
  58. <Tooltip popupContent={MODEL_STATUS_TEXT[model.status][language]}>
  59. <AlertTriangle className='w-4 h-4 text-[#F79009]' />
  60. </Tooltip>
  61. )
  62. : (
  63. <RiArrowDownSLine
  64. className='w-3.5 h-3.5 text-gray-500'
  65. />
  66. )
  67. }
  68. </div>
  69. )}
  70. </div>
  71. )
  72. }
  73. export default ModelTrigger