feat: stabilize agentic workspace management with robust ID resolution and persistent logging

Fixes persistent deletion failures caused by name-to-UUID resolution errors and missing IPC logging handlers. Updates ChatPanel to maintain reactive workspace state sync and hardens SQLite foreign key handling during node creation.

Author: Tim Sinaeve

App.tsx

@@ -181,7 +181,7 @@ interface UpdateToastState {
 
 export const MainApp: React.FC = () => {
     const { settings, saveSettings, loaded: settingsLoaded } = useSettings();
-    const { nodes, items, addDocument, addFolder, updateItem, commitVersion, deleteItems, moveItems, getDescendantIds, duplicateItems, addDocumentsFromFiles, importNodesFromTransfer, createDocumentFromClipboard, setItemLock, isLoading: areDocumentsLoading } = useDocuments();
+    const { nodes, items, addDocument, addFolder, updateItem, commitVersion, deleteItems, moveItems, getDescendantIds, duplicateItems, addDocumentsFromFiles, importNodesFromTransfer, createDocumentFromClipboard, setItemLock, refreshNodes, getLatestItems, isLoading: areDocumentsLoading } = useDocuments();
     const { templates, addTemplate, updateTemplate, deleteTemplate, deleteTemplates } = useTemplates();
     const { theme } = useTheme();
 
@@ -196,6 +196,8 @@ export const MainApp: React.FC = () => {
     const [pendingRevealId, setPendingRevealId] = useState<string | null>(null);
     const [renamingNodeId, setRenamingNodeId] = useState<string | null>(null);
     const [selectedText, setSelectedText] = useState<string | undefined>(undefined);
+    const [chatContextNodeIds, setChatContextNodeIds] = useState(new Set<string>());
+    const [hasLoadedChatContext, setHasLoadedChatContext] = useState(false);
     const [pendingInsertText, setPendingInsertText] = useState<string | null>(null);
 
     const [view, setView] = useState<'editor' | 'info' | 'settings'>('editor');
@@ -432,6 +434,38 @@ export const MainApp: React.FC = () => {
         return unsubscribe;
     }, [addLog]);
 
+    const addToChatContextAction = useCallback((ids: string[]) => {
+        addLog('INFO', `User action: Adding ${ids.length} item(s) to chat context.`);
+        setChatContextNodeIds(prev => {
+            const next = new Set(prev);
+            ids.forEach(id => {
+                const item = items.find(i => i.id === id);
+                if (item?.type === 'folder') {
+                    // Recursively find all documents within this folder
+                    const descendantIds = getDescendantIds(id);
+                    descendantIds.forEach(dId => {
+                        const dNode = items.find(n => n.id === dId);
+                        if (dNode?.type === 'document') {
+                            next.add(dId);
+                        }
+                    });
+                } else if (item?.type === 'document') {
+                    next.add(id);
+                }
+            });
+            return next;
+        });
+        if (!isChatPanelVisible) {
+            setIsChatPanelVisible(true);
+        }
+    }, [addLog, isChatPanelVisible, items, getDescendantIds]);
+
+    useEffect(() => {
+        if (settingsLoaded && hasLoadedChatContext) {
+            storageService.save(LOCAL_STORAGE_KEYS.CHAT_CONTEXT_NODE_IDS, Array.from(chatContextNodeIds));
+        }
+    }, [chatContextNodeIds, settingsLoaded, hasLoadedChatContext]);
+
     useEffect(() => {
         if (!isElectron || !window.electronAPI?.dbGetPath) {
             setDatabaseStatus({ message: 'Database actions unavailable in this environment.', tone: 'info' });
@@ -1501,13 +1535,22 @@ export const MainApp: React.FC = () => {
                 }
                 setExpandedFolderIds(new Set(ids));
             })
-            .catch(() => {
-                // Loading failed; we'll fall back to the default empty set.
-            })
+            .catch(() => {})
             .finally(() => {
-                if (!isCancelled) {
-                    setHasLoadedExpandedFolders(true);
+                if (!isCancelled) setHasLoadedExpandedFolders(true);
+            });
+
+        storageService
+            .load<string[]>(LOCAL_STORAGE_KEYS.CHAT_CONTEXT_NODE_IDS, [])
+            .then(ids => {
+                if (isCancelled) return;
+                if (Array.isArray(ids)) {
+                    setChatContextNodeIds(new Set(ids));
                 }
+            })
+            .catch(() => {})
+            .finally(() => {
+                if (!isCancelled) setHasLoadedChatContext(true);
             });
 
         return () => {
@@ -1866,6 +1909,52 @@ export const MainApp: React.FC = () => {
         setRenamingNodeId(newDoc.id);
     }, [addDocument, getParentIdForNewItem, ensureNodeVisible, addLog, activateDocumentTab]);
 
+    const handleAgentAddNode = useCallback(async (params: any) => {
+        addLog('INFO', `Agent action: Creating new ${params.node_type} "${params.title}"...`);
+        
+        let resultNode: any;
+        if (params.node_type === 'folder') {
+            resultNode = await addFolder(params.parent_id || params.parentId, params.title);
+        } else {
+            resultNode = await addDocument({
+                parentId: params.parent_id || params.parentId,
+                title: params.title,
+                content: params.document?.content || params.content,
+                doc_type: params.document?.doc_type || params.docType,
+                language_hint: params.document?.language_hint || params.languageHint
+            });
+            
+            if (params.document?.content || params.content) {
+                await commitVersion(resultNode.id, params.document?.content || params.content);
+            }
+        }
+        
+        if (resultNode) {
+            ensureNodeVisible(resultNode);
+            activateDocumentTab(resultNode.id);
+            setSelectedIds(new Set([resultNode.id]));
+            setLastClickedId(resultNode.id);
+            
+            // Map back to the 'Node' structure expected by agentService.ts
+            return {
+                node_id: resultNode.id,
+                parent_id: resultNode.parentId,
+                node_type: resultNode.type,
+                title: resultNode.title,
+                created_at: resultNode.createdAt,
+                updated_at: resultNode.updatedAt,
+                locked: resultNode.locked,
+                document: resultNode.type === 'document' ? {
+                    node_id: resultNode.id,
+                    content: resultNode.content,
+                    doc_type: resultNode.doc_type,
+                    language_hint: resultNode.language_hint,
+                } : undefined
+            };
+        }
+        throw new Error('Failed to create node via agent.');
+    }, [addDocument, addFolder, commitVersion, ensureNodeVisible, activateDocumentTab, addLog]);
+
     const handleNewDocumentFromClipboard = useCallback(async (parentId?: string | null) => {
         addLog('INFO', 'User action: Create New Document from Clipboard.');
         const effectiveParentId = parentId !== undefined ? parentId : getParentIdForNewItem();
@@ -2405,6 +2494,11 @@ export const MainApp: React.FC = () => {
         handleDeleteSelection(idsToDelete, { force: shiftKey });
     }, [items, selectedIds, handleDeleteSelection, addLog]);
 
+    const handleAgentDeleteNodes = useCallback(async (ids: string[]) => {
+        addLog('INFO', `Agent action: Deleting ${ids.length} nodes.`);
+        await handleDeleteSelection(new Set(ids), { force: true });
+    }, [handleDeleteSelection, addLog]);
+
     const handleDeleteTemplate = useCallback((id: string, shiftKey: boolean = false) => {
         const templateToDelete = templates.find(t => t.template_id === id);
         if (!templateToDelete) return;
@@ -2872,6 +2966,11 @@ export const MainApp: React.FC = () => {
                 { label: 'New Folder', icon: FolderPlusIcon, action: () => handleNewFolder(parentIdForNewItem), shortcut: getCommand('new-folder')?.shortcutString },
                 { label: 'New from Template...', icon: DocumentDuplicateIcon, action: newFromTemplateAction, shortcut: getCommand('new-from-template')?.shortcutString },
                 { type: 'separator' },
+                { label: 'Add to Chat Context', icon: SearchIcon, action: () => {
+                    const docIds = selectedNodes.filter(n => n.type === 'document').map(n => n.id);
+                    if (docIds.length > 0) addToChatContextAction(docIds);
+                }, disabled: !hasDocuments },
+                { type: 'separator' },
                 { label: 'Format', icon: FormatIcon, action: handleFormatDocument, disabled: !isFormattable || currentSelection.size !== 1, shortcut: getCommand('format-document')?.shortcutString },
                 { label: firstSelectedNode?.locked ? 'Unlock Document' : 'Lock Document', icon: firstSelectedNode?.locked ? LockOpenIcon : LockClosedIcon, action: () => { if (firstSelectedNode && firstSelectedNode.type === 'document') { void handleSetNodeLockState(firstSelectedNode.id, !firstSelectedNode.locked); } }, disabled: !isDocument || currentSelection.size !== 1, shortcut: getCommand('toggle-document-lock')?.shortcutString },
                 { label: 'Rename', icon: PencilIcon, action: () => handleStartRenamingNode(nodeId), disabled: currentSelection.size !== 1, shortcut: getCommand('rename-item')?.shortcutString },
@@ -2897,7 +2996,7 @@ export const MainApp: React.FC = () => {
             position: { x: e.clientX, y: e.clientY },
             items: menuItems
         });
-    }, [selectedIds, items, handleNewDocument, handleNewFolder, handleDuplicateSelection, handleDeleteSelection, handleCopyNodeContent, addLog, enrichedCommands, handleOpenNewCodeFileModal, handleFormatDocument, handleStartRenamingNode, handleNewDocumentFromClipboard, handleSaveNodeToFile, handleSetNodeLockState]);
+    }, [selectedIds, items, handleNewDocument, handleNewFolder, handleDuplicateSelection, handleDeleteSelection, handleCopyNodeContent, addLog, enrichedCommands, handleOpenNewCodeFileModal, handleFormatDocument, handleStartRenamingNode, handleNewDocumentFromClipboard, handleSaveNodeToFile, handleSetNodeLockState, addToChatContextAction]);
 
 
     const handleSidebarMouseDown = useCallback((e: React.MouseEvent) => {
@@ -3278,18 +3377,35 @@ export const MainApp: React.FC = () => {
                                         onApplyToEditor={handleApplyToEditor}
                                         onCreateDocument={handleCreateDocumentFromChat}
                                         activeDocument={activeNode ? { title: activeNode.title, content: activeNode.content ?? '' } : undefined}
+                                        chatContextNodeIds={chatContextNodeIds}
+                                        onRemoveNodeFromContext={(id) => setChatContextNodeIds(prev => {
+                                            const next = new Set(prev);
+                                            next.delete(id);
+                                            return next;
+                                        })}
+                                        onAddNodesToContext={addToChatContextAction}
+                                        onClearAllContext={() => setChatContextNodeIds(new Set())}
                                         selectedText={selectedText}
                                         addLog={addLog}
                                         nodes={nodes}
-                                        addNode={handleNewDocument}
-                                        updateNode={handleRenameNode}
+                                        addNode={handleAgentAddNode}
+                                        updateNode={async (id, updates) => {
+                                            await handleRenameNode(id, updates.title);
+                                        }}
                                         updateDocumentContent={handleCommitVersion}
-                                        deleteNodes={handleDeleteNode}
+                                        deleteNodes={handleAgentDeleteNodes}
                                         moveNodes={moveItems}
+                                        getLatestItems={async () => {
+                                            await refreshNodes();
+                                            const latestNodes = await repository.getNodeTree();
+                                            return latestNodes;
+                                        }}
                                         runPython={async (code, nodeId) => {
+                                            // Real implementation would go here, currently handled by ChatPanel's inner context
                                             return ""; 
                                         }}
                                         runScript={async (lang, code, nodeId) => {
+                                            // Real implementation would go here
                                             return "";
                                         }}
                                     />

FUNCTIONAL_MANUAL.md

@@ -148,13 +148,36 @@ Import PDFs by dragging `.pdf` files from your operating system into the sidebar
 
 #### RAG (Chat with Workspace)
 
-The "Chat with Workspace" feature allows you to query your entire collection of documents using a local AI model.
+The "Chat with Workspace" feature allows you to query your entire collection of documents using a local AI model. It uses semantic search to find answers even when you don't use the exact keywords.
+
+-   **Indexing**: Before you can chat, DocForge needs to index your workspace. This process creates a semantic representation of your documents. 
+    -   Click **Build Index** (or **Rebuild**) in the Chat Panel to start.
+    -   A progress bar will show the number of documents being processed.
+    -   Once complete, the "indexed" badge in the chat header will reflect the current state of your workspace.
+-   **Semantic Retrieval**: When you ask a question, DocForge searches for the most relevant "chunks" of text across all your indexed files.
+-   **Configurable Retrieval**: In Settings, you can adjust the **RAG Similarity Threshold**. 
+    -   A **lower value** (e.g., 0.8) makes the search more restrictive, returning only highly relevant matches.
+    -   A **higher value** (e.g., 1.5) returns more broad context.
+-   **Context Limit**: You can configure the chat to analyze up to **500 documents** at once for deep workspace insights.
+-   **Source Attribution**: When the AI answers a question, it lists the specific documents it used as context. 
+    -   Click a source badge to instantly open and navigate to that document.
+    -   Sources are streamed as soon as the search completes, before the AI starts typing its answer.
+-   **Smart Privacy**: If the AI determines that none of the retrieved documents contain the answer to your question, the source list is automatically hidden to keep the chat interface clean.
+
+#### Multi-Document Context
+
+Beyond searching your entire workspace, you can explicitly "pin" documents to a conversation for focused reasoning.
+
+-   **Adding Context**: 
+    -   **Right-Click**: Select one or more documents in the sidebar, right-click, and choose **Add to Chat Context**.
+    -   **Drag and Drop**: Drag one or more nodes from the sidebar and drop them directly onto the Chat Panel. A "Drop to add Context" overlay will appear.
+-   **Context Badges**: Active context is displayed as badges at the top of the chat area.
+    -   **Active Document**: Automatically includes the file you are currently editing.
+    -   **Pinned Documents**: Indicated with a 📌 icon. These stay in the context even if you switch tabs.
+    -   **Selection active**: If you select text in the editor, it is automatically included in the next chat message.
+-   **Removing Context**: Click the **X** on any context badge to remove that specific document from the current session.
+-   **Cross-Document Reasoning**: Use this to ask questions like "Compare the requirements in Document A with the implementation in Document B."
 
-- **Indexing**: Before you can chat, DocForge needs to index your workspace. This process creates a semantic representation of your documents, allowing the AI to find relevant context even if the exact keywords don't match.
-- **Configurable Retrieval**: In Settings, you can adjust the **RAG Similarity Threshold**. A lower value (e.g., 0.8) makes the search more restrictive, returning only highly relevant matches, while a higher value (e.g., 1.5) returns more context.
-- **Context Limit**: You can configure the chat to analyze up to **500 documents** at once for deep workspace insights.
-- **Source Attribution**: When the AI answers a question, it lists the specific documents it used as context. These sources are streamed immediately as soon as the search completes.
-- **Smart Privacy**: If the AI determines that none of the retrieved documents contain the answer to your question, the source list is automatically hidden to keep the chat interface clean.
 
 #### Agentic Chat & Workspace Orchestration
 
@@ -199,6 +222,8 @@ You can drop an item (or a group of items):
 
 **Importing from your computer:** You can also drag files and folders directly from your operating system's file explorer into the sidebar. Dropping them on a folder will import them into that folder, while dropping them in an empty area will import them to the root. The original folder structure is preserved.
 
+**Adding to Chat Context:** Drag and drop is also used to augment your AI conversations. Drag one or more documents from the sidebar and drop them onto the **Chat Panel** to instantly pin them as context for your next question.
+
 ### AI-Powered Refinement
 
 Clicking the **Refine with AI** (sparkles) button in the editor toolbar sends your current document content to your configured local LLM. The AI's task is not to *answer* the document's request, but to *improve* the document itself. A modal will appear with the suggested refinement, which you can then accept or discard. The "Accept" button is the default and can be triggered by pressing `Enter`. This feature is available for Markdown and plaintext documents.