限制每天使用量

2025-11-17 14:33:41 +08:00 · 2024-05-25 21:14:21 +08:00
parent b3654cb4cf
commit a697822445
7 changed files with 54 additions and 12 deletions
--- a/app/store/chat.ts
+++ b/app/store/chat.ts
@@ -514,6 +514,26 @@ export const useChatStore = createPersistStore(
            botMessage,
          ]);
        });
+
+        const current_day_token = parseInt(
+          localStorage.getItem("current_day_token") ?? "0",
+        );
+        // console.log('---------', current_day_token)
+        if (current_day_token >= 100000) {
+          botMessage.content +=
+            "\n\n" +
+            prettyObject({
+              error: true,
+              message: "当日请求过多。",
+            });
+          botMessage.streaming = false;
+          get().onNewMessage(botMessage);
+          set(() => ({}));
+          extAttr?.setAutoScroll(true);
+
+          return;
+        }
+
        if (
          content.toLowerCase().startsWith("/mj") ||
          content.toLowerCase().startsWith("/MJ")