From 6626fc9cc053f4c8d924a39a0fb694a3c5cf3811 Mon Sep 17 00:00:00 2001
From: Patryk Kopycinski <patryk.kopycinski@elastic.co>
Date: Tue, 25 Feb 2025 10:16:42 +0100
Subject: [PATCH] Fix lang

---
 .../get_comments/custom_codeblock/custom_code_block.tsx       | 4 ++--
 .../public/assistant/get_comments/stream/message_text.tsx     | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/custom_codeblock/custom_code_block.tsx b/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/custom_codeblock/custom_code_block.tsx
index 90e406f500ba8..0731d06a5dd26 100644
--- a/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/custom_codeblock/custom_code_block.tsx
+++ b/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/custom_codeblock/custom_code_block.tsx
@@ -8,7 +8,7 @@ import { EuiCodeBlock, EuiFlexGroup, EuiFlexItem, EuiPanel, useEuiTheme } from '
 import { css } from '@emotion/css';
 import React from 'react';
 
-export const CustomCodeBlock = ({ value }: { value: string }) => {
+export const CustomCodeBlock = ({ value, lang }: { value: string; lang: string }) => {
   const theme = useEuiTheme();
 
   return (
@@ -31,7 +31,7 @@ export const CustomCodeBlock = ({ value }: { value: string }) => {
     >
       <EuiFlexGroup direction="column" gutterSize="xs">
         <EuiFlexItem grow={false}>
-          <EuiCodeBlock isCopyable fontSize="m">
+          <EuiCodeBlock isCopyable fontSize="m" language={lang}>
             {value}
           </EuiCodeBlock>
         </EuiFlexItem>
diff --git a/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/stream/message_text.tsx b/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/stream/message_text.tsx
index 60020f8b1abff..8db63019636d2 100644
--- a/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/stream/message_text.tsx
+++ b/x-pack/solutions/security/plugins/security_solution/public/assistant/get_comments/stream/message_text.tsx
@@ -139,7 +139,7 @@ const getPluginDependencies = ({
     customCodeBlock: (props) => {
       return (
         <>
-          <CustomCodeBlock value={props.value} />
+          <CustomCodeBlock value={props.value} lang={props.lang} />
           <EuiSpacer size="m" />
         </>
       );