[Observability AI Assistant] Return errorDoc instead of error in …

…GetApmErrorDocument function (#165588)
elastic · Sep 4, 2023 · 046cfc8 · 046cfc8
1 parent 4b4ba09
commit 046cfc8
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/x-pack/plugins/apm/server/routes/assistant_functions/get_apm_error_document/index.ts b/x-pack/plugins/apm/server/routes/assistant_functions/get_apm_error_document/index.ts
@@ -54,14 +54,14 @@ export async function getApmErrorDocument({
     },
   });
 
-  const error = response.hits.hits[0]?._source as APMError;
+  const errorDoc = response.hits.hits[0]?._source as APMError;
 
-  if (!error) {
+  if (!errorDoc) {
     return undefined;
   }
 
-  return pick(
-    error,
+  const formattedResponse = pick(
+    errorDoc,
     'message',
     'error',
     '@timestamp',
@@ -71,4 +71,11 @@ export async function getApmErrorDocument({
     'span.type',
     'span.subtype'
   );
+
+  const { error, ...rest } = formattedResponse;
+
+  return {
+    ...rest,
+    errorDoc: formattedResponse.error,
+  };
 }