@magemetrics/core 0.3.1 → 0.3.3

package/dist/index.js

@@ -54,7 +54,7 @@ var addApiKeyHeader = (apiKey) => {
   };
 };
 
-// ../../node_modules/.pnpm/@asteasolutions+zod-to-openapi@7.3.4_zod@3.25.76/node_modules/@asteasolutions/zod-to-openapi/dist/index.mjs
+// ../../node_modules/.pnpm/@asteasolutions+zod-to-openapi@8.1.0_zod@4.1.12/node_modules/@asteasolutions/zod-to-openapi/dist/index.mjs
 function __rest(s, e) {
   var t = {};
   for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
@@ -66,15 +66,242 @@ function __rest(s, e) {
     }
   return t;
 }
-function isZodType(schema, typeName) {
-  var _a;
-  return ((_a = schema === null || schema === void 0 ? void 0 : schema._def) === null || _a === void 0 ? void 0 : _a.typeName) === typeName;
+var ZodTypeKeys = {
+  ZodAny: "any",
+  ZodArray: "array",
+  ZodBigInt: "bigint",
+  ZodBoolean: "boolean",
+  ZodDefault: "default",
+  ZodTransform: "transform",
+  ZodEnum: "enum",
+  ZodIntersection: "intersection",
+  ZodLiteral: "literal",
+  ZodNever: "never",
+  ZodNull: "null",
+  ZodNullable: "nullable",
+  ZodNumber: "number",
+  ZodNonOptional: "nonoptional",
+  ZodObject: "object",
+  ZodOptional: "optional",
+  ZodPipe: "pipe",
+  ZodReadonly: "readonly",
+  ZodRecord: "record",
+  ZodString: "string",
+  ZodTuple: "tuple",
+  ZodType: "type",
+  ZodUnion: "union",
+  ZodDiscriminatedUnion: "union",
+  ZodUnknown: "unknown",
+  ZodVoid: "void",
+  ZodDate: "date"
+};
+function isZodType(schema, typeNames) {
+  const typeNamesArray = Array.isArray(typeNames) ? typeNames : [typeNames];
+  return typeNamesArray.some((typeName) => {
+    var _a;
+    const typeNameMatch = ((_a = schema === null || schema === void 0 ? void 0 : schema.def) === null || _a === void 0 ? void 0 : _a.type) === ZodTypeKeys[typeName];
+    if (typeName === "ZodDiscriminatedUnion") {
+      return typeNameMatch && "discriminator" in schema.def;
+    }
+    return typeNameMatch;
+  });
+}
+function isAnyZodType(schema) {
+  return "def" in schema;
+}
+var $ZodRegistry = class {
+  constructor() {
+    this._map = /* @__PURE__ */ new Map();
+    this._idmap = /* @__PURE__ */ new Map();
+  }
+  add(schema, ..._meta) {
+    const meta = _meta[0];
+    this._map.set(schema, meta);
+    if (meta && typeof meta === "object" && "id" in meta) {
+      if (this._idmap.has(meta.id)) {
+        throw new Error(`ID ${meta.id} already exists in the registry`);
+      }
+      this._idmap.set(meta.id, schema);
+    }
+    return this;
+  }
+  clear() {
+    this._map = /* @__PURE__ */ new Map();
+    this._idmap = /* @__PURE__ */ new Map();
+    return this;
+  }
+  remove(schema) {
+    const meta = this._map.get(schema);
+    if (meta && typeof meta === "object" && "id" in meta) {
+      this._idmap.delete(meta.id);
+    }
+    this._map.delete(schema);
+    return this;
+  }
+  get(schema) {
+    const p = schema._zod.parent;
+    if (p) {
+      const pm = { ...this.get(p) ?? {} };
+      delete pm.id;
+      return { ...pm, ...this._map.get(schema) };
+    }
+    return this._map.get(schema);
+  }
+  has(schema) {
+    return this._map.has(schema);
+  }
+};
+function registry() {
+  return new $ZodRegistry();
+}
+function isUndefined(value) {
+  return value === void 0;
+}
+function omit(object, keys) {
+  const result = {};
+  Object.entries(object).forEach(([key, value]) => {
+    if (!keys.some((keyToOmit) => keyToOmit === key)) {
+      result[key] = value;
+    }
+  });
+  return result;
 }
-function preserveMetadataFromModifier(zod, modifier) {
-  const zodModifier = zod.ZodType.prototype[modifier];
-  zod.ZodType.prototype[modifier] = function(...args) {
+function omitBy(object, predicate) {
+  const result = {};
+  Object.entries(object).forEach(([key, value]) => {
+    if (!predicate(value, key)) {
+      result[key] = value;
+    }
+  });
+  return result;
+}
+var zodToOpenAPIRegistry = registry();
+var Metadata = class {
+  static collectMetadata(schema, metadata) {
+    const currentMetadata = this.getMetadataFromRegistry(schema);
+    const _internal = Object.assign(Object.assign({}, currentMetadata === null || currentMetadata === void 0 ? void 0 : currentMetadata._internal), metadata === null || metadata === void 0 ? void 0 : metadata._internal);
+    const param = Object.assign(Object.assign({}, currentMetadata === null || currentMetadata === void 0 ? void 0 : currentMetadata.param), metadata === null || metadata === void 0 ? void 0 : metadata.param);
+    const totalMetadata = Object.assign(Object.assign(Object.assign(Object.assign({}, Object.keys(_internal).length > 0 ? { _internal } : {}), currentMetadata), metadata), Object.keys(param).length > 0 ? { param } : {});
+    if (isZodType(schema, [
+      "ZodOptional",
+      "ZodNullable",
+      "ZodDefault",
+      "ZodReadonly",
+      "ZodNonOptional"
+    ]) && isAnyZodType(schema._zod.def.innerType)) {
+      return this.collectMetadata(schema._zod.def.innerType, totalMetadata);
+    }
+    if (isZodType(schema, "ZodPipe")) {
+      const inSchema = schema._zod.def.in;
+      const outSchema = schema._zod.def.out;
+      if (isZodType(inSchema, "ZodTransform") && isAnyZodType(outSchema)) {
+        return this.collectMetadata(outSchema, totalMetadata);
+      }
+      if (isAnyZodType(inSchema)) {
+        return this.collectMetadata(inSchema, totalMetadata);
+      }
+    }
+    return totalMetadata;
+  }
+  /**
+   * @deprecated Use one of `getOpenApiMetadata` or `getInternalMetadata` instead
+   */
+  static getMetadata(zodSchema) {
+    return this.collectMetadata(zodSchema);
+  }
+  static getOpenApiMetadata(zodSchema) {
+    const metadata = this.collectMetadata(zodSchema);
+    const _a = metadata !== null && metadata !== void 0 ? metadata : {}, rest = __rest(_a, ["_internal"]);
+    return rest;
+  }
+  static getInternalMetadata(zodSchema) {
+    var _a;
+    return (_a = this.collectMetadata(zodSchema)) === null || _a === void 0 ? void 0 : _a._internal;
+  }
+  static getParamMetadata(zodSchema) {
+    const metadata = this.collectMetadata(zodSchema);
+    return Object.assign(Object.assign({}, metadata), {
+      // A description provided from .openapi() should be taken with higher precedence
+      param: Object.assign(Object.assign({}, (metadata === null || metadata === void 0 ? void 0 : metadata.description) ? { description: metadata.description } : {}), metadata === null || metadata === void 0 ? void 0 : metadata.param)
+    });
+  }
+  /**
+   * A method that omits all custom keys added to the regular OpenAPI
+   * metadata properties
+   */
+  static buildSchemaMetadata(metadata) {
+    return omitBy(omit(metadata, ["param", "_internal"]), isUndefined);
+  }
+  static buildParameterMetadata(metadata) {
+    return omitBy(metadata, isUndefined);
+  }
+  static applySchemaMetadata(initialData, metadata) {
+    return omitBy(Object.assign(Object.assign({}, initialData), this.buildSchemaMetadata(metadata)), isUndefined);
+  }
+  static getRefId(zodSchema) {
+    var _a;
+    return (_a = this.getInternalMetadata(zodSchema)) === null || _a === void 0 ? void 0 : _a.refId;
+  }
+  static unwrapChained(schema) {
+    return this.unwrapUntil(schema);
+  }
+  static getDefaultValue(zodSchema) {
+    const unwrapped = this.unwrapUntil(zodSchema, "ZodDefault");
+    return unwrapped === null || unwrapped === void 0 ? void 0 : unwrapped._zod.def.defaultValue;
+  }
+  static unwrapUntil(schema, typeName) {
+    if (typeName && isZodType(schema, typeName)) {
+      return schema;
+    }
+    if (isZodType(schema, [
+      "ZodOptional",
+      "ZodNullable",
+      "ZodDefault",
+      "ZodReadonly",
+      "ZodNonOptional"
+    ]) && isAnyZodType(schema._zod.def.innerType)) {
+      return this.unwrapUntil(schema._zod.def.innerType, typeName);
+    }
+    if (isZodType(schema, "ZodPipe")) {
+      const inSchema = schema._zod.def.in;
+      const outSchema = schema._zod.def.out;
+      if (isZodType(inSchema, "ZodTransform") && isAnyZodType(outSchema)) {
+        return this.unwrapUntil(outSchema, typeName);
+      }
+      if (isAnyZodType(inSchema)) {
+        return this.unwrapUntil(inSchema, typeName);
+      }
+    }
+    return typeName ? void 0 : schema;
+  }
+  static getMetadataFromInternalRegistry(zodSchema) {
+    return zodToOpenAPIRegistry.get(zodSchema);
+  }
+  static getMetadataFromRegistry(zodSchema) {
+    const internal = this.getMetadataFromInternalRegistry(zodSchema);
+    const general = zodSchema.meta();
+    if (!internal) {
+      return general;
+    }
+    const { _internal } = internal, rest = __rest(internal, ["_internal"]);
+    const _a = general !== null && general !== void 0 ? general : {}, { id, title } = _a, restGeneral = __rest(_a, ["id", "title"]);
+    return Object.assign(Object.assign(Object.assign({ _internal: Object.assign(Object.assign({}, id ? { refId: id } : {}), _internal) }, rest), title ? { description: title } : {}), restGeneral);
+  }
+  static setMetadataInRegistry(zodSchema, metadata) {
+    zodToOpenAPIRegistry.add(zodSchema, metadata);
+  }
+};
+function preserveMetadataFromModifier(zodSchema, modifier) {
+  const zodModifier = zodSchema[modifier];
+  if (typeof zodModifier !== "function") {
+    return;
+  }
+  zodSchema[modifier] = function(...args) {
     const result = zodModifier.apply(this, args);
-    result._def.openapi = this._def.openapi;
+    const meta = Metadata.getMetadataFromRegistry(this);
+    if (meta) {
+      Metadata.setMetadataInRegistry(result, meta);
+    }
     return result;
   };
 }
@@ -82,59 +309,66 @@ function extendZodWithOpenApi(zod) {
   if (typeof zod.ZodType.prototype.openapi !== "undefined") {
     return;
   }
-  zod.ZodType.prototype.openapi = function(refOrOpenapi, metadata) {
-    var _a, _b, _c, _d, _e, _f;
-    const openapi = typeof refOrOpenapi === "string" ? metadata : refOrOpenapi;
-    const _g = openapi !== null && openapi !== void 0 ? openapi : {}, { param } = _g, restOfOpenApi = __rest(_g, ["param"]);
-    const _internal = Object.assign(Object.assign({}, (_a = this._def.openapi) === null || _a === void 0 ? void 0 : _a._internal), typeof refOrOpenapi === "string" ? { refId: refOrOpenapi } : void 0);
-    const resultMetadata = Object.assign(Object.assign(Object.assign({}, (_b = this._def.openapi) === null || _b === void 0 ? void 0 : _b.metadata), restOfOpenApi), ((_d = (_c = this._def.openapi) === null || _c === void 0 ? void 0 : _c.metadata) === null || _d === void 0 ? void 0 : _d.param) || param ? {
-      param: Object.assign(Object.assign({}, (_f = (_e = this._def.openapi) === null || _e === void 0 ? void 0 : _e.metadata) === null || _f === void 0 ? void 0 : _f.param), param)
+  zod.ZodType.prototype.openapi = function(...args) {
+    const { refId, metadata, options } = getOpenApiConfiguration(...args);
+    const _a = metadata !== null && metadata !== void 0 ? metadata : {}, { param } = _a, restOfOpenApi = __rest(_a, ["param"]);
+    const allMetadata = Metadata.getMetadataFromRegistry(this);
+    const _b = allMetadata !== null && allMetadata !== void 0 ? allMetadata : {}, { _internal: internalMetadata } = _b, currentMetadata = __rest(_b, ["_internal"]);
+    const _internal = Object.assign(Object.assign(Object.assign({}, internalMetadata), options), refId ? { refId } : void 0);
+    const resultMetadata = Object.assign(Object.assign(Object.assign({}, currentMetadata), restOfOpenApi), (currentMetadata === null || currentMetadata === void 0 ? void 0 : currentMetadata.param) || param ? {
+      param: Object.assign(Object.assign({}, currentMetadata === null || currentMetadata === void 0 ? void 0 : currentMetadata.param), param)
     } : void 0);
-    const result = new this.constructor(Object.assign(Object.assign({}, this._def), { openapi: Object.assign(Object.assign({}, Object.keys(_internal).length > 0 ? { _internal } : void 0), Object.keys(resultMetadata).length > 0 ? { metadata: resultMetadata } : void 0) }));
-    if (isZodType(this, "ZodObject")) {
-      const originalExtend = this.extend;
-      result.extend = function(...args) {
-        var _a2, _b2, _c2, _d2, _e2, _f2, _g2;
-        const extendedResult = originalExtend.apply(this, args);
-        extendedResult._def.openapi = {
+    const result = new this.constructor(this._def);
+    Metadata.setMetadataInRegistry(result, Object.assign(Object.assign({}, Object.keys(_internal).length > 0 ? { _internal } : void 0), resultMetadata));
+    if (isZodType(result, "ZodObject")) {
+      const currentMetadata2 = Metadata.getMetadataFromRegistry(result);
+      const originalExtend = result.extend;
+      result.extend = function(...args2) {
+        const extendedResult = originalExtend.apply(result, args2);
+        const _a2 = currentMetadata2 !== null && currentMetadata2 !== void 0 ? currentMetadata2 : {}, { _internal: _internal2 } = _a2, rest = __rest(_a2, ["_internal"]);
+        Metadata.setMetadataInRegistry(extendedResult, {
           _internal: {
-            extendedFrom: ((_b2 = (_a2 = this._def.openapi) === null || _a2 === void 0 ? void 0 : _a2._internal) === null || _b2 === void 0 ? void 0 : _b2.refId) ? { refId: (_d2 = (_c2 = this._def.openapi) === null || _c2 === void 0 ? void 0 : _c2._internal) === null || _d2 === void 0 ? void 0 : _d2.refId, schema: this } : (_f2 = (_e2 = this._def.openapi) === null || _e2 === void 0 ? void 0 : _e2._internal) === null || _f2 === void 0 ? void 0 : _f2.extendedFrom
-          },
-          metadata: (_g2 = extendedResult._def.openapi) === null || _g2 === void 0 ? void 0 : _g2.metadata
-        };
-        return extendedResult;
+            extendedFrom: (_internal2 === null || _internal2 === void 0 ? void 0 : _internal2.refId) ? { refId: _internal2.refId, schema: result } : _internal2 === null || _internal2 === void 0 ? void 0 : _internal2.extendedFrom
+          }
+        });
+        return extendedResult.openapi(rest);
       };
-    }
-    return result;
-  };
-  preserveMetadataFromModifier(zod, "optional");
-  preserveMetadataFromModifier(zod, "nullable");
-  preserveMetadataFromModifier(zod, "default");
-  preserveMetadataFromModifier(zod, "transform");
-  preserveMetadataFromModifier(zod, "refine");
-  const zodDeepPartial = zod.ZodObject.prototype.deepPartial;
-  zod.ZodObject.prototype.deepPartial = function() {
-    const initialShape = this._def.shape();
-    const result = zodDeepPartial.apply(this);
-    const resultShape = result._def.shape();
-    Object.entries(resultShape).forEach(([key, value]) => {
-      var _a, _b;
-      value._def.openapi = (_b = (_a = initialShape[key]) === null || _a === void 0 ? void 0 : _a._def) === null || _b === void 0 ? void 0 : _b.openapi;
-    });
-    result._def.openapi = void 0;
-    return result;
-  };
-  const zodPick = zod.ZodObject.prototype.pick;
-  zod.ZodObject.prototype.pick = function(...args) {
-    const result = zodPick.apply(this, args);
-    result._def.openapi = void 0;
+      preserveMetadataFromModifier(result, "catchall");
+    }
+    preserveMetadataFromModifier(result, "optional");
+    preserveMetadataFromModifier(result, "nullable");
+    preserveMetadataFromModifier(result, "default");
+    preserveMetadataFromModifier(result, "transform");
+    preserveMetadataFromModifier(result, "refine");
+    preserveMetadataFromModifier(result, "length");
+    preserveMetadataFromModifier(result, "min");
+    preserveMetadataFromModifier(result, "max");
+    const originalMeta = result.meta;
+    result.meta = function(...args2) {
+      const result2 = originalMeta.apply(this, args2);
+      if (args2[0]) {
+        const meta = Metadata.getMetadataFromInternalRegistry(this);
+        if (meta) {
+          Metadata.setMetadataInRegistry(result2, Object.assign(Object.assign({}, meta), args2[0]));
+        }
+      }
+      return result2;
+    };
     return result;
   };
-  const zodOmit = zod.ZodObject.prototype.omit;
-  zod.ZodObject.prototype.omit = function(...args) {
-    const result = zodOmit.apply(this, args);
-    result._def.openapi = void 0;
-    return result;
+}
+function getOpenApiConfiguration(refOrOpenapi, metadataOrOptions, options) {
+  if (typeof refOrOpenapi === "string") {
+    return {
+      refId: refOrOpenapi,
+      metadata: metadataOrOptions,
+      options
+    };
+  }
+  return {
+    refId: void 0,
+    metadata: refOrOpenapi,
+    options: metadataOrOptions
   };
 }
 new Set(".\\+*[^]$()");
@@ -155,11 +389,11 @@ var TanStackColumnSchema = z.object({
   accessorKey: z.string(),
   header: z.string(),
   cell: z.any().optional(),
-  meta: z.record(z.any()).optional()
+  meta: z.record(z.string(), z.any()).optional()
 });
 var SqlPreviewResponseSchema = z.object({
   columns: z.array(TanStackColumnSchema),
-  data: z.array(z.record(z.any())),
+  data: z.array(z.record(z.string(), z.any())),
   metadata: z.object({
     wasSampled: z.boolean(),
     originalCount: z.number(),
@@ -337,7 +571,7 @@ var BusinessExplanationSchema = z.object({
 var FlowStepSchema = z.object({
   name: z.string(),
   status: z.enum(["running", "completed", "failed", "awaiting_input"]),
-  output: z.object({}).passthrough().or(z.unknown().array()).nullable()
+  output: z.looseObject({}).or(z.unknown().array()).nullable()
 });
 var DatabaseFlowSchema = z.object({
   created_at: z.string(),
@@ -363,13 +597,13 @@ var FrontendRecentFlowsSchema = DatabaseFlowSchema.omit({
   flow_steps: true,
   flow_chat_messages: true
 });
-var ReportColumnSchema = z.object({
+var ReportColumnSchema = z.looseObject({
   position: z.number().nonnegative().optional(),
   data_type: z.string(),
   null_count: z.number().nullable(),
   unique_count: z.number().nullable(),
   unique_percentage: z.number().nullable()
-}).passthrough();
+});
 var ReportSchema = z.object({
   created_at: z.string(),
   flow_id: z.string(),
@@ -380,10 +614,10 @@ var ReportSchema = z.object({
   table: z.string(),
   title: z.string(),
   request: z.string().nullable(),
-  data_sample: z.array(z.record(z.unknown())),
-  data_summary: z.object({
+  data_sample: z.array(z.record(z.string(), z.unknown())),
+  data_summary: z.looseObject({
     columns: z.record(z.string(), ReportColumnSchema)
-  }).passthrough(),
+  }),
   bookmarked: z.boolean(),
   status: z.string().nullable(),
   is_removed: z.boolean()
@@ -395,7 +629,7 @@ var FrontendReportSchema = ReportSchema.omit({
   is_sample: true,
   schema: true
 });
-var ReportDataSchema = z.object({}).passthrough().array().openapi("ReportData");
+var ReportDataSchema = z.looseObject({}).array().openapi("ReportData");
 var ReportColumnsSchema = z.object({
   name: z.string(),
   data_type: z.string()
@@ -404,11 +638,12 @@ z.object({
   render_type: z.string().nullish(),
   unit: z.string().nullish()
 });
-var FrontendReportColumnsSchema = ReportColumnsSchema.element.merge(z.object({
+var FrontendReportColumnsSchema = z.object({
+  ...ReportColumnsSchema.element.shape,
   dataType: z.string(),
   renderType: z.string().optional(),
   unit: z.string().optional()
-})).array().openapi("ReportColumns");
+}).array().openapi("ReportColumns");
 z.object({
   goal: z.string(),
   user_friendly_goal: z.string().optional()
@@ -708,16 +943,18 @@ var BaseVisualizationSchema = z.object({
   created_at: z.string(),
   sql: z.string(),
   bookmarked: z.boolean(),
-  data_sample: z.array(z.record(z.unknown())),
+  data_sample: z.array(z.record(z.string(), z.unknown())),
   is_sample: z.boolean(),
-  data_summary: z.record(z.unknown())
+  data_summary: z.record(z.string(), z.unknown())
 });
-var V1VisualizationSchema = BaseVisualizationSchema.merge(z.object({
+var V1VisualizationSchema = z.object({
+  ...BaseVisualizationSchema.shape,
   configuration: V1VisualizationConfigurationSchema
-}));
-var VisualizationSchema = BaseVisualizationSchema.merge(z.object({
+});
+var VisualizationSchema = z.object({
+  ...BaseVisualizationSchema.shape,
   configuration: VisualizationConfigurationSchema
-}));
+});
 var VisualizationDataSchema = z.array(z.record(
   z.string(),
   // any key
@@ -1588,12 +1825,13 @@ createRoute({
       description: "The visualizations for the report",
       content: {
         "application/json": {
-          schema: FrontendReportSchema.merge(z.object({
+          schema: z.object({
+            ...FrontendReportSchema.shape,
             visualizations: z.union([
               V1FrontendVisualizationSchema,
               FrontendVisualizationSchema
             ]).array()
-          }))
+          })
         }
       }
     },
@@ -2644,7 +2882,7 @@ var ChatMessages = createRoute({
       content: {
         "application/json": {
           // we don't want to have to write a parser for Vercel AI messages
-          schema: z.object({ messages: z.object({}).passthrough().array() })
+          schema: z.object({ messages: z.looseObject({}).array() })
         }
       }
     },
@@ -2870,7 +3108,7 @@ var toSearchParams = ({ cursor, filters, sorting, limit }) => {
 
 // package.json
 var package_default = {
-  version: "0.3.1"};
+  version: "0.3.3"};
 var MageMetricsChatTransport = class extends DefaultChatTransport {
   constructor(apiUrl, flowId, options) {
     super({