diff --git a/frontend/src/components/custom-tools/prompt-card/PromptCardItems.jsx b/frontend/src/components/custom-tools/prompt-card/PromptCardItems.jsx
index cc1b88fd2..1f8463a9c 100644
--- a/frontend/src/components/custom-tools/prompt-card/PromptCardItems.jsx
+++ b/frontend/src/components/custom-tools/prompt-card/PromptCardItems.jsx
@@ -18,7 +18,7 @@ import { useCustomToolStore } from "../../../store/custom-tool-store";
 import { Header } from "./Header";
 import { OutputForIndex } from "./OutputForIndex";
 import { PromptOutput } from "./PromptOutput";
-import { TABLE_ENFORCE_TYPE } from "./constants";
+import { TABLE_ENFORCE_TYPE, RECORD_ENFORCE_TYPE } from "./constants";
 
 let TableExtractionSettingsBtn;
 try {
@@ -234,10 +234,12 @@ function PromptCardItems({
                       </Button>
                     </Space>
                     <Space>
-                      {enforceType === TABLE_ENFORCE_TYPE &&
+                      {(enforceType === TABLE_ENFORCE_TYPE ||
+                        enforceType === RECORD_ENFORCE_TYPE) &&
                         TableExtractionSettingsBtn && (
                           <TableExtractionSettingsBtn
                             promptId={promptDetails?.prompt_id}
+                            enforceType={enforceType}
                           />
                         )}
                       <Select
diff --git a/frontend/src/components/custom-tools/prompt-card/constants.js b/frontend/src/components/custom-tools/prompt-card/constants.js
index 951e737ee..9e35aa5b0 100644
--- a/frontend/src/components/custom-tools/prompt-card/constants.js
+++ b/frontend/src/components/custom-tools/prompt-card/constants.js
@@ -1,4 +1,5 @@
 const TABLE_ENFORCE_TYPE = "table";
+const RECORD_ENFORCE_TYPE = "record";
 
 const handleUpdateStatus = (isUpdate, promptId, value, setUpdateStatus) => {
   if (!isUpdate) {
@@ -10,4 +11,4 @@ const handleUpdateStatus = (isUpdate, promptId, value, setUpdateStatus) => {
   });
 };
 
-export { TABLE_ENFORCE_TYPE, handleUpdateStatus };
+export { TABLE_ENFORCE_TYPE, handleUpdateStatus, RECORD_ENFORCE_TYPE };