[FEAT] Table extractor Enhancements FE (#743)

* Select choices for cloud repo * Remove choices static file * Support for Input type * Reverting support for TSV * Addition of records * Choices field addition * Variable renames * Adding helper text for record enforce type * added new enforce type --------- Signed-off-by: harini-venkataraman <[email protected]> Co-authored-by: harini-venkataraman <[email protected]> Co-authored-by: harini-venkataraman <[email protected]>
Zipstack · Sep 26, 2024 · 5582a63 · 5582a63
1 parent 9754247
commit 5582a63
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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 };