successfully merged documentationtext functionality with new version of agent, however still minor issues with the agent not selecting the proper tool for documentation generation without the additional context from the topbar

1 files changed, 23 insertions, 16 deletions

diff --git a/src/client/views/nodes/chatbot/tools/TutorialTool.ts b/src/client/views/nodes/chatbot/tools/TutorialTool.ts
index 08e4e1409..1624f0439 100644
--- a/src/client/views/nodes/chatbot/tools/TutorialTool.ts
+++ b/src/client/views/nodes/chatbot/tools/TutorialTool.ts
@@ -11,7 +11,9 @@ import { RichTextField } from '../../../../../fields/RichTextField';
 import { DocumentViewInternal } from '../../DocumentView';
 import { Docs } from '../../../../documents/Documents';
 import { OpenWhere } from '../../OpenWhere';
-import { CollectionFreeFormView } from '../../../collections/collectionFreeForm';
+import { CollectionFreeFormView } from '../../../collections/collectionFreeForm/CollectionFreeFormView';
+import { AgentDocumentManager } from '../utils/AgentDocumentManager';
+import { Node as ProseMirrorNode } from 'prosemirror-model';
 
 const generateTutorialNodeToolParams = [
     {
@@ -28,20 +30,26 @@ const generateTutorialNodeToolInfo: ToolInfo<typeof generateTutorialNodeToolPara
     parameterRules: generateTutorialNodeToolParams,
     citationRules: "No citation needed for this tool's output.",
 };
-const applyFormatting = (markdownText: string): { doc: any; plainText: string } => {
+
+interface FormattedDocument {
+    doc: ProseMirrorNode;
+    plainText: string;
+}
+
+const applyFormatting = (markdownText: string): FormattedDocument => {
     const lines = markdownText.split('\n');
-    const nodes: any[] = [];
+    const nodes: ProseMirrorNode[] = [];
     let plainText = '';
     let i = 0;
-    let currentListItems: any[] = [];
-    let currentParagraph: any[] = [];
-    let currentOrderedListItems: any[] = [];
+    let currentListItems: ProseMirrorNode[] = [];
+    let currentParagraph: ProseMirrorNode[] = [];
+    let currentOrderedListItems: ProseMirrorNode[] = [];
     let inOrderedList = false;
     let inBulletList = false;
 
-    const processBoldText = (text: string) => {
+    const processBoldText = (text: string): ProseMirrorNode[] => {
         const boldRegex = /\*\*(.*?)\*\*/g;
-        const parts = [];
+        const parts: ProseMirrorNode[] = [];
         let lastIndex = 0;
         let match;
 
@@ -58,7 +66,7 @@ const applyFormatting = (markdownText: string): { doc: any; plainText: string }
         return parts.length > 0 ? parts : [schema.text(text)];
     };
 
-    const flushListItems = () => {
+    const flushListItems = (): void => {
         if (currentListItems.length > 0) {
             nodes.push(schema.nodes.ordered_list.create({ mapStyle: 'bullet' }, currentListItems));
             nodes.push(schema.nodes.paragraph.create());
@@ -73,14 +81,14 @@ const applyFormatting = (markdownText: string): { doc: any; plainText: string }
         }
     };
 
-    const flushParagraph = () => {
+    const flushParagraph = (): void => {
         if (currentParagraph.length > 0) {
             nodes.push(schema.nodes.paragraph.create({}, currentParagraph));
             currentParagraph = [];
         }
     };
 
-    const processHeader = (line: string) => {
+    const processHeader = (line: string): boolean => {
         const headerMatch = line.match(/^(#{1,6})\s+(.+)$/);
         if (headerMatch) {
             const level = Math.min(headerMatch[1].length, 6); // Cap at h6
@@ -138,12 +146,11 @@ const applyFormatting = (markdownText: string): { doc: any; plainText: string }
 };
 
 export class GPTTutorialTool extends BaseTool<typeof generateTutorialNodeToolParams> {
-    private _createDocInDash: (doc: parsedDoc) => Doc | undefined;
+    private _docManager: AgentDocumentManager;
 
-    constructor(createDocInDash: (doc: parsedDoc) => Doc | undefined) {
+    constructor(docManager: AgentDocumentManager) {
         super(generateTutorialNodeToolInfo);
-
-        this._createDocInDash = createDocInDash;
+        this._docManager = docManager;
     }
 
     async execute(args: ParametersType<typeof generateTutorialNodeToolParams>): Promise<Observation[]> {
@@ -158,7 +165,7 @@ export class GPTTutorialTool extends BaseTool<typeof generateTutorialNodeToolPar
 
             // Build the ProseMirror‐in‐JSON + plain-text for RichTextField
             const rtfData = {
-                doc: (doc as any).toJSON ? (doc as any).toJSON() : doc,
+                doc: doc.toJSON ? doc.toJSON() : doc,
                 selection: { type: 'text', anchor: 0, head: 0 },
                 storedMarks: [],
             };