1 files changed, 13 insertions, 1 deletions
diff --git a/src/server/ApiManagers/AssistantManager.ts b/src/server/ApiManagers/AssistantManager.ts
index a35708ccd..8a5f12c2b 100644
--- a/src/server/ApiManagers/AssistantManager.ts
+++ b/src/server/ApiManagers/AssistantManager.ts
@@ -173,7 +173,19 @@ export default class AssistantManager extends ApiManager {
                         }
                     );
 
-                    res.send({ document: response.data });
+                    const jobId = response.data.job_id;
+
+                    // Poll for results
+                    let result;
+                    while (!result) {
+                        await new Promise(resolve => setTimeout(resolve, 5000)); // Wait for 1 second
+                        const resultResponse = await axios.get(`http://localhost:8080/getResult/${jobId}`);
+                        if (resultResponse.status === 200) {
+                            result = resultResponse.data;
+                        }
+                    }
+
+                    res.send({ document_json: result });
                 } catch (error: any) {
                     console.error('Error communicating with chatbot:', error);
                     res.status(500).send({ error: 'Failed to communicate with the chatbot', details: error.message });