apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-10 16:22:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Simplify live message sync using Electric as source of truth

Browse source
This commit is contained in:
CREDO23 2026-01-22 20:40:49 +02:00
parent 39d434b00c
commit b1b63c6740
1 changed files with 17 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -50,7 +50,6 @@ import {
type MessageRecord,
type ThreadRecord,
} from "@/lib/chat/thread-persistence";
import { chatSessionStateAtom } from "@/atoms/chat/chat-session-state.atom";
import { useChatSessionStateSync } from "@/hooks/use-chat-session-state";
import { useMessagesElectric } from "@/hooks/use-messages-electric";
import {
@ -263,36 +262,25 @@ export default function NewChatPage() {
// Live collaboration: sync session state and messages via Electric SQL
useChatSessionStateSync(threadId);
const sessionState = useAtomValue(chatSessionStateAtom);
const isAiResponding = sessionState?.isAiResponding ?? false;
const respondingToUserId = sessionState?.respondingToUserId ?? null;
const { data: membersData } = useAtomValue(membersAtom);
const handleElectricMessagesUpdate = useCallback(
(electricMessages: { id: number; thread_id: number; role: string; content: unknown; author_id: string | null; created_at: string }[]) => {
// Skip sync while AI is responding to us or during local streaming
if ((isAiResponding && respondingToUserId === currentUser?.id) || isRunning) {
if (isRunning) {
return;
}
setMessages((prev) => {
const existingIds = new Set(
prev.map((m) => {
const match = String(m.id).match(/^msg-(\d+)$/);
return match ? Number.parseInt(match[1], 10) : null;
}).filter((id): id is number => id !== null)
);
const newConverted: ReturnType<typeof convertToThreadMessage>[] = [];
for (const msg of electricMessages) {
if (existingIds.has(msg.id)) continue;
if (electricMessages.length < prev.length) {
return prev;
}
return electricMessages.map((msg) => {
const member = msg.author_id
? membersData?.find((m) => m.user_id === msg.author_id)
: null;
newConverted.push(
convertToThreadMessage({
return convertToThreadMessage({
id: msg.id,
thread_id: msg.thread_id,
role: msg.role.toLowerCase() as "user" | "assistant" | "system",
@ -301,18 +289,11 @@ export default function NewChatPage() {
created_at: msg.created_at,
author_display_name: member?.user_display_name ?? null,
author_avatar_url: member?.user_avatar_url ?? null,
})
);
}
if (newConverted.length === 0) {
return prev;
}
return [...prev, ...newConverted];
});
});
});
},
[isAiResponding, respondingToUserId, currentUser?.id, isRunning, membersData]
[isRunning, membersData]
);
useMessagesElectric(threadId, handleElectricMessagesUpdate);
@ -646,8 +627,6 @@ export default function NewChatPage() {
content: persistContent,
})
.then(() => {
// For new threads, the backend updates the title from the first user message
// Invalidate threads query so sidebar shows the updated title in real-time
if (isNewThread) {
queryClient.invalidateQueries({ queryKey: ["threads", String(searchSpaceId)] });
}