From 3d290c126623404b56958d16becc244d04548658 Mon Sep 17 00:00:00 2001
From: "A.J. Shulman" <Shulman.aj@gmail.com>
Date: Thu, 22 Aug 2024 11:17:28 -0400
Subject: fixed table citations and now formatted chunks show table info as
 well

---
 src/server/ApiManagers/AssistantManager.ts | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

(limited to 'src/server/ApiManagers')

diff --git a/src/server/ApiManagers/AssistantManager.ts b/src/server/ApiManagers/AssistantManager.ts
index a59a2d22d..9bc5bf128 100644
--- a/src/server/ApiManagers/AssistantManager.ts
+++ b/src/server/ApiManagers/AssistantManager.ts
@@ -189,7 +189,7 @@ export default class AssistantManager extends ApiManager {
 
                 try {
                     // Read file data and convert to base64
-                    const file_data = fs.readFileSync(public_path, { encoding: 'base64' });
+                    const file_data: string = fs.readFileSync(public_path, { encoding: 'base64' });
 
                     const response = await axios.post(
                         'http://localhost:8080/createDocument',
@@ -265,12 +265,12 @@ export default class AssistantManager extends ApiManager {
                 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'}>`,
+                        text: `<chunk chunk_id=${chunk.id} chunk_type=${chunk.metadata.type}>`,
                     });
 
                     if (chunk.metadata.type === 'image' || chunk.metadata.type === 'table') {
                         try {
-                            const filePath = serverPathToFile(Directory.parsed_files, chunk.metadata.file_path);
+                            const filePath = serverPathToFile(Directory.chunk_images, chunk.metadata.file_path);
                             const imageBuffer = await readFileAsync(filePath);
                             const base64Image = imageBuffer.toString('base64');
                             if (base64Image) {
@@ -286,10 +286,8 @@ export default class AssistantManager extends ApiManager {
                         } catch (error) {
                             console.error(`Error reading image file for chunk ${chunk.id}:`, error);
                         }
-                        content.push({ type: 'text', text: `\n</chunk>\n` });
-                    } else {
-                        content.push({ type: 'text', text: `${chunk.metadata.text}\n</chunk>\n` });
                     }
+                    content.push({ type: 'text', text: `${chunk.metadata.text}\n</chunk>\n` });
                 }
 
                 content.push({ type: 'text', text: '</chunks>' });
-- 
cgit v1.2.3-70-g09d2