changed to generic addLinkedDoc

1 files changed, 4 insertions, 3 deletions

diff --git a/src/client/views/nodes/chatbot/agentsystem/Agent.ts b/src/client/views/nodes/chatbot/agentsystem/Agent.ts
index df307bc21..9253175d5 100644
--- a/src/client/views/nodes/chatbot/agentsystem/Agent.ts
+++ b/src/client/views/nodes/chatbot/agentsystem/Agent.ts
@@ -15,8 +15,9 @@ import { AgentMessage, AssistantMessage, Observation, PROCESSING_TYPE, Processin
 import { Vectorstore } from '../vectorstore/Vectorstore';
 import { getReactPrompt } from './prompts';
 import { BaseTool } from '../tools/BaseTool';
-import { Parameter, ParametersType } from '../tools/ToolTypes';
+import { Parameter, ParametersType } from '../types/tool_types';
 import { CreateTextDocTool } from '../tools/CreateTextDocumentTool';
+import { DocumentOptions } from '../../../../documents/Documents';
 
 dotenv.config();
 
@@ -55,7 +56,7 @@ export class Agent {
         history: () => string,
         csvData: () => { filename: string; id: string; text: string }[],
         addLinkedUrlDoc: (url: string, id: string) => void,
-        addLinkedTextDoc: (text_content: string, options: {}, id: string) => void,
+        addLinkedDoc: (doc_type: string, data: string, options: DocumentOptions, id: string) => void,
         createCSVInDash: (url: string, title: string, id: string, data: string) => void
     ) {
         // Initialize OpenAI client with API key from environment
@@ -74,7 +75,7 @@ export class Agent {
             searchTool: new SearchTool(addLinkedUrlDoc),
             createCSV: new CreateCSVTool(createCSVInDash),
             noTool: new NoTool(),
-            createTextDoc: new CreateTextDocTool(addLinkedTextDoc),
+            createTextDoc: new CreateTextDocTool(addLinkedDoc),
         };
     }