working much better still working on adding images thouhg

1 files changed, 12 insertions, 31 deletions

diff --git a/src/client/views/nodes/ChatBox/tools/RAGTool.ts b/src/client/views/nodes/ChatBox/tools/RAGTool.ts
index 90f7bebfe..0a4529974 100644
--- a/src/client/views/nodes/ChatBox/tools/RAGTool.ts
+++ b/src/client/views/nodes/ChatBox/tools/RAGTool.ts
@@ -2,6 +2,7 @@ import { BaseTool } from './BaseTool';
 import { Vectorstore } from '../vectorstore/VectorstoreUpload';
 import { Chunk } from '../types';
 import * as fs from 'fs';
+import { Networking } from '../../../../Network';
 
 export class RAGTool extends BaseTool<{ hypothetical_document_chunk: string }> {
     constructor(
@@ -52,42 +53,22 @@ export class RAGTool extends BaseTool<{ hypothetical_document_chunk: string }> {
 
     async execute(args: { hypothetical_document_chunk: string }): Promise<any> {
         const relevantChunks = await this.vectorstore.retrieve(args.hypothetical_document_chunk);
-        return this.getFormattedChunks(relevantChunks);
+        const formatted_chunks = await this.getFormattedChunks(relevantChunks);
+        return formatted_chunks;
     }
 
-    private getFormattedChunks(relevantChunks: Chunk[]): { type: string; text?: string; image_url?: { url: string } }[] {
-        const content: { type: string; text?: string; image_url?: { url: string } }[] = [{ type: 'text', text: '<chunks>' }];
+    async getFormattedChunks(relevantChunks: Chunk[]): Promise<{ type: string; text?: string; image_url?: { url: string } }[]> {
+        try {
+            const { formattedChunks } = await Networking.PostToServer('/formatChunks', { relevantChunks });
 
-        for (const chunk of relevantChunks) {
-            content.push({
-                type: 'text',
-                text: `<chunk chunk_id=${chunk.id} chunk_type=${chunk.metadata.type === 'image' || chunk.metadata.type === 'table' ? 'image' : 'text'}>`,
-            });
-
-            if (chunk.metadata.type === 'image' || chunk.metadata.type === 'table') {
-                try {
-                    const imageBuffer = fs.readFileSync(chunk.metadata.file_path);
-                    const base64Image = imageBuffer.toString('base64');
-                    if (base64Image) {
-                        content.push({
-                            type: 'image_url',
-                            image_url: {
-                                url: `data:image/jpeg;base64,${base64Image}`,
-                            },
-                        });
-                    } else {
-                        console.log(`Failed to encode image for chunk ${chunk.id}`);
-                    }
-                } catch (error) {
-                    console.error(`Error reading image file for chunk ${chunk.id}:`, error);
-                }
+            if (!formattedChunks) {
+                throw new Error('Failed to format chunks');
             }
 
-            content.push({ type: 'text', text: `${chunk.metadata.text}\n</chunk>\n` });
+            return formattedChunks;
+        } catch (error) {
+            console.error('Error formatting chunks:', error);
+            throw error;
         }
-
-        content.push({ type: 'text', text: '</chunks>' });
-
-        return content;
     }
 }