1 files changed, 15 insertions, 0 deletions
diff --git a/src/client/views/nodes/chatbot/utils/AgentDocumentManager.ts b/src/client/views/nodes/chatbot/utils/AgentDocumentManager.ts
index a96d93a25..485430403 100644
--- a/src/client/views/nodes/chatbot/utils/AgentDocumentManager.ts
+++ b/src/client/views/nodes/chatbot/utils/AgentDocumentManager.ts
@@ -165,6 +165,21 @@ export class AgentDocumentManager {
     }
 
     /**
+     * Initialize documents based on current mode (canvas or linked)
+     * This should be called by tools instead of hardcoding initializeFindDocsFreeform
+     */
+    @action
+    public initializeDocuments() {
+        if (this._useCanvasMode) {
+            console.log('[AgentDocumentManager] Initializing canvas documents (canvas mode enabled)');
+            this.initializeCanvasDocuments();
+        } else {
+            console.log('[AgentDocumentManager] Initializing linked documents (canvas mode disabled)');
+            this.initializeFindDocsFreeform();
+        }
+    }
+
+    /**
      * Initialize documents from the entire canvas
      */
     @action