From ca0d8ba8d179375b1ec9fd6a3e7f90433c1f6e41 Mon Sep 17 00:00:00 2001 From: Tyler Ohlsen Date: Fri, 13 Sep 2024 11:15:35 -0700 Subject: [PATCH] Fix parsing of ML resp processor Signed-off-by: Tyler Ohlsen --- .../processor_inputs/output_transform_modal.tsx | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/public/pages/workflow_detail/workflow_inputs/processor_inputs/output_transform_modal.tsx b/public/pages/workflow_detail/workflow_inputs/processor_inputs/output_transform_modal.tsx index 0d709bca..47f87f2d 100644 --- a/public/pages/workflow_detail/workflow_inputs/processor_inputs/output_transform_modal.tsx +++ b/public/pages/workflow_detail/workflow_inputs/processor_inputs/output_transform_modal.tsx @@ -5,7 +5,7 @@ import React, { useState, useEffect } from 'react'; import { useFormikContext, getIn } from 'formik'; -import { cloneDeep, get, isEmpty, set } from 'lodash'; +import { cloneDeep, isEmpty, set } from 'lodash'; import { EuiCodeEditor, EuiFlexGroup, @@ -168,7 +168,7 @@ export function OutputTransformModal(props: OutputTransformModalProps) { const docObjs = unwrapTransformedDocs(resp); if (docObjs.length > 0) { const sampleModelResult = - docObjs[0]?.inference_results; + docObjs[0]?.inference_results || {}; setSourceInput( customStringify(sampleModelResult) ); @@ -224,11 +224,8 @@ export function OutputTransformModal(props: OutputTransformModalProps) { (hit: SearchHit) => hit._source ) as any[]; if (hits.length > 0) { - const sampleModelResult = get( - hits, - '0.inference_results.0', - {} - ); + const sampleModelResult = + hits[0].inference_results || {}; setSourceInput(customStringify(sampleModelResult)); } })