(query assist) extract ppl from error response

Signed-off-by: Joshua Li <[email protected]>
joshuali925 · Jan 19, 2024 · 0e926c8 · 0e926c8
1 parent 8f1f83a
commit 0e926c8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/public/components/event_analytics/explorer/query_assist/input.tsx b/public/components/event_analytics/explorer/query_assist/input.tsx
@@ -99,7 +99,8 @@ export const QueryAssistInput: React.FC<Props> = (props) => {
             },
           })
         );
-        if (explorerData.total > 0) generateSummary();
+        if (explorerData.total > 0 || summaryData.responseForSummaryStatus === 'failure')
+          generateSummary();
       })();
     }
   }, [summaryData.responseForSummaryStatus]);

diff --git a/server/routes/query_assist/routes.ts b/server/routes/query_assist/routes.ts
@@ -69,10 +69,17 @@ export function registerQueryAssistRoutes(router: IRouter, config: Observability
           .replace(/\bSPAN\(/g, 'span('); // https://github.com/opensearch-project/dashboards-observability/issues/759
         return response.ok({ body: ppl });
       } catch (error) {
-        return response.custom({
-          statusCode: error.statusCode || 500,
-          body: error.message,
-        });
+        // parse PPL query from error response if exists
+        // TODO remove after https://github.com/opensearch-project/skills/issues/138
+        if (error.name === 'ResponseError') {
+          try {
+            const ppl = error.body.error.reason.match(/execute ppl:(.+), get error:/);
+            return response.ok({ body: ppl[1] });
+          } catch (parseError) {
+            return response.custom({ statusCode: error.statusCode || 500, body: error.message });
+          }
+        }
+        return response.custom({ statusCode: error.statusCode || 500, body: error.message });
       }
     }
   );