Revert "fix credit card gateway error" (#1232)
This commit is contained in:
parent
2c36a54c94
commit
8fbfc253fa
1 changed files with 15 additions and 6 deletions
|
|
@ -14,6 +14,7 @@ import type { VisibilityType } from './visibility-selector';
|
||||||
import { useArtifactSelector } from '@/hooks/use-artifact';
|
import { useArtifactSelector } from '@/hooks/use-artifact';
|
||||||
import { unstable_serialize } from 'swr/infinite';
|
import { unstable_serialize } from 'swr/infinite';
|
||||||
import { getChatHistoryPaginationKey } from './sidebar-history';
|
import { getChatHistoryPaginationKey } from './sidebar-history';
|
||||||
|
import { toast } from './toast';
|
||||||
import type { Session } from 'next-auth';
|
import type { Session } from 'next-auth';
|
||||||
import { useSearchParams } from 'next/navigation';
|
import { useSearchParams } from 'next/navigation';
|
||||||
import { useChatVisibility } from '@/hooks/use-chat-visibility';
|
import { useChatVisibility } from '@/hooks/use-chat-visibility';
|
||||||
|
|
@ -62,8 +63,7 @@ export function Chat({
|
||||||
|
|
||||||
const [input, setInput] = useState<string>('');
|
const [input, setInput] = useState<string>('');
|
||||||
const [usage, setUsage] = useState<AppUsage | undefined>(initialLastContext);
|
const [usage, setUsage] = useState<AppUsage | undefined>(initialLastContext);
|
||||||
const [showErrorAlert, setShowErrorAlert] = useState(false);
|
const [showCreditCardAlert, setShowCreditCardAlert] = useState(false);
|
||||||
const [errorMessage, setErrorMessage] = useState('');
|
|
||||||
const [currentModelId, setCurrentModelId] = useState(initialChatModel);
|
const [currentModelId, setCurrentModelId] = useState(initialChatModel);
|
||||||
const currentModelIdRef = useRef(currentModelId);
|
const currentModelIdRef = useRef(currentModelId);
|
||||||
|
|
||||||
|
|
@ -108,8 +108,17 @@ export function Chat({
|
||||||
},
|
},
|
||||||
onError: (error) => {
|
onError: (error) => {
|
||||||
if (error instanceof ChatSDKError) {
|
if (error instanceof ChatSDKError) {
|
||||||
setErrorMessage(error.message);
|
// Check if it's a credit card error
|
||||||
setShowErrorAlert(true);
|
if (
|
||||||
|
error.message?.includes('AI Gateway requires a valid credit card')
|
||||||
|
) {
|
||||||
|
setShowCreditCardAlert(true);
|
||||||
|
} else {
|
||||||
|
toast({
|
||||||
|
type: 'error',
|
||||||
|
description: error.message,
|
||||||
|
});
|
||||||
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|
@ -209,8 +218,8 @@ export function Chat({
|
||||||
/>
|
/>
|
||||||
|
|
||||||
<AlertDialog
|
<AlertDialog
|
||||||
open={showErrorAlert}
|
open={showCreditCardAlert}
|
||||||
onOpenChange={setShowErrorAlert}
|
onOpenChange={setShowCreditCardAlert}
|
||||||
>
|
>
|
||||||
<AlertDialogContent>
|
<AlertDialogContent>
|
||||||
<AlertDialogHeader>
|
<AlertDialogHeader>
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue