From 16e7cfcac3d41bd86ef953f131bb0fecba11f299 Mon Sep 17 00:00:00 2001
From: sharkiecodes <lanyi_stroud@brown.edu>
Date: Tue, 22 Jul 2025 21:05:47 -0400
Subject: adjusted agent to include UI control tool

---
 src/client/views/nodes/chatbot/tools/DocumentMetadataTool.ts | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/client/views/nodes/chatbot/tools/DocumentMetadataTool.ts')

diff --git a/src/client/views/nodes/chatbot/tools/DocumentMetadataTool.ts b/src/client/views/nodes/chatbot/tools/DocumentMetadataTool.ts
index 6568766c5..941b875a2 100644
--- a/src/client/views/nodes/chatbot/tools/DocumentMetadataTool.ts
+++ b/src/client/views/nodes/chatbot/tools/DocumentMetadataTool.ts
@@ -491,7 +491,7 @@ export class DocumentMetadataTool extends BaseTool<DocumentMetadataToolParamsTyp
     constructor(docManager: AgentDocumentManager) {
         super(documentMetadataToolInfo);
         this._docManager = docManager;
-        this._docManager.initializeFindDocsFreeform();
+        this._docManager.initializeDocuments();
     }
 
     /**
@@ -502,8 +502,8 @@ export class DocumentMetadataTool extends BaseTool<DocumentMetadataToolParamsTyp
     async execute(args: ParametersType<DocumentMetadataToolParamsType>): Promise<Observation[]> {
         console.log('DocumentMetadataTool: Executing with args:', args);
 
-        // Find all documents in the Freeform view
-        this._docManager.initializeFindDocsFreeform();
+        // Find all documents based on current mode (canvas or linked)
+        this._docManager.initializeDocuments();
 
         try {
             // Validate required input parameters based on action
-- 
cgit v1.2.3-70-g09d2