estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Model Runtime (#1858)

Browse Source 
Co-authored-by: StyleZhang <jasonapring2015@outlook.com>
Co-authored-by: Garfield Dai <dai.hai@foxmail.com>
Co-authored-by: chenhe <guchenhe@gmail.com>
Co-authored-by: jyong <jyong@dify.ai>
Co-authored-by: Joel <iamjoel007@gmail.com>
Co-authored-by: Yeuoly <admin@srmxy.cn>
This commit is contained in:
takatost
2024-01-02 23:42:00 +08:00
committed by GitHub
parent e91dd28a76
commit d069c668f8
807 changed files with 171310 additions and 23806 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
web/hooks/use-pay.tsx
View File

@@ -6,7 +6,6 @@ import { useTranslation } from 'react-i18next'
import useSWR from 'swr'
import { useContext } from 'use-context-selector'
import I18n from '@/context/i18n'
import { ProviderEnum } from '@/app/components/header/account-setting/model-page/declarations'
import {
fetchDataSourceNotionBinding,
fetchFreeQuotaVerify,
@@ -24,7 +23,7 @@ export const useAnthropicCheckPay = () => {
const paymentResult = searchParams.get('payment_result')
useEffect(() => {
if (providerName === ProviderEnum.anthropic && (paymentResult === 'succeeded' || paymentResult === 'cancelled')) {
if (providerName === 'anthropic' && (paymentResult === 'succeeded' || paymentResult === 'cancelled')) {
setConfirm({
type: paymentResult === 'succeeded' ? 'success' : 'danger',
title: paymentResult === 'succeeded' ? t('common.actionMsg.paySucceeded') : t('common.actionMsg.payCancelled'),
@@ -54,8 +53,8 @@ export const useBillingPay = () => {
return confirm
}
const QUOTA_RECEIVE_STATUS = {
[ProviderEnum.spark]: {
const QUOTA_RECEIVE_STATUS: Record<string, any> = {
spark: {
success: {
'en': 'Successful collection, the quota will be automatically increased after 5 minutes.',
'zh-Hans': '领取成功，将在 5 分钟后自动增加配额',
@@ -65,7 +64,7 @@ const QUOTA_RECEIVE_STATUS = {
'zh-Hans': '领取失败',
},
},
[ProviderEnum.zhipuai]: {
zhipuai: {
success: {
'en': 'Successful collection',
'zh-Hans': '领取成功',
@@ -77,14 +76,14 @@ const QUOTA_RECEIVE_STATUS = {
},
}
const FREE_CHECK_PROVIDER = [ProviderEnum.spark, ProviderEnum.zhipuai]
const FREE_CHECK_PROVIDER = ['spark', 'zhipuai']
export const useCheckFreeQuota = () => {
const { locale } = useContext(I18n)
const router = useRouter()
const [shouldVerify, setShouldVerify] = useState(false)
const searchParams = useSearchParams()
const type = searchParams.get('type')
const provider = searchParams.get('provider') as (ProviderEnum.spark | ProviderEnum.zhipuai)
const provider = searchParams.get('provider')
const result = searchParams.get('result')
const token = searchParams.get('token')
@@ -101,14 +100,14 @@ export const useCheckFreeQuota = () => {
}, [error, router])
useEffect(() => {
if (type === 'provider_apply_callback' && FREE_CHECK_PROVIDER.includes(provider) && result === 'success')
if (type === 'provider_apply_callback' && FREE_CHECK_PROVIDER.includes(provider as string) && result === 'success')
setShouldVerify(true)
}, [type, provider, result])
return (data && provider)
? {
type: data.flag ? 'success' : 'danger',
title: data.flag ? QUOTA_RECEIVE_STATUS[provider].success[locale] : QUOTA_RECEIVE_STATUS[provider].fail[locale],
title: data.flag ? QUOTA_RECEIVE_STATUS[provider as string].success[locale] : QUOTA_RECEIVE_STATUS[provider].fail[locale],
desc: !data.flag ? data.reason : undefined,
}
: null