Fix #151 - UI: Document Schema Editor Regex Fields Not Persisting

Author: Bob Strahan

CHANGELOG.md

@@ -63,6 +63,7 @@ SPDX-License-Identifier: MIT-0
 
 ### Fixed
 
+- **UI: Document Schema Editor Regex Fields Not Persisting** - Fixed issue where Document Name Regex and Page Content Regex fields were not being saved in configuration or restored after page refresh. Fixes #151
 - **Document Schema Builder Enum Support** - Fixed enum value handling in schema builder to properly support enumeration constraints for attribute definitions
 - **Agentic Extraction Parameter Passing** - Fixed temperature and top_p parameters now correctly passed to agentic extraction service, enabling proper model behavior control
 - **Document Schema Builder UI Labels** - Enhanced field labels and formats in document schema builder for improved clarity and user experience

src/ui/src/hooks/useSchemaDesigner.js

@@ -1,6 +1,11 @@
 import { useState, useCallback, useEffect } from 'react';
 import { produce } from 'immer';
-import { X_AWS_IDP_DOCUMENT_TYPE, X_AWS_IDP_EXAMPLES } from '../constants/schemaConstants';
+import {
+  X_AWS_IDP_DOCUMENT_TYPE,
+  X_AWS_IDP_EXAMPLES,
+  X_AWS_IDP_DOCUMENT_NAME_REGEX,
+  X_AWS_IDP_PAGE_CONTENT_REGEX,
+} from '../constants/schemaConstants';
 
 const extractInlineObjectsToClasses = (properties, extractedClasses, timestamp) => {
   const updatedProperties = {};
@@ -129,6 +134,9 @@ const convertJsonSchemaToClasses = (jsonSchema) => {
         },
         // Preserve examples if they exist in the schema
         ...(schema[X_AWS_IDP_EXAMPLES] ? { [X_AWS_IDP_EXAMPLES]: schema[X_AWS_IDP_EXAMPLES] } : {}),
+        // Preserve regex fields if they exist in the schema
+        ...(schema[X_AWS_IDP_DOCUMENT_NAME_REGEX] ? { [X_AWS_IDP_DOCUMENT_NAME_REGEX]: schema[X_AWS_IDP_DOCUMENT_NAME_REGEX] } : {}),
+        ...(schema[X_AWS_IDP_PAGE_CONTENT_REGEX] ? { [X_AWS_IDP_PAGE_CONTENT_REGEX]: schema[X_AWS_IDP_PAGE_CONTENT_REGEX] } : {}),
       };
       allClasses.push(docTypeClass);
 
@@ -192,6 +200,9 @@ const convertJsonSchemaToClasses = (jsonSchema) => {
     },
     // Preserve examples if they exist in the schema
     ...(jsonSchema[X_AWS_IDP_EXAMPLES] ? { [X_AWS_IDP_EXAMPLES]: jsonSchema[X_AWS_IDP_EXAMPLES] } : {}),
+    // Preserve regex fields if they exist in the schema
+    ...(jsonSchema[X_AWS_IDP_DOCUMENT_NAME_REGEX] ? { [X_AWS_IDP_DOCUMENT_NAME_REGEX]: jsonSchema[X_AWS_IDP_DOCUMENT_NAME_REGEX] } : {}),
+    ...(jsonSchema[X_AWS_IDP_PAGE_CONTENT_REGEX] ? { [X_AWS_IDP_PAGE_CONTENT_REGEX]: jsonSchema[X_AWS_IDP_PAGE_CONTENT_REGEX] } : {}),
   };
   classes.push(mainClass);
 
@@ -619,6 +630,12 @@ export const useSchemaDesigner = (initialSchema = []) => {
         ...(docTypeClass.attributes.required?.length > 0 ? { required: docTypeClass.attributes.required } : {}),
         ...(Object.keys(defs).length > 0 ? { $defs: defs } : {}),
         ...(docTypeClass[X_AWS_IDP_EXAMPLES]?.length > 0 ? { [X_AWS_IDP_EXAMPLES]: docTypeClass[X_AWS_IDP_EXAMPLES] } : {}),
+        ...(docTypeClass[X_AWS_IDP_DOCUMENT_NAME_REGEX]
+          ? { [X_AWS_IDP_DOCUMENT_NAME_REGEX]: docTypeClass[X_AWS_IDP_DOCUMENT_NAME_REGEX] }
+          : {}),
+        ...(docTypeClass[X_AWS_IDP_PAGE_CONTENT_REGEX]
+          ? { [X_AWS_IDP_PAGE_CONTENT_REGEX]: docTypeClass[X_AWS_IDP_PAGE_CONTENT_REGEX] }
+          : {}),
       };
 
       console.log('Final schema has $defs?', '$defs' in result);