diff --git a/x-pack/plugins/observability_solution/logs_shared/public/components/log_ai_assistant/index.tsx b/x-pack/plugins/observability_solution/logs_shared/public/components/log_ai_assistant/index.tsx
index 465f1f3f66eff..484af6e4a0809 100644
--- a/x-pack/plugins/observability_solution/logs_shared/public/components/log_ai_assistant/index.tsx
+++ b/x-pack/plugins/observability_solution/logs_shared/public/components/log_ai_assistant/index.tsx
@@ -22,12 +22,12 @@ export function createLogAIAssistant({
 export const createLogsAIAssistantRenderer =
   (LogAIAssistantRender: ReturnType<typeof createLogAIAssistant>) =>
   ({ doc }: ObservabilityLogsAIAssistantFeatureRenderDeps) => {
-    const mappedDoc = useMemo(
+    const mappedDoc = useMemo<LogAIAssistantDocument>(
       () => ({
         fields: Object.entries(doc.flattened).map(([field, value]) => ({
           field,
-          value,
-        })) as LogAIAssistantDocument['fields'],
+          value: Array.isArray(value) ? value : [value],
+        })),
       }),
       [doc]
     );
diff --git a/x-pack/plugins/observability_solution/logs_shared/public/components/log_ai_assistant/log_ai_assistant.tsx b/x-pack/plugins/observability_solution/logs_shared/public/components/log_ai_assistant/log_ai_assistant.tsx
index 49d4fbc01c2e4..3e1b6fced3337 100644
--- a/x-pack/plugins/observability_solution/logs_shared/public/components/log_ai_assistant/log_ai_assistant.tsx
+++ b/x-pack/plugins/observability_solution/logs_shared/public/components/log_ai_assistant/log_ai_assistant.tsx
@@ -51,8 +51,7 @@ export const LogAIAssistant = ({
       return undefined;
     }
 
-    const messageValue = doc.fields.find((field) => field.field === 'message')?.value;
-    const message = Array.isArray(messageValue) ? messageValue[0] : messageValue;
+    const message = doc.fields.find((field) => field.field === 'message')?.value[0];
 
     return getContextualInsightMessages({
       message: `I'm looking at a log entry. Can you construct a Kibana KQL query that I can enter in the search bar that gives me similar log entries, based on the message field?`,