fix(chat): handle guest total limit

2026-06-23 15:14:17 +08:00
parent c4a33576bb
commit f14b8a48d2
5 changed files with 74 additions and 10 deletions
@@ -103,7 +103,12 @@ export function applyHttpSendOutput(
 ): Partial<ChatState> {
  const { response, reply } = output;

-  if (response.blocked === true && response.blockReason === "daily_limit") {
+  const isMessageLimitBlocked =
+    response.blocked === true &&
+    (response.blockReason === "daily_limit" ||
+      response.blockReason === "total_limit");
+
+  if (isMessageLimitBlocked) {
    const detail = response.blockDetail;
    const lastMessage = context.messages[context.messages.length - 1];
    const messages =
@@ -115,9 +120,19 @@ export function applyHttpSendOutput(
      messages,
      isReplyingAI: false,
      paywallTriggered: true,
-      paywallReason: "daily_limit",
+      paywallReason:
+        response.blockReason === "total_limit" ? "total_limit" : "daily_limit",
      paywallDetail: detail
-        ? { usedToday: detail.usedToday, limit: detail.limit }
+        ? {
+            type: detail.type,
+            ...(detail.usedToday != null
+              ? { usedToday: detail.usedToday }
+              : {}),
+            ...(detail.usedTotal != null
+              ? { usedTotal: detail.usedTotal }
+              : {}),
+            limit: detail.limit,
+          }
        : null,
    };
  }