@paro.io/expert-shared-components 1.14.73 → 1.14.75

package/lib/components/TaxAxis/TaxAxisShell.js

@@ -195,20 +195,68 @@ const TaxAxisShell = ({ taxAxisApi, userContext = 'expert', initialSessionId, in
         }
     }), [taxAxisApi]);
     // On mount: if a sessionId was injected (e.g. returning to an existing session),
-    // reload the last GENERATION llmRun so the report screens get live data.
+    // fetch the session stage and resume at the correct step.
     (0, react_1.useEffect)(() => {
-        if (!initialSessionId || !taxAxisApi.getLlmRun)
+        if (!initialSessionId)
             return;
-        taxAxisApi.getLlmRun(initialSessionId, 'GENERATION')
-            .then((run) => {
-            if (run === null || run === void 0 ? void 0 : run.outputPayload) {
-                setLlmResult(run.outputPayload);
+        taxAxisApi.getSession(initialSessionId)
+            .then((session) => __awaiter(void 0, void 0, void 0, function* () {
+            var _a;
+            if (!session)
+                return;
+            // Restore profile from session fields so DOCUMENT_UPLOAD header renders correctly
+            setSessionId(initialSessionId);
+            setProfile((prev) => {
+                var _a, _b, _c, _d, _e, _f, _g, _h;
+                return prev !== null && prev !== void 0 ? prev : {
+                    bizName: (_a = session.businessName) !== null && _a !== void 0 ? _a : '',
+                    entity: (_b = session.entityType) !== null && _b !== void 0 ? _b : '',
+                    industry: (_c = session.industry) !== null && _c !== void 0 ? _c : '',
+                    year: String((_d = session.taxYear) !== null && _d !== void 0 ? _d : new Date().getFullYear()),
+                    revenue: String((_e = session.annualRevenue) !== null && _e !== void 0 ? _e : ''),
+                    ownerComp: String((_f = session.w2Compensation) !== null && _f !== void 0 ? _f : ''),
+                    employees: String((_g = session.employeeCount) !== null && _g !== void 0 ? _g : ''),
+                    states: (_h = session.states) !== null && _h !== void 0 ? _h : [],
+                };
+            });
+            const stage = (_a = session.stage) !== null && _a !== void 0 ? _a : '';
+            if (stage === 'REPORT_READY' || stage === 'COMPLETED') {
+                // Load LLM run data and jump to dashboard
+                if (taxAxisApi.getLlmRun) {
+                    try {
+                        const run = yield taxAxisApi.getLlmRun(initialSessionId, 'GENERATION');
+                        if (run === null || run === void 0 ? void 0 : run.outputPayload) {
+                            setLlmResult(run.outputPayload);
+                        }
+                    }
+                    catch ( /* non-blocking */_b) { /* non-blocking */ }
+                }
+                yield fetchAndSetParsedDocuments(initialSessionId);
                 setStep('DASHBOARD');
-                fetchAndSetParsedDocuments(initialSessionId);
+                if (onSessionChange)
+                    onSessionChange(initialSessionId);
+                return;
+            }
+            if (stage === 'LLM_RUNNING' || stage === 'EVAL_RUNNING') {
+                // Jump straight to processing screen and start polling
+                setStep('PROCESSING');
+                pollForResult(initialSessionId);
+                if (onSessionChange)
+                    onSessionChange(initialSessionId);
+                return;
             }
-        })
-            .catch(() => { });
-        // fetchAndSetParsedDocuments intentionally excluded: only runs on mount for initial session restore
+            // INITIALIZED, DOCUMENTS_UPLOADED, PARSING, PARSED, EXTRACTION_REVIEW, FAILED, EVAL_FAILED
+            // Resume at document upload so the user can review/add documents or retry
+            setStep('DOCUMENT_UPLOAD');
+            if (onSessionChange)
+                onSessionChange(initialSessionId);
+        }))
+            .catch(() => {
+            // On error fall back to SESSION_SETUP — clear skeleton so page isn't stuck
+            if (onSessionChange)
+                onSessionChange(null);
+        });
+        // fetchAndSetParsedDocuments and pollForResult intentionally excluded: only runs on mount
         // eslint-disable-next-line react-hooks/exhaustive-deps
     }, [initialSessionId]);
     const updateSessionId = (0, react_1.useCallback)((nextSessionId) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@paro.io/expert-shared-components",
-  "version": "1.14.73",
+  "version": "1.14.75",
   "description": "",
   "main": "lib/index.js",
   "scripts": {