specific error for pro user without enough credits

app/api/ask/route.ts

@@ -141,7 +141,8 @@ export async function POST(request: Request) {
           try {
             let errorPayload = "";
             if (
-              errorMessage?.includes("exceeded your monthly included credits")
+              errorMessage?.includes("exceeded your monthly included credits") ||
+              errorMessage?.includes("reached the free monthly usage limit")
             ) {
               errorPayload = JSON.stringify({
                 messageError: errorMessage,

components/ask-ai/useGeneration.ts

@@ -4,11 +4,12 @@ import { useRef } from "react";
 import { toast } from "sonner";
 import { useRouter } from "next/navigation";
 import { v4 as uuidv4 } from "uuid";
+import { useLocalStorage } from "react-use";
+import { useSession } from "next-auth/react";
 
 import { formatResponse } from "@/lib/format";
 import { File, Message, MessageActionType, ProviderType } from "@/lib/type";
 import { getContextFilesFromPrompt } from "@/lib/utils";
-import { useLocalStorage } from "react-use";
 
 const MESSAGES_QUERY_KEY = (projectName: string) =>
   ["messages", projectName] as const;
@@ -22,6 +23,7 @@ export const useGeneration = (projectName: string) => {
     `messages-${projectName}`,
     []
   );
+  const { data: session } = useSession();
 
   const { data: isLoading } = useQuery({
     queryKey: ["ai.generation.isLoading"],
@@ -348,10 +350,14 @@ export const useGeneration = (projectName: string) => {
                   isThinking: false,
                   isAborted: true,
                   content: errorData?.showProMessage
-                    ? `You have exceeded your monthly included credits with Hugging Face inference provider. Please consider upgrading to a pro plan.`
+                    ? session?.user?.isPro ? "You have already reached your monthly included credits with Hugging Face Pro plan. Please consider adding more credits to your account." : "You have exceeded your monthly included credits with Hugging Face inference provider. Please consider upgrading to a pro plan."
                     : `Error: ${errorData.messageError}`,
                   actions: errorData?.showProMessage
-                    ? [
+                    ? session?.user?.isPro ? [{
+                      label: "Add more credits",
+                      variant: "default",
+                      type: MessageActionType.ADD_CREDITS,
+                    }] : [
                         {
                           label: "Upgrade to Pro",
                           variant: "pro",

components/chat/index.tsx

@@ -19,7 +19,7 @@ import { MessageAction, MessageActionType, File } from "@/lib/type";
 import { Button } from "@/components/ui/button";
 import { getFileIcon } from "@/components/ask-ai/input-mentions";
 import ProIcon from "@/assets/pro.svg";
-import ProModal from "../pro-modal";
+import { ProModal } from "../pro-modal";
 
 export function AppEditorChat({
   isNew,
@@ -67,6 +67,11 @@ export function AppEditorChat({
         );
       case MessageActionType.UPGRADE_TO_PRO:
         return setOpenProModal(true);
+      case MessageActionType.ADD_CREDITS:
+        return window.open(
+          "https://huggingface.co/settings/billing?add-credits=true",
+          "_blank"
+        );
     }
   };
 

components/pro-modal/index.tsx

@@ -96,4 +96,3 @@ export const ProTag = ({
     PRO
   </span>
 );
-export default ProModal;

lib/type.ts

@@ -10,6 +10,7 @@ export enum MessageActionType {
   PUBLISH_PROJECT = "PUBLISH_PROJECT",
   SEE_LIVE_PREVIEW = "SEE_LIVE_PREVIEW",
   UPGRADE_TO_PRO = "UPGRADE_TO_PRO",
+  ADD_CREDITS = "ADD_CREDITS",
 }
 export type MessageAction = {
   label: string;