Completed agent functionality - recognition of all documents on canvas

1 files changed, 4 insertions, 1 deletions

diff --git a/src/client/views/nodes/chatbot/agentsystem/Agent.ts b/src/client/views/nodes/chatbot/agentsystem/Agent.ts
index 47e2e8fd3..6a15d0c1d 100644
--- a/src/client/views/nodes/chatbot/agentsystem/Agent.ts
+++ b/src/client/views/nodes/chatbot/agentsystem/Agent.ts
@@ -33,7 +33,8 @@ import { TakeQuizTool } from '../tools/TakeQuizTool';
 import { GPTTutorialTool } from '../tools/TutorialTool';
 import { WebsiteInfoScraperTool } from '../tools/WebsiteInfoScraperTool';
 import { AgentDocumentManager } from '../utils/AgentDocumentManager';
-import { FilterDocsTool } from '../tools/FilterDocTool';
+import { FilterDocsTool } from '../tools/FilterDocsTool';
+import { CanvasDocsTool } from '../tools/CanvasDocsTool';
 
 dotenv.config();
 
@@ -91,6 +92,7 @@ export class Agent {
         this._history = history;
         this._csvData = csvData;
         this._docManager = docManager;
+        this.is_dash_doc_assistant = true; // Initialize to default value
 
         // Initialize dynamic tool registry
         this.dynamicToolRegistry = new Map();
@@ -114,6 +116,7 @@ export class Agent {
             tagDocs: new TagDocsTool(this._docManager),
             filterDocs: new FilterDocsTool(this._docManager, this.parentView),
             takeQuiz: new TakeQuizTool(this._docManager),
+            canvasDocs: new CanvasDocsTool(),
 
         };