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

Added source connector integration to chat input options

Browse source
This commit is contained in:
Utkarsh-Patel-13 2025-07-22 16:06:50 -07:00
parent b8c23030c5
commit cf162101a6
6 changed files with 2828 additions and 2157 deletions
Download patch file Download diff file Expand all files Collapse all files

4168
surfsense_web/app/dashboard/[search_space_id]/researcher/[chat_id]/page.tsx
View file

File diff suppressed because it is too large Load diff

26
surfsense_web/app/dashboard/[search_space_id]/v2/[[...chat_id]]/page.tsx
View file

@ -42,9 +42,15 @@ export default function ResearchChatPageV2() {
return selectedDocuments.map((doc) => doc.id);
}, [selectedDocuments]);
// Memoize connector types to prevent infinite re-renders
const connectorTypes = useMemo(() => {
return selectedConnectors;
}, [selectedConnectors]);
// Unified localStorage management for chat state
interface ChatState {
selectedDocuments: Document[];
selectedConnectors: string[];
searchMode: "DOCUMENTS" | "CHUNKS";
researchMode: ResearchMode;
}
@ -89,7 +95,7 @@ export default function ResearchChatPageV2() {
body: {
data: {
search_space_id: search_space_id,
selected_connectors: selectedConnectors,
selected_connectors: connectorTypes,
research_mode: researchMode,
search_mode: searchMode,
document_ids_to_add_in_context: documentIds,
@ -104,11 +110,16 @@ export default function ResearchChatPageV2() {
message: Message | CreateMessage,
chatRequestOptions?: { data?: any }
) => {
const newChatId = await createChat(message.content, researchMode);
const newChatId = await createChat(
message.content,
researchMode,
selectedConnectors
);
if (newChatId) {
// Store chat state before navigation
storeChatState(search_space_id as string, newChatId, {
selectedDocuments,
selectedConnectors,
searchMode,
researchMode,
});
@ -133,6 +144,7 @@ export default function ResearchChatPageV2() {
);
if (restoredState) {
setSelectedDocuments(restoredState.selectedDocuments);
setSelectedConnectors(restoredState.selectedConnectors);
setSearchMode(restoredState.searchMode);
setResearchMode(restoredState.researchMode);
}
@ -141,6 +153,7 @@ export default function ResearchChatPageV2() {
chatIdParam,
search_space_id,
setSelectedDocuments,
setSelectedConnectors,
setSearchMode,
setResearchMode,
]);
@ -192,7 +205,12 @@ export default function ResearchChatPageV2() {
handler.messages.length > 0 &&
handler.messages[handler.messages.length - 1]?.role === "assistant"
) {
updateChat(chatIdParam, handler.messages, researchMode);
updateChat(
chatIdParam,
handler.messages,
researchMode,
selectedConnectors
);
}
}, [handler.messages, handler.status, chatIdParam, isNewChat]);
@ -212,6 +230,8 @@ export default function ResearchChatPageV2() {
}}
onDocumentSelectionChange={setSelectedDocuments}
selectedDocuments={selectedDocuments}
onConnectorSelectionChange={setSelectedConnectors}
selectedConnectors={selectedConnectors}
searchMode={searchMode}
onSearchModeChange={setSearchMode}
researchMode={researchMode}