-
+
{t("dont_have_account")}{" "}
{t("sign_up")}
diff --git a/surfsense_web/app/dashboard/[search_space_id]/client-layout.tsx b/surfsense_web/app/dashboard/[search_space_id]/client-layout.tsx
index 25e4e990b..1715e525f 100644
--- a/surfsense_web/app/dashboard/[search_space_id]/client-layout.tsx
+++ b/surfsense_web/app/dashboard/[search_space_id]/client-layout.tsx
@@ -183,6 +183,10 @@ export function DashboardClientLayout({
);
}
+ if (isOnboardingPage) {
+ return <>{children};
+ }
+
return (
diff --git a/surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/DocumentsTableShell.tsx b/surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/DocumentsTableShell.tsx
index 8a5e3cd96..92ced6e47 100644
--- a/surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/DocumentsTableShell.tsx
+++ b/surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/DocumentsTableShell.tsx
@@ -990,9 +990,10 @@ export function DocumentsTableShell({
handleDeleteFromMenu();
}}
disabled={isDeleting}
- className="bg-destructive text-destructive-foreground hover:bg-destructive/90"
+ className="relative bg-destructive text-destructive-foreground hover:bg-destructive/90"
>
- {isDeleting ? : "Delete"}
+ Delete
+ {isDeleting && }
@@ -1108,9 +1109,10 @@ export function DocumentsTableShell({
handleBulkDelete();
}}
disabled={isBulkDeleting}
- className="bg-destructive text-destructive-foreground hover:bg-destructive/90"
+ className="relative bg-destructive text-destructive-foreground hover:bg-destructive/90"
>
- {isBulkDeleting ? : "Delete"}
+ Delete
+ {isBulkDeleting && }
diff --git a/surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/RowActions.tsx b/surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/RowActions.tsx
index a8b85e20b..5b7451c61 100644
--- a/surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/RowActions.tsx
+++ b/surfsense_web/app/dashboard/[search_space_id]/documents/(manage)/components/RowActions.tsx
@@ -24,7 +24,7 @@ import {
} from "@/components/ui/dropdown-menu";
import type { Document } from "./types";
-const EDITABLE_DOCUMENT_TYPES = ["NOTE"] as const;
+const EDITABLE_DOCUMENT_TYPES = ["FILE", "NOTE"] as const;
// SURFSENSE_DOCS are system-managed and cannot be deleted
const NON_DELETABLE_DOCUMENT_TYPES = ["SURFSENSE_DOCS"] as const;
diff --git a/surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx b/surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx
index 29bbc0c5c..8928974d9 100644
--- a/surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx
+++ b/surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx
@@ -33,7 +33,7 @@ import { closeReportPanelAtom } from "@/atoms/chat/report-panel.atom";
import { type AgentCreatedDocument, agentCreatedDocumentsAtom } from "@/atoms/documents/ui.atoms";
import { closeEditorPanelAtom } from "@/atoms/editor/editor-panel.atom";
import { membersAtom } from "@/atoms/members/members-query.atoms";
-import { updateChatTabTitleAtom } from "@/atoms/tabs/tabs.atom";
+import { removeChatTabAtom, updateChatTabTitleAtom } from "@/atoms/tabs/tabs.atom";
import { currentUserAtom } from "@/atoms/user/user-query.atoms";
import { ThinkingStepsDataUI } from "@/components/assistant-ui/thinking-steps";
import { Thread } from "@/components/assistant-ui/thread";
@@ -56,6 +56,7 @@ import {
buildContentForPersistence,
buildContentForUI,
type ContentPartsState,
+ FrameBatchedUpdater,
readSSEStream,
type ThinkingStepData,
updateThinkingSteps,
@@ -69,6 +70,7 @@ import {
getThreadMessages,
type ThreadRecord,
} from "@/lib/chat/thread-persistence";
+import { NotFoundError } from "@/lib/error";
import {
trackChatCreated,
trackChatError,
@@ -130,6 +132,7 @@ function extractMentionedDocuments(content: unknown): MentionedDocumentInfo[] {
* Tools that should render custom UI in the chat.
*/
const TOOLS_WITH_UI = new Set([
+ "web_search",
"generate_podcast",
"generate_report",
"generate_video_presentation",
@@ -193,6 +196,7 @@ export default function NewChatPage() {
const closeReportPanel = useSetAtom(closeReportPanelAtom);
const closeEditorPanel = useSetAtom(closeEditorPanelAtom);
const updateChatTabTitle = useSetAtom(updateChatTabTitleAtom);
+ const removeChatTab = useSetAtom(removeChatTabAtom);
const setAgentCreatedDocuments = useSetAtom(agentCreatedDocumentsAtom);
// Get current user for author info in shared chats
@@ -272,7 +276,6 @@ export default function NewChatPage() {
// Initialize thread and load messages
// For new chats (no urlChatId), we use lazy creation - thread is created on first message
- // biome-ignore lint/correctness/useExhaustiveDependencies: searchSpaceId triggers re-init when switching spaces with the same urlChatId
const initializeThread = useCallback(async () => {
setIsInitializing(true);
@@ -323,6 +326,14 @@ export default function NewChatPage() {
// This improves UX (instant load) and avoids orphan threads
} catch (error) {
console.error("[NewChatPage] Failed to initialize thread:", error);
+ if (urlChatId > 0 && error instanceof NotFoundError) {
+ removeChatTab(urlChatId);
+ if (typeof window !== "undefined") {
+ window.history.replaceState(null, "", `/dashboard/${searchSpaceId}/new-chat`);
+ }
+ toast.error("This chat was deleted.");
+ return;
+ }
// Keep threadId as null - don't use Date.now() as it creates an invalid ID
// that will cause 404 errors on subsequent API calls
setThreadId(null);
@@ -333,15 +344,16 @@ export default function NewChatPage() {
}
}, [
urlChatId,
- searchSpaceId,
setMessageDocumentsMap,
setMentionedDocuments,
setSidebarDocuments,
closeReportPanel,
closeEditorPanel,
+ removeChatTab,
+ searchSpaceId,
]);
- // Initialize on mount
+ // Initialize on mount, and re-init when switching search spaces (even if urlChatId is the same)
useEffect(() => {
initializeThread();
}, [initializeThread]);
@@ -484,18 +496,17 @@ export default function NewChatPage() {
// Add user message to state
const userMsgId = `msg-user-${Date.now()}`;
- // Include author metadata for shared chats
- const authorMetadata =
- currentThread?.visibility === "SEARCH_SPACE" && currentUser
- ? {
- custom: {
- author: {
- displayName: currentUser.display_name ?? null,
- avatarUrl: currentUser.avatar_url ?? null,
- },
+ // Always include author metadata so the UI layer can decide visibility
+ const authorMetadata = currentUser
+ ? {
+ custom: {
+ author: {
+ displayName: currentUser.display_name ?? null,
+ avatarUrl: currentUser.avatar_url ?? null,
},
- }
- : undefined;
+ },
+ }
+ : undefined;
const userMessage: ThreadMessageLike = {
id: userMsgId,
@@ -571,6 +582,7 @@ export default function NewChatPage() {
// Prepare assistant message
const assistantMsgId = `msg-assistant-${Date.now()}`;
const currentThinkingSteps = new Map();
+ const batcher = new FrameBatchedUpdater();
const contentPartsState: ContentPartsState = {
contentParts: [],
@@ -642,33 +654,30 @@ export default function NewChatPage() {
throw new Error(`Backend error: ${response.status}`);
}
+ const flushMessages = () => {
+ setMessages((prev) =>
+ prev.map((m) =>
+ m.id === assistantMsgId
+ ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
+ : m
+ )
+ );
+ };
+ const scheduleFlush = () => batcher.schedule(flushMessages);
+
for await (const parsed of readSSEStream(response)) {
switch (parsed.type) {
case "text-delta":
appendText(contentPartsState, parsed.delta);
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ scheduleFlush();
break;
case "tool-input-start":
- // Add tool call inline - this breaks the current text segment
addToolCall(contentPartsState, TOOLS_WITH_UI, parsed.toolCallId, parsed.toolName, {});
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ batcher.flush();
break;
case "tool-input-available": {
- // Update existing tool call's args, or add if not exists
if (toolCallIndices.has(parsed.toolCallId)) {
updateToolCall(contentPartsState, parsed.toolCallId, { args: parsed.input || {} });
} else {
@@ -680,23 +689,14 @@ export default function NewChatPage() {
parsed.input || {}
);
}
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ batcher.flush();
break;
}
case "tool-output-available": {
- // Update the tool call with its result
updateToolCall(contentPartsState, parsed.toolCallId, { result: parsed.output });
markInterruptsCompleted(contentParts);
- // Handle podcast-specific logic
if (parsed.output?.status === "pending" && parsed.output?.podcast_id) {
- // Check if this is a podcast tool by looking at the content part
const idx = toolCallIndices.get(parsed.toolCallId);
if (idx !== undefined) {
const part = contentParts[idx];
@@ -705,13 +705,7 @@ export default function NewChatPage() {
}
}
}
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ batcher.flush();
break;
}
@@ -719,14 +713,10 @@ export default function NewChatPage() {
const stepData = parsed.data as ThinkingStepData;
if (stepData?.id) {
currentThinkingSteps.set(stepData.id, stepData);
- updateThinkingSteps(contentPartsState, currentThinkingSteps);
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ const didUpdate = updateThinkingSteps(contentPartsState, currentThinkingSteps);
+ if (didUpdate) {
+ scheduleFlush();
+ }
}
break;
}
@@ -803,6 +793,8 @@ export default function NewChatPage() {
}
}
+ batcher.flush();
+
// Skip persistence for interrupted messages -- handleResume will persist the final version
const finalContent = buildContentForPersistence(contentPartsState, TOOLS_WITH_UI);
if (contentParts.length > 0 && !wasInterrupted) {
@@ -832,6 +824,7 @@ export default function NewChatPage() {
trackChatResponseReceived(searchSpaceId, currentThreadId);
}
} catch (error) {
+ batcher.dispose();
if (error instanceof Error && error.name === "AbortError") {
// Request was cancelled by user - persist partial response if any content was received
const hasContent = contentParts.some(
@@ -899,8 +892,8 @@ export default function NewChatPage() {
setMentionedDocuments,
setSidebarDocuments,
setMessageDocumentsMap,
+ setAgentCreatedDocuments,
queryClient,
- currentThread,
currentUser,
disabledTools,
updateChatTabTitle,
@@ -931,6 +924,7 @@ export default function NewChatPage() {
abortControllerRef.current = controller;
const currentThinkingSteps = new Map();
+ const batcher = new FrameBatchedUpdater();
const contentPartsState: ContentPartsState = {
contentParts: [],
@@ -1018,28 +1012,27 @@ export default function NewChatPage() {
throw new Error(`Backend error: ${response.status}`);
}
+ const flushMessages = () => {
+ setMessages((prev) =>
+ prev.map((m) =>
+ m.id === assistantMsgId
+ ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
+ : m
+ )
+ );
+ };
+ const scheduleFlush = () => batcher.schedule(flushMessages);
+
for await (const parsed of readSSEStream(response)) {
switch (parsed.type) {
case "text-delta":
appendText(contentPartsState, parsed.delta);
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ scheduleFlush();
break;
case "tool-input-start":
addToolCall(contentPartsState, TOOLS_WITH_UI, parsed.toolCallId, parsed.toolName, {});
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ batcher.flush();
break;
case "tool-input-available":
@@ -1056,13 +1049,7 @@ export default function NewChatPage() {
parsed.input || {}
);
}
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ batcher.flush();
break;
case "tool-output-available":
@@ -1070,27 +1057,17 @@ export default function NewChatPage() {
result: parsed.output,
});
markInterruptsCompleted(contentParts);
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ batcher.flush();
break;
case "data-thinking-step": {
const stepData = parsed.data as ThinkingStepData;
if (stepData?.id) {
currentThinkingSteps.set(stepData.id, stepData);
- updateThinkingSteps(contentPartsState, currentThinkingSteps);
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ const didUpdate = updateThinkingSteps(contentPartsState, currentThinkingSteps);
+ if (didUpdate) {
+ scheduleFlush();
+ }
}
break;
}
@@ -1144,6 +1121,8 @@ export default function NewChatPage() {
}
}
+ batcher.flush();
+
const finalContent = buildContentForPersistence(contentPartsState, TOOLS_WITH_UI);
if (contentParts.length > 0) {
try {
@@ -1160,6 +1139,7 @@ export default function NewChatPage() {
}
}
} catch (error) {
+ batcher.dispose();
if (error instanceof Error && error.name === "AbortError") {
return;
}
@@ -1305,6 +1285,7 @@ export default function NewChatPage() {
toolCallIndices: new Map(),
};
const { contentParts, toolCallIndices } = contentPartsState;
+ const batcher = new FrameBatchedUpdater();
// Add placeholder messages to UI
// Always add back the user message (with new query for edit, or original content for reload)
@@ -1349,28 +1330,27 @@ export default function NewChatPage() {
throw new Error(`Backend error: ${response.status}`);
}
+ const flushMessages = () => {
+ setMessages((prev) =>
+ prev.map((m) =>
+ m.id === assistantMsgId
+ ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
+ : m
+ )
+ );
+ };
+ const scheduleFlush = () => batcher.schedule(flushMessages);
+
for await (const parsed of readSSEStream(response)) {
switch (parsed.type) {
case "text-delta":
appendText(contentPartsState, parsed.delta);
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ scheduleFlush();
break;
case "tool-input-start":
addToolCall(contentPartsState, TOOLS_WITH_UI, parsed.toolCallId, parsed.toolName, {});
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ batcher.flush();
break;
case "tool-input-available":
@@ -1385,13 +1365,7 @@ export default function NewChatPage() {
parsed.input || {}
);
}
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ batcher.flush();
break;
case "tool-output-available":
@@ -1406,27 +1380,17 @@ export default function NewChatPage() {
}
}
}
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ batcher.flush();
break;
case "data-thinking-step": {
const stepData = parsed.data as ThinkingStepData;
if (stepData?.id) {
currentThinkingSteps.set(stepData.id, stepData);
- updateThinkingSteps(contentPartsState, currentThinkingSteps);
- setMessages((prev) =>
- prev.map((m) =>
- m.id === assistantMsgId
- ? { ...m, content: buildContentForUI(contentPartsState, TOOLS_WITH_UI) }
- : m
- )
- );
+ const didUpdate = updateThinkingSteps(contentPartsState, currentThinkingSteps);
+ if (didUpdate) {
+ scheduleFlush();
+ }
}
break;
}
@@ -1436,6 +1400,8 @@ export default function NewChatPage() {
}
}
+ batcher.flush();
+
// Persist messages after streaming completes
const finalContent = buildContentForPersistence(contentPartsState, TOOLS_WITH_UI);
if (contentParts.length > 0) {
@@ -1477,6 +1443,7 @@ export default function NewChatPage() {
if (error instanceof Error && error.name === "AbortError") {
return;
}
+ batcher.dispose();
console.error("[NewChatPage] Regeneration error:", error);
trackChatError(
searchSpaceId,
@@ -1484,7 +1451,6 @@ export default function NewChatPage() {
error instanceof Error ? error.message : "Unknown error"
);
toast.error("Failed to regenerate response. Please try again.");
- // Update assistant message with error
setMessages((prev) =>
prev.map((m) =>
m.id === assistantMsgId
diff --git a/surfsense_web/app/dashboard/[search_space_id]/onboard/page.tsx b/surfsense_web/app/dashboard/[search_space_id]/onboard/page.tsx
index b188d7c8f..4dba3bbb6 100644
--- a/surfsense_web/app/dashboard/[search_space_id]/onboard/page.tsx
+++ b/surfsense_web/app/dashboard/[search_space_id]/onboard/page.tsx
@@ -1,7 +1,6 @@
"use client";
-import { useAtomValue, useSetAtom } from "jotai";
-import { motion } from "motion/react";
+import { useAtomValue } from "jotai";
import { useParams, useRouter } from "next/navigation";
import { useEffect, useRef, useState } from "react";
import { toast } from "sonner";
@@ -13,19 +12,17 @@ import {
globalNewLLMConfigsAtom,
llmPreferencesAtom,
} from "@/atoms/new-llm-config/new-llm-config-query.atoms";
-import { searchSpaceSettingsDialogAtom } from "@/atoms/settings/settings-dialog.atoms";
import { Logo } from "@/components/Logo";
import { LLMConfigForm, type LLMConfigFormData } from "@/components/shared/llm-config-form";
-import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card";
+import { Button } from "@/components/ui/button";
import { Spinner } from "@/components/ui/spinner";
+import { useGlobalLoadingEffect } from "@/hooks/use-global-loading";
import { getBearerToken, redirectToLogin } from "@/lib/auth-utils";
export default function OnboardPage() {
const router = useRouter();
const params = useParams();
const searchSpaceId = Number(params.search_space_id);
- const setSearchSpaceSettingsDialog = useSetAtom(searchSpaceSettingsDialogAtom);
-
// Queries
const {
data: globalConfigs = [],
@@ -62,14 +59,12 @@ export default function OnboardPage() {
preferences.document_summary_llm_id !== null &&
preferences.document_summary_llm_id !== undefined;
- // If onboarding is already complete, redirect immediately
useEffect(() => {
if (!preferencesLoading && isOnboardingComplete) {
router.push(`/dashboard/${searchSpaceId}/new-chat`);
}
}, [preferencesLoading, isOnboardingComplete, router, searchSpaceId]);
- // Auto-configure if global configs are available
useEffect(() => {
const autoConfigureWithGlobal = async () => {
if (hasAttemptedAutoConfig.current) return;
@@ -77,7 +72,6 @@ export default function OnboardPage() {
if (!globalConfigsLoaded) return;
if (isOnboardingComplete) return;
- // Only auto-configure if we have global configs
if (globalConfigs.length > 0) {
hasAttemptedAutoConfig.current = true;
setIsAutoConfiguring(true);
@@ -97,7 +91,6 @@ export default function OnboardPage() {
description: `Using ${firstGlobalConfig.name}. You can customize this later in Settings.`,
});
- // Redirect to new-chat
router.push(`/dashboard/${searchSpaceId}/new-chat`);
} catch (error) {
console.error("Auto-configuration failed:", error);
@@ -119,13 +112,10 @@ export default function OnboardPage() {
router,
]);
- // Handle form submission
const handleSubmit = async (formData: LLMConfigFormData) => {
try {
- // Create the config
const newConfig = await createConfig(formData);
- // Auto-assign to all roles
await updatePreferences({
search_space_id: searchSpaceId,
data: {
@@ -138,7 +128,6 @@ export default function OnboardPage() {
description: "Redirecting to chat...",
});
- // Redirect to new-chat
router.push(`/dashboard/${searchSpaceId}/new-chat`);
} catch (error) {
console.error("Failed to create config:", error);
@@ -150,124 +139,59 @@ export default function OnboardPage() {
const isSubmitting = isCreating || isUpdatingPreferences;
- // Loading state
- if (globalConfigsLoading || preferencesLoading || isAutoConfiguring) {
- return (
-
-
-
-
-
- {isAutoConfiguring ? "Setting up your AI..." : "Loading..."}
-
-
- {isAutoConfiguring
- ? "Auto-configuring with available settings"
- : "Please wait while we check your configuration"}
+ const isLoading = globalConfigsLoading || preferencesLoading || isAutoConfiguring;
+ useGlobalLoadingEffect(isLoading);
+
+ if (isLoading) {
+ return null;
+ }
+
+ if (globalConfigs.length > 0 && !isAutoConfiguring) {
+ return null;
+ }
+
+ return (
+
+
+ {/* Header */}
+
+
+
+
Configure Your AI
+
+ Add your LLM provider to get started with SurfSense
-
- {[0, 1, 2].map((i) => (
-
- ))}
-
-
-
- );
- }
+
- // If global configs exist but auto-config failed, show simple message
- if (globalConfigs.length > 0 && !isAutoConfiguring) {
- return null; // Will redirect via useEffect
- }
+ {/* Form card */}
+
+
+
- // No global configs - show the config form
- return (
-
-
-
- {/* Header */}
-
-
-
-
-
-
-
Configure Your AI
-
- Add your LLM provider to get started with SurfSense
-
-
-
+
Start Using SurfSense
+ {isSubmitting &&
}
+
+
You can add more configurations later
+
);
diff --git a/surfsense_web/app/dashboard/[search_space_id]/user-settings/components/PromptsContent.tsx b/surfsense_web/app/dashboard/[search_space_id]/user-settings/components/PromptsContent.tsx
index 476fd05a6..e4ba1426d 100644
--- a/surfsense_web/app/dashboard/[search_space_id]/user-settings/components/PromptsContent.tsx
+++ b/surfsense_web/app/dashboard/[search_space_id]/user-settings/components/PromptsContent.tsx
@@ -229,8 +229,9 @@ export function PromptsContent() {
-
+ {citations.map((citation) => (
+
+ ))}
+
{
create_confluence_page: CreateConfluencePageToolUI,
update_confluence_page: UpdateConfluencePageToolUI,
delete_confluence_page: DeleteConfluencePageToolUI,
+ web_search: () => null,
link_preview: () => null,
multi_link_preview: () => null,
scrape_webpage: () => null,
@@ -131,10 +286,16 @@ const AssistantMessageInner: FC = () => {
+ {isMobile && (
+
+
+
+ )}
+
-
+
);
};
diff --git a/surfsense_web/components/assistant-ui/citation-metadata-context.tsx b/surfsense_web/components/assistant-ui/citation-metadata-context.tsx
new file mode 100644
index 000000000..0bf5dd946
--- /dev/null
+++ b/surfsense_web/components/assistant-ui/citation-metadata-context.tsx
@@ -0,0 +1,69 @@
+"use client";
+
+import { useAuiState } from "@assistant-ui/react";
+import { createContext, type FC, type ReactNode, useContext, useMemo } from "react";
+
+export interface CitationMeta {
+ title: string;
+ snippet?: string;
+}
+
+type CitationMetadataMap = ReadonlyMap;
+
+const CitationMetadataContext = createContext(new Map());
+
+interface ToolCallResult {
+ status?: string;
+ citations?: Record;
+}
+
+interface MessageContent {
+ type: string;
+ toolName?: string;
+ result?: unknown;
+}
+
+export const CitationMetadataProvider: FC<{ children: ReactNode }> = ({ children }) => {
+ const content = useAuiState(
+ ({ message }) => (message as { content?: MessageContent[] })?.content
+ );
+
+ const metadataMap = useMemo(() => {
+ if (!content || !Array.isArray(content)) return new Map();
+
+ const merged = new Map();
+
+ for (const part of content) {
+ if (part.type !== "tool-call" || part.toolName !== "web_search" || !part.result) {
+ continue;
+ }
+
+ const result = part.result as ToolCallResult;
+ const citations = result.citations;
+ if (!citations || typeof citations !== "object") continue;
+
+ for (const [url, meta] of Object.entries(citations)) {
+ if (url.startsWith("http") && meta.title && !merged.has(url)) {
+ merged.set(url, { title: meta.title, snippet: meta.snippet });
+ }
+ }
+ }
+
+ return merged;
+ }, [content]);
+
+ return (
+
+ {children}
+
+ );
+};
+
+export function useCitationMetadata(url: string): CitationMeta | undefined {
+ const map = useContext(CitationMetadataContext);
+ return map.get(url);
+}
+
+export function useAllCitationMetadata(): CitationMetadataMap {
+ return useContext(CitationMetadataContext);
+}
diff --git a/surfsense_web/components/assistant-ui/connector-popup.tsx b/surfsense_web/components/assistant-ui/connector-popup.tsx
index c8175fcb7..5a35016f5 100644
--- a/surfsense_web/components/assistant-ui/connector-popup.tsx
+++ b/surfsense_web/components/assistant-ui/connector-popup.tsx
@@ -1,7 +1,7 @@
"use client";
import { useAtomValue, useSetAtom } from "jotai";
-import { AlertTriangle, Cable, Settings } from "lucide-react";
+import { AlertTriangle, Settings } from "lucide-react";
import { forwardRef, useEffect, useImperativeHandle, useMemo, useState } from "react";
import { createPortal } from "react-dom";
import { statusInboxItemsAtom } from "@/atoms/inbox/status-inbox.atom";
@@ -12,17 +12,14 @@ import {
import { activeSearchSpaceIdAtom } from "@/atoms/search-spaces/search-space-query.atoms";
import { searchSpaceSettingsDialogAtom } from "@/atoms/settings/settings-dialog.atoms";
import { currentUserAtom } from "@/atoms/user/user-query.atoms";
-import { TooltipIconButton } from "@/components/assistant-ui/tooltip-icon-button";
import { Alert, AlertDescription, AlertTitle } from "@/components/ui/alert";
import { Button } from "@/components/ui/button";
import { Dialog, DialogContent, DialogTitle } from "@/components/ui/dialog";
-import { Spinner } from "@/components/ui/spinner";
import { Tabs, TabsContent } from "@/components/ui/tabs";
import type { SearchSourceConnector } from "@/contracts/types/connector.types";
import { useConnectorsSync } from "@/hooks/use-connectors-sync";
import { PICKER_CLOSE_EVENT, PICKER_OPEN_EVENT } from "@/hooks/use-google-picker";
import { useZeroDocumentTypeCounts } from "@/hooks/use-zero-document-type-counts";
-import { cn } from "@/lib/utils";
import { ConnectorDialogHeader } from "./connector-popup/components/connector-dialog-header";
import { ConnectorConnectView } from "./connector-popup/connector-configs/views/connector-connect-view";
import { ConnectorEditView } from "./connector-popup/connector-configs/views/connector-edit-view";
@@ -47,7 +44,7 @@ interface ConnectorIndicatorProps {
}
export const ConnectorIndicator = forwardRef(
- ({ showTrigger = true }, ref) => {
+ (_props, ref) => {
const searchSpaceId = useAtomValue(activeSearchSpaceIdAtom);
const setSearchSpaceSettingsDialog = useSetAtom(searchSpaceSettingsDialogAtom);
useAtomValue(currentUserAtom);
@@ -74,8 +71,6 @@ export const ConnectorIndicator = forwardRef count > 0)
@@ -205,41 +198,6 @@ export const ConnectorIndicator = forwardRef
- {showTrigger && (
- handleOpenChange(true)}
- >
- {isLoading ? (
-
- ) : (
- <>
-
- {activeConnectorsCount > 0 && (
-
- {activeConnectorsCount > 99 ? "99+" : activeConnectorsCount}
-
- )}
-
- )}
-
- )}
-
{isOpen &&
createPortal(
= ({
size="sm"
variant={isConnected ? "secondary" : "default"}
className={cn(
- "h-8 text-[11px] px-3 rounded-lg shrink-0 font-medium",
+ "relative h-8 text-[11px] px-3 rounded-lg shrink-0 font-medium items-center justify-center",
isConnected &&
"bg-white text-slate-700 hover:bg-slate-50 border-0 shadow-xs dark:bg-secondary dark:text-secondary-foreground dark:hover:bg-secondary/80",
!isConnected && "shadow-xs"
@@ -151,19 +151,18 @@ export const ConnectorCard: FC = ({
onClick={isConnected ? onManage : onConnect}
disabled={isConnecting || !isEnabled}
>
- {isConnecting ? (
-
- ) : !isEnabled ? (
- "Unavailable"
- ) : isConnected ? (
- "Manage"
- ) : id === "youtube-crawler" ? (
- "Add"
- ) : connectorType ? (
- "Connect"
- ) : (
- "Add"
- )}
+
+ {!isEnabled
+ ? "Unavailable"
+ : isConnected
+ ? "Manage"
+ : id === "youtube-crawler"
+ ? "Add"
+ : connectorType
+ ? "Connect"
+ : "Add"}
+
+ {isConnecting && }
);
diff --git a/surfsense_web/components/assistant-ui/connector-popup/connect-forms/components/mcp-connect-form.tsx b/surfsense_web/components/assistant-ui/connector-popup/connect-forms/components/mcp-connect-form.tsx
index 2a5bb2121..18dcbebc3 100644
--- a/surfsense_web/components/assistant-ui/connector-popup/connect-forms/components/mcp-connect-form.tsx
+++ b/surfsense_web/components/assistant-ui/connector-popup/connect-forms/components/mcp-connect-form.tsx
@@ -243,7 +243,7 @@ export const MCPConnectForm: FC = ({ onSubmit, isSubmitting })
onClick={(e) => {
e.preventDefault();
e.stopPropagation();
- setShowDetails(!showDetails);
+ setShowDetails(prev => !prev);
}}
>
{showDetails ? (
diff --git a/surfsense_web/components/assistant-ui/connector-popup/connector-configs/components/composio-drive-config.tsx b/surfsense_web/components/assistant-ui/connector-popup/connector-configs/components/composio-drive-config.tsx
index 0f6044050..dcaadb34d 100644
--- a/surfsense_web/components/assistant-ui/connector-popup/connector-configs/components/composio-drive-config.tsx
+++ b/surfsense_web/components/assistant-ui/connector-popup/connector-configs/components/composio-drive-config.tsx
@@ -250,7 +250,7 @@ export const ComposioDriveConfig: FC = ({ connector, onCon