From 6f9b8f9b393d411a17f7954b6cc36618efe698e2 Mon Sep 17 00:00:00 2001
From: "A.J. Shulman" <Shulman.aj@gmail.com>
Date: Thu, 15 Aug 2024 13:16:32 -0400
Subject: implemented search tool and other tools but scraping doesn't work

---
 .../views/nodes/ChatBox/tools/DataAnalysisTool.ts       | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

(limited to 'src/client/views/nodes/ChatBox/tools/DataAnalysisTool.ts')

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>' }];
     }
 }
-- 
cgit v1.2.3-70-g09d2