1 files changed, 13 insertions, 4 deletions
diff --git a/src/client/views/nodes/ChatBox/tools/DataAnalysisTool.ts b/src/client/views/nodes/ChatBox/tools/DataAnalysisTool.ts
index d2edc4847..b45733639 100644
--- a/src/client/views/nodes/ChatBox/tools/DataAnalysisTool.ts
+++ b/src/client/views/nodes/ChatBox/tools/DataAnalysisTool.ts
@@ -1,8 +1,9 @@
 import { BaseTool } from './BaseTool';
 
 export class DataAnalysisTool extends BaseTool<{ csv_file_name: string }> {
-    private csv_files_function: () => { [filename: string]: string };
-    constructor(csv_files: () => { [filename: string]: string }) {
+    private csv_files_function: () => { filename: string; id: string; text: string }[];
+
+    constructor(csv_files: () => { filename: string; id: string; text: string }[]) {
         super(
             'dataAnalysis',
             'Analyzes, and provides insights, from a CSV file',
@@ -21,10 +22,18 @@ export class DataAnalysisTool extends BaseTool<{ csv_file_name: string }> {
 
     getFileContent(filename: string): string | undefined {
         const files = this.csv_files_function();
-        return files[filename];
+        const file = files.find(f => f.filename === filename);
+        return file?.text;
+    }
+
+    getFileID(filename: string): string | undefined {
+        const files = this.csv_files_function();
+        const file = files.find(f => f.filename === filename);
+        return file?.id;
     }
 
     async execute(args: { csv_file_name: string }): Promise<any> {
-        return [{ type: 'text', text: this.getFileContent(args.csv_file_name) }];
+        console.log(this.csv_files_function());
+        return [{ type: 'text', text: `<chunk chunk_id=${this.getFileID(args.csv_file_name)} chunk_type=csv}>` + this.getFileContent(args.csv_file_name) + '</chunk>' }];
     }
 }