apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-08 23:32:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(story-3.5): add cloud-mode LLM model selection with token quota enforcement

Browse source 
Implement system-managed model catalog, subscription tier enforcement,
atomic token quota tracking, and frontend cloud/self-hosted conditional
rendering. Apply all 20 BMAD code review patches including security
fixes (cross-user API key hijack), race condition mitigation (atomic SQL
UPDATE), and SSE mid-stream quota error handling.

Co-Authored-By: Claude Sonnet 4 <noreply@anthropic.com>
This commit is contained in:
Vonic 2026-04-14 17:01:21 +07:00
parent e7382b26de
commit c1776b3ec8
19 changed files with 1003 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

60
surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx
View file

@ -14,6 +14,8 @@ import { useCallback, useEffect, useMemo, useRef, useState } from "react";
import { toast } from "sonner";
import { z } from "zod";
import { disabledToolsAtom } from "@/atoms/agent-tools/agent-tools.atoms";
import { selectedSystemModelIdAtom } from "@/atoms/new-llm-config/system-models-query.atoms";
import { isCloud } from "@/lib/env-config";
import {
clearTargetCommentIdAtom,
currentThreadAtom,
@ -173,6 +175,16 @@ function extractMentionedDocuments(content: unknown): MentionedDocumentInfo[] {
return [];
}
/**
* Throw this when the backend returns 402 Payment Required (quota exceeded).
*/
class QuotaExceededError extends Error {
constructor() {
super("Token quota exceeded");
this.name = "QuotaExceededError";
}
}
/**
* Tools that should render custom UI in the chat.
*/
@ -230,6 +242,9 @@ export default function NewChatPage() {
// Get disabled tools from the tool toggle UI
const disabledTools = useAtomValue(disabledToolsAtom);
// Cloud mode: selected system model ID (null = backend default)
const selectedSystemModelId = useAtomValue(selectedSystemModelIdAtom);
// Get mentioned document IDs from the composer (derived from @ mentions + sidebar selections)
const mentionedDocumentIds = useAtomValue(mentionedDocumentIdsAtom);
const mentionedDocuments = useAtomValue(mentionedDocumentsAtom);
@ -704,11 +719,13 @@ export default function NewChatPage() {
? mentionedDocumentIds.surfsense_doc_ids
: undefined,
disabled_tools: disabledTools.length > 0 ? disabledTools : undefined,
...(isCloud() && selectedSystemModelId != null && { model_id: selectedSystemModelId }),
}),
signal: controller.signal,
});
if (!response.ok) {
if (response.status === 402) throw new QuotaExceededError();
throw new Error(`Backend error: ${response.status}`);
}
@ -847,6 +864,9 @@ export default function NewChatPage() {
}
case "error":
if (parsed.errorText?.includes("quota") || parsed.errorText?.includes("token_quota_exceeded")) {
throw new QuotaExceededError();
}
throw new Error(parsed.errorText || "Server error");
}
}
@ -909,6 +929,15 @@ export default function NewChatPage() {
}
return;
}
if (error instanceof QuotaExceededError) {
toast.error("Monthly token quota exceeded. Upgrade your plan to continue.", {
action: {
label: "Upgrade",
onClick: () => window.open("/pricing", "_blank"),
},
});
return;
}
console.error("[NewChatPage] Chat error:", error);
// Track chat error
@ -955,6 +984,7 @@ export default function NewChatPage() {
currentUser,
disabledTools,
updateChatTabTitle,
selectedSystemModelId,
]
);
@ -1062,11 +1092,13 @@ export default function NewChatPage() {
body: JSON.stringify({
search_space_id: searchSpaceId,
decisions,
...(isCloud() && selectedSystemModelId != null && { model_id: selectedSystemModelId }),
}),
signal: controller.signal,
});
if (!response.ok) {
if (response.status === 402) throw new QuotaExceededError();
throw new Error(`Backend error: ${response.status}`);
}
@ -1175,6 +1207,9 @@ export default function NewChatPage() {
}
case "error":
if (parsed.errorText?.includes("quota") || parsed.errorText?.includes("token_quota_exceeded")) {
throw new QuotaExceededError();
}
throw new Error(parsed.errorText || "Server error");
}
}
@ -1201,6 +1236,15 @@ export default function NewChatPage() {
if (error instanceof Error && error.name === "AbortError") {
return;
}
if (error instanceof QuotaExceededError) {
toast.error("Monthly token quota exceeded. Upgrade your plan to continue.", {
action: {
label: "Upgrade",
onClick: () => window.open("/pricing", "_blank"),
},
});
return;
}
console.error("[NewChatPage] Resume error:", error);
toast.error("Failed to resume. Please try again.");
} finally {
@ -1380,11 +1424,13 @@ export default function NewChatPage() {
search_space_id: searchSpaceId,
user_query: newUserQuery || null,
disabled_tools: disabledTools.length > 0 ? disabledTools : undefined,
...(isCloud() && selectedSystemModelId != null && { model_id: selectedSystemModelId }),
}),
signal: controller.signal,
});
if (!response.ok) {
if (response.status === 402) throw new QuotaExceededError();
throw new Error(`Backend error: ${response.status}`);
}
@ -1454,6 +1500,9 @@ export default function NewChatPage() {
}
case "error":
if (parsed.errorText?.includes("quota") || parsed.errorText?.includes("token_quota_exceeded")) {
throw new QuotaExceededError();
}
throw new Error(parsed.errorText || "Server error");
}
}
@ -1502,6 +1551,15 @@ export default function NewChatPage() {
return;
}
batcher.dispose();
if (error instanceof QuotaExceededError) {
toast.error("Monthly token quota exceeded. Upgrade your plan to continue.", {
action: {
label: "Upgrade",
onClick: () => window.open("/pricing", "_blank"),
},
});
return;
}
console.error("[NewChatPage] Regeneration error:", error);
trackChatError(
searchSpaceId,
@ -1524,7 +1582,7 @@ export default function NewChatPage() {
abortControllerRef.current = null;
}
},
[threadId, searchSpaceId, messages, disabledTools]
[threadId, searchSpaceId, messages, disabledTools, selectedSystemModelId]
);
// Handle editing a message - truncates history and regenerates with new query