@rebasepro/plugin-data-enhancement 0.5.0 → 0.6.1

package/dist/index.umd.js

@@ -1,750 +1,772 @@
 (function(global, factory) {
-  typeof exports === "object" && typeof module !== "undefined" ? factory(exports, require("react-compiler-runtime"), require("react/jsx-runtime"), require("react"), require("@rebasepro/core"), require("@rebasepro/admin"), require("@rebasepro/common"), require("@rebasepro/utils"), require("@rebasepro/ui")) : typeof define === "function" && define.amd ? define(["exports", "react-compiler-runtime", "react/jsx-runtime", "react", "@rebasepro/core", "@rebasepro/admin", "@rebasepro/common", "@rebasepro/utils", "@rebasepro/ui"], factory) : (global = typeof globalThis !== "undefined" ? globalThis : global || self, factory(global.Rebase = {}, global.reactCompilerRuntime, global.jsxRuntime, global.React, global.core, global.admin, global.common, global.utils, global.ui));
-})(this, (function(exports2, reactCompilerRuntime, jsxRuntime, React, core, admin, common, utils, ui) {
-  "use strict";
-  function flatMapEntityValues(values, path = "") {
-    if (!values) return {};
-    return Object.entries(values).flatMap(([key, value]) => {
-      const currentPath = path ? `${path}.${key}` : key;
-      if (typeof value === "object") {
-        return flatMapEntityValues(value, currentPath);
-      } else {
-        return {
-          [currentPath]: value
-        };
-      }
-    }).reduce((acc, curr) => ({
-      ...acc,
-      ...curr
-    }), {});
-  }
-  const DEFAULT_SERVER = "https://api.rebase.pro";
-  async function enhanceDataAPIStream(props) {
-    const flatValues = flatMapEntityValues(props.values);
-    const properties = props.properties;
-    const inputEntity = {
-      entityId: props.entityId,
-      values: flatValues
-    };
-    const request = {
-      inputEntity,
-      properties,
-      entityName: props.entityName,
-      entityDescription: props.entityDescription,
-      propertyKey: props.propertyKey,
-      propertyInstructions: props.propertyInstructions,
-      instructions: props.instructions
-    };
-    console.debug("enhanceDataAPIStream", request);
-    return fetch((props.host ?? DEFAULT_SERVER) + "/data/enhance_stream/", {
-      // mode: "no-cors",
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        Authorization: `Basic ${props.firebaseToken}`,
-        "x-de-api-key": `Basic ${props.apiKey}`
-        // "x-de-version": version
-      },
-      body: JSON.stringify(request)
-    }).then(async (res) => {
-      if (!res.ok) {
-        console.error("enhanceDataAPIStream error", res);
-        throw await res.json();
-      }
-      const reader = res.body?.getReader();
-      if (!reader) {
-        throw new Error("No reader");
-      }
-      for await (const chunk of readChunks(reader)) {
-        const str = new TextDecoder().decode(chunk);
-        try {
-          str.split("&$# ").forEach((s) => {
-            if (s && s.length > 0) {
-              const data = JSON.parse(s.trim());
-              if (data.type === "suggestion_delta") props.onUpdateDelta(data.data.propertyKey, data.data.partialValue);
-              else if (data.type === "suggestion") props.onUpdate(data.data);
-              else if (data.type === "result") props.onEnd(data.data);
-            }
-          });
-        } catch (e) {
-          console.error("str", str);
-          console.error("Error parsing stream", e);
-          props.onError(e instanceof Error ? e : new Error(String(e)));
-        }
-      }
-    });
-  }
-  function readChunks(reader) {
-    return {
-      async *[Symbol.asyncIterator]() {
-        let readResult = await reader.read();
-        while (!readResult.done) {
-          yield readResult.value;
-          readResult = await reader.read();
-        }
-      }
-    };
-  }
-  async function fetchEntityPromptSuggestion(props) {
-    return fetch((props.host ?? DEFAULT_SERVER) + "/data/prompt_autocomplete/", {
-      // mode: "no-cors",
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        Authorization: `Basic ${props.firebaseToken}`,
-        "x-de-api-key": `Basic ${props.apiKey}`
-      },
-      body: JSON.stringify({
-        entityName: props.entityName,
-        input: props.input ?? null
-      })
-    }).then(async (res) => {
-      const data = await res.json();
-      if (!res.ok) {
-        console.error("fetchEntityPromptSuggestion", data);
-        throw Error(data.message);
-      }
-      return {
-        prompts: data.data.prompts.map((e) => ({
-          prompt: e,
-          type: "sample"
-        }))
-      };
-    });
-  }
-  async function autocompleteStream(props) {
-    let result = "";
-    return fetch((props.host ?? DEFAULT_SERVER) + "/data/autocomplete/", {
-      // mode: "no-cors",
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        Authorization: `Basic ${props.firebaseToken}`
-        // "x-de-version": version
-      },
-      body: JSON.stringify({
-        textBefore: props.textBefore,
-        textAfter: props.textAfter
-      })
-    }).then(async (res) => {
-      if (!res.ok) {
-        console.error("enhanceDataAPIStream error", res);
-        throw await res.json();
-      }
-      const reader = res.body?.getReader();
-      if (!reader) {
-        throw new Error("No reader");
-      }
-      for await (const chunk of readChunks(reader)) {
-        const str = new TextDecoder().decode(chunk);
-        result += str;
-        console.debug("Autocomplete update:", str);
-        props.onUpdate(str);
-      }
-    }).then(() => {
-      console.debug("Autocomplete result:", result);
-      return result;
-    });
-  }
-  function getAppendableSuggestion(suggestion, value) {
-    const suggestionIncludesValue = typeof suggestion === "string" && typeof value === "string" && suggestion.toLowerCase().trim().startsWith(value.toLowerCase().trim());
-    return typeof value === "string" && suggestionIncludesValue ? suggestion.substring(suggestion.toLowerCase().trim().indexOf(value.toLowerCase().trim()) + value.trim().length) : void 0;
-  }
-  function getSimplifiedProperties(properties, values, path = "") {
-    if (!properties) return {};
-    return Object.entries(properties).map(([key, property]) => {
-      if (common.isPropertyBuilder(property)) return {};
-      const fullKey = path ? `${path}.${key}` : key;
-      const valueInPath = utils.getValueInPath(values, fullKey);
-      return getSimplifiedProperty(property, fullKey, valueInPath);
-    }).reduce((a, b) => ({
-      ...a,
-      ...b
-    }), {});
-  }
-  function getSimpleProperty(property) {
-    const fieldId = admin.getFieldId(property);
-    if (!fieldId) {
-      console.error("No fieldId found for property", property);
-      throw new Error("Field id not found");
-    }
-    return {
-      name: property.name,
-      description: property.description,
-      type: property.type,
-      fieldConfigId: fieldId,
-      enum: "enum" in property && property.enum ? getSimpleEnumValues(property.enum) : void 0,
-      disabled: Boolean(property.ui?.disabled || property.ui?.readOnly)
-    };
-  }
-  function getSimplifiedProperty(property, path, value) {
-    if (common.isPropertyBuilder(property)) return {};
-    if (property.type === "array") {
-      if (property.of && !Array.isArray(property.of) && !common.isPropertyBuilder(property.of)) {
-        const arrayParentProperty = {
-          name: property.name,
-          description: property.description,
-          type: property.type,
-          fieldConfigId: "repeat",
-          disabled: Boolean(property.ui?.disabled || property.ui?.readOnly),
-          of: getSimpleProperty(property.of)
-        };
-        const result = {
-          [path]: arrayParentProperty
-        };
-        return result;
-      } else if (property.oneOf) {
-        const arrayParentProperty = {
-          name: property.name,
-          description: property.description,
-          type: property.type,
-          fieldConfigId: "block",
-          disabled: Boolean(property.ui?.disabled || property.ui?.readOnly),
-          oneOf: {
-            typeField: property.oneOf.typeField,
-            valueField: property.oneOf.valueField,
-            properties: Object.entries(property.oneOf.properties).map(([key, prop]) => ({
-              [key]: getSimpleProperty(prop)
-            })).reduce((a, b) => ({
-              ...a,
-              ...b
-            }), {})
-          }
-        };
-        if (!Array.isArray(value)) {
-          return {
-            [path]: arrayParentProperty
-          };
-        }
-        return value.map((v, i) => {
-          if (v == null) return {};
-          const typeKey = property.oneOf.typeField ?? "type";
-          const oneOfType = v[typeKey];
-          const valueKey = property.oneOf.valueField ?? "value";
-          const oneOfValue = v[valueKey];
-          const childProperty = property.oneOf.properties[oneOfType];
-          if (childProperty === void 0) {
-            console.error(`No property found for type ${oneOfType}`, property.oneOf.properties);
-            return {};
-          }
-          const simplifiedProperty = getSimplifiedProperty(childProperty, `${path}.${i}.${valueKey}`, oneOfValue);
-          return {
-            [`${path}.${i}.${typeKey}`]: oneOfType,
-            ...simplifiedProperty
-          };
-        }).reduce((a, b) => ({
-          ...a,
-          ...b
-        }), {
-          [path]: arrayParentProperty
-        });
-      }
-    } else if (property.type === "map") {
-      if (property.properties) {
-        const mapProperties = Object.entries(property.properties).map(([key, childProperty]) => {
-          const childValue = value && typeof value === "object" ? value[key] : void 0;
-          return getSimplifiedProperty(childProperty, key, childValue);
-        }).map((o) => attachPathToKeys(o, path)).reduce((a, b) => ({
-          ...a,
-          ...b
-        }), {});
-        if (Object.keys(mapProperties).length === 0) return {};
-        const mapParentProperty = {
-          name: property.name,
-          description: property.description,
-          type: property.type,
-          fieldConfigId: "group",
-          disabled: Boolean(property.ui?.disabled || property.ui?.readOnly)
-        };
-        return {
-          [path]: mapParentProperty,
-          ...mapProperties
-        };
-      }
-    } else {
-      const fieldId = admin.getFieldId(property);
-      if (!fieldId) {
-        console.warn(`No fieldId found for property ${path} with type ${property.type}`);
-        return {};
-      }
-      return {
-        [path]: getSimpleProperty(property)
-      };
-    }
-    return {};
-  }
-  function attachPathToKeys(obj, path = "") {
-    return Object.entries(obj).map(([key, value]) => {
-      const fullKey = path ? `${path}.${key}` : key;
-      return {
-        [fullKey]: value
-      };
-    }).reduce((a, b) => ({
-      ...a,
-      ...b
-    }), {});
-  }
-  function getSimpleEnumValues(enumValues) {
-    if (Array.isArray(enumValues)) return enumValues.map((v) => String(v.id));
-    if (typeof enumValues === "object") return Object.keys(enumValues);
-    throw Error("getSimpleEnumValues: Invalid enumValues");
-  }
-  function useEditorAIController({
-    getAuthToken
-  }) {
-    const autocomplete = async (textBefore, textAfter, onUpdate) => {
-      if (!getAuthToken) {
-        throw new Error("Firebase token is required");
-      }
-      const firebaseToken = await getAuthToken();
-      return autocompleteStream({
-        firebaseToken,
-        textBefore,
-        textAfter,
-        onUpdate
-      });
-    };
-    return {
-      autocomplete
-    };
-  }
-  const DataEnhancementControllerContext = React.createContext(null);
-  const useDataEnhancementController = () => {
-    return React.useContext(DataEnhancementControllerContext);
-  };
-  function getPropertyFromKey(properties, propertyKey) {
-    if (propertyKey in properties) {
-      return properties[propertyKey];
-    } else {
-      const split = propertyKey.split(".");
-      if (split.length === 1) {
-        return void 0;
-      }
-      const parentKey = split.slice(0, split.length - 1).join(".");
-      return getPropertyFromKey(properties, parentKey);
-    }
-  }
-  function DataEnhancementControllerProvider({
-    apiKey,
-    getConfigForPath,
-    children,
-    host,
-    path,
-    collection,
-    formContext
-  }) {
-    const [enabled, setEnabled] = React.useState(false);
-    const [suggestions, setSuggestions] = React.useState({});
-    const [loadingSuggestions, setLoadingSuggestions] = React.useState([]);
-    const enhancingInProgress = React.useRef(false);
-    const authController = core.useAuthController();
-    const snackbarController = core.useSnackbarController();
-    const properties = React.useMemo(() => getSimplifiedProperties(collection.properties, formContext?.values ?? {}), [formContext?.values]);
-    const valuesRef = React.useRef(formContext?.values ?? {});
-    React.useEffect(() => {
-      if (!enhancingInProgress.current) valuesRef.current = formContext?.values ?? {};
-    }, [formContext?.values]);
-    const allowReferenceDataSelection = false;
-    const updateConfig = React.useCallback(async () => {
-      if (!getConfigForPath) return;
-      const config = getConfigForPath({
-        path,
-        collection
-      });
-      if (config) {
-        setEnabled(true);
-      }
-    }, [collection, getConfigForPath, path]);
-    React.useEffect(() => {
-      if (!getConfigForPath) {
-        setEnabled(true);
-      } else {
-        updateConfig();
-      }
-    }, [getConfigForPath, updateConfig]);
-    const urlController = admin.useUrlController();
-    const clearSuggestion = React.useCallback((propertyKey) => {
-      setSuggestions((prev) => {
-        const {
-          [propertyKey]: _,
-          ...rest
-        } = prev;
-        return rest;
-      });
-    }, []);
-    const appendValueDelta = React.useCallback((propertyKey_0, delta) => {
-      const property = getPropertyFromKey(properties, propertyKey_0);
-      if (delta === null || property?.disabled) {
-        return;
-      }
-      const value = utils.getValueInPath(valuesRef.current, propertyKey_0);
-      const currentValue = value ? value + "" : "";
-      const updatedValue = currentValue + delta;
-      valuesRef.current = {
-        ...valuesRef.current,
-        [propertyKey_0]: updatedValue
-      };
-      formContext?.setFieldValue(propertyKey_0, updatedValue, false);
-      setSuggestions((prev_0) => ({
-        ...prev_0,
-        [propertyKey_0]: (prev_0[propertyKey_0] ?? "") + delta
-      }));
-    }, [properties, formContext]);
-    const updateSuggestedValues = React.useCallback((currentValues, updatedValues, replaceValues) => {
-      setLoadingSuggestions((prev_1) => {
-        return prev_1.filter((p) => !Object.keys(updatedValues).includes(p));
-      });
-      Object.entries(updatedValues).forEach(([propertyKey_1, suggestion]) => {
-        const value_0 = utils.getValueInPath(currentValues, propertyKey_1);
-        const property_0 = getPropertyFromKey(properties, propertyKey_1);
-        if (!property_0 || suggestion === null || property_0?.disabled) {
-          return;
-        }
-        if (typeof suggestion === "number") {
-          formContext?.setFieldValue(propertyKey_1, suggestion);
-          return;
-        }
-        if (replaceValues) {
-          formContext?.setFieldValue(propertyKey_1, suggestion);
-          return;
-        }
-        const appendableValue = getAppendableSuggestion(suggestion, value_0);
-        const currentValue_0 = value_0 ? value_0 + "" : "";
-        if (appendableValue) {
-          formContext?.setFieldValue(propertyKey_1, suggestion);
-        } else {
-          const multiline = property_0?.fieldConfigId === "multiline" || property_0?.fieldConfigId === "markdown";
-          const trimmedValue = currentValue_0.trimEnd();
-          if (multiline && (trimmedValue.endsWith(".") || trimmedValue.endsWith("?") || trimmedValue.endsWith("!") || trimmedValue.endsWith(":"))) {
-            formContext?.setFieldValue(propertyKey_1, trimmedValue + "\n\n" + suggestion.trimStart());
-          } else {
-            formContext?.setFieldValue(propertyKey_1, trimmedValue + (trimmedValue.length > 0 ? " " : "") + suggestion);
-          }
-        }
-      });
-      setSuggestions((prev_2) => ({
-        ...prev_2,
-        ...Object.keys(updatedValues).reduce((acc, key) => {
-          const value_1 = utils.getValueInPath(formContext?.values, key);
-          const suggestion_0 = updatedValues[key];
-          return {
-            ...acc,
-            [key]: getAppendableSuggestion(suggestion_0, value_1) ?? suggestion_0
-          };
-        }, {})
-      }));
-    }, [properties, formContext]);
-    const displayNeededSubscriptionSnackbar = React.useCallback((projectId) => {
-      snackbarController.open({
-        type: "warning",
-        message: "A valid subscription is needed in order to use this function.",
-        autoHideDuration: 4e3
-      });
-    }, [snackbarController]);
-    const editorAIController = useEditorAIController({
-      getAuthToken: authController.getAuthToken
-    });
-    const clearAllSuggestions = React.useCallback(() => {
-      setSuggestions({});
-    }, []);
-    const enhance = React.useCallback(async (props) => {
-      if (!authController.user) {
-        snackbarController.open({
-          type: "warning",
-          message: "You need to be logged in to enhance data"
-        });
-        return Promise.reject(new Error("Not logged in"));
-      }
-      const resolvedPath = urlController.resolveDatabasePathsFrom(path);
-      const firebaseToken = await authController.getAuthToken();
-      if (props.propertyKey) {
-        clearSuggestion(props.propertyKey);
-      } else {
-        clearAllSuggestions();
-      }
-      setLoadingSuggestions((prev_3) => [...prev_3, ...props.propertyKey ? [props.propertyKey] : Object.keys(properties)]);
-      enhancingInProgress.current = true;
-      const currentValues_0 = valuesRef.current ?? {};
-      return new Promise((resolve, reject) => {
-        function onError(e) {
-          setLoadingSuggestions([]);
-          const err = e instanceof Error ? e : typeof e === "object" && e !== null ? e : new Error(String(e));
-          const errorObj = err;
-          if (errorObj.code === "payment-required") {
-            const data = errorObj.data;
-            const projectId_0 = data?.projectId;
-            displayNeededSubscriptionSnackbar(projectId_0);
-          } else {
-            console.error("Enhance error", e);
-          }
-          reject(e);
-          enhancingInProgress.current = false;
-        }
-        try {
-          enhanceDataAPIStream({
-            ...props,
-            host,
-            apiKey,
-            properties,
-            path: resolvedPath,
-            entityName: collection.singularName ?? collection.name,
-            entityDescription: collection.description,
-            firebaseToken,
-            onUpdate: (suggestions_0) => {
-              console.debug("de onUpdate", suggestions_0);
-              updateSuggestedValues(currentValues_0, suggestions_0, props.replaceValues ?? false);
-            },
-            onUpdateDelta: (propertyKey_2, partialValue) => {
-              appendValueDelta(propertyKey_2, partialValue);
-            },
-            onError,
-            onEnd: (result) => {
-              console.debug("de onEnd", result);
-              if (result.errors) {
-                result.errors.forEach((error) => {
-                  snackbarController.open({
-                    type: "warning",
-                    message: error
-                  });
-                });
-              }
-              if (Object.keys(result.suggestions).length === 0) {
-                snackbarController.open({
-                  type: "info",
-                  autoHideDuration: 1800,
-                  message: "No fields were updated"
-                });
-              }
-              setLoadingSuggestions([]);
-              resolve(result);
-              enhancingInProgress.current = false;
-            }
-          }).catch(onError);
-        } catch (e_0) {
-          onError(e_0);
-        }
-      });
-    }, [authController, urlController, path, clearSuggestion, clearAllSuggestions, properties, host, apiKey, collection, updateSuggestedValues, appendValueDelta, displayNeededSubscriptionSnackbar, snackbarController]);
-    const getSamplePrompts = React.useCallback(async (entityName, input) => {
-      const firebaseToken_0 = await authController.getAuthToken();
-      return fetchEntityPromptSuggestion({
-        host,
-        entityName,
-        firebaseToken: firebaseToken_0,
-        apiKey,
-        input
-      });
-    }, [apiKey, authController.getAuthToken, host]);
-    const dataEnhancementController = React.useMemo(() => ({
-      enabled,
-      suggestions,
-      clearSuggestion,
-      enhance,
-      allowReferenceDataSelection,
-      clearAllSuggestions,
-      getSamplePrompts,
-      loadingSuggestions,
-      editorAIController
-    }), [enabled, suggestions, clearSuggestion, enhance, allowReferenceDataSelection, clearAllSuggestions, getSamplePrompts, loadingSuggestions, editorAIController]);
-    return /* @__PURE__ */ jsxRuntime.jsx(DataEnhancementControllerContext.Provider, { value: dataEnhancementController, children });
-  }
-  function FormEnhanceAction({
-    entityId,
-    path,
-    status,
-    collection,
-    formContext,
-    openEntityMode
-  }) {
-    const largeLayout = core.useLargeLayout();
-    const storageKey = createLocalStorageKey(path, status);
-    const [loading, setLoading] = React.useState(false);
-    const dataEnhancementController = useDataEnhancementController();
-    const [samplePrompts, setSamplePrompts] = React.useState(void 0);
-    const [instructions, setInstructions] = React.useState("");
-    const getSamplePrompts = dataEnhancementController?.getSamplePrompts;
-    const loadingPrompts = React.useRef(false);
-    const updateSuggestedPrompts = React.useCallback(async function updateSuggestedPrompts2(instructions_0) {
-      if (!getSamplePrompts) return;
-      if (loadingPrompts.current) return;
-      loadingPrompts.current = true;
-      const prompts = status === "new" ? (await getSamplePrompts(collection.singularName ?? collection.name, instructions_0)).prompts : getPromptsForExistingEntities(collection.properties);
-      const recentPromptsFromStorage = getRecentPromptsFromStorage(storageKey);
-      const recentPrompts = recentPromptsFromStorage.map((prompt) => prompt.prompt);
-      setSamplePrompts([...recentPromptsFromStorage, ...prompts.filter((p) => !recentPrompts.includes(p.prompt))].slice(0, 5));
-      loadingPrompts.current = false;
-    }, [collection.name, collection.singularName, getSamplePrompts, status]);
-    React.useDeferredValue(formContext?.values);
-    React.useEffect(() => {
-      if (!dataEnhancementController) return;
-      if (!samplePrompts) {
-        setSamplePrompts(getRecentPromptsFromStorage(storageKey));
-        updateSuggestedPrompts().then();
-      }
-    }, [dataEnhancementController, samplePrompts, storageKey, updateSuggestedPrompts, instructions, status]);
-    React.useEffect(() => {
-      if (!dataEnhancementController) return;
-      updateSuggestedPrompts().then();
-    }, [dataEnhancementController, status]);
-    const enhance = (prompt_0) => {
-      if (!dataEnhancementController || !formContext?.values) return;
-      setLoading(true);
-      if (prompt_0) {
-        addRecentPrompt(storageKey, prompt_0);
-        setSamplePrompts([{
-          prompt: prompt_0,
-          type: "recent"
-        }, ...(samplePrompts ?? []).slice(0, 5)]);
-      }
-      return dataEnhancementController.enhance({
-        entityId,
-        values: formContext.values,
-        instructions: prompt_0,
-        replaceValues: true
-      }).finally(() => {
-        setLoading(false);
-      });
-    };
-    if (!dataEnhancementController?.enabled) return null;
-    const suggestions = dataEnhancementController.suggestions;
-    Object.values(suggestions).filter(Boolean).length > 0;
-    (samplePrompts ?? []).length > 0 && instructions.length === 0;
-    function submit() {
-      enhance(instructions);
-    }
-    return /* @__PURE__ */ jsxRuntime.jsxs(ui.Menu, { align: "end", sideOffset: 8, className: "max-w-[100vw]", trigger: /* @__PURE__ */ jsxRuntime.jsxs(ui.Button, { variant: "filled", color: "neutral", fullWidth: largeLayout && openEntityMode === "full_screen", size: "small", disabled: loading, children: [
-      !loading && /* @__PURE__ */ jsxRuntime.jsx(core.AIIcon, { size: "small" }),
-      loading && /* @__PURE__ */ jsxRuntime.jsx(ui.CircularProgress, { size: "small" }),
-      "Autofill"
-    ] }), children: [
-      /* @__PURE__ */ jsxRuntime.jsxs(ui.MenuItem, { className: "py-4", onClick: () => {
-        enhance();
-      }, children: [
-        /* @__PURE__ */ jsxRuntime.jsx(core.AIIcon, { size: "small" }),
-        "Autofill based on the current content"
-      ] }),
-      /* @__PURE__ */ jsxRuntime.jsx(ui.Separator, { orientation: "horizontal", className: "mt-2" }),
-      samplePrompts?.map((samplePrompt, index) => {
-        return /* @__PURE__ */ jsxRuntime.jsxs(ui.MenuItem, { onClick: () => {
-          setInstructions(samplePrompt.prompt);
-          enhance(samplePrompt.prompt);
-        }, children: [
-          /* @__PURE__ */ jsxRuntime.jsx("div", { className: "pl-9 grow text-text-secondary dark:text-text-secondary-dark", children: samplePrompt.prompt }),
-          samplePrompt.type === "recent" && /* @__PURE__ */ jsxRuntime.jsx(ui.IconButton, { onClick: (e) => {
-            e.preventDefault();
-            e.stopPropagation();
-            removeRecentPrompt(storageKey, samplePrompt.prompt);
-            setSamplePrompts((samplePrompts ?? []).filter((p_0) => p_0.prompt !== samplePrompt.prompt));
-          }, size: "smallest", children: /* @__PURE__ */ jsxRuntime.jsx(ui.XIcon, { size: ui.iconSize.smallest }) })
-        ] }, index + "_" + samplePrompt.prompt);
-      }),
-      /* @__PURE__ */ jsxRuntime.jsx(ui.Separator, { orientation: "horizontal" }),
-      /* @__PURE__ */ jsxRuntime.jsxs("div", { className: ui.cls("my-2 w-[500px] max-w-full flex items-start text-surface-700 dark:text-surface-200"), children: [
-        /* @__PURE__ */ jsxRuntime.jsx(ui.TextareaAutosize, { className: ui.cls("p-4 rounded-lg resize-none bg-surface-100 dark:bg-surface-950 mx-2 w-full grow outline-hidden max-h-[300px] overflow-auto", ui.focusedDisabled), value: instructions, autoFocus: status === "new", disabled: loading, onFocus: (event) => {
-          event.stopPropagation();
-        }, placeholder: "...or provide instructions", onKeyDown: (e_0) => {
-          e_0.stopPropagation();
-          if (e_0.key === "Enter" && !e_0.shiftKey) {
-            e_0.preventDefault();
-            submit();
-          }
-        }, onChange: (e_1) => {
-          setInstructions(e_1.target.value);
-        } }),
-        /* @__PURE__ */ jsxRuntime.jsx(ui.IconButton, { size: "small", onClick: () => {
-          setInstructions("");
-        }, color: !instructions ? "primary" : void 0, disabled: loading || !instructions, children: /* @__PURE__ */ jsxRuntime.jsx(ui.XIcon, { size: ui.iconSize.small }) }),
-        /* @__PURE__ */ jsxRuntime.jsxs(ui.IconButton, { onClick: () => enhance(instructions), size: "small", color: !instructions ? "primary" : void 0, disabled: loading || !instructions, children: [
-          loading && /* @__PURE__ */ jsxRuntime.jsx(ui.CircularProgress, { size: "smallest" }),
-          !loading && /* @__PURE__ */ jsxRuntime.jsx(ui.SendIcon, { color: "primary" })
-        ] })
-      ] })
-    ] });
-  }
-  function getPromptsForExistingEntities(properties) {
-    const multilineProperties = Object.values(properties).filter((p) => {
-      if (common.isPropertyBuilder(p)) {
-        return false;
-      }
-      return p.type === "string" && (p.ui?.markdown || p.ui?.multiline);
-    });
-    const multilinePrompt = multilineProperties.length > 0 ? multilineProperties[Math.floor(Math.random() * multilineProperties.length)] : void 0;
-    const prompts = ["Fill the missing fields", "Translate the missing content"];
-    if (multilinePrompt) {
-      prompts.push(`Add 2 paragraphs to '${multilinePrompt.name}'`);
-    }
-    return prompts.map((p) => ({
-      prompt: p,
-      type: "sample"
-    }));
-  }
-  const createLocalStorageKey = (path, status) => {
-    const statusString = status === "new" ? "new" : "existing";
-    return `data_enhancement::${statusString}::${common.stripCollectionPath(path)}`;
-  };
-  const getRecentPromptsFromStorage = (storageKey) => {
-    const item = localStorage.getItem(storageKey);
-    return item ? JSON.parse(item).map((e) => ({
-      prompt: e,
-      type: "recent"
-    })) : [];
-  };
-  const addRecentPrompt = (storageKey, prompt) => {
-    if (!prompt || prompt.trim().length === 0) {
-      return;
-    }
-    const recentPrompts = getRecentPromptsFromStorage(storageKey);
-    localStorage.setItem(storageKey, JSON.stringify([prompt, ...recentPrompts.map((e) => e.prompt).filter((e) => e !== prompt).slice(0, 5)]));
-  };
-  const removeRecentPrompt = (storageKey, prompt) => {
-    localStorage.setItem(storageKey, JSON.stringify(getRecentPromptsFromStorage(storageKey).map((e) => e.prompt).filter((e) => e !== prompt)));
-  };
-  const DEFAULT_API_KEY = "fcms-U9jdDii0xXWSDC34asfrf54lbkFJBfKfRWcEDEwdc4V5wDWEDF";
-  function useDataEnhancementPlugin(props) {
-    const $ = reactCompilerRuntime.c(5);
-    const apiKey = props?.apiKey ?? DEFAULT_API_KEY;
-    const getConfigForPath = props?.getConfigForPath;
-    props?.host;
-    let t0;
-    let t1;
-    if ($[0] === /* @__PURE__ */ Symbol.for("react.memo_cache_sentinel")) {
-      t1 = [{
-        slot: "form.actions",
-        Component: FormEnhanceAction,
-        order: 40
-      }];
-      $[0] = t1;
-    } else {
-      t1 = $[0];
-    }
-    const t2 = props?.host;
-    let t3;
-    if ($[1] !== apiKey || $[2] !== getConfigForPath || $[3] !== t2) {
-      t3 = {
-        key: "data_enhancement",
-        slots: t1,
-        providers: [{
-          scope: "form",
-          Component: DataEnhancementControllerProvider,
-          props: {
-            apiKey,
-            getConfigForPath,
-            host: t2
-          }
-        }]
-      };
-      $[1] = apiKey;
-      $[2] = getConfigForPath;
-      $[3] = t2;
-      $[4] = t3;
-    } else {
-      t3 = $[4];
-    }
-    t0 = t3;
-    return t0;
-  }
-  exports2.useDataEnhancementPlugin = useDataEnhancementPlugin;
-  exports2.useEditorAIController = useEditorAIController;
-  Object.defineProperty(exports2, Symbol.toStringTag, { value: "Module" });
-}));
-//# sourceMappingURL=index.umd.js.map
+	typeof exports === "object" && typeof module !== "undefined" ? factory(exports, require("react"), require("@rebasepro/core"), require("@rebasepro/admin"), require("@rebasepro/common"), require("@rebasepro/utils"), require("react/jsx-runtime"), require("@rebasepro/ui")) : typeof define === "function" && define.amd ? define([
+		"exports",
+		"react",
+		"@rebasepro/core",
+		"@rebasepro/admin",
+		"@rebasepro/common",
+		"@rebasepro/utils",
+		"react/jsx-runtime",
+		"@rebasepro/ui"
+	], factory) : (global = typeof globalThis !== "undefined" ? globalThis : global || self, factory(global.Rebase = {}, global.react, global._rebasepro_core, global._rebasepro_admin, global._rebasepro_common, global._rebasepro_utils, global.react_jsx_runtime, global._rebasepro_ui));
+})(this, function(exports, react, _rebasepro_core, _rebasepro_admin, _rebasepro_common, _rebasepro_utils, react_jsx_runtime, _rebasepro_ui) {
+	Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+	//#region \0rolldown/runtime.js
+	var __create = Object.create;
+	var __defProp = Object.defineProperty;
+	var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+	var __getOwnPropNames = Object.getOwnPropertyNames;
+	var __getProtoOf = Object.getPrototypeOf;
+	var __hasOwnProp = Object.prototype.hasOwnProperty;
+	var __copyProps = (to, from, except, desc) => {
+		if (from && typeof from === "object" || typeof from === "function") for (var keys = __getOwnPropNames(from), i = 0, n = keys.length, key; i < n; i++) {
+			key = keys[i];
+			if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, {
+				get: ((k) => from[k]).bind(null, key),
+				enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
+			});
+		}
+		return to;
+	};
+	var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", {
+		value: mod,
+		enumerable: true
+	}) : target, mod));
+	//#endregion
+	react = __toESM(react, 1);
+	//#region src/utils/values.ts
+	function flatMapEntityValues(values, path = "") {
+		if (!values) return {};
+		return Object.entries(values).flatMap(([key, value]) => {
+			const currentPath = path ? `${path}.${key}` : key;
+			if (typeof value === "object") return flatMapEntityValues(value, currentPath);
+			else return { [currentPath]: value };
+		}).reduce((acc, curr) => ({
+			...acc,
+			...curr
+		}), {});
+	}
+	//#endregion
+	//#region src/api.ts
+	var DEFAULT_SERVER = "https://api.rebase.pro";
+	async function enhanceDataAPIStream(props) {
+		const flatValues = flatMapEntityValues(props.values);
+		const properties = props.properties;
+		const request = {
+			inputEntity: {
+				entityId: props.entityId,
+				values: flatValues
+			},
+			properties,
+			entityName: props.entityName,
+			entityDescription: props.entityDescription,
+			propertyKey: props.propertyKey,
+			propertyInstructions: props.propertyInstructions,
+			instructions: props.instructions
+		};
+		console.debug("enhanceDataAPIStream", request);
+		return fetch((props.host ?? DEFAULT_SERVER) + "/data/enhance_stream/", {
+			method: "POST",
+			headers: {
+				"Content-Type": "application/json",
+				Authorization: `Basic ${props.firebaseToken}`,
+				"x-de-api-key": `Basic ${props.apiKey}`
+			},
+			body: JSON.stringify(request)
+		}).then(async (res) => {
+			if (!res.ok) {
+				console.error("enhanceDataAPIStream error", res);
+				throw await res.json();
+			}
+			const reader = res.body?.getReader();
+			if (!reader) throw new Error("No reader");
+			for await (const chunk of readChunks(reader)) {
+				const str = new TextDecoder().decode(chunk);
+				try {
+					str.split("&$# ").forEach((s) => {
+						if (s && s.length > 0) {
+							const data = JSON.parse(s.trim());
+							if (data.type === "suggestion_delta") props.onUpdateDelta(data.data.propertyKey, data.data.partialValue);
+							else if (data.type === "suggestion") props.onUpdate(data.data);
+							else if (data.type === "result") props.onEnd(data.data);
+						}
+					});
+				} catch (e) {
+					console.error("str", str);
+					console.error("Error parsing stream", e);
+					props.onError(e instanceof Error ? e : new Error(String(e)));
+				}
+			}
+		});
+	}
+	function readChunks(reader) {
+		return { async *[Symbol.asyncIterator]() {
+			let readResult = await reader.read();
+			while (!readResult.done) {
+				yield readResult.value;
+				readResult = await reader.read();
+			}
+		} };
+	}
+	async function fetchEntityPromptSuggestion(props) {
+		return fetch((props.host ?? DEFAULT_SERVER) + "/data/prompt_autocomplete/", {
+			method: "POST",
+			headers: {
+				"Content-Type": "application/json",
+				Authorization: `Basic ${props.firebaseToken}`,
+				"x-de-api-key": `Basic ${props.apiKey}`
+			},
+			body: JSON.stringify({
+				entityName: props.entityName,
+				input: props.input ?? null
+			})
+		}).then(async (res) => {
+			const data = await res.json();
+			if (!res.ok) {
+				console.error("fetchEntityPromptSuggestion", data);
+				throw Error(data.message);
+			}
+			return { prompts: data.data.prompts.map((e) => ({
+				prompt: e,
+				type: "sample"
+			})) };
+		});
+	}
+	async function autocompleteStream(props) {
+		let result = "";
+		return fetch((props.host ?? DEFAULT_SERVER) + "/data/autocomplete/", {
+			method: "POST",
+			headers: {
+				"Content-Type": "application/json",
+				Authorization: `Basic ${props.firebaseToken}`
+			},
+			body: JSON.stringify({
+				textBefore: props.textBefore,
+				textAfter: props.textAfter
+			})
+		}).then(async (res) => {
+			if (!res.ok) {
+				console.error("enhanceDataAPIStream error", res);
+				throw await res.json();
+			}
+			const reader = res.body?.getReader();
+			if (!reader) throw new Error("No reader");
+			for await (const chunk of readChunks(reader)) {
+				const str = new TextDecoder().decode(chunk);
+				result += str;
+				console.debug("Autocomplete update:", str);
+				props.onUpdate(str);
+			}
+		}).then(() => {
+			console.debug("Autocomplete result:", result);
+			return result;
+		});
+	}
+	//#endregion
+	//#region src/utils/suggestions.ts
+	function getAppendableSuggestion(suggestion, value) {
+		const suggestionIncludesValue = typeof suggestion === "string" && typeof value === "string" && suggestion.toLowerCase().trim().startsWith(value.toLowerCase().trim());
+		return typeof value === "string" && suggestionIncludesValue ? suggestion.substring(suggestion.toLowerCase().trim().indexOf(value.toLowerCase().trim()) + value.trim().length) : void 0;
+	}
+	//#endregion
+	//#region src/utils/properties.ts
+	function getSimplifiedProperties(properties, values, path = "") {
+		if (!properties) return {};
+		return Object.entries(properties).map(([key, property]) => {
+			if ((0, _rebasepro_common.isPropertyBuilder)(property)) return {};
+			const fullKey = path ? `${path}.${key}` : key;
+			return getSimplifiedProperty(property, fullKey, (0, _rebasepro_utils.getValueInPath)(values, fullKey));
+		}).reduce((a, b) => ({
+			...a,
+			...b
+		}), {});
+	}
+	function getSimpleProperty(property) {
+		const fieldId = (0, _rebasepro_admin.getFieldId)(property);
+		if (!fieldId) {
+			console.error("No fieldId found for property", property);
+			throw new Error("Field id not found");
+		}
+		return {
+			name: property.name,
+			description: property.description,
+			type: property.type,
+			fieldConfigId: fieldId,
+			enum: "enum" in property && property.enum ? getSimpleEnumValues(property.enum) : void 0,
+			disabled: Boolean(property.ui?.disabled || property.ui?.readOnly)
+		};
+	}
+	function getSimplifiedProperty(property, path, value) {
+		if ((0, _rebasepro_common.isPropertyBuilder)(property)) return {};
+		if (property.type === "array") {
+			if (property.of && !Array.isArray(property.of) && !(0, _rebasepro_common.isPropertyBuilder)(property.of)) {
+				const arrayParentProperty = {
+					name: property.name,
+					description: property.description,
+					type: property.type,
+					fieldConfigId: "repeat",
+					disabled: Boolean(property.ui?.disabled || property.ui?.readOnly),
+					of: getSimpleProperty(property.of)
+				};
+				return { [path]: arrayParentProperty };
+			} else if (property.oneOf) {
+				const arrayParentProperty = {
+					name: property.name,
+					description: property.description,
+					type: property.type,
+					fieldConfigId: "block",
+					disabled: Boolean(property.ui?.disabled || property.ui?.readOnly),
+					oneOf: {
+						typeField: property.oneOf.typeField,
+						valueField: property.oneOf.valueField,
+						properties: Object.entries(property.oneOf.properties).map(([key, prop]) => ({ [key]: getSimpleProperty(prop) })).reduce((a, b) => ({
+							...a,
+							...b
+						}), {})
+					}
+				};
+				if (!Array.isArray(value)) return { [path]: arrayParentProperty };
+				return value.map((v, i) => {
+					if (v == null) return {};
+					const typeKey = property.oneOf.typeField ?? "type";
+					const oneOfType = v[typeKey];
+					const valueKey = property.oneOf.valueField ?? "value";
+					const oneOfValue = v[valueKey];
+					const childProperty = property.oneOf.properties[oneOfType];
+					if (childProperty === void 0) {
+						console.error(`No property found for type ${oneOfType}`, property.oneOf.properties);
+						return {};
+					}
+					const simplifiedProperty = getSimplifiedProperty(childProperty, `${path}.${i}.${valueKey}`, oneOfValue);
+					return {
+						[`${path}.${i}.${typeKey}`]: oneOfType,
+						...simplifiedProperty
+					};
+				}).reduce((a, b) => ({
+					...a,
+					...b
+				}), { [path]: arrayParentProperty });
+			}
+		} else if (property.type === "map") {
+			if (property.properties) {
+				const mapProperties = Object.entries(property.properties).map(([key, childProperty]) => {
+					return getSimplifiedProperty(childProperty, key, value && typeof value === "object" ? value[key] : void 0);
+				}).map((o) => attachPathToKeys(o, path)).reduce((a, b) => ({
+					...a,
+					...b
+				}), {});
+				if (Object.keys(mapProperties).length === 0) return {};
+				const mapParentProperty = {
+					name: property.name,
+					description: property.description,
+					type: property.type,
+					fieldConfigId: "group",
+					disabled: Boolean(property.ui?.disabled || property.ui?.readOnly)
+				};
+				return {
+					[path]: mapParentProperty,
+					...mapProperties
+				};
+			}
+		} else {
+			if (!(0, _rebasepro_admin.getFieldId)(property)) {
+				console.warn(`No fieldId found for property ${path} with type ${property.type}`);
+				return {};
+			}
+			return { [path]: getSimpleProperty(property) };
+		}
+		return {};
+	}
+	function attachPathToKeys(obj, path = "") {
+		return Object.entries(obj).map(([key, value]) => {
+			return { [path ? `${path}.${key}` : key]: value };
+		}).reduce((a, b) => ({
+			...a,
+			...b
+		}), {});
+	}
+	function getSimpleEnumValues(enumValues) {
+		if (Array.isArray(enumValues)) return enumValues.map((v) => String(v.id));
+		if (typeof enumValues === "object") return Object.keys(enumValues);
+		throw Error("getSimpleEnumValues: Invalid enumValues");
+	}
+	//#endregion
+	//#region src/editor/useEditorAIController.tsx
+	function useEditorAIController({ getAuthToken }) {
+		const autocomplete = async (textBefore, textAfter, onUpdate) => {
+			if (!getAuthToken) throw new Error("Firebase token is required");
+			return autocompleteStream({
+				firebaseToken: await getAuthToken(),
+				textBefore,
+				textAfter,
+				onUpdate
+			});
+		};
+		return { autocomplete };
+	}
+	//#endregion
+	//#region src/components/DataEnhancementControllerProvider.tsx
+	var DataEnhancementControllerContext = react.default.createContext(null);
+	var useDataEnhancementController = () => (0, react.useContext)(DataEnhancementControllerContext);
+	function getPropertyFromKey(properties, propertyKey) {
+		if (propertyKey in properties) return properties[propertyKey];
+		else {
+			const split = propertyKey.split(".");
+			if (split.length === 1) return;
+			return getPropertyFromKey(properties, split.slice(0, split.length - 1).join("."));
+		}
+	}
+	function DataEnhancementControllerProvider({ apiKey, getConfigForPath, children, host, path, collection, formContext }) {
+		const [enabled, setEnabled] = (0, react.useState)(false);
+		const [suggestions, setSuggestions] = (0, react.useState)({});
+		const [loadingSuggestions, setLoadingSuggestions] = (0, react.useState)([]);
+		const enhancingInProgress = (0, react.useRef)(false);
+		const authController = (0, _rebasepro_core.useAuthController)();
+		const snackbarController = (0, _rebasepro_core.useSnackbarController)();
+		const properties = (0, react.useMemo)(() => getSimplifiedProperties(collection.properties, formContext?.values ?? {}), [formContext?.values]);
+		const valuesRef = react.default.useRef(formContext?.values ?? {});
+		(0, react.useEffect)(() => {
+			if (!enhancingInProgress.current) valuesRef.current = formContext?.values ?? {};
+		}, [formContext?.values]);
+		const allowReferenceDataSelection = false;
+		const updateConfig = (0, react.useCallback)(async () => {
+			if (!getConfigForPath) return;
+			if (getConfigForPath({
+				path,
+				collection
+			})) setEnabled(true);
+		}, [
+			collection,
+			getConfigForPath,
+			path
+		]);
+		(0, react.useEffect)(() => {
+			if (!getConfigForPath) setEnabled(true);
+			else updateConfig();
+		}, [getConfigForPath, updateConfig]);
+		const urlController = (0, _rebasepro_admin.useUrlController)();
+		const clearSuggestion = (0, react.useCallback)((propertyKey) => {
+			setSuggestions((prev) => {
+				const { [propertyKey]: _, ...rest } = prev;
+				return rest;
+			});
+		}, []);
+		const appendValueDelta = (0, react.useCallback)((propertyKey, delta) => {
+			const property = getPropertyFromKey(properties, propertyKey);
+			if (delta === null || property?.disabled) return;
+			const value = (0, _rebasepro_utils.getValueInPath)(valuesRef.current, propertyKey);
+			const updatedValue = (value ? value + "" : "") + delta;
+			valuesRef.current = {
+				...valuesRef.current,
+				[propertyKey]: updatedValue
+			};
+			formContext?.setFieldValue(propertyKey, updatedValue, false);
+			setSuggestions((prev) => ({
+				...prev,
+				[propertyKey]: (prev[propertyKey] ?? "") + delta
+			}));
+		}, [properties, formContext]);
+		const updateSuggestedValues = (0, react.useCallback)((currentValues, updatedValues, replaceValues) => {
+			setLoadingSuggestions((prev) => {
+				return prev.filter((p) => !Object.keys(updatedValues).includes(p));
+			});
+			Object.entries(updatedValues).forEach(([propertyKey, suggestion]) => {
+				const value = (0, _rebasepro_utils.getValueInPath)(currentValues, propertyKey);
+				const property = getPropertyFromKey(properties, propertyKey);
+				if (!property || suggestion === null || property?.disabled) return;
+				if (typeof suggestion === "number") {
+					formContext?.setFieldValue(propertyKey, suggestion);
+					return;
+				}
+				if (replaceValues) {
+					formContext?.setFieldValue(propertyKey, suggestion);
+					return;
+				}
+				const appendableValue = getAppendableSuggestion(suggestion, value);
+				const currentValue = value ? value + "" : "";
+				if (appendableValue) formContext?.setFieldValue(propertyKey, suggestion);
+				else {
+					const multiline = property?.fieldConfigId === "multiline" || property?.fieldConfigId === "markdown";
+					const trimmedValue = currentValue.trimEnd();
+					if (multiline && (trimmedValue.endsWith(".") || trimmedValue.endsWith("?") || trimmedValue.endsWith("!") || trimmedValue.endsWith(":"))) formContext?.setFieldValue(propertyKey, trimmedValue + "\n\n" + suggestion.trimStart());
+					else formContext?.setFieldValue(propertyKey, trimmedValue + (trimmedValue.length > 0 ? " " : "") + suggestion);
+				}
+			});
+			setSuggestions((prev) => ({
+				...prev,
+				...Object.keys(updatedValues).reduce((acc, key) => {
+					const value = (0, _rebasepro_utils.getValueInPath)(formContext?.values, key);
+					const suggestion = updatedValues[key];
+					return {
+						...acc,
+						[key]: getAppendableSuggestion(suggestion, value) ?? suggestion
+					};
+				}, {})
+			}));
+		}, [properties, formContext]);
+		const displayNeededSubscriptionSnackbar = (0, react.useCallback)((projectId) => {
+			snackbarController.open({
+				type: "warning",
+				message: "A valid subscription is needed in order to use this function.",
+				autoHideDuration: 4e3
+			});
+		}, [snackbarController]);
+		const editorAIController = useEditorAIController({ getAuthToken: authController.getAuthToken });
+		const clearAllSuggestions = (0, react.useCallback)(() => {
+			setSuggestions({});
+		}, []);
+		const enhance = (0, react.useCallback)(async (props) => {
+			if (!authController.user) {
+				snackbarController.open({
+					type: "warning",
+					message: "You need to be logged in to enhance data"
+				});
+				return Promise.reject(/* @__PURE__ */ new Error("Not logged in"));
+			}
+			const resolvedPath = urlController.resolveDatabasePathsFrom(path);
+			const firebaseToken = await authController.getAuthToken();
+			if (props.propertyKey) clearSuggestion(props.propertyKey);
+			else clearAllSuggestions();
+			setLoadingSuggestions((prev) => [...prev, ...props.propertyKey ? [props.propertyKey] : Object.keys(properties)]);
+			enhancingInProgress.current = true;
+			const currentValues = valuesRef.current ?? {};
+			return new Promise((resolve, reject) => {
+				function onError(e) {
+					setLoadingSuggestions([]);
+					const errorObj = e instanceof Error ? e : typeof e === "object" && e !== null ? e : new Error(String(e));
+					if (errorObj.code === "payment-required") {
+						const projectId = errorObj.data?.projectId;
+						displayNeededSubscriptionSnackbar(projectId);
+					} else console.error("Enhance error", e);
+					reject(e);
+					enhancingInProgress.current = false;
+				}
+				try {
+					enhanceDataAPIStream({
+						...props,
+						host,
+						apiKey,
+						properties,
+						path: resolvedPath,
+						entityName: collection.singularName ?? collection.name,
+						entityDescription: collection.description,
+						firebaseToken,
+						onUpdate: (suggestions) => {
+							console.debug("de onUpdate", suggestions);
+							updateSuggestedValues(currentValues, suggestions, props.replaceValues ?? false);
+						},
+						onUpdateDelta: (propertyKey, partialValue) => {
+							appendValueDelta(propertyKey, partialValue);
+						},
+						onError,
+						onEnd: (result) => {
+							console.debug("de onEnd", result);
+							if (result.errors) result.errors.forEach((error) => {
+								snackbarController.open({
+									type: "warning",
+									message: error
+								});
+							});
+							if (Object.keys(result.suggestions).length === 0) snackbarController.open({
+								type: "info",
+								autoHideDuration: 1800,
+								message: "No fields were updated"
+							});
+							setLoadingSuggestions([]);
+							resolve(result);
+							enhancingInProgress.current = false;
+						}
+					}).catch(onError);
+				} catch (e) {
+					onError(e);
+				}
+			});
+		}, [
+			authController,
+			urlController,
+			path,
+			clearSuggestion,
+			clearAllSuggestions,
+			properties,
+			host,
+			apiKey,
+			collection,
+			updateSuggestedValues,
+			appendValueDelta,
+			displayNeededSubscriptionSnackbar,
+			snackbarController
+		]);
+		const getSamplePrompts = (0, react.useCallback)(async (entityName, input) => {
+			return fetchEntityPromptSuggestion({
+				host,
+				entityName,
+				firebaseToken: await authController.getAuthToken(),
+				apiKey,
+				input
+			});
+		}, [
+			apiKey,
+			authController.getAuthToken,
+			host
+		]);
+		const dataEnhancementController = (0, react.useMemo)(() => ({
+			enabled,
+			suggestions,
+			clearSuggestion,
+			enhance,
+			allowReferenceDataSelection,
+			clearAllSuggestions,
+			getSamplePrompts,
+			loadingSuggestions,
+			editorAIController
+		}), [
+			enabled,
+			suggestions,
+			clearSuggestion,
+			enhance,
+			allowReferenceDataSelection,
+			clearAllSuggestions,
+			getSamplePrompts,
+			loadingSuggestions,
+			editorAIController
+		]);
+		return /* @__PURE__ */ (0, react_jsx_runtime.jsx)(DataEnhancementControllerContext.Provider, {
+			value: dataEnhancementController,
+			children
+		});
+	}
+	//#endregion
+	//#region src/components/FormEnhanceAction.tsx
+	function FormEnhanceAction({ entityId, path, status, collection, formContext, openEntityMode }) {
+		const largeLayout = (0, _rebasepro_core.useLargeLayout)();
+		const storageKey = createLocalStorageKey(path, status);
+		const [loading, setLoading] = react.default.useState(false);
+		const dataEnhancementController = useDataEnhancementController();
+		const [samplePrompts, setSamplePrompts] = react.default.useState(void 0);
+		const [instructions, setInstructions] = react.default.useState("");
+		const getSamplePrompts = dataEnhancementController?.getSamplePrompts;
+		const loadingPrompts = (0, react.useRef)(false);
+		const updateSuggestedPrompts = (0, react.useCallback)(async function updateSuggestedPrompts(instructions) {
+			if (!getSamplePrompts) return;
+			if (loadingPrompts.current) return;
+			loadingPrompts.current = true;
+			const prompts = status === "new" ? (await getSamplePrompts(collection.singularName ?? collection.name, instructions)).prompts : getPromptsForExistingEntities(collection.properties);
+			const recentPromptsFromStorage = getRecentPromptsFromStorage(storageKey);
+			const recentPrompts = recentPromptsFromStorage.map((prompt) => prompt.prompt);
+			setSamplePrompts([...recentPromptsFromStorage, ...prompts.filter((p) => !recentPrompts.includes(p.prompt))].slice(0, 5));
+			loadingPrompts.current = false;
+		}, [
+			collection.name,
+			collection.singularName,
+			getSamplePrompts,
+			status
+		]);
+		(0, react.useDeferredValue)(formContext?.values);
+		(0, react.useEffect)(() => {
+			if (!dataEnhancementController) return;
+			if (!samplePrompts) {
+				setSamplePrompts(getRecentPromptsFromStorage(storageKey));
+				updateSuggestedPrompts().then();
+			}
+		}, [
+			dataEnhancementController,
+			samplePrompts,
+			storageKey,
+			updateSuggestedPrompts,
+			instructions,
+			status
+		]);
+		(0, react.useEffect)(() => {
+			if (!dataEnhancementController) return;
+			updateSuggestedPrompts().then();
+		}, [dataEnhancementController, status]);
+		const enhance = (prompt) => {
+			if (!dataEnhancementController || !formContext?.values) return;
+			setLoading(true);
+			if (prompt) {
+				addRecentPrompt(storageKey, prompt);
+				setSamplePrompts([{
+					prompt,
+					type: "recent"
+				}, ...(samplePrompts ?? []).slice(0, 5)]);
+			}
+			return dataEnhancementController.enhance({
+				entityId,
+				values: formContext.values,
+				instructions: prompt,
+				replaceValues: true
+			}).finally(() => {
+				setLoading(false);
+			});
+		};
+		if (!dataEnhancementController?.enabled) return null;
+		const suggestions = dataEnhancementController.suggestions;
+		Object.values(suggestions).filter(Boolean).length;
+		(samplePrompts ?? []).length > 0 && instructions.length;
+		function submit() {
+			enhance(instructions);
+		}
+		return /* @__PURE__ */ (0, react_jsx_runtime.jsxs)(_rebasepro_ui.Menu, {
+			align: "end",
+			sideOffset: 8,
+			className: "max-w-[100vw]",
+			trigger: /* @__PURE__ */ (0, react_jsx_runtime.jsxs)(_rebasepro_ui.Button, {
+				variant: "filled",
+				color: "neutral",
+				fullWidth: largeLayout && openEntityMode === "full_screen",
+				size: "small",
+				disabled: loading,
+				children: [
+					!loading && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_core.AIIcon, { size: "small" }),
+					loading && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.CircularProgress, { size: "small" }),
+					"Autofill"
+				]
+			}),
+			children: [
+				/* @__PURE__ */ (0, react_jsx_runtime.jsxs)(_rebasepro_ui.MenuItem, {
+					className: "py-4",
+					onClick: () => {
+						enhance();
+					},
+					children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_core.AIIcon, { size: "small" }), "Autofill based on the current content"]
+				}),
+				/* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.Separator, {
+					orientation: "horizontal",
+					className: "mt-2"
+				}),
+				samplePrompts?.map((samplePrompt, index) => {
+					return /* @__PURE__ */ (0, react_jsx_runtime.jsxs)(_rebasepro_ui.MenuItem, {
+						onClick: () => {
+							setInstructions(samplePrompt.prompt);
+							enhance(samplePrompt.prompt);
+						},
+						children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("div", {
+							className: "pl-9 grow text-text-secondary dark:text-text-secondary-dark",
+							children: samplePrompt.prompt
+						}), samplePrompt.type === "recent" && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.IconButton, {
+							onClick: (e) => {
+								e.preventDefault();
+								e.stopPropagation();
+								removeRecentPrompt(storageKey, samplePrompt.prompt);
+								setSamplePrompts((samplePrompts ?? []).filter((p) => p.prompt !== samplePrompt.prompt));
+							},
+							size: "smallest",
+							children: /* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.XIcon, { size: _rebasepro_ui.iconSize.smallest })
+						})]
+					}, index + "_" + samplePrompt.prompt);
+				}),
+				/* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.Separator, { orientation: "horizontal" }),
+				/* @__PURE__ */ (0, react_jsx_runtime.jsxs)("div", {
+					className: (0, _rebasepro_ui.cls)("my-2 w-[500px] max-w-full flex items-start text-surface-700 dark:text-surface-200"),
+					children: [
+						/* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.TextareaAutosize, {
+							className: (0, _rebasepro_ui.cls)("p-4 rounded-lg resize-none bg-surface-100 dark:bg-surface-950 mx-2 w-full grow outline-hidden max-h-[300px] overflow-auto", _rebasepro_ui.focusedDisabled),
+							value: instructions,
+							autoFocus: status === "new",
+							disabled: loading,
+							onFocus: (event) => {
+								event.stopPropagation();
+							},
+							placeholder: "...or provide instructions",
+							onKeyDown: (e) => {
+								e.stopPropagation();
+								if (e.key === "Enter" && !e.shiftKey) {
+									e.preventDefault();
+									submit();
+								}
+							},
+							onChange: (e) => {
+								setInstructions(e.target.value);
+							}
+						}),
+						/* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.IconButton, {
+							size: "small",
+							onClick: () => {
+								setInstructions("");
+							},
+							color: !instructions ? "primary" : void 0,
+							disabled: loading || !instructions,
+							children: /* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.XIcon, { size: _rebasepro_ui.iconSize.small })
+						}),
+						/* @__PURE__ */ (0, react_jsx_runtime.jsxs)(_rebasepro_ui.IconButton, {
+							onClick: () => enhance(instructions),
+							size: "small",
+							color: !instructions ? "primary" : void 0,
+							disabled: loading || !instructions,
+							children: [loading && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.CircularProgress, { size: "smallest" }), !loading && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(_rebasepro_ui.SendIcon, { color: "primary" })]
+						})
+					]
+				})
+			]
+		});
+	}
+	function getPromptsForExistingEntities(properties) {
+		const multilineProperties = Object.values(properties).filter((p) => {
+			if ((0, _rebasepro_common.isPropertyBuilder)(p)) return false;
+			return p.type === "string" && (p.ui?.markdown || p.ui?.multiline);
+		});
+		const multilinePrompt = multilineProperties.length > 0 ? multilineProperties[Math.floor(Math.random() * multilineProperties.length)] : void 0;
+		const prompts = ["Fill the missing fields", "Translate the missing content"];
+		if (multilinePrompt) prompts.push(`Add 2 paragraphs to '${multilinePrompt.name}'`);
+		return prompts.map((p) => ({
+			prompt: p,
+			type: "sample"
+		}));
+	}
+	var createLocalStorageKey = (path, status) => {
+		return `data_enhancement::${status === "new" ? "new" : "existing"}::${(0, _rebasepro_common.stripCollectionPath)(path)}`;
+	};
+	var getRecentPromptsFromStorage = (storageKey) => {
+		const item = localStorage.getItem(storageKey);
+		return item ? JSON.parse(item).map((e) => ({
+			prompt: e,
+			type: "recent"
+		})) : [];
+	};
+	var addRecentPrompt = (storageKey, prompt) => {
+		if (!prompt || prompt.trim().length === 0) return;
+		const recentPrompts = getRecentPromptsFromStorage(storageKey);
+		localStorage.setItem(storageKey, JSON.stringify([prompt, ...recentPrompts.map((e) => e.prompt).filter((e) => e !== prompt).slice(0, 5)]));
+	};
+	var removeRecentPrompt = (storageKey, prompt) => {
+		localStorage.setItem(storageKey, JSON.stringify(getRecentPromptsFromStorage(storageKey).map((e) => e.prompt).filter((e) => e !== prompt)));
+	};
+	//#endregion
+	//#region src/useDataEnhancementPlugin.tsx
+	var DEFAULT_API_KEY = "fcms-U9jdDii0xXWSDC34asfrf54lbkFJBfKfRWcEDEwdc4V5wDWEDF";
+	/**
+	* Use this hook to initialise the data enhancement plugin.
+	* This is likely the only hook you will need to use.
+	* @param props
+	*/
+	function useDataEnhancementPlugin(props) {
+		const apiKey = props?.apiKey ?? DEFAULT_API_KEY;
+		const getConfigForPath = props?.getConfigForPath;
+		return react.default.useMemo(() => ({
+			key: "data_enhancement",
+			slots: [{
+				slot: "form.actions",
+				Component: FormEnhanceAction,
+				order: 40
+			}],
+			providers: [{
+				scope: "form",
+				Component: DataEnhancementControllerProvider,
+				props: {
+					apiKey,
+					getConfigForPath,
+					host: props?.host
+				}
+			}]
+		}), [
+			apiKey,
+			getConfigForPath,
+			props?.host
+		]);
+	}
+	//#endregion
+	exports.useDataEnhancementPlugin = useDataEnhancementPlugin;
+	exports.useEditorAIController = useEditorAIController;
+});
+
+//# sourceMappingURL=index.umd.js.map