draft: pass through plugins

packages/visual-editor/THIRD-PARTY-NOTICES

@@ -1163,7 +1163,8 @@ OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 
 The following npm packages may be included in this product:
 
- - @measured/puck@0.20.2
+ - @measured/puck-plugin-heading-analyzer@0.20.2
+ - @measured/puck@0.21.0-canary.c0db75c1
  - next-themes@0.3.0
 
 These packages each contain the following license:

packages/visual-editor/package.json

@@ -61,10 +61,13 @@
     "i18n:check-empty": "pnpm exec tsx scripts/checkEmptyTranslations.ts"
   },
   "dependencies": {
-    "@measured/puck": "0.20.2",
+    "@measured/puck": "0.21.0-canary.c0db75c1",
+    "@measured/puck-plugin-heading-analyzer": "^0.20.2",
     "@microsoft/api-documenter": "^7.26.29",
     "@microsoft/api-extractor": "^7.52.8",
     "@microsoft/api-extractor-model": "^7.30.6",
+    "@puckeditor/cloud-client": "^0.3.1",
+    "@puckeditor/plugin-ai": "^0.3.1",
     "@radix-ui/react-accordion": "^1.2.3",
     "@radix-ui/react-alert-dialog": "^1.1.1",
     "@radix-ui/react-dialog": "^1.1.13",
@@ -155,7 +158,7 @@
     "vite": "^5.3.5",
     "vitepress": "^1.6.3",
     "vitest": "^3.2.4",
-    "zod": "^3.23.8"
+    "zod": "^4.1.13"
   },
   "peerDependencies": {
     "@yext/pages-components": "1.1.16",

packages/visual-editor/src/components/categories/SlotsCategory.tsx

@@ -192,81 +192,82 @@ const ExpandedHeaderComponents = {
   },
 };
 
-export const SlotsCategoryComponents = {
-  AddressSlot: { ...Address, permissions: lockedPermissions },
-  BodyTextSlot: { ...BodyText, permissions: lockedPermissions },
-  BreadcrumbsSlot: { ...BreadcrumbsSection, permissions: lockedPermissions },
-  CopyrightMessageSlot: {
-    ...CopyrightMessageSlot,
-    permissions: lockedPermissions,
-  },
-  CTASlot: { ...CTAWrapper, permissions: lockedPermissions },
-  DirectoryCard: { ...DirectoryCard, permissions: lockedPermissions },
-  DirectoryGrid: { ...DirectoryGrid, permissions: lockedPermissions },
-  EmailsSlot: { ...Emails, permissions: lockedPermissions },
-  EventCard: { ...EventCard, permissions: lockedPermissions },
-  EventCardsWrapper: { ...EventCardsWrapper, permissions: lockedPermissions },
-  ...ExpandedHeaderComponents,
-  FAQsWrapperSlot: { ...FAQsWrapperSlot, permissions: lockedPermissions },
-  FAQSlot: { ...FAQSlot, permissions: lockedPermissions },
-  FooterExpandedLinkSectionSlot: {
-    ...FooterExpandedLinkSectionSlot,
-    permissions: lockedPermissions,
-  },
-  FooterExpandedLinksWrapper: {
-    ...FooterExpandedLinksWrapper,
-    permissions: lockedPermissions,
-  },
-  FooterLinksSlot: { ...FooterLinksSlot, permissions: lockedPermissions },
-  FooterLogoSlot: { ...FooterLogoSlot, permissions: lockedPermissions },
-  FooterSocialLinksSlot: {
-    ...FooterSocialLinksSlot,
-    permissions: lockedPermissions,
-  },
-  FooterUtilityImagesSlot: {
-    ...FooterUtilityImagesSlot,
-    permissions: lockedPermissions,
-  },
-  HeadingTextSlot: { ...HeadingText, permissions: lockedPermissions },
-  HeroImageSlot: { ...HeroImage, permissions: lockedPermissions },
-  HoursStatusSlot: { ...HoursStatus, permissions: lockedPermissions },
-  HoursTableSlot: { ...HoursTable, permissions: lockedPermissions },
-  ImageSlot: { ...ImageWrapper, permissions: lockedPermissions },
-  InsightCardsWrapper: {
-    ...InsightCardsWrapper,
-    permissions: lockedPermissions,
-  },
-  InsightCard: { ...InsightCard, permissions: lockedPermissions },
-  NearbyLocationCardsWrapper: {
-    ...NearbyLocationCardsWrapper,
-    permissions: lockedPermissions,
-  },
-  PhoneNumbersSlot: { ...PhoneList, permissions: lockedPermissions },
-  PhoneSlot: { ...Phone, permissions: lockedPermissions },
-  PhotoGalleryWrapper: {
-    ...PhotoGalleryWrapper,
-    permissions: lockedPermissions,
-  },
-  ProductCardsWrapper: {
-    ...ProductCardsWrapper,
-    permissions: lockedPermissions,
-  },
-  ProductCard: { ...ProductCard, permissions: lockedPermissions },
-  SecondaryFooterSlot: {
-    ...SecondaryFooterSlot,
-    permissions: lockedPermissions,
-  },
-  TeamCard: { ...TeamCard, permissions: lockedPermissions },
-  TeamCardsWrapper: { ...TeamCardsWrapper, permissions: lockedPermissions },
-  TestimonialCard: { ...TestimonialCard, permissions: lockedPermissions },
-  TestimonialCardsWrapper: {
-    ...TestimonialCardsWrapper,
-    permissions: lockedPermissions,
-  },
-  TextListSlot: { ...TextList, permissions: lockedPermissions },
-  Timestamp: { ...Timestamp, permissions: lockedPermissions },
-  VideoSlot: { ...Video, permissions: lockedPermissions },
-};
+export const SlotsCategoryComponents: { [K in keyof SlotsCategoryProps]: any } =
+  {
+    AddressSlot: { ...Address, permissions: lockedPermissions },
+    BodyTextSlot: { ...BodyText, permissions: lockedPermissions },
+    BreadcrumbsSlot: { ...BreadcrumbsSection, permissions: lockedPermissions },
+    CopyrightMessageSlot: {
+      ...CopyrightMessageSlot,
+      permissions: lockedPermissions,
+    },
+    CTASlot: { ...CTAWrapper, permissions: lockedPermissions },
+    DirectoryCard: { ...DirectoryCard, permissions: lockedPermissions },
+    DirectoryGrid: { ...DirectoryGrid, permissions: lockedPermissions },
+    EmailsSlot: { ...Emails, permissions: lockedPermissions },
+    EventCard: { ...EventCard, permissions: lockedPermissions },
+    EventCardsWrapper: { ...EventCardsWrapper, permissions: lockedPermissions },
+    ...ExpandedHeaderComponents,
+    FAQsWrapperSlot: { ...FAQsWrapperSlot, permissions: lockedPermissions },
+    FAQSlot: { ...FAQSlot, permissions: lockedPermissions },
+    FooterExpandedLinkSectionSlot: {
+      ...FooterExpandedLinkSectionSlot,
+      permissions: lockedPermissions,
+    },
+    FooterExpandedLinksWrapper: {
+      ...FooterExpandedLinksWrapper,
+      permissions: lockedPermissions,
+    },
+    FooterLinksSlot: { ...FooterLinksSlot, permissions: lockedPermissions },
+    FooterLogoSlot: { ...FooterLogoSlot, permissions: lockedPermissions },
+    FooterSocialLinksSlot: {
+      ...FooterSocialLinksSlot,
+      permissions: lockedPermissions,
+    },
+    FooterUtilityImagesSlot: {
+      ...FooterUtilityImagesSlot,
+      permissions: lockedPermissions,
+    },
+    HeadingTextSlot: { ...HeadingText, permissions: lockedPermissions },
+    HeroImageSlot: { ...HeroImage, permissions: lockedPermissions },
+    HoursStatusSlot: { ...HoursStatus, permissions: lockedPermissions },
+    HoursTableSlot: { ...HoursTable, permissions: lockedPermissions },
+    ImageSlot: { ...ImageWrapper, permissions: lockedPermissions },
+    InsightCardsWrapper: {
+      ...InsightCardsWrapper,
+      permissions: lockedPermissions,
+    },
+    InsightCard: { ...InsightCard, permissions: lockedPermissions },
+    NearbyLocationCardsWrapper: {
+      ...NearbyLocationCardsWrapper,
+      permissions: lockedPermissions,
+    },
+    PhoneNumbersSlot: { ...PhoneList, permissions: lockedPermissions },
+    PhoneSlot: { ...Phone, permissions: lockedPermissions },
+    PhotoGalleryWrapper: {
+      ...PhotoGalleryWrapper,
+      permissions: lockedPermissions,
+    },
+    ProductCardsWrapper: {
+      ...ProductCardsWrapper,
+      permissions: lockedPermissions,
+    },
+    ProductCard: { ...ProductCard, permissions: lockedPermissions },
+    SecondaryFooterSlot: {
+      ...SecondaryFooterSlot,
+      permissions: lockedPermissions,
+    },
+    TeamCard: { ...TeamCard, permissions: lockedPermissions },
+    TeamCardsWrapper: { ...TeamCardsWrapper, permissions: lockedPermissions },
+    TestimonialCard: { ...TestimonialCard, permissions: lockedPermissions },
+    TestimonialCardsWrapper: {
+      ...TestimonialCardsWrapper,
+      permissions: lockedPermissions,
+    },
+    TextListSlot: { ...TextList, permissions: lockedPermissions },
+    Timestamp: { ...Timestamp, permissions: lockedPermissions },
+    VideoSlot: { ...Video, permissions: lockedPermissions },
+  };
 
 export const SlotsCategory = Object.keys(
   SlotsCategoryComponents

packages/visual-editor/src/editor/Editor.tsx

@@ -1,5 +1,5 @@
 import "./index.css";
-import React, { ErrorInfo, useEffect, useState } from "react";
+import { ErrorInfo, useEffect, useState } from "react";
 import { ErrorBoundary } from "react-error-boundary";
 import { LoadingScreen } from "../internal/puck/components/LoadingScreen.tsx";
 import { Toaster } from "../internal/puck/ui/Toaster.tsx";

packages/visual-editor/src/editor/YextEntityFieldSelector.tsx

@@ -1,5 +1,11 @@
 import React from "react";
-import { AutoField, FieldLabel, Field, CustomField } from "@measured/puck";
+import {
+  AutoField,
+  FieldLabel,
+  Field,
+  CustomField,
+  UiState,
+} from "@measured/puck";
 import {
   ConstantValueTypes,
   EntityFieldTypes,
@@ -44,11 +50,10 @@ import { EmbeddedFieldStringInputFromEntity } from "./EmbeddedFieldStringInput.t
 import { ComboboxOption } from "../internal/puck/ui/Combobox.tsx";
 import { DATE_TIME_CONSTANT_CONFIG } from "../internal/puck/components/DateTimeSelector.tsx";
 import { FAQ_SECTION_CONSTANT_CONFIG } from "../internal/puck/constant-value-fields/FAQsSection";
+import { getSchemaForYextEntityField, FieldAiConfig } from "./aiSchemas.ts";
 
 const devLogger = new DevLogger();
 
-type RenderProps = Parameters<CustomField<any>["render"]>[0];
-
 /** Represents data that can either be from the Yext Knowledge Graph or statically defined */
 export type YextEntityField<T> = {
   /** The api name of the Yext field */
@@ -94,6 +99,8 @@ export type RenderYextEntityFieldSelectorProps<T extends Record<string, any>> =
     disableConstantValueToggle?: boolean;
     disallowTranslation?: boolean;
     typeSelectorConfig?: TypeSelectorConfigProps;
+    /** AI configuration for Puck AI plugin */
+    ai?: FieldAiConfig;
   };
 
 export const TYPE_TO_CONSTANT_CONFIG: Record<string, Field<any>> = {
@@ -196,16 +203,36 @@ const returnConstantFieldConfig = (
   }
   return fieldConfiguration;
 };
-
 /**
  * Allows the user to select an entity field from the document and set a constant value.
  */
 export const YextEntityFieldSelector = <T extends Record<string, any>, U>(
   props: RenderYextEntityFieldSelectorProps<T>
 ): Field<YextEntityField<U>> => {
+  // Generate default schema based on filter types
+  const defaultSchema = getSchemaForYextEntityField(props.filter);
+
+  // Build AI config with default schema (can be overridden by props.ai)
+  // If exclude is true, don't include schema (workaround for Puck bug)
+  const aiConfig: FieldAiConfig = props.ai?.exclude
+    ? { ...props.ai }
+    : {
+        ...(defaultSchema ? { schema: defaultSchema } : {}),
+        ...props.ai,
+      };
+
   return {
     type: "custom",
-    render: ({ value, onChange }: RenderProps) => {
+    // Include AI config if we have a schema or user provided config
+    ...(Object.keys(aiConfig).length > 0 ? { ai: aiConfig } : {}),
+
+    render: ({
+      value,
+      onChange,
+    }: {
+      value: YextEntityField<U>;
+      onChange: (value: YextEntityField<U>) => void;
+    }) => {
       const toggleConstantValueEnabled = (constantValueEnabled: boolean) => {
         onChange({
           ...value,
@@ -217,7 +244,7 @@ export const YextEntityFieldSelector = <T extends Record<string, any>, U>(
         <>
           <ConstantValueModeToggler
             fieldTypeFilter={props.filter.types ?? []}
-            constantValueEnabled={value?.constantValueEnabled}
+            constantValueEnabled={value?.constantValueEnabled ?? false}
             toggleConstantValueEnabled={toggleConstantValueEnabled}
             disableConstantValue={props.disableConstantValueToggle}
             label={pt(props.label)}
@@ -366,7 +393,7 @@ export const ConstantValueInput = <T extends Record<string, any>>({
   ) : (
     <AutoField
       key={value?.selectedType} // reset when type changes
-      onChange={(newConstantValue, uiState) =>
+      onChange={(newConstantValue: T, uiState?: Partial<UiState>) =>
         onChange(
           {
             ...value,
@@ -375,8 +402,8 @@ export const ConstantValueInput = <T extends Record<string, any>>({
           uiState
         )
       }
-      value={value?.constantValue}
-      field={constantFieldConfig}
+      value={value?.constantValue as T}
+      field={constantFieldConfig as Field<T>}
     />
   );
 
@@ -499,7 +526,7 @@ export const EntityFieldInput = <T extends Record<string, any>>({
       {typeSelectorConfig && (
         <AutoField
           field={typeSelector!}
-          onChange={(selectedType, uiState) => {
+          onChange={(selectedType: string, uiState?: Partial<UiState>) => {
             onChange(
               {
                 ...value,
@@ -515,7 +542,7 @@ export const EntityFieldInput = <T extends Record<string, any>>({
       <AutoField
         key={value?.selectedType}
         field={entityFieldSelector}
-        onChange={(selectedEntityField, uiState) => {
+        onChange={(selectedEntityField: string, uiState?: Partial<UiState>) => {
           onChange(
             {
               ...value,