adjusted agent to include UI control tool

1 files changed, 3 insertions, 3 deletions

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