From f473137253caab6ddcab8b35bf487460360f9183 Mon Sep 17 00:00:00 2001
From: deep1401 <gandhi0869@gmail.com>
Date: Fri, 21 Feb 2025 08:01:41 -0800
Subject: [PATCH] Fix bug which caused document preview issues within folders

---
 src/renderer/components/Experiment/Rag/Documents.tsx | 3 ++-
 src/renderer/lib/transformerlab-api-sdk.ts           | 6 ++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/renderer/components/Experiment/Rag/Documents.tsx b/src/renderer/components/Experiment/Rag/Documents.tsx
index bc03db64..28cf6a11 100644
--- a/src/renderer/components/Experiment/Rag/Documents.tsx
+++ b/src/renderer/components/Experiment/Rag/Documents.tsx
@@ -420,7 +420,8 @@ export default function Documents({
           <iframe
             src={chatAPI.Endpoints.Documents.Open(
               experimentInfo?.id,
-              previewFile
+              previewFile,
+              currentFolder
             )}
             style={{ width: '100%', height: '100%' }}
           ></iframe>
diff --git a/src/renderer/lib/transformerlab-api-sdk.ts b/src/renderer/lib/transformerlab-api-sdk.ts
index f49b7a5e..cbf52369 100644
--- a/src/renderer/lib/transformerlab-api-sdk.ts
+++ b/src/renderer/lib/transformerlab-api-sdk.ts
@@ -1116,12 +1116,14 @@ Endpoints.Config = {
 Endpoints.Documents = {
   List: (experimentId: string, currentFolder: string = '') =>
     API_URL() + 'experiment/' + experimentId + '/documents/list?folder=' + currentFolder,
-  Open: (experimentId: string, document_name: string) =>
+  Open: (experimentId: string, document_name: string, folder: string) =>
     API_URL() +
     'experiment/' +
     experimentId +
     '/documents/open/' +
-    document_name,
+    document_name +
+    '?folder=' +
+    folder,
   Upload: (experimentId: string, currentFolder: string = '') =>
     API_URL() + 'experiment/' + experimentId + '/documents/upload?folder=' + currentFolder,
   Delete: (experimentId: string, document_name: string) =>
-- 
GitLab