Optimize the response

Signed-off-by: gaobinlong <[email protected]>
opensearch-project · Dec 2, 2024 · 035d673 · 035d673
1 parent fcbd5d2
commit 035d673
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/server/routes/text2viz_routes.ts b/server/routes/text2viz_routes.ts
@@ -86,11 +86,20 @@ export function registerText2VizRoutes(router: IRouter, assistantService: Assist
         }
         return res.badRequest();
       } catch (e) {
-        return res.customError({
-          body: e.body || 'execute agent failed',
-          statusCode: e.statusCode || 500,
-          headers: e.headers,
-        });
+        context.assistant_plugin.logger.error('Execute agent failed!', e);
+        if (e.statusCode >= 400 && e.statusCode <= 499) {
+          return res.customError({
+            body: e.body,
+            statusCode: e.statusCode,
+            headers: e.headers,
+          });
+        } else {
+          return res.customError({
+            body: 'Execute agent failed!',
+            statusCode: 500,
+            headers: e.headers,
+          });
+        }
       }
     })
   );