mirror of
https://github.com/langgenius/dify.git
synced 2026-09-03 15:27:49 +08:00
46 lines
1.6 KiB
TypeScript
46 lines
1.6 KiB
TypeScript
'use client'
|
|
import type { FC } from 'react'
|
|
import { RiHardDrive3Line } from '@remixicon/react'
|
|
import { useQuery } from '@tanstack/react-query'
|
|
import * as React from 'react'
|
|
import { useTranslation } from 'react-i18next'
|
|
import { useProviderContext } from '@/context/provider-context'
|
|
import { consoleQuery } from '@/service/client'
|
|
import UsageInfo from '../usage-info'
|
|
import { getPlanVectorSpaceLimitMB } from '../utils'
|
|
|
|
type Props = Readonly<{
|
|
className?: string
|
|
}>
|
|
|
|
// Storage threshold in MB - usage below this shows as "< 50 MB"
|
|
const STORAGE_THRESHOLD_MB = getPlanVectorSpaceLimitMB('sandbox')
|
|
|
|
const VectorSpaceInfo: FC<Props> = ({ className }) => {
|
|
const { t } = useTranslation()
|
|
const { plan } = useProviderContext()
|
|
const { data: vectorSpace } = useQuery(consoleQuery.features.vectorSpace.get.queryOptions())
|
|
const vectorSpaceUsage = vectorSpace?.size ?? plan.usage.vectorSpace
|
|
const vectorSpaceLimit = vectorSpace?.limit ?? getPlanVectorSpaceLimitMB(plan.type)
|
|
const isSandbox = plan.type === 'sandbox'
|
|
|
|
return (
|
|
<UsageInfo
|
|
className={className}
|
|
Icon={RiHardDrive3Line}
|
|
name={t(($) => $['usagePage.vectorSpace'], { ns: 'billing' })}
|
|
tooltip={t(($) => $['usagePage.vectorSpaceTooltip'], { ns: 'billing' }) as string}
|
|
usage={vectorSpaceUsage}
|
|
total={vectorSpaceLimit}
|
|
unit="MB"
|
|
unitPosition="inline"
|
|
storageMode
|
|
storageThreshold={STORAGE_THRESHOLD_MB}
|
|
storageTooltip={t(($) => $['usagePage.storageThresholdTooltip'], { ns: 'billing' }) as string}
|
|
isSandboxPlan={isSandbox}
|
|
usageUnknown={vectorSpace?.usage_unknown}
|
|
/>
|
|
)
|
|
}
|
|
export default React.memo(VectorSpaceInfo)
|