@databricks/appkit 0.72.0 → 0.73.0

package/dist/database/runtime/engine/drizzle-data-path.js.map

@@ -1 +1 @@
-{"version":3,"file":"drizzle-data-path.js","names":[],"sources":["../../../../src/database/runtime/engine/drizzle-data-path.ts"],"sourcesContent":["import { and, eq, isSQLWrapper, type SQL, sql } from \"drizzle-orm\";\nimport { drizzle, type NodePgDatabase } from \"drizzle-orm/node-postgres\";\nimport type { PgTable } from \"drizzle-orm/pg-core\";\nimport type { Pool } from \"pg\";\n\nimport { createLogger } from \"../../../logging/logger\";\nimport {\n  type DatabaseErrorCategory,\n  DatabasePluginError,\n  invalidDatabaseRequest,\n} from \"../../errors\";\nimport type { AppKitTable, ColumnMeta, Schema } from \"../../schema-builder\";\nimport { buildEngineRelations } from \"../../schema-builder/engine/relations\";\nimport { columnValueSchema } from \"../../schema-builder/validators\";\nimport {\n  conflictTargetMeta,\n  type DataPath,\n  type IdValue,\n  limitOrDefault,\n  primaryKeyMeta,\n  type Row,\n  validateOffset,\n  type WhereClause,\n} from \"../data-path\";\nimport {\n  columnOf,\n  type ColumnAccess,\n  defaultColumns,\n  publicColumnNames,\n  returningColumns,\n  selectToColumns,\n  translateInclude,\n  translateOrder,\n  translateWhere,\n} from \"./translate\";\n\nconst logger = createLogger(\"database\");\n\n/** Concrete Drizzle seam shared by the adapter and its focused tests. */\nexport type DrizzleDb = NodePgDatabase<Record<string, never>>;\n\nexport interface DrizzleDataPathOptions {\n  /** Explicit capability for trusted server code that must read private data. */\n  readonly columnAccess?: ColumnAccess;\n}\n\n/** Bind finalized AppKit metadata to a relational Drizzle database. */\nexport function createDrizzleDb(pool: Pool, schema: Schema): DrizzleDb {\n  const completeSchema: Record<string, unknown> = Object.assign(\n    Object.create(null),\n    schema.$engine,\n    buildEngineRelations(schema.$tables),\n  );\n  return drizzle(pool, { schema: completeSchema }) as unknown as DrizzleDb;\n}\n\ninterface RelationalQueryBuilder {\n  findMany(config: Record<string, unknown>): Promise<Row[]>;\n  findFirst(config: Record<string, unknown>): Promise<Row | undefined>;\n}\n\n/** Reject same-name or forged tables by requiring finalized object identity. */\nfunction assertRegisteredTable(schema: Schema, table: AppKitTable): void {\n  if (schema.$tables[table.$name] !== table) {\n    throw invalidDatabaseRequest(`Table \"${table.$name}\" is not registered`);\n  }\n}\n\nfunction relationalQueryBuilder(\n  db: DrizzleDb,\n  schema: Schema,\n  table: AppKitTable,\n): RelationalQueryBuilder {\n  assertRegisteredTable(schema, table);\n  const query = (db.query as unknown as Record<string, RelationalQueryBuilder>)[\n    table.$name\n  ];\n  if (!query) {\n    throw invalidDatabaseRequest(`Table \"${table.$name}\" is not registered`);\n  }\n  return query;\n}\n\nfunction selectedColumns(\n  table: AppKitTable,\n  select?: readonly string[],\n  access: ColumnAccess = \"public\",\n): Record<string, true> {\n  return select === undefined\n    ? defaultColumns(table)\n    : selectToColumns(table, select, access);\n}\n\ntype MutationKind = \"insert\" | \"update\";\n\ninterface MutationSchema {\n  safeParse(\n    value: unknown,\n  ):\n    | { readonly success: true; readonly data: unknown }\n    | { readonly success: false };\n}\n\n/** Apply the finalized write schema again at the persistence boundary. */\nfunction mutationValues(\n  table: AppKitTable,\n  values: Row,\n  kind: MutationKind,\n): Row {\n  if (values === null || typeof values !== \"object\" || Array.isArray(values)) {\n    throw invalidDatabaseRequest(\"Database mutation values must be an object\");\n  }\n  for (const [key, value] of Object.entries(values)) {\n    if (!Object.hasOwn(table.$columns, key)) {\n      throw invalidDatabaseRequest(`Unknown column \"${table.$name}.${key}\"`);\n    }\n    if (isSQLWrapper(value)) {\n      throw invalidDatabaseRequest(\n        \"Database mutation values cannot contain SQL\",\n      );\n    }\n  }\n  const schema = (\n    kind === \"insert\" ? table.$insertSchema : table.$updateSchema\n  ) as MutationSchema;\n  const result = schema.safeParse(values);\n  if (!result.success) {\n    throw invalidDatabaseRequest(`Invalid database ${kind} values`);\n  }\n  return result.data as Row;\n}\n\n/** Never change row identity or the conflict key during the update half. */\nfunction upsertUpdateValues(\n  table: AppKitTable,\n  parameters: Row,\n  conflictTarget: string,\n): Row {\n  const updates: Row = {};\n  for (const [key, value] of Object.entries(parameters)) {\n    const column = table.$columns[key];\n    if (\n      column.primaryKey ||\n      column.serverGenerated ||\n      column.columnName === conflictTarget\n    ) {\n      continue;\n    }\n    updates[key] = value;\n  }\n  if (Object.keys(updates).length > 0) return updates;\n\n  // A safe self-assignment preserves the exactly-one-row upsert contract.\n  return { [conflictTarget]: columnOf(table, conflictTarget) };\n}\n\n// Drizzle wraps driver failures in DrizzleQueryError, so the SQLSTATE sits on a\n// nested `cause` rather than the thrown error. Walk a bounded chain to find it.\nconst MAX_CAUSE_DEPTH = 5;\n\nfunction sqlStateOf(error: unknown): string | undefined {\n  let current = error;\n  for (let depth = 0; depth < MAX_CAUSE_DEPTH; depth++) {\n    if (!current || typeof current !== \"object\") return undefined;\n    try {\n      const candidate = Reflect.get(current, \"code\");\n      // SQLSTATE is always a five-character alphanumeric class code.\n      if (typeof candidate === \"string\" && /^[0-9A-Z]{5}$/.test(candidate)) {\n        return candidate;\n      }\n      current = Reflect.get(current, \"cause\");\n    } catch {\n      return undefined;\n    }\n  }\n  return undefined;\n}\n\n/** Classify SQLSTATE without retaining the driver error or its properties. */\nfunction classifyDriverError(error: unknown): DatabasePluginError {\n  const code = sqlStateOf(error);\n  const category: DatabaseErrorCategory =\n    code === \"40001\" || code === \"40P01\" || code === \"57014\"\n      ? \"TRANSIENT\"\n      : code === \"42501\"\n        ? \"FORBIDDEN\"\n        : code?.startsWith(\"23\")\n          ? \"CONFLICT\"\n          : \"INTERNAL\";\n  logger.error(\n    \"Database driver error classified as %s (SQLSTATE %s)\",\n    category,\n    code ?? \"unknown\",\n  );\n  return new DatabasePluginError(category, \"runtime\");\n}\n\nasync function runDatabaseOperation<T>(\n  operation: () => Promise<T>,\n): Promise<T> {\n  try {\n    return await operation();\n  } catch (error) {\n    if (error instanceof DatabasePluginError) throw error;\n    throw classifyDriverError(error);\n  }\n}\n\n/** Apply the adapter's column capability before rows reach cardinality checks. */\nfunction mutationRows(\n  table: AppKitTable,\n  rows: Row[],\n  access: ColumnAccess,\n): Row[] {\n  if (access === \"trusted\") return rows;\n  const names = publicColumnNames(table);\n  return rows.map((row) => {\n    const projected: Row = {};\n    for (const name of names) {\n      if (Object.hasOwn(row, name)) projected[name] = row[name];\n    }\n    return projected;\n  });\n}\n\n/** Resolve and validate the sole key once before a keyed operation executes. */\nfunction validatedPrimaryKey(\n  table: AppKitTable,\n  id: unknown,\n): { readonly meta: ColumnMeta; readonly value: IdValue } {\n  const meta = primaryKeyMeta(table);\n  const result = columnValueSchema(meta).safeParse(id);\n  if (!result.success) {\n    throw invalidDatabaseRequest(\n      `Invalid primary-key value for \"${table.$name}.${meta.columnName}\"`,\n    );\n  }\n  return { meta, value: result.data as IdValue };\n}\n\n// Enforce the single-row DataPath contract before results reach callers.\nfunction expectExactlyOne(rows: Row[]): Row {\n  if (rows.length !== 1) {\n    throw new DatabasePluginError(\"INTERNAL\", \"runtime\");\n  }\n  return rows[0];\n}\n\nfunction expectZeroOrOne(rows: Row[]): Row | null {\n  if (rows.length > 1) {\n    throw new DatabasePluginError(\"INTERNAL\", \"runtime\");\n  }\n  return rows[0] ?? null;\n}\n\n/** Adapt a Drizzle database to AppKit's internal execution port. */\nexport function createDrizzleDataPath(\n  db: DrizzleDb,\n  schema: Schema,\n  options: DrizzleDataPathOptions = {},\n): DataPath {\n  const columnAccess = options.columnAccess ?? \"public\";\n  const pgTable = (table: AppKitTable): PgTable => {\n    assertRegisteredTable(schema, table);\n    return table.$engine as unknown as PgTable;\n  };\n  const whereSql = (\n    table: AppKitTable,\n    where?: WhereClause,\n  ): SQL | undefined =>\n    where === undefined\n      ? undefined\n      : translateWhere(table, where, columnAccess);\n\n  return {\n    async select(table, spec) {\n      return runDatabaseOperation(() =>\n        relationalQueryBuilder(db, schema, table).findMany({\n          where: whereSql(table, spec.where),\n          orderBy:\n            spec.order === undefined\n              ? undefined\n              : translateOrder(table, spec.order, columnAccess),\n          columns: selectedColumns(table, spec.select, columnAccess),\n          with:\n            spec.include === undefined\n              ? undefined\n              : translateInclude(table, schema, spec.include, columnAccess),\n          limit: limitOrDefault(spec.limit),\n          offset:\n            spec.offset === undefined ? undefined : validateOffset(spec.offset),\n        }),\n      );\n    },\n\n    async findOne(table, id, spec) {\n      const { meta: primaryKey, value: validatedId } = validatedPrimaryKey(\n        table,\n        id,\n      );\n      const primaryKeyPredicate = eq(\n        columnOf(table, primaryKey.columnName),\n        validatedId,\n      );\n      const additionalPredicate =\n        spec?.where === undefined\n          ? undefined\n          : translateWhere(table, spec.where, columnAccess);\n      const row = await runDatabaseOperation(() =>\n        relationalQueryBuilder(db, schema, table).findFirst({\n          where:\n            additionalPredicate === undefined\n              ? primaryKeyPredicate\n              : and(primaryKeyPredicate, additionalPredicate),\n          columns: selectedColumns(table, spec?.select, columnAccess),\n          with:\n            spec?.include === undefined\n              ? undefined\n              : translateInclude(table, schema, spec.include, columnAccess),\n        }),\n      );\n      return row ?? null;\n    },\n\n    async count(table, where) {\n      return runDatabaseOperation(() =>\n        db.$count(pgTable(table), whereSql(table, where)),\n      );\n    },\n\n    async insert(table, values) {\n      const engineTable = pgTable(table);\n      const parameters = mutationValues(table, values, \"insert\");\n      const rows = await runDatabaseOperation(() =>\n        db\n          .insert(engineTable)\n          .values(parameters)\n          .returning(returningColumns(table, columnAccess)),\n      );\n      return expectExactlyOne(mutationRows(table, rows as Row[], columnAccess));\n    },\n\n    async update(table, id, values) {\n      const engineTable = pgTable(table);\n      const parameters = mutationValues(table, values, \"update\");\n      const { meta: primaryKey, value: validatedId } = validatedPrimaryKey(\n        table,\n        id,\n      );\n      const rows = await runDatabaseOperation(() =>\n        db\n          .update(engineTable)\n          .set(parameters)\n          .where(eq(columnOf(table, primaryKey.columnName), validatedId))\n          .returning(returningColumns(table, columnAccess)),\n      );\n      return expectZeroOrOne(mutationRows(table, rows as Row[], columnAccess));\n    },\n\n    async upsert(table, values, onConflict) {\n      const engineTable = pgTable(table);\n      const parameters = mutationValues(table, values, \"insert\");\n      const target = conflictTargetMeta(table, onConflict);\n      const updates = upsertUpdateValues(table, parameters, target.columnName);\n      const rows = await runDatabaseOperation(() =>\n        db\n          .insert(engineTable)\n          .values(parameters)\n          .onConflictDoUpdate({\n            target: columnOf(table, target.columnName),\n            set: updates,\n          })\n          .returning(returningColumns(table, columnAccess)),\n      );\n      return expectExactlyOne(mutationRows(table, rows as Row[], columnAccess));\n    },\n\n    async delete(table, id) {\n      const { meta: primaryKey, value: validatedId } = validatedPrimaryKey(\n        table,\n        id,\n      );\n      const rows = await runDatabaseOperation(() =>\n        db\n          .delete(pgTable(table))\n          .where(eq(columnOf(table, primaryKey.columnName), validatedId))\n          .returning({ id: columnOf(table, primaryKey.columnName) }),\n      );\n      return expectZeroOrOne(rows as Row[]) !== null;\n    },\n\n    async raw<T = Row>(\n      strings: TemplateStringsArray,\n      ...values: unknown[]\n    ): Promise<T[]> {\n      // SQL wrappers carry structure; tagged interpolations may carry values only.\n      if (values.some((value) => isSQLWrapper(value))) {\n        throw invalidDatabaseRequest(\n          \"Tagged SQL interpolations must be parameter values\",\n        );\n      }\n      const result = await runDatabaseOperation(() =>\n        db.execute(sql(strings, ...values.map((value) => sql.param(value)))),\n      );\n      return ((result as { rows?: unknown }).rows ?? result) as T[];\n    },\n\n    async transaction<T>(callback: (tx: DataPath) => Promise<T>): Promise<T> {\n      let callbackFailed = false;\n      let callbackError: unknown;\n      try {\n        return await db.transaction(async (transaction) => {\n          try {\n            return await callback(\n              createDrizzleDataPath(\n                transaction as unknown as DrizzleDb,\n                schema,\n                { columnAccess },\n              ),\n            );\n          } catch (error) {\n            callbackFailed = true;\n            callbackError = error;\n            throw error;\n          }\n        });\n      } catch (error) {\n        // Callback errors are application-owned; sanitize only tx lifecycle errors.\n        if (\n          callbackFailed &&\n          (error === callbackError ||\n            callbackError instanceof DatabasePluginError)\n        ) {\n          throw callbackError;\n        }\n        if (error instanceof DatabasePluginError) throw error;\n        throw classifyDriverError(error);\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;AAoCA,MAAM,SAAS,aAAa,WAAW;;AAWvC,SAAgB,gBAAgB,MAAY,QAA2B;AAMrE,QAAO,QAAQ,MAAM,EAAE,QALyB,OAAO,OACrD,OAAO,OAAO,KAAK,EACnB,OAAO,SACP,qBAAqB,OAAO,QAAQ,CACrC,EAC8C,CAAC;;;AASlD,SAAS,sBAAsB,QAAgB,OAA0B;AACvE,KAAI,OAAO,QAAQ,MAAM,WAAW,MAClC,OAAM,uBAAuB,UAAU,MAAM,MAAM,qBAAqB;;AAI5E,SAAS,uBACP,IACA,QACA,OACwB;AACxB,uBAAsB,QAAQ,MAAM;CACpC,MAAM,QAAS,GAAG,MAChB,MAAM;AAER,KAAI,CAAC,MACH,OAAM,uBAAuB,UAAU,MAAM,MAAM,qBAAqB;AAE1E,QAAO;;AAGT,SAAS,gBACP,OACA,QACA,SAAuB,UACD;AACtB,QAAO,WAAW,SACd,eAAe,MAAM,GACrB,gBAAgB,OAAO,QAAQ,OAAO;;;AAc5C,SAAS,eACP,OACA,QACA,MACK;AACL,KAAI,WAAW,QAAQ,OAAO,WAAW,YAAY,MAAM,QAAQ,OAAO,CACxE,OAAM,uBAAuB,6CAA6C;AAE5E,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,OAAO,EAAE;AACjD,MAAI,CAAC,OAAO,OAAO,MAAM,UAAU,IAAI,CACrC,OAAM,uBAAuB,mBAAmB,MAAM,MAAM,GAAG,IAAI,GAAG;AAExE,MAAI,aAAa,MAAM,CACrB,OAAM,uBACJ,8CACD;;CAML,MAAM,UAFJ,SAAS,WAAW,MAAM,gBAAgB,MAAM,eAE5B,UAAU,OAAO;AACvC,KAAI,CAAC,OAAO,QACV,OAAM,uBAAuB,oBAAoB,KAAK,SAAS;AAEjE,QAAO,OAAO;;;AAIhB,SAAS,mBACP,OACA,YACA,gBACK;CACL,MAAM,UAAe,EAAE;AACvB,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,WAAW,EAAE;EACrD,MAAM,SAAS,MAAM,SAAS;AAC9B,MACE,OAAO,cACP,OAAO,mBACP,OAAO,eAAe,eAEtB;AAEF,UAAQ,OAAO;;AAEjB,KAAI,OAAO,KAAK,QAAQ,CAAC,SAAS,EAAG,QAAO;AAG5C,QAAO,GAAG,iBAAiB,SAAS,OAAO,eAAe,EAAE;;AAK9D,MAAM,kBAAkB;AAExB,SAAS,WAAW,OAAoC;CACtD,IAAI,UAAU;AACd,MAAK,IAAI,QAAQ,GAAG,QAAQ,iBAAiB,SAAS;AACpD,MAAI,CAAC,WAAW,OAAO,YAAY,SAAU,QAAO;AACpD,MAAI;GACF,MAAM,YAAY,QAAQ,IAAI,SAAS,OAAO;AAE9C,OAAI,OAAO,cAAc,YAAY,gBAAgB,KAAK,UAAU,CAClE,QAAO;AAET,aAAU,QAAQ,IAAI,SAAS,QAAQ;UACjC;AACN;;;;;AAON,SAAS,oBAAoB,OAAqC;CAChE,MAAM,OAAO,WAAW,MAAM;CAC9B,MAAM,WACJ,SAAS,WAAW,SAAS,WAAW,SAAS,UAC7C,cACA,SAAS,UACP,cACA,MAAM,WAAW,KAAK,GACpB,aACA;AACV,QAAO,MACL,wDACA,UACA,QAAQ,UACT;AACD,QAAO,IAAI,oBAAoB,UAAU,UAAU;;AAGrD,eAAe,qBACb,WACY;AACZ,KAAI;AACF,SAAO,MAAM,WAAW;UACjB,OAAO;AACd,MAAI,iBAAiB,oBAAqB,OAAM;AAChD,QAAM,oBAAoB,MAAM;;;;AAKpC,SAAS,aACP,OACA,MACA,QACO;AACP,KAAI,WAAW,UAAW,QAAO;CACjC,MAAM,QAAQ,kBAAkB,MAAM;AACtC,QAAO,KAAK,KAAK,QAAQ;EACvB,MAAM,YAAiB,EAAE;AACzB,OAAK,MAAM,QAAQ,MACjB,KAAI,OAAO,OAAO,KAAK,KAAK,CAAE,WAAU,QAAQ,IAAI;AAEtD,SAAO;GACP;;;AAIJ,SAAS,oBACP,OACA,IACwD;CACxD,MAAM,OAAO,eAAe,MAAM;CAClC,MAAM,SAAS,kBAAkB,KAAK,CAAC,UAAU,GAAG;AACpD,KAAI,CAAC,OAAO,QACV,OAAM,uBACJ,kCAAkC,MAAM,MAAM,GAAG,KAAK,WAAW,GAClE;AAEH,QAAO;EAAE;EAAM,OAAO,OAAO;EAAiB;;AAIhD,SAAS,iBAAiB,MAAkB;AAC1C,KAAI,KAAK,WAAW,EAClB,OAAM,IAAI,oBAAoB,YAAY,UAAU;AAEtD,QAAO,KAAK;;AAGd,SAAS,gBAAgB,MAAyB;AAChD,KAAI,KAAK,SAAS,EAChB,OAAM,IAAI,oBAAoB,YAAY,UAAU;AAEtD,QAAO,KAAK,MAAM;;;AAIpB,SAAgB,sBACd,IACA,QACA,UAAkC,EAAE,EAC1B;CACV,MAAM,eAAe,QAAQ,gBAAgB;CAC7C,MAAM,WAAW,UAAgC;AAC/C,wBAAsB,QAAQ,MAAM;AACpC,SAAO,MAAM;;CAEf,MAAM,YACJ,OACA,UAEA,UAAU,SACN,SACA,eAAe,OAAO,OAAO,aAAa;AAEhD,QAAO;EACL,MAAM,OAAO,OAAO,MAAM;AACxB,UAAO,2BACL,uBAAuB,IAAI,QAAQ,MAAM,CAAC,SAAS;IACjD,OAAO,SAAS,OAAO,KAAK,MAAM;IAClC,SACE,KAAK,UAAU,SACX,SACA,eAAe,OAAO,KAAK,OAAO,aAAa;IACrD,SAAS,gBAAgB,OAAO,KAAK,QAAQ,aAAa;IAC1D,MACE,KAAK,YAAY,SACb,SACA,iBAAiB,OAAO,QAAQ,KAAK,SAAS,aAAa;IACjE,OAAO,eAAe,KAAK,MAAM;IACjC,QACE,KAAK,WAAW,SAAY,SAAY,eAAe,KAAK,OAAO;IACtE,CAAC,CACH;;EAGH,MAAM,QAAQ,OAAO,IAAI,MAAM;GAC7B,MAAM,EAAE,MAAM,YAAY,OAAO,gBAAgB,oBAC/C,OACA,GACD;GACD,MAAM,sBAAsB,GAC1B,SAAS,OAAO,WAAW,WAAW,EACtC,YACD;GACD,MAAM,sBACJ,MAAM,UAAU,SACZ,SACA,eAAe,OAAO,KAAK,OAAO,aAAa;AAcrD,UAbY,MAAM,2BAChB,uBAAuB,IAAI,QAAQ,MAAM,CAAC,UAAU;IAClD,OACE,wBAAwB,SACpB,sBACA,IAAI,qBAAqB,oBAAoB;IACnD,SAAS,gBAAgB,OAAO,MAAM,QAAQ,aAAa;IAC3D,MACE,MAAM,YAAY,SACd,SACA,iBAAiB,OAAO,QAAQ,KAAK,SAAS,aAAa;IAClE,CAAC,CACH,IACa;;EAGhB,MAAM,MAAM,OAAO,OAAO;AACxB,UAAO,2BACL,GAAG,OAAO,QAAQ,MAAM,EAAE,SAAS,OAAO,MAAM,CAAC,CAClD;;EAGH,MAAM,OAAO,OAAO,QAAQ;GAC1B,MAAM,cAAc,QAAQ,MAAM;GAClC,MAAM,aAAa,eAAe,OAAO,QAAQ,SAAS;AAO1D,UAAO,iBAAiB,aAAa,OANxB,MAAM,2BACjB,GACG,OAAO,YAAY,CACnB,OAAO,WAAW,CAClB,UAAU,iBAAiB,OAAO,aAAa,CAAC,CACpD,EAC0D,aAAa,CAAC;;EAG3E,MAAM,OAAO,OAAO,IAAI,QAAQ;GAC9B,MAAM,cAAc,QAAQ,MAAM;GAClC,MAAM,aAAa,eAAe,OAAO,QAAQ,SAAS;GAC1D,MAAM,EAAE,MAAM,YAAY,OAAO,gBAAgB,oBAC/C,OACA,GACD;AAQD,UAAO,gBAAgB,aAAa,OAPvB,MAAM,2BACjB,GACG,OAAO,YAAY,CACnB,IAAI,WAAW,CACf,MAAM,GAAG,SAAS,OAAO,WAAW,WAAW,EAAE,YAAY,CAAC,CAC9D,UAAU,iBAAiB,OAAO,aAAa,CAAC,CACpD,EACyD,aAAa,CAAC;;EAG1E,MAAM,OAAO,OAAO,QAAQ,YAAY;GACtC,MAAM,cAAc,QAAQ,MAAM;GAClC,MAAM,aAAa,eAAe,OAAO,QAAQ,SAAS;GAC1D,MAAM,SAAS,mBAAmB,OAAO,WAAW;GACpD,MAAM,UAAU,mBAAmB,OAAO,YAAY,OAAO,WAAW;AAWxE,UAAO,iBAAiB,aAAa,OAVxB,MAAM,2BACjB,GACG,OAAO,YAAY,CACnB,OAAO,WAAW,CAClB,mBAAmB;IAClB,QAAQ,SAAS,OAAO,OAAO,WAAW;IAC1C,KAAK;IACN,CAAC,CACD,UAAU,iBAAiB,OAAO,aAAa,CAAC,CACpD,EAC0D,aAAa,CAAC;;EAG3E,MAAM,OAAO,OAAO,IAAI;GACtB,MAAM,EAAE,MAAM,YAAY,OAAO,gBAAgB,oBAC/C,OACA,GACD;AAOD,UAAO,gBANM,MAAM,2BACjB,GACG,OAAO,QAAQ,MAAM,CAAC,CACtB,MAAM,GAAG,SAAS,OAAO,WAAW,WAAW,EAAE,YAAY,CAAC,CAC9D,UAAU,EAAE,IAAI,SAAS,OAAO,WAAW,WAAW,EAAE,CAAC,CAC7D,CACoC,KAAK;;EAG5C,MAAM,IACJ,SACA,GAAG,QACW;AAEd,OAAI,OAAO,MAAM,UAAU,aAAa,MAAM,CAAC,CAC7C,OAAM,uBACJ,qDACD;GAEH,MAAM,SAAS,MAAM,2BACnB,GAAG,QAAQ,IAAI,SAAS,GAAG,OAAO,KAAK,UAAU,IAAI,MAAM,MAAM,CAAC,CAAC,CAAC,CACrE;AACD,UAAS,OAA8B,QAAQ;;EAGjD,MAAM,YAAe,UAAoD;GACvE,IAAI,iBAAiB;GACrB,IAAI;AACJ,OAAI;AACF,WAAO,MAAM,GAAG,YAAY,OAAO,gBAAgB;AACjD,SAAI;AACF,aAAO,MAAM,SACX,sBACE,aACA,QACA,EAAE,cAAc,CACjB,CACF;cACM,OAAO;AACd,uBAAiB;AACjB,sBAAgB;AAChB,YAAM;;MAER;YACK,OAAO;AAEd,QACE,mBACC,UAAU,iBACT,yBAAyB,qBAE3B,OAAM;AAER,QAAI,iBAAiB,oBAAqB,OAAM;AAChD,UAAM,oBAAoB,MAAM;;;EAGrC"}
+{"version":3,"file":"drizzle-data-path.js","names":[],"sources":["../../../../src/database/runtime/engine/drizzle-data-path.ts"],"sourcesContent":["import { and, eq, isSQLWrapper, type SQL, sql } from \"drizzle-orm\";\nimport { drizzle, type NodePgDatabase } from \"drizzle-orm/node-postgres\";\nimport type { PgTable } from \"drizzle-orm/pg-core\";\nimport type { Pool } from \"pg\";\n\nimport { createLogger } from \"../../../logging/logger\";\nimport {\n  type DatabaseErrorCategory,\n  DatabasePluginError,\n  invalidDatabaseRequest,\n} from \"../../errors\";\nimport type { AppKitTable, ColumnMeta, Schema } from \"../../schema-builder\";\nimport { buildEngineRelations } from \"../../schema-builder/engine/relations\";\nimport { columnValueSchema } from \"../../schema-builder/validators\";\nimport {\n  andWhere,\n  conflictTargetMeta,\n  type DataPath,\n  type IdValue,\n  limitOrDefault,\n  primaryKeyMeta,\n  type Row,\n  validateOffset,\n  type WhereClause,\n} from \"../data-path\";\nimport {\n  columnOf,\n  type ColumnAccess,\n  defaultColumns,\n  publicColumnNames,\n  returningColumns,\n  selectToColumns,\n  translateInclude,\n  translateOrder,\n  translateWhere,\n} from \"./translate\";\n\nconst logger = createLogger(\"database\");\n\n/** Concrete Drizzle seam shared by the adapter and its focused tests. */\nexport type DrizzleDb = NodePgDatabase<Record<string, never>>;\n\nexport interface DrizzleDataPathOptions {\n  /** Explicit capability for trusted server code that must read private data. */\n  readonly columnAccess?: ColumnAccess;\n}\n\n/** Bind finalized AppKit metadata to a relational Drizzle database. */\nexport function createDrizzleDb(pool: Pool, schema: Schema): DrizzleDb {\n  const completeSchema: Record<string, unknown> = Object.assign(\n    Object.create(null),\n    schema.$engine,\n    buildEngineRelations(schema.$tables),\n  );\n  return drizzle(pool, { schema: completeSchema }) as unknown as DrizzleDb;\n}\n\ninterface RelationalQueryBuilder {\n  findMany(config: Record<string, unknown>): Promise<Row[]>;\n  findFirst(config: Record<string, unknown>): Promise<Row | undefined>;\n}\n\n/** Reject same-name or forged tables by requiring finalized object identity. */\nfunction assertRegisteredTable(schema: Schema, table: AppKitTable): void {\n  if (schema.$tables[table.$name] !== table) {\n    throw invalidDatabaseRequest(`Table \"${table.$name}\" is not registered`);\n  }\n}\n\nfunction relationalQueryBuilder(\n  db: DrizzleDb,\n  schema: Schema,\n  table: AppKitTable,\n): RelationalQueryBuilder {\n  assertRegisteredTable(schema, table);\n  const query = (db.query as unknown as Record<string, RelationalQueryBuilder>)[\n    table.$name\n  ];\n  if (!query) {\n    throw invalidDatabaseRequest(`Table \"${table.$name}\" is not registered`);\n  }\n  return query;\n}\n\nfunction selectedColumns(\n  table: AppKitTable,\n  select?: readonly string[],\n  access: ColumnAccess = \"public\",\n): Record<string, true> {\n  return select === undefined\n    ? defaultColumns(table)\n    : selectToColumns(table, select, access);\n}\n\ntype MutationKind = \"insert\" | \"update\";\n\ninterface MutationSchema {\n  safeParse(\n    value: unknown,\n  ):\n    | { readonly success: true; readonly data: unknown }\n    | { readonly success: false };\n}\n\n/** Apply the finalized write schema again at the persistence boundary. */\nfunction mutationValues(\n  table: AppKitTable,\n  values: Row,\n  kind: MutationKind,\n): Row {\n  if (values === null || typeof values !== \"object\" || Array.isArray(values)) {\n    throw invalidDatabaseRequest(\"Database mutation values must be an object\");\n  }\n  for (const [key, value] of Object.entries(values)) {\n    if (!Object.hasOwn(table.$columns, key)) {\n      throw invalidDatabaseRequest(`Unknown column \"${table.$name}.${key}\"`);\n    }\n    if (isSQLWrapper(value)) {\n      throw invalidDatabaseRequest(\n        \"Database mutation values cannot contain SQL\",\n      );\n    }\n  }\n  const schema = (\n    kind === \"insert\" ? table.$insertSchema : table.$updateSchema\n  ) as MutationSchema;\n  const result = schema.safeParse(values);\n  if (!result.success) {\n    throw invalidDatabaseRequest(`Invalid database ${kind} values`);\n  }\n  return result.data as Row;\n}\n\n/** Never change row identity or the conflict key during the update half. */\nfunction upsertUpdateValues(\n  table: AppKitTable,\n  parameters: Row,\n  conflictTarget: string,\n): Row {\n  const updates: Row = {};\n  for (const [key, value] of Object.entries(parameters)) {\n    const column = table.$columns[key];\n    if (\n      column.primaryKey ||\n      column.serverGenerated ||\n      column.columnName === conflictTarget\n    ) {\n      continue;\n    }\n    updates[key] = value;\n  }\n  if (Object.keys(updates).length > 0) return updates;\n\n  // A safe self-assignment preserves the exactly-one-row upsert contract.\n  return { [conflictTarget]: columnOf(table, conflictTarget) };\n}\n\n// Drizzle wraps driver failures in DrizzleQueryError, so the SQLSTATE sits on a\n// nested `cause` rather than the thrown error. Walk a bounded chain to find it.\nconst MAX_CAUSE_DEPTH = 5;\n\nfunction sqlStateOf(error: unknown): string | undefined {\n  let current = error;\n  for (let depth = 0; depth < MAX_CAUSE_DEPTH; depth++) {\n    if (!current || typeof current !== \"object\") return undefined;\n    try {\n      const candidate = Reflect.get(current, \"code\");\n      // SQLSTATE is always a five-character alphanumeric class code.\n      if (typeof candidate === \"string\" && /^[0-9A-Z]{5}$/.test(candidate)) {\n        return candidate;\n      }\n      current = Reflect.get(current, \"cause\");\n    } catch {\n      return undefined;\n    }\n  }\n  return undefined;\n}\n\n/** Classify SQLSTATE without retaining the driver error or its properties. */\nfunction classifyDriverError(error: unknown): DatabasePluginError {\n  const code = sqlStateOf(error);\n  const category: DatabaseErrorCategory =\n    code === \"40001\" || code === \"40P01\" || code === \"57014\"\n      ? \"TRANSIENT\"\n      : code === \"42501\"\n        ? \"FORBIDDEN\"\n        : code?.startsWith(\"23\")\n          ? \"CONFLICT\"\n          : \"INTERNAL\";\n  logger.error(\n    \"Database driver error classified as %s (SQLSTATE %s)\",\n    category,\n    code ?? \"unknown\",\n  );\n  return new DatabasePluginError(category, \"runtime\");\n}\n\nasync function runDatabaseOperation<T>(\n  operation: () => Promise<T>,\n): Promise<T> {\n  try {\n    return await operation();\n  } catch (error) {\n    if (error instanceof DatabasePluginError) throw error;\n    throw classifyDriverError(error);\n  }\n}\n\n/** Apply the adapter's column capability before rows reach cardinality checks. */\nfunction mutationRows(\n  table: AppKitTable,\n  rows: Row[],\n  access: ColumnAccess,\n): Row[] {\n  if (access === \"trusted\") return rows;\n  const names = publicColumnNames(table);\n  return rows.map((row) => {\n    const projected: Row = {};\n    for (const name of names) {\n      if (Object.hasOwn(row, name)) projected[name] = row[name];\n    }\n    return projected;\n  });\n}\n\n/** Resolve and validate the sole key once before a keyed operation executes. */\nfunction validatedPrimaryKey(\n  table: AppKitTable,\n  id: unknown,\n): { readonly meta: ColumnMeta; readonly value: IdValue } {\n  const meta = primaryKeyMeta(table);\n  const result = columnValueSchema(meta).safeParse(id);\n  if (!result.success) {\n    throw invalidDatabaseRequest(\n      `Invalid primary-key value for \"${table.$name}.${meta.columnName}\"`,\n    );\n  }\n  return { meta, value: result.data as IdValue };\n}\n\n// Enforce the single-row DataPath contract before results reach callers.\nfunction expectExactlyOne(rows: Row[]): Row {\n  if (rows.length !== 1) {\n    throw new DatabasePluginError(\"INTERNAL\", \"runtime\");\n  }\n  return rows[0];\n}\n\nfunction expectZeroOrOne(rows: Row[]): Row | null {\n  if (rows.length > 1) {\n    throw new DatabasePluginError(\"INTERNAL\", \"runtime\");\n  }\n  return rows[0] ?? null;\n}\n\n/** Adapt a Drizzle database to AppKit's internal execution port. */\nexport function createDrizzleDataPath(\n  db: DrizzleDb,\n  schema: Schema,\n  options: DrizzleDataPathOptions = {},\n): DataPath {\n  const columnAccess = options.columnAccess ?? \"public\";\n  const pgTable = (table: AppKitTable): PgTable => {\n    assertRegisteredTable(schema, table);\n    return table.$engine as unknown as PgTable;\n  };\n  const whereSql = (\n    table: AppKitTable,\n    where?: WhereClause,\n  ): SQL | undefined =>\n    where === undefined\n      ? undefined\n      : translateWhere(table, where, columnAccess);\n  /** Narrow by the primary key while preserving an accumulated predicate. */\n  const keyedWhere = (\n    table: AppKitTable,\n    primaryKey: ColumnMeta,\n    id: IdValue,\n    where?: WhereClause,\n  ): SQL | undefined =>\n    where === undefined\n      ? eq(columnOf(table, primaryKey.columnName), id)\n      : translateWhere(\n          table,\n          andWhere({ [primaryKey.columnName]: { eq: id } }, where),\n          columnAccess,\n        );\n\n  return {\n    async select(table, spec) {\n      return runDatabaseOperation(() =>\n        relationalQueryBuilder(db, schema, table).findMany({\n          where: whereSql(table, spec.where),\n          orderBy:\n            spec.order === undefined\n              ? undefined\n              : translateOrder(table, spec.order, columnAccess),\n          columns: selectedColumns(table, spec.select, columnAccess),\n          with:\n            spec.include === undefined\n              ? undefined\n              : translateInclude(table, schema, spec.include, columnAccess),\n          limit: limitOrDefault(spec.limit),\n          offset:\n            spec.offset === undefined ? undefined : validateOffset(spec.offset),\n        }),\n      );\n    },\n\n    async findOne(table, id, spec) {\n      const { meta: primaryKey, value: validatedId } = validatedPrimaryKey(\n        table,\n        id,\n      );\n      const primaryKeyPredicate = eq(\n        columnOf(table, primaryKey.columnName),\n        validatedId,\n      );\n      const additionalPredicate =\n        spec?.where === undefined\n          ? undefined\n          : translateWhere(table, spec.where, columnAccess);\n      const row = await runDatabaseOperation(() =>\n        relationalQueryBuilder(db, schema, table).findFirst({\n          where:\n            additionalPredicate === undefined\n              ? primaryKeyPredicate\n              : and(primaryKeyPredicate, additionalPredicate),\n          columns: selectedColumns(table, spec?.select, columnAccess),\n          with:\n            spec?.include === undefined\n              ? undefined\n              : translateInclude(table, schema, spec.include, columnAccess),\n        }),\n      );\n      return row ?? null;\n    },\n\n    async count(table, where) {\n      return runDatabaseOperation(() =>\n        db.$count(pgTable(table), whereSql(table, where)),\n      );\n    },\n\n    async insert(table, values) {\n      const engineTable = pgTable(table);\n      const parameters = mutationValues(table, values, \"insert\");\n      const rows = await runDatabaseOperation(() =>\n        db\n          .insert(engineTable)\n          .values(parameters)\n          .returning(returningColumns(table, columnAccess)),\n      );\n      return expectExactlyOne(mutationRows(table, rows as Row[], columnAccess));\n    },\n\n    async update(table, id, values, where) {\n      const engineTable = pgTable(table);\n      const parameters = mutationValues(table, values, \"update\");\n      const { meta: primaryKey, value: validatedId } = validatedPrimaryKey(\n        table,\n        id,\n      );\n      const rows = await runDatabaseOperation(() =>\n        db\n          .update(engineTable)\n          .set(parameters)\n          .where(keyedWhere(table, primaryKey, validatedId, where))\n          .returning(returningColumns(table, columnAccess)),\n      );\n      return expectZeroOrOne(mutationRows(table, rows as Row[], columnAccess));\n    },\n\n    async upsert(table, values, onConflict) {\n      const engineTable = pgTable(table);\n      const parameters = mutationValues(table, values, \"insert\");\n      const target = conflictTargetMeta(table, onConflict);\n      const updates = upsertUpdateValues(table, parameters, target.columnName);\n      const rows = await runDatabaseOperation(() =>\n        db\n          .insert(engineTable)\n          .values(parameters)\n          .onConflictDoUpdate({\n            target: columnOf(table, target.columnName),\n            set: updates,\n          })\n          .returning(returningColumns(table, columnAccess)),\n      );\n      return expectExactlyOne(mutationRows(table, rows as Row[], columnAccess));\n    },\n\n    async delete(table, id, where) {\n      const { meta: primaryKey, value: validatedId } = validatedPrimaryKey(\n        table,\n        id,\n      );\n      const rows = await runDatabaseOperation(() =>\n        db\n          .delete(pgTable(table))\n          .where(keyedWhere(table, primaryKey, validatedId, where))\n          .returning({ id: columnOf(table, primaryKey.columnName) }),\n      );\n      return expectZeroOrOne(rows as Row[]) !== null;\n    },\n\n    async raw<T = Row>(\n      strings: TemplateStringsArray,\n      ...values: unknown[]\n    ): Promise<T[]> {\n      // SQL wrappers carry structure; tagged interpolations may carry values only.\n      if (values.some((value) => isSQLWrapper(value))) {\n        throw invalidDatabaseRequest(\n          \"Tagged SQL interpolations must be parameter values\",\n        );\n      }\n      const result = await runDatabaseOperation(() =>\n        db.execute(sql(strings, ...values.map((value) => sql.param(value)))),\n      );\n      return ((result as { rows?: unknown }).rows ?? result) as T[];\n    },\n\n    async transaction<T>(callback: (tx: DataPath) => Promise<T>): Promise<T> {\n      let callbackFailed = false;\n      let callbackError: unknown;\n      try {\n        return await db.transaction(async (transaction) => {\n          try {\n            return await callback(\n              createDrizzleDataPath(\n                transaction as unknown as DrizzleDb,\n                schema,\n                { columnAccess },\n              ),\n            );\n          } catch (error) {\n            callbackFailed = true;\n            callbackError = error;\n            throw error;\n          }\n        });\n      } catch (error) {\n        // Callback errors are application-owned; sanitize only tx lifecycle errors.\n        if (\n          callbackFailed &&\n          (error === callbackError ||\n            callbackError instanceof DatabasePluginError)\n        ) {\n          throw callbackError;\n        }\n        if (error instanceof DatabasePluginError) throw error;\n        throw classifyDriverError(error);\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;AAqCA,MAAM,SAAS,aAAa,WAAW;;AAWvC,SAAgB,gBAAgB,MAAY,QAA2B;AAMrE,QAAO,QAAQ,MAAM,EAAE,QALyB,OAAO,OACrD,OAAO,OAAO,KAAK,EACnB,OAAO,SACP,qBAAqB,OAAO,QAAQ,CACrC,EAC8C,CAAC;;;AASlD,SAAS,sBAAsB,QAAgB,OAA0B;AACvE,KAAI,OAAO,QAAQ,MAAM,WAAW,MAClC,OAAM,uBAAuB,UAAU,MAAM,MAAM,qBAAqB;;AAI5E,SAAS,uBACP,IACA,QACA,OACwB;AACxB,uBAAsB,QAAQ,MAAM;CACpC,MAAM,QAAS,GAAG,MAChB,MAAM;AAER,KAAI,CAAC,MACH,OAAM,uBAAuB,UAAU,MAAM,MAAM,qBAAqB;AAE1E,QAAO;;AAGT,SAAS,gBACP,OACA,QACA,SAAuB,UACD;AACtB,QAAO,WAAW,SACd,eAAe,MAAM,GACrB,gBAAgB,OAAO,QAAQ,OAAO;;;AAc5C,SAAS,eACP,OACA,QACA,MACK;AACL,KAAI,WAAW,QAAQ,OAAO,WAAW,YAAY,MAAM,QAAQ,OAAO,CACxE,OAAM,uBAAuB,6CAA6C;AAE5E,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,OAAO,EAAE;AACjD,MAAI,CAAC,OAAO,OAAO,MAAM,UAAU,IAAI,CACrC,OAAM,uBAAuB,mBAAmB,MAAM,MAAM,GAAG,IAAI,GAAG;AAExE,MAAI,aAAa,MAAM,CACrB,OAAM,uBACJ,8CACD;;CAML,MAAM,UAFJ,SAAS,WAAW,MAAM,gBAAgB,MAAM,eAE5B,UAAU,OAAO;AACvC,KAAI,CAAC,OAAO,QACV,OAAM,uBAAuB,oBAAoB,KAAK,SAAS;AAEjE,QAAO,OAAO;;;AAIhB,SAAS,mBACP,OACA,YACA,gBACK;CACL,MAAM,UAAe,EAAE;AACvB,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,WAAW,EAAE;EACrD,MAAM,SAAS,MAAM,SAAS;AAC9B,MACE,OAAO,cACP,OAAO,mBACP,OAAO,eAAe,eAEtB;AAEF,UAAQ,OAAO;;AAEjB,KAAI,OAAO,KAAK,QAAQ,CAAC,SAAS,EAAG,QAAO;AAG5C,QAAO,GAAG,iBAAiB,SAAS,OAAO,eAAe,EAAE;;AAK9D,MAAM,kBAAkB;AAExB,SAAS,WAAW,OAAoC;CACtD,IAAI,UAAU;AACd,MAAK,IAAI,QAAQ,GAAG,QAAQ,iBAAiB,SAAS;AACpD,MAAI,CAAC,WAAW,OAAO,YAAY,SAAU,QAAO;AACpD,MAAI;GACF,MAAM,YAAY,QAAQ,IAAI,SAAS,OAAO;AAE9C,OAAI,OAAO,cAAc,YAAY,gBAAgB,KAAK,UAAU,CAClE,QAAO;AAET,aAAU,QAAQ,IAAI,SAAS,QAAQ;UACjC;AACN;;;;;AAON,SAAS,oBAAoB,OAAqC;CAChE,MAAM,OAAO,WAAW,MAAM;CAC9B,MAAM,WACJ,SAAS,WAAW,SAAS,WAAW,SAAS,UAC7C,cACA,SAAS,UACP,cACA,MAAM,WAAW,KAAK,GACpB,aACA;AACV,QAAO,MACL,wDACA,UACA,QAAQ,UACT;AACD,QAAO,IAAI,oBAAoB,UAAU,UAAU;;AAGrD,eAAe,qBACb,WACY;AACZ,KAAI;AACF,SAAO,MAAM,WAAW;UACjB,OAAO;AACd,MAAI,iBAAiB,oBAAqB,OAAM;AAChD,QAAM,oBAAoB,MAAM;;;;AAKpC,SAAS,aACP,OACA,MACA,QACO;AACP,KAAI,WAAW,UAAW,QAAO;CACjC,MAAM,QAAQ,kBAAkB,MAAM;AACtC,QAAO,KAAK,KAAK,QAAQ;EACvB,MAAM,YAAiB,EAAE;AACzB,OAAK,MAAM,QAAQ,MACjB,KAAI,OAAO,OAAO,KAAK,KAAK,CAAE,WAAU,QAAQ,IAAI;AAEtD,SAAO;GACP;;;AAIJ,SAAS,oBACP,OACA,IACwD;CACxD,MAAM,OAAO,eAAe,MAAM;CAClC,MAAM,SAAS,kBAAkB,KAAK,CAAC,UAAU,GAAG;AACpD,KAAI,CAAC,OAAO,QACV,OAAM,uBACJ,kCAAkC,MAAM,MAAM,GAAG,KAAK,WAAW,GAClE;AAEH,QAAO;EAAE;EAAM,OAAO,OAAO;EAAiB;;AAIhD,SAAS,iBAAiB,MAAkB;AAC1C,KAAI,KAAK,WAAW,EAClB,OAAM,IAAI,oBAAoB,YAAY,UAAU;AAEtD,QAAO,KAAK;;AAGd,SAAS,gBAAgB,MAAyB;AAChD,KAAI,KAAK,SAAS,EAChB,OAAM,IAAI,oBAAoB,YAAY,UAAU;AAEtD,QAAO,KAAK,MAAM;;;AAIpB,SAAgB,sBACd,IACA,QACA,UAAkC,EAAE,EAC1B;CACV,MAAM,eAAe,QAAQ,gBAAgB;CAC7C,MAAM,WAAW,UAAgC;AAC/C,wBAAsB,QAAQ,MAAM;AACpC,SAAO,MAAM;;CAEf,MAAM,YACJ,OACA,UAEA,UAAU,SACN,SACA,eAAe,OAAO,OAAO,aAAa;;CAEhD,MAAM,cACJ,OACA,YACA,IACA,UAEA,UAAU,SACN,GAAG,SAAS,OAAO,WAAW,WAAW,EAAE,GAAG,GAC9C,eACE,OACA,SAAS,GAAG,WAAW,aAAa,EAAE,IAAI,IAAI,EAAE,EAAE,MAAM,EACxD,aACD;AAEP,QAAO;EACL,MAAM,OAAO,OAAO,MAAM;AACxB,UAAO,2BACL,uBAAuB,IAAI,QAAQ,MAAM,CAAC,SAAS;IACjD,OAAO,SAAS,OAAO,KAAK,MAAM;IAClC,SACE,KAAK,UAAU,SACX,SACA,eAAe,OAAO,KAAK,OAAO,aAAa;IACrD,SAAS,gBAAgB,OAAO,KAAK,QAAQ,aAAa;IAC1D,MACE,KAAK,YAAY,SACb,SACA,iBAAiB,OAAO,QAAQ,KAAK,SAAS,aAAa;IACjE,OAAO,eAAe,KAAK,MAAM;IACjC,QACE,KAAK,WAAW,SAAY,SAAY,eAAe,KAAK,OAAO;IACtE,CAAC,CACH;;EAGH,MAAM,QAAQ,OAAO,IAAI,MAAM;GAC7B,MAAM,EAAE,MAAM,YAAY,OAAO,gBAAgB,oBAC/C,OACA,GACD;GACD,MAAM,sBAAsB,GAC1B,SAAS,OAAO,WAAW,WAAW,EACtC,YACD;GACD,MAAM,sBACJ,MAAM,UAAU,SACZ,SACA,eAAe,OAAO,KAAK,OAAO,aAAa;AAcrD,UAbY,MAAM,2BAChB,uBAAuB,IAAI,QAAQ,MAAM,CAAC,UAAU;IAClD,OACE,wBAAwB,SACpB,sBACA,IAAI,qBAAqB,oBAAoB;IACnD,SAAS,gBAAgB,OAAO,MAAM,QAAQ,aAAa;IAC3D,MACE,MAAM,YAAY,SACd,SACA,iBAAiB,OAAO,QAAQ,KAAK,SAAS,aAAa;IAClE,CAAC,CACH,IACa;;EAGhB,MAAM,MAAM,OAAO,OAAO;AACxB,UAAO,2BACL,GAAG,OAAO,QAAQ,MAAM,EAAE,SAAS,OAAO,MAAM,CAAC,CAClD;;EAGH,MAAM,OAAO,OAAO,QAAQ;GAC1B,MAAM,cAAc,QAAQ,MAAM;GAClC,MAAM,aAAa,eAAe,OAAO,QAAQ,SAAS;AAO1D,UAAO,iBAAiB,aAAa,OANxB,MAAM,2BACjB,GACG,OAAO,YAAY,CACnB,OAAO,WAAW,CAClB,UAAU,iBAAiB,OAAO,aAAa,CAAC,CACpD,EAC0D,aAAa,CAAC;;EAG3E,MAAM,OAAO,OAAO,IAAI,QAAQ,OAAO;GACrC,MAAM,cAAc,QAAQ,MAAM;GAClC,MAAM,aAAa,eAAe,OAAO,QAAQ,SAAS;GAC1D,MAAM,EAAE,MAAM,YAAY,OAAO,gBAAgB,oBAC/C,OACA,GACD;AAQD,UAAO,gBAAgB,aAAa,OAPvB,MAAM,2BACjB,GACG,OAAO,YAAY,CACnB,IAAI,WAAW,CACf,MAAM,WAAW,OAAO,YAAY,aAAa,MAAM,CAAC,CACxD,UAAU,iBAAiB,OAAO,aAAa,CAAC,CACpD,EACyD,aAAa,CAAC;;EAG1E,MAAM,OAAO,OAAO,QAAQ,YAAY;GACtC,MAAM,cAAc,QAAQ,MAAM;GAClC,MAAM,aAAa,eAAe,OAAO,QAAQ,SAAS;GAC1D,MAAM,SAAS,mBAAmB,OAAO,WAAW;GACpD,MAAM,UAAU,mBAAmB,OAAO,YAAY,OAAO,WAAW;AAWxE,UAAO,iBAAiB,aAAa,OAVxB,MAAM,2BACjB,GACG,OAAO,YAAY,CACnB,OAAO,WAAW,CAClB,mBAAmB;IAClB,QAAQ,SAAS,OAAO,OAAO,WAAW;IAC1C,KAAK;IACN,CAAC,CACD,UAAU,iBAAiB,OAAO,aAAa,CAAC,CACpD,EAC0D,aAAa,CAAC;;EAG3E,MAAM,OAAO,OAAO,IAAI,OAAO;GAC7B,MAAM,EAAE,MAAM,YAAY,OAAO,gBAAgB,oBAC/C,OACA,GACD;AAOD,UAAO,gBANM,MAAM,2BACjB,GACG,OAAO,QAAQ,MAAM,CAAC,CACtB,MAAM,WAAW,OAAO,YAAY,aAAa,MAAM,CAAC,CACxD,UAAU,EAAE,IAAI,SAAS,OAAO,WAAW,WAAW,EAAE,CAAC,CAC7D,CACoC,KAAK;;EAG5C,MAAM,IACJ,SACA,GAAG,QACW;AAEd,OAAI,OAAO,MAAM,UAAU,aAAa,MAAM,CAAC,CAC7C,OAAM,uBACJ,qDACD;GAEH,MAAM,SAAS,MAAM,2BACnB,GAAG,QAAQ,IAAI,SAAS,GAAG,OAAO,KAAK,UAAU,IAAI,MAAM,MAAM,CAAC,CAAC,CAAC,CACrE;AACD,UAAS,OAA8B,QAAQ;;EAGjD,MAAM,YAAe,UAAoD;GACvE,IAAI,iBAAiB;GACrB,IAAI;AACJ,OAAI;AACF,WAAO,MAAM,GAAG,YAAY,OAAO,gBAAgB;AACjD,SAAI;AACF,aAAO,MAAM,SACX,sBACE,aACA,QACA,EAAE,cAAc,CACjB,CACF;cACM,OAAO;AACd,uBAAiB;AACjB,sBAAgB;AAChB,YAAM;;MAER;YACK,OAAO;AAEd,QACE,mBACC,UAAU,iBACT,yBAAyB,qBAE3B,OAAM;AAER,QAAI,iBAAiB,oBAAqB,OAAM;AAChD,UAAM,oBAAoB,MAAM;;;EAGrC"}

package/dist/database/schema-builder/define-schema.d.ts

@@ -8,7 +8,7 @@ interface SchemaBuilderContext {
 }
 /**
  * Compile one declared schema. The returned type keeps the table names the
- * builder returned, so `crudRoutes` and `hooks` can name only real tables.
+ * builder returned, so `api.tables` and `hooks` can name only real tables.
  */
 declare function defineSchema<TTables extends Record<string, AppKitTable>>(builder: (context: SchemaBuilderContext) => TTables, options?: DefineSchemaOptions): Schema<Extract<keyof TTables, string>>;
 //#endregion

package/dist/database/schema-builder/define-schema.js

@@ -186,7 +186,7 @@ function assertFinalizedSchema(value) {
 }
 /**
 * Compile one declared schema. The returned type keeps the table names the
-* builder returned, so `crudRoutes` and `hooks` can name only real tables.
+* builder returned, so `api.tables` and `hooks` can name only real tables.
 */
 function defineSchema(builder, options) {
 	const schemaName = options?.schemaName ?? "public";

package/dist/database/schema-builder/define-schema.js.map

@@ -1 +1 @@
-{"version":3,"file":"define-schema.js","names":[],"sources":["../../../src/database/schema-builder/define-schema.ts"],"sourcesContent":["import { ColumnBuilder, enumColumn, validateLiteralDefaults } from \"./columns\";\nimport { buildEngineTables } from \"./engine/tables\";\nimport { resolveForeignKeys } from \"./fk\";\nimport { buildRelations } from \"./relations\";\nimport {\n  type AppKitTable,\n  type ColumnMeta,\n  type ColumnRef,\n  type DefineSchemaOptions,\n  type MutableColumnMeta,\n  type ResolvedRelation,\n  type Schema,\n  SchemaBuildError,\n  type TableHandle,\n} from \"./types\";\nimport { deriveInsertSchema, deriveUpdateSchema } from \"./validators\";\n\ninterface RawTable {\n  readonly name: string;\n  readonly metas: Record<string, MutableColumnMeta>;\n  readonly handle: Record<string, ColumnRef>;\n}\n\ninterface FinalizedTableCandidate {\n  readonly columns: Readonly<Record<string, ColumnMeta>>;\n  readonly engine: Schema[\"$engine\"][string];\n  readonly relations: readonly ResolvedRelation[];\n  readonly insertSchema: unknown;\n  readonly updateSchema: unknown;\n}\n\ninterface DeclarationState {\n  readonly raw: Map<string, RawTable>;\n  readonly handleNames: Map<object, string>;\n}\n\nexport interface SchemaBuilderContext {\n  table<C extends Record<string, ColumnBuilder>>(\n    name: string,\n    columns: C,\n  ): TableHandle<C>;\n  enum(name: string, values: readonly string[]): ColumnBuilder;\n}\n\nconst RESERVED_OBJECT_KEYS = new Set([\"__proto__\", \"prototype\", \"constructor\"]);\nconst RESERVED_DATABASE_EXPORT_KEYS = new Set([\"sql\", \"transaction\"]);\n// A global symbol lets typegen recognize schemas loaded through another AppKit instance.\nconst FINALIZED_SCHEMA = Symbol.for(\"@databricks/appkit.database.schema\");\nconst TABLE_METADATA_KEYS = [\n  \"$name\",\n  \"$schemaName\",\n  \"$columns\",\n  \"$engine\",\n  \"$relations\",\n  \"$insertSchema\",\n  \"$updateSchema\",\n] as const;\n\nfunction nullRecord<T>(): Record<string, T> {\n  return Object.create(null) as Record<string, T>;\n}\n\nfunction assertRecord(value: unknown, label: string): asserts value is object {\n  if (value === null || typeof value !== \"object\" || Array.isArray(value)) {\n    throw new SchemaBuildError(`${label} must be an object`);\n  }\n}\n\nfunction assertName(value: string, label: string): void {\n  if (!value || RESERVED_OBJECT_KEYS.has(value)) {\n    throw new SchemaBuildError(`${label} \"${value}\" is reserved`);\n  }\n}\n\nfunction assertTableName(value: string): void {\n  assertName(value, \"Table name\");\n  if (RESERVED_DATABASE_EXPORT_KEYS.has(value)) {\n    throw new SchemaBuildError(`Table name \"${value}\" is reserved`);\n  }\n}\n\nfunction finalizeColumn(meta: MutableColumnMeta): ColumnMeta {\n  if (!meta.engineColumn) {\n    throw new SchemaBuildError(\n      `Engine column \"${meta.columnName}\" is missing during finalization`,\n    );\n  }\n  const { fkRef: _fkRef, ...published } = meta;\n  return Object.freeze({\n    ...published,\n    enumValues: meta.enumValues\n      ? Object.freeze([...meta.enumValues])\n      : undefined,\n    fk: meta.fk ? Object.freeze({ ...meta.fk }) : undefined,\n    engineColumn: meta.engineColumn,\n  }) as ColumnMeta;\n}\n\nfunction finalizeRelations(\n  relations: readonly ResolvedRelation[],\n): readonly ResolvedRelation[] {\n  return Object.freeze(\n    relations.map((relation) => Object.freeze({ ...relation })),\n  );\n}\n\n/** Clone builders into per-table metadata and create stable column handles. */\nfunction declareTable<C extends Record<string, ColumnBuilder>>(\n  state: DeclarationState,\n  name: string,\n  columns: C,\n): TableHandle<C> {\n  assertTableName(name);\n  if (state.raw.has(name)) {\n    throw new SchemaBuildError(`Duplicate table \"${name}\"`);\n  }\n  assertRecord(columns, `Columns for table \"${name}\"`);\n\n  const metas = nullRecord<MutableColumnMeta>();\n  const handle = Object.create(null) as Record<string, ColumnRef>;\n  for (const [key, column] of Object.entries(columns)) {\n    assertName(key, `Column name on table \"${name}\"`);\n    if (key.startsWith(\"$\") || key === \"and\" || key === \"or\") {\n      throw new SchemaBuildError(\n        `Column \"${name}.${key}\" collides with AppKit runtime metadata`,\n      );\n    }\n    if (!(column instanceof ColumnBuilder)) {\n      throw new SchemaBuildError(\n        `Column \"${name}.${key}\" was not created by an AppKit column builder`,\n      );\n    }\n\n    const meta = column._cloneMeta();\n    meta.name = key;\n    meta.columnName = key;\n    metas[key] = meta;\n\n    const reference: ColumnRef = Object.freeze({\n      __isColumnRef: true,\n      tableName: name,\n      columnName: key,\n    });\n    Object.defineProperty(handle, key, {\n      enumerable: true,\n      value: reference,\n    });\n  }\n\n  state.raw.set(name, { name, metas, handle });\n  state.handleNames.set(handle, name);\n  return handle as unknown as TableHandle<C>;\n}\n\nfunction createBuilderContext(state: DeclarationState): SchemaBuilderContext {\n  return {\n    table(name, columns) {\n      return declareTable(state, name, columns);\n    },\n    enum(name, values) {\n      return enumColumn(name, values);\n    },\n  };\n}\n\n/** Require every exact table handle once under its declared identity. */\nfunction validateReturnedTables(\n  returned: unknown,\n  state: DeclarationState,\n): void {\n  assertRecord(returned, \"defineSchema() return value\");\n  const returnedHandles = new Set<object>();\n  for (const [key, value] of Object.entries(returned)) {\n    const name =\n      value !== null && typeof value === \"object\"\n        ? state.handleNames.get(value)\n        : undefined;\n    if (!name) {\n      throw new SchemaBuildError(\n        `defineSchema returned a value for \"${key}\" that was not produced by ctx.table()`,\n      );\n    }\n    if (key !== name) {\n      throw new SchemaBuildError(\n        `defineSchema returned table \"${name}\" under key \"${key}\"; aliases are not supported`,\n      );\n    }\n    if (returnedHandles.has(value)) {\n      throw new SchemaBuildError(`Table \"${name}\" was returned more than once`);\n    }\n    returnedHandles.add(value);\n  }\n  if (returnedHandles.size !== state.raw.size) {\n    const omitted = [...state.raw.values()]\n      .filter((table) => !returnedHandles.has(table.handle))\n      .map((table) => table.name);\n    throw new SchemaBuildError(\n      `defineSchema omitted declared table${omitted.length === 1 ? \"\" : \"s\"}: ${omitted.join(\", \")}`,\n    );\n  }\n}\n\n/** Detect declaration-handle tampering before metadata publication. */\nfunction validateHandles(raw: ReadonlyMap<string, RawTable>): void {\n  for (const table of raw.values()) {\n    const ownKeys = Reflect.ownKeys(table.handle);\n    const hasOnlyDeclaredColumns =\n      ownKeys.length === Object.keys(table.metas).length &&\n      ownKeys.every(\n        (key) => typeof key === \"string\" && Object.hasOwn(table.metas, key),\n      );\n    const hasMetadataCollision = TABLE_METADATA_KEYS.some((key) =>\n      Object.hasOwn(table.handle, key),\n    );\n    if (\n      Object.getPrototypeOf(table.handle) !== null ||\n      !Object.isExtensible(table.handle) ||\n      !hasOnlyDeclaredColumns ||\n      hasMetadataCollision\n    ) {\n      throw new SchemaBuildError(\n        `Table handle \"${table.name}\" was modified during schema declaration`,\n      );\n    }\n  }\n}\n\n/** Reject composite primary keys, which keyed operations cannot represent. */\nfunction validatePrimaryKeys(raw: ReadonlyMap<string, RawTable>): void {\n  for (const table of raw.values()) {\n    const primaryKeys = Object.values(table.metas).filter(\n      (meta) => meta.primaryKey,\n    );\n    if (primaryKeys.length > 1) {\n      throw new SchemaBuildError(\n        `Table \"${table.name}\" declares multiple primary-key columns; composite primary keys are not supported`,\n      );\n    }\n  }\n}\n\n/** Reject table names that collide with generated Drizzle relation keys. */\nfunction validateRelationKeys(\n  raw: ReadonlyMap<string, RawTable>,\n  relations: ReadonlyMap<string, readonly ResolvedRelation[]>,\n): void {\n  for (const [name, tableRelations] of relations) {\n    if (tableRelations.length > 0 && raw.has(`${name}Relations`)) {\n      throw new SchemaBuildError(\n        `Table \"${name}Relations\" collides with generated relation metadata for \"${name}\"`,\n      );\n    }\n  }\n}\n\n/** Prepare every engine object and validator without mutating handles. */\nfunction prepareTables(\n  raw: ReadonlyMap<string, RawTable>,\n  schemaName: string,\n  relations: ReadonlyMap<string, readonly ResolvedRelation[]>,\n): Map<string, FinalizedTableCandidate> {\n  const built = buildEngineTables(raw.values(), schemaName);\n  const candidates = new Map<string, FinalizedTableCandidate>();\n  for (const table of raw.values()) {\n    const builtTable = built.get(table.name);\n    if (!builtTable) {\n      throw new SchemaBuildError(\n        `Engine table \"${table.name}\" was not constructed`,\n      );\n    }\n    const columns = nullRecord<ColumnMeta>();\n    for (const [key, meta] of Object.entries(builtTable.columns)) {\n      columns[key] = finalizeColumn(meta);\n    }\n    Object.freeze(columns);\n    const tableRelations = finalizeRelations(relations.get(table.name) ?? []);\n    const validatorTable = { $columns: columns } as AppKitTable;\n    candidates.set(table.name, {\n      columns,\n      engine: builtTable.engine,\n      relations: tableRelations,\n      insertSchema: deriveInsertSchema(validatorTable),\n      updateSchema: deriveUpdateSchema(validatorTable),\n    });\n  }\n  return candidates;\n}\n\n/** Atomically publish prepared metadata as the final schema transition. */\nfunction publishSchema(\n  raw: ReadonlyMap<string, RawTable>,\n  candidates: ReadonlyMap<string, FinalizedTableCandidate>,\n  schemaName: string,\n): Schema {\n  const publications = [...raw.values()].map((table) => {\n    const candidate = candidates.get(table.name);\n    if (!candidate) {\n      throw new SchemaBuildError(\n        `Table \"${table.name}\" was not prepared for finalization`,\n      );\n    }\n    return { table, candidate };\n  });\n\n  const tables = nullRecord<AppKitTable>();\n  const engine = nullRecord<Schema[\"$engine\"][string]>();\n  for (const { table, candidate } of publications) {\n    Object.defineProperties(table.handle, {\n      $name: { value: table.name },\n      $schemaName: { value: schemaName },\n      $columns: { value: candidate.columns },\n      $engine: { value: candidate.engine },\n      $relations: { value: candidate.relations },\n      $insertSchema: { value: candidate.insertSchema },\n      $updateSchema: { value: candidate.updateSchema },\n    });\n    const finalized = Object.freeze(table.handle) as unknown as AppKitTable;\n    tables[table.name] = finalized;\n    engine[table.name] = candidate.engine;\n  }\n\n  const schema = {\n    $schemaName: schemaName,\n    $tables: Object.freeze(tables),\n    $engine: Object.freeze(engine),\n  };\n  Object.defineProperty(schema, FINALIZED_SCHEMA, { value: true });\n  return Object.freeze(schema);\n}\n\n/** @internal Reject values that were not finalized by `defineSchema()`. */\nexport function assertFinalizedSchema(value: unknown): asserts value is Schema {\n  if (\n    value === null ||\n    typeof value !== \"object\" ||\n    (value as Record<PropertyKey, unknown>)[FINALIZED_SCHEMA] !== true\n  ) {\n    throw new TypeError(\"Expected a finalized AppKit database schema\");\n  }\n}\n\n/**\n * Compile one declared schema. The returned type keeps the table names the\n * builder returned, so `crudRoutes` and `hooks` can name only real tables.\n */\nexport function defineSchema<TTables extends Record<string, AppKitTable>>(\n  builder: (context: SchemaBuilderContext) => TTables,\n  options?: DefineSchemaOptions,\n): Schema<Extract<keyof TTables, string>> {\n  const schemaName = options?.schemaName ?? \"public\";\n  if (!schemaName) throw new SchemaBuildError(\"Schema name cannot be empty\");\n\n  const state: DeclarationState = {\n    raw: new Map(),\n    handleNames: new Map(),\n  };\n  const returned = builder(createBuilderContext(state));\n\n  validateReturnedTables(returned, state);\n  validateHandles(state.raw);\n  validatePrimaryKeys(state.raw);\n  resolveForeignKeys(state.raw);\n  // FK literals are checked against the inherited target storage, not the placeholder.\n  validateLiteralDefaults(state.raw.values());\n\n  const relations = buildRelations(state.raw);\n  validateRelationKeys(state.raw, relations);\n  const candidates = prepareTables(state.raw, schemaName, relations);\n  return publishSchema(state.raw, candidates, schemaName);\n}\n"],"mappings":";;;;;;;;AA4CA,MAAM,uBAAuB,IAAI,IAAI;CAAC;CAAa;CAAa;CAAc,CAAC;AAC/E,MAAM,gCAAgC,IAAI,IAAI,CAAC,OAAO,cAAc,CAAC;AAErE,MAAM,mBAAmB,OAAO,IAAI,qCAAqC;AACzE,MAAM,sBAAsB;CAC1B;CACA;CACA;CACA;CACA;CACA;CACA;CACD;AAED,SAAS,aAAmC;AAC1C,QAAO,OAAO,OAAO,KAAK;;AAG5B,SAAS,aAAa,OAAgB,OAAwC;AAC5E,KAAI,UAAU,QAAQ,OAAO,UAAU,YAAY,MAAM,QAAQ,MAAM,CACrE,OAAM,IAAI,iBAAiB,GAAG,MAAM,oBAAoB;;AAI5D,SAAS,WAAW,OAAe,OAAqB;AACtD,KAAI,CAAC,SAAS,qBAAqB,IAAI,MAAM,CAC3C,OAAM,IAAI,iBAAiB,GAAG,MAAM,IAAI,MAAM,eAAe;;AAIjE,SAAS,gBAAgB,OAAqB;AAC5C,YAAW,OAAO,aAAa;AAC/B,KAAI,8BAA8B,IAAI,MAAM,CAC1C,OAAM,IAAI,iBAAiB,eAAe,MAAM,eAAe;;AAInE,SAAS,eAAe,MAAqC;AAC3D,KAAI,CAAC,KAAK,aACR,OAAM,IAAI,iBACR,kBAAkB,KAAK,WAAW,kCACnC;CAEH,MAAM,EAAE,OAAO,QAAQ,GAAG,cAAc;AACxC,QAAO,OAAO,OAAO;EACnB,GAAG;EACH,YAAY,KAAK,aACb,OAAO,OAAO,CAAC,GAAG,KAAK,WAAW,CAAC,GACnC;EACJ,IAAI,KAAK,KAAK,OAAO,OAAO,EAAE,GAAG,KAAK,IAAI,CAAC,GAAG;EAC9C,cAAc,KAAK;EACpB,CAAC;;AAGJ,SAAS,kBACP,WAC6B;AAC7B,QAAO,OAAO,OACZ,UAAU,KAAK,aAAa,OAAO,OAAO,EAAE,GAAG,UAAU,CAAC,CAAC,CAC5D;;;AAIH,SAAS,aACP,OACA,MACA,SACgB;AAChB,iBAAgB,KAAK;AACrB,KAAI,MAAM,IAAI,IAAI,KAAK,CACrB,OAAM,IAAI,iBAAiB,oBAAoB,KAAK,GAAG;AAEzD,cAAa,SAAS,sBAAsB,KAAK,GAAG;CAEpD,MAAM,QAAQ,YAA+B;CAC7C,MAAM,SAAS,OAAO,OAAO,KAAK;AAClC,MAAK,MAAM,CAAC,KAAK,WAAW,OAAO,QAAQ,QAAQ,EAAE;AACnD,aAAW,KAAK,yBAAyB,KAAK,GAAG;AACjD,MAAI,IAAI,WAAW,IAAI,IAAI,QAAQ,SAAS,QAAQ,KAClD,OAAM,IAAI,iBACR,WAAW,KAAK,GAAG,IAAI,yCACxB;AAEH,MAAI,EAAE,kBAAkB,eACtB,OAAM,IAAI,iBACR,WAAW,KAAK,GAAG,IAAI,+CACxB;EAGH,MAAM,OAAO,OAAO,YAAY;AAChC,OAAK,OAAO;AACZ,OAAK,aAAa;AAClB,QAAM,OAAO;EAEb,MAAM,YAAuB,OAAO,OAAO;GACzC,eAAe;GACf,WAAW;GACX,YAAY;GACb,CAAC;AACF,SAAO,eAAe,QAAQ,KAAK;GACjC,YAAY;GACZ,OAAO;GACR,CAAC;;AAGJ,OAAM,IAAI,IAAI,MAAM;EAAE;EAAM;EAAO;EAAQ,CAAC;AAC5C,OAAM,YAAY,IAAI,QAAQ,KAAK;AACnC,QAAO;;AAGT,SAAS,qBAAqB,OAA+C;AAC3E,QAAO;EACL,MAAM,MAAM,SAAS;AACnB,UAAO,aAAa,OAAO,MAAM,QAAQ;;EAE3C,KAAK,MAAM,QAAQ;AACjB,UAAO,WAAW,MAAM,OAAO;;EAElC;;;AAIH,SAAS,uBACP,UACA,OACM;AACN,cAAa,UAAU,8BAA8B;CACrD,MAAM,kCAAkB,IAAI,KAAa;AACzC,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,SAAS,EAAE;EACnD,MAAM,OACJ,UAAU,QAAQ,OAAO,UAAU,WAC/B,MAAM,YAAY,IAAI,MAAM,GAC5B;AACN,MAAI,CAAC,KACH,OAAM,IAAI,iBACR,sCAAsC,IAAI,wCAC3C;AAEH,MAAI,QAAQ,KACV,OAAM,IAAI,iBACR,gCAAgC,KAAK,eAAe,IAAI,8BACzD;AAEH,MAAI,gBAAgB,IAAI,MAAM,CAC5B,OAAM,IAAI,iBAAiB,UAAU,KAAK,+BAA+B;AAE3E,kBAAgB,IAAI,MAAM;;AAE5B,KAAI,gBAAgB,SAAS,MAAM,IAAI,MAAM;EAC3C,MAAM,UAAU,CAAC,GAAG,MAAM,IAAI,QAAQ,CAAC,CACpC,QAAQ,UAAU,CAAC,gBAAgB,IAAI,MAAM,OAAO,CAAC,CACrD,KAAK,UAAU,MAAM,KAAK;AAC7B,QAAM,IAAI,iBACR,sCAAsC,QAAQ,WAAW,IAAI,KAAK,IAAI,IAAI,QAAQ,KAAK,KAAK,GAC7F;;;;AAKL,SAAS,gBAAgB,KAA0C;AACjE,MAAK,MAAM,SAAS,IAAI,QAAQ,EAAE;EAChC,MAAM,UAAU,QAAQ,QAAQ,MAAM,OAAO;EAC7C,MAAM,yBACJ,QAAQ,WAAW,OAAO,KAAK,MAAM,MAAM,CAAC,UAC5C,QAAQ,OACL,QAAQ,OAAO,QAAQ,YAAY,OAAO,OAAO,MAAM,OAAO,IAAI,CACpE;EACH,MAAM,uBAAuB,oBAAoB,MAAM,QACrD,OAAO,OAAO,MAAM,QAAQ,IAAI,CACjC;AACD,MACE,OAAO,eAAe,MAAM,OAAO,KAAK,QACxC,CAAC,OAAO,aAAa,MAAM,OAAO,IAClC,CAAC,0BACD,qBAEA,OAAM,IAAI,iBACR,iBAAiB,MAAM,KAAK,0CAC7B;;;;AAMP,SAAS,oBAAoB,KAA0C;AACrE,MAAK,MAAM,SAAS,IAAI,QAAQ,CAI9B,KAHoB,OAAO,OAAO,MAAM,MAAM,CAAC,QAC5C,SAAS,KAAK,WAChB,CACe,SAAS,EACvB,OAAM,IAAI,iBACR,UAAU,MAAM,KAAK,mFACtB;;;AAMP,SAAS,qBACP,KACA,WACM;AACN,MAAK,MAAM,CAAC,MAAM,mBAAmB,UACnC,KAAI,eAAe,SAAS,KAAK,IAAI,IAAI,GAAG,KAAK,WAAW,CAC1D,OAAM,IAAI,iBACR,UAAU,KAAK,4DAA4D,KAAK,GACjF;;;AAMP,SAAS,cACP,KACA,YACA,WACsC;CACtC,MAAM,QAAQ,kBAAkB,IAAI,QAAQ,EAAE,WAAW;CACzD,MAAM,6BAAa,IAAI,KAAsC;AAC7D,MAAK,MAAM,SAAS,IAAI,QAAQ,EAAE;EAChC,MAAM,aAAa,MAAM,IAAI,MAAM,KAAK;AACxC,MAAI,CAAC,WACH,OAAM,IAAI,iBACR,iBAAiB,MAAM,KAAK,uBAC7B;EAEH,MAAM,UAAU,YAAwB;AACxC,OAAK,MAAM,CAAC,KAAK,SAAS,OAAO,QAAQ,WAAW,QAAQ,CAC1D,SAAQ,OAAO,eAAe,KAAK;AAErC,SAAO,OAAO,QAAQ;EACtB,MAAM,iBAAiB,kBAAkB,UAAU,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;EACzE,MAAM,iBAAiB,EAAE,UAAU,SAAS;AAC5C,aAAW,IAAI,MAAM,MAAM;GACzB;GACA,QAAQ,WAAW;GACnB,WAAW;GACX,cAAc,mBAAmB,eAAe;GAChD,cAAc,mBAAmB,eAAe;GACjD,CAAC;;AAEJ,QAAO;;;AAIT,SAAS,cACP,KACA,YACA,YACQ;CACR,MAAM,eAAe,CAAC,GAAG,IAAI,QAAQ,CAAC,CAAC,KAAK,UAAU;EACpD,MAAM,YAAY,WAAW,IAAI,MAAM,KAAK;AAC5C,MAAI,CAAC,UACH,OAAM,IAAI,iBACR,UAAU,MAAM,KAAK,qCACtB;AAEH,SAAO;GAAE;GAAO;GAAW;GAC3B;CAEF,MAAM,SAAS,YAAyB;CACxC,MAAM,SAAS,YAAuC;AACtD,MAAK,MAAM,EAAE,OAAO,eAAe,cAAc;AAC/C,SAAO,iBAAiB,MAAM,QAAQ;GACpC,OAAO,EAAE,OAAO,MAAM,MAAM;GAC5B,aAAa,EAAE,OAAO,YAAY;GAClC,UAAU,EAAE,OAAO,UAAU,SAAS;GACtC,SAAS,EAAE,OAAO,UAAU,QAAQ;GACpC,YAAY,EAAE,OAAO,UAAU,WAAW;GAC1C,eAAe,EAAE,OAAO,UAAU,cAAc;GAChD,eAAe,EAAE,OAAO,UAAU,cAAc;GACjD,CAAC;EACF,MAAM,YAAY,OAAO,OAAO,MAAM,OAAO;AAC7C,SAAO,MAAM,QAAQ;AACrB,SAAO,MAAM,QAAQ,UAAU;;CAGjC,MAAM,SAAS;EACb,aAAa;EACb,SAAS,OAAO,OAAO,OAAO;EAC9B,SAAS,OAAO,OAAO,OAAO;EAC/B;AACD,QAAO,eAAe,QAAQ,kBAAkB,EAAE,OAAO,MAAM,CAAC;AAChE,QAAO,OAAO,OAAO,OAAO;;;AAI9B,SAAgB,sBAAsB,OAAyC;AAC7E,KACE,UAAU,QACV,OAAO,UAAU,YAChB,MAAuC,sBAAsB,KAE9D,OAAM,IAAI,UAAU,8CAA8C;;;;;;AAQtE,SAAgB,aACd,SACA,SACwC;CACxC,MAAM,aAAa,SAAS,cAAc;AAC1C,KAAI,CAAC,WAAY,OAAM,IAAI,iBAAiB,8BAA8B;CAE1E,MAAM,QAA0B;EAC9B,qBAAK,IAAI,KAAK;EACd,6BAAa,IAAI,KAAK;EACvB;AAGD,wBAFiB,QAAQ,qBAAqB,MAAM,CAAC,EAEpB,MAAM;AACvC,iBAAgB,MAAM,IAAI;AAC1B,qBAAoB,MAAM,IAAI;AAC9B,oBAAmB,MAAM,IAAI;AAE7B,yBAAwB,MAAM,IAAI,QAAQ,CAAC;CAE3C,MAAM,YAAY,eAAe,MAAM,IAAI;AAC3C,sBAAqB,MAAM,KAAK,UAAU;CAC1C,MAAM,aAAa,cAAc,MAAM,KAAK,YAAY,UAAU;AAClE,QAAO,cAAc,MAAM,KAAK,YAAY,WAAW"}
+{"version":3,"file":"define-schema.js","names":[],"sources":["../../../src/database/schema-builder/define-schema.ts"],"sourcesContent":["import { ColumnBuilder, enumColumn, validateLiteralDefaults } from \"./columns\";\nimport { buildEngineTables } from \"./engine/tables\";\nimport { resolveForeignKeys } from \"./fk\";\nimport { buildRelations } from \"./relations\";\nimport {\n  type AppKitTable,\n  type ColumnMeta,\n  type ColumnRef,\n  type DefineSchemaOptions,\n  type MutableColumnMeta,\n  type ResolvedRelation,\n  type Schema,\n  SchemaBuildError,\n  type TableHandle,\n} from \"./types\";\nimport { deriveInsertSchema, deriveUpdateSchema } from \"./validators\";\n\ninterface RawTable {\n  readonly name: string;\n  readonly metas: Record<string, MutableColumnMeta>;\n  readonly handle: Record<string, ColumnRef>;\n}\n\ninterface FinalizedTableCandidate {\n  readonly columns: Readonly<Record<string, ColumnMeta>>;\n  readonly engine: Schema[\"$engine\"][string];\n  readonly relations: readonly ResolvedRelation[];\n  readonly insertSchema: unknown;\n  readonly updateSchema: unknown;\n}\n\ninterface DeclarationState {\n  readonly raw: Map<string, RawTable>;\n  readonly handleNames: Map<object, string>;\n}\n\nexport interface SchemaBuilderContext {\n  table<C extends Record<string, ColumnBuilder>>(\n    name: string,\n    columns: C,\n  ): TableHandle<C>;\n  enum(name: string, values: readonly string[]): ColumnBuilder;\n}\n\nconst RESERVED_OBJECT_KEYS = new Set([\"__proto__\", \"prototype\", \"constructor\"]);\nconst RESERVED_DATABASE_EXPORT_KEYS = new Set([\"sql\", \"transaction\"]);\n// A global symbol lets typegen recognize schemas loaded through another AppKit instance.\nconst FINALIZED_SCHEMA = Symbol.for(\"@databricks/appkit.database.schema\");\nconst TABLE_METADATA_KEYS = [\n  \"$name\",\n  \"$schemaName\",\n  \"$columns\",\n  \"$engine\",\n  \"$relations\",\n  \"$insertSchema\",\n  \"$updateSchema\",\n] as const;\n\nfunction nullRecord<T>(): Record<string, T> {\n  return Object.create(null) as Record<string, T>;\n}\n\nfunction assertRecord(value: unknown, label: string): asserts value is object {\n  if (value === null || typeof value !== \"object\" || Array.isArray(value)) {\n    throw new SchemaBuildError(`${label} must be an object`);\n  }\n}\n\nfunction assertName(value: string, label: string): void {\n  if (!value || RESERVED_OBJECT_KEYS.has(value)) {\n    throw new SchemaBuildError(`${label} \"${value}\" is reserved`);\n  }\n}\n\nfunction assertTableName(value: string): void {\n  assertName(value, \"Table name\");\n  if (RESERVED_DATABASE_EXPORT_KEYS.has(value)) {\n    throw new SchemaBuildError(`Table name \"${value}\" is reserved`);\n  }\n}\n\nfunction finalizeColumn(meta: MutableColumnMeta): ColumnMeta {\n  if (!meta.engineColumn) {\n    throw new SchemaBuildError(\n      `Engine column \"${meta.columnName}\" is missing during finalization`,\n    );\n  }\n  const { fkRef: _fkRef, ...published } = meta;\n  return Object.freeze({\n    ...published,\n    enumValues: meta.enumValues\n      ? Object.freeze([...meta.enumValues])\n      : undefined,\n    fk: meta.fk ? Object.freeze({ ...meta.fk }) : undefined,\n    engineColumn: meta.engineColumn,\n  }) as ColumnMeta;\n}\n\nfunction finalizeRelations(\n  relations: readonly ResolvedRelation[],\n): readonly ResolvedRelation[] {\n  return Object.freeze(\n    relations.map((relation) => Object.freeze({ ...relation })),\n  );\n}\n\n/** Clone builders into per-table metadata and create stable column handles. */\nfunction declareTable<C extends Record<string, ColumnBuilder>>(\n  state: DeclarationState,\n  name: string,\n  columns: C,\n): TableHandle<C> {\n  assertTableName(name);\n  if (state.raw.has(name)) {\n    throw new SchemaBuildError(`Duplicate table \"${name}\"`);\n  }\n  assertRecord(columns, `Columns for table \"${name}\"`);\n\n  const metas = nullRecord<MutableColumnMeta>();\n  const handle = Object.create(null) as Record<string, ColumnRef>;\n  for (const [key, column] of Object.entries(columns)) {\n    assertName(key, `Column name on table \"${name}\"`);\n    if (key.startsWith(\"$\") || key === \"and\" || key === \"or\") {\n      throw new SchemaBuildError(\n        `Column \"${name}.${key}\" collides with AppKit runtime metadata`,\n      );\n    }\n    if (!(column instanceof ColumnBuilder)) {\n      throw new SchemaBuildError(\n        `Column \"${name}.${key}\" was not created by an AppKit column builder`,\n      );\n    }\n\n    const meta = column._cloneMeta();\n    meta.name = key;\n    meta.columnName = key;\n    metas[key] = meta;\n\n    const reference: ColumnRef = Object.freeze({\n      __isColumnRef: true,\n      tableName: name,\n      columnName: key,\n    });\n    Object.defineProperty(handle, key, {\n      enumerable: true,\n      value: reference,\n    });\n  }\n\n  state.raw.set(name, { name, metas, handle });\n  state.handleNames.set(handle, name);\n  return handle as unknown as TableHandle<C>;\n}\n\nfunction createBuilderContext(state: DeclarationState): SchemaBuilderContext {\n  return {\n    table(name, columns) {\n      return declareTable(state, name, columns);\n    },\n    enum(name, values) {\n      return enumColumn(name, values);\n    },\n  };\n}\n\n/** Require every exact table handle once under its declared identity. */\nfunction validateReturnedTables(\n  returned: unknown,\n  state: DeclarationState,\n): void {\n  assertRecord(returned, \"defineSchema() return value\");\n  const returnedHandles = new Set<object>();\n  for (const [key, value] of Object.entries(returned)) {\n    const name =\n      value !== null && typeof value === \"object\"\n        ? state.handleNames.get(value)\n        : undefined;\n    if (!name) {\n      throw new SchemaBuildError(\n        `defineSchema returned a value for \"${key}\" that was not produced by ctx.table()`,\n      );\n    }\n    if (key !== name) {\n      throw new SchemaBuildError(\n        `defineSchema returned table \"${name}\" under key \"${key}\"; aliases are not supported`,\n      );\n    }\n    if (returnedHandles.has(value)) {\n      throw new SchemaBuildError(`Table \"${name}\" was returned more than once`);\n    }\n    returnedHandles.add(value);\n  }\n  if (returnedHandles.size !== state.raw.size) {\n    const omitted = [...state.raw.values()]\n      .filter((table) => !returnedHandles.has(table.handle))\n      .map((table) => table.name);\n    throw new SchemaBuildError(\n      `defineSchema omitted declared table${omitted.length === 1 ? \"\" : \"s\"}: ${omitted.join(\", \")}`,\n    );\n  }\n}\n\n/** Detect declaration-handle tampering before metadata publication. */\nfunction validateHandles(raw: ReadonlyMap<string, RawTable>): void {\n  for (const table of raw.values()) {\n    const ownKeys = Reflect.ownKeys(table.handle);\n    const hasOnlyDeclaredColumns =\n      ownKeys.length === Object.keys(table.metas).length &&\n      ownKeys.every(\n        (key) => typeof key === \"string\" && Object.hasOwn(table.metas, key),\n      );\n    const hasMetadataCollision = TABLE_METADATA_KEYS.some((key) =>\n      Object.hasOwn(table.handle, key),\n    );\n    if (\n      Object.getPrototypeOf(table.handle) !== null ||\n      !Object.isExtensible(table.handle) ||\n      !hasOnlyDeclaredColumns ||\n      hasMetadataCollision\n    ) {\n      throw new SchemaBuildError(\n        `Table handle \"${table.name}\" was modified during schema declaration`,\n      );\n    }\n  }\n}\n\n/** Reject composite primary keys, which keyed operations cannot represent. */\nfunction validatePrimaryKeys(raw: ReadonlyMap<string, RawTable>): void {\n  for (const table of raw.values()) {\n    const primaryKeys = Object.values(table.metas).filter(\n      (meta) => meta.primaryKey,\n    );\n    if (primaryKeys.length > 1) {\n      throw new SchemaBuildError(\n        `Table \"${table.name}\" declares multiple primary-key columns; composite primary keys are not supported`,\n      );\n    }\n  }\n}\n\n/** Reject table names that collide with generated Drizzle relation keys. */\nfunction validateRelationKeys(\n  raw: ReadonlyMap<string, RawTable>,\n  relations: ReadonlyMap<string, readonly ResolvedRelation[]>,\n): void {\n  for (const [name, tableRelations] of relations) {\n    if (tableRelations.length > 0 && raw.has(`${name}Relations`)) {\n      throw new SchemaBuildError(\n        `Table \"${name}Relations\" collides with generated relation metadata for \"${name}\"`,\n      );\n    }\n  }\n}\n\n/** Prepare every engine object and validator without mutating handles. */\nfunction prepareTables(\n  raw: ReadonlyMap<string, RawTable>,\n  schemaName: string,\n  relations: ReadonlyMap<string, readonly ResolvedRelation[]>,\n): Map<string, FinalizedTableCandidate> {\n  const built = buildEngineTables(raw.values(), schemaName);\n  const candidates = new Map<string, FinalizedTableCandidate>();\n  for (const table of raw.values()) {\n    const builtTable = built.get(table.name);\n    if (!builtTable) {\n      throw new SchemaBuildError(\n        `Engine table \"${table.name}\" was not constructed`,\n      );\n    }\n    const columns = nullRecord<ColumnMeta>();\n    for (const [key, meta] of Object.entries(builtTable.columns)) {\n      columns[key] = finalizeColumn(meta);\n    }\n    Object.freeze(columns);\n    const tableRelations = finalizeRelations(relations.get(table.name) ?? []);\n    const validatorTable = { $columns: columns } as AppKitTable;\n    candidates.set(table.name, {\n      columns,\n      engine: builtTable.engine,\n      relations: tableRelations,\n      insertSchema: deriveInsertSchema(validatorTable),\n      updateSchema: deriveUpdateSchema(validatorTable),\n    });\n  }\n  return candidates;\n}\n\n/** Atomically publish prepared metadata as the final schema transition. */\nfunction publishSchema(\n  raw: ReadonlyMap<string, RawTable>,\n  candidates: ReadonlyMap<string, FinalizedTableCandidate>,\n  schemaName: string,\n): Schema {\n  const publications = [...raw.values()].map((table) => {\n    const candidate = candidates.get(table.name);\n    if (!candidate) {\n      throw new SchemaBuildError(\n        `Table \"${table.name}\" was not prepared for finalization`,\n      );\n    }\n    return { table, candidate };\n  });\n\n  const tables = nullRecord<AppKitTable>();\n  const engine = nullRecord<Schema[\"$engine\"][string]>();\n  for (const { table, candidate } of publications) {\n    Object.defineProperties(table.handle, {\n      $name: { value: table.name },\n      $schemaName: { value: schemaName },\n      $columns: { value: candidate.columns },\n      $engine: { value: candidate.engine },\n      $relations: { value: candidate.relations },\n      $insertSchema: { value: candidate.insertSchema },\n      $updateSchema: { value: candidate.updateSchema },\n    });\n    const finalized = Object.freeze(table.handle) as unknown as AppKitTable;\n    tables[table.name] = finalized;\n    engine[table.name] = candidate.engine;\n  }\n\n  const schema = {\n    $schemaName: schemaName,\n    $tables: Object.freeze(tables),\n    $engine: Object.freeze(engine),\n  };\n  Object.defineProperty(schema, FINALIZED_SCHEMA, { value: true });\n  return Object.freeze(schema);\n}\n\n/** @internal Reject values that were not finalized by `defineSchema()`. */\nexport function assertFinalizedSchema(value: unknown): asserts value is Schema {\n  if (\n    value === null ||\n    typeof value !== \"object\" ||\n    (value as Record<PropertyKey, unknown>)[FINALIZED_SCHEMA] !== true\n  ) {\n    throw new TypeError(\"Expected a finalized AppKit database schema\");\n  }\n}\n\n/**\n * Compile one declared schema. The returned type keeps the table names the\n * builder returned, so `api.tables` and `hooks` can name only real tables.\n */\nexport function defineSchema<TTables extends Record<string, AppKitTable>>(\n  builder: (context: SchemaBuilderContext) => TTables,\n  options?: DefineSchemaOptions,\n): Schema<Extract<keyof TTables, string>> {\n  const schemaName = options?.schemaName ?? \"public\";\n  if (!schemaName) throw new SchemaBuildError(\"Schema name cannot be empty\");\n\n  const state: DeclarationState = {\n    raw: new Map(),\n    handleNames: new Map(),\n  };\n  const returned = builder(createBuilderContext(state));\n\n  validateReturnedTables(returned, state);\n  validateHandles(state.raw);\n  validatePrimaryKeys(state.raw);\n  resolveForeignKeys(state.raw);\n  // FK literals are checked against the inherited target storage, not the placeholder.\n  validateLiteralDefaults(state.raw.values());\n\n  const relations = buildRelations(state.raw);\n  validateRelationKeys(state.raw, relations);\n  const candidates = prepareTables(state.raw, schemaName, relations);\n  return publishSchema(state.raw, candidates, schemaName);\n}\n"],"mappings":";;;;;;;;AA4CA,MAAM,uBAAuB,IAAI,IAAI;CAAC;CAAa;CAAa;CAAc,CAAC;AAC/E,MAAM,gCAAgC,IAAI,IAAI,CAAC,OAAO,cAAc,CAAC;AAErE,MAAM,mBAAmB,OAAO,IAAI,qCAAqC;AACzE,MAAM,sBAAsB;CAC1B;CACA;CACA;CACA;CACA;CACA;CACA;CACD;AAED,SAAS,aAAmC;AAC1C,QAAO,OAAO,OAAO,KAAK;;AAG5B,SAAS,aAAa,OAAgB,OAAwC;AAC5E,KAAI,UAAU,QAAQ,OAAO,UAAU,YAAY,MAAM,QAAQ,MAAM,CACrE,OAAM,IAAI,iBAAiB,GAAG,MAAM,oBAAoB;;AAI5D,SAAS,WAAW,OAAe,OAAqB;AACtD,KAAI,CAAC,SAAS,qBAAqB,IAAI,MAAM,CAC3C,OAAM,IAAI,iBAAiB,GAAG,MAAM,IAAI,MAAM,eAAe;;AAIjE,SAAS,gBAAgB,OAAqB;AAC5C,YAAW,OAAO,aAAa;AAC/B,KAAI,8BAA8B,IAAI,MAAM,CAC1C,OAAM,IAAI,iBAAiB,eAAe,MAAM,eAAe;;AAInE,SAAS,eAAe,MAAqC;AAC3D,KAAI,CAAC,KAAK,aACR,OAAM,IAAI,iBACR,kBAAkB,KAAK,WAAW,kCACnC;CAEH,MAAM,EAAE,OAAO,QAAQ,GAAG,cAAc;AACxC,QAAO,OAAO,OAAO;EACnB,GAAG;EACH,YAAY,KAAK,aACb,OAAO,OAAO,CAAC,GAAG,KAAK,WAAW,CAAC,GACnC;EACJ,IAAI,KAAK,KAAK,OAAO,OAAO,EAAE,GAAG,KAAK,IAAI,CAAC,GAAG;EAC9C,cAAc,KAAK;EACpB,CAAC;;AAGJ,SAAS,kBACP,WAC6B;AAC7B,QAAO,OAAO,OACZ,UAAU,KAAK,aAAa,OAAO,OAAO,EAAE,GAAG,UAAU,CAAC,CAAC,CAC5D;;;AAIH,SAAS,aACP,OACA,MACA,SACgB;AAChB,iBAAgB,KAAK;AACrB,KAAI,MAAM,IAAI,IAAI,KAAK,CACrB,OAAM,IAAI,iBAAiB,oBAAoB,KAAK,GAAG;AAEzD,cAAa,SAAS,sBAAsB,KAAK,GAAG;CAEpD,MAAM,QAAQ,YAA+B;CAC7C,MAAM,SAAS,OAAO,OAAO,KAAK;AAClC,MAAK,MAAM,CAAC,KAAK,WAAW,OAAO,QAAQ,QAAQ,EAAE;AACnD,aAAW,KAAK,yBAAyB,KAAK,GAAG;AACjD,MAAI,IAAI,WAAW,IAAI,IAAI,QAAQ,SAAS,QAAQ,KAClD,OAAM,IAAI,iBACR,WAAW,KAAK,GAAG,IAAI,yCACxB;AAEH,MAAI,EAAE,kBAAkB,eACtB,OAAM,IAAI,iBACR,WAAW,KAAK,GAAG,IAAI,+CACxB;EAGH,MAAM,OAAO,OAAO,YAAY;AAChC,OAAK,OAAO;AACZ,OAAK,aAAa;AAClB,QAAM,OAAO;EAEb,MAAM,YAAuB,OAAO,OAAO;GACzC,eAAe;GACf,WAAW;GACX,YAAY;GACb,CAAC;AACF,SAAO,eAAe,QAAQ,KAAK;GACjC,YAAY;GACZ,OAAO;GACR,CAAC;;AAGJ,OAAM,IAAI,IAAI,MAAM;EAAE;EAAM;EAAO;EAAQ,CAAC;AAC5C,OAAM,YAAY,IAAI,QAAQ,KAAK;AACnC,QAAO;;AAGT,SAAS,qBAAqB,OAA+C;AAC3E,QAAO;EACL,MAAM,MAAM,SAAS;AACnB,UAAO,aAAa,OAAO,MAAM,QAAQ;;EAE3C,KAAK,MAAM,QAAQ;AACjB,UAAO,WAAW,MAAM,OAAO;;EAElC;;;AAIH,SAAS,uBACP,UACA,OACM;AACN,cAAa,UAAU,8BAA8B;CACrD,MAAM,kCAAkB,IAAI,KAAa;AACzC,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,SAAS,EAAE;EACnD,MAAM,OACJ,UAAU,QAAQ,OAAO,UAAU,WAC/B,MAAM,YAAY,IAAI,MAAM,GAC5B;AACN,MAAI,CAAC,KACH,OAAM,IAAI,iBACR,sCAAsC,IAAI,wCAC3C;AAEH,MAAI,QAAQ,KACV,OAAM,IAAI,iBACR,gCAAgC,KAAK,eAAe,IAAI,8BACzD;AAEH,MAAI,gBAAgB,IAAI,MAAM,CAC5B,OAAM,IAAI,iBAAiB,UAAU,KAAK,+BAA+B;AAE3E,kBAAgB,IAAI,MAAM;;AAE5B,KAAI,gBAAgB,SAAS,MAAM,IAAI,MAAM;EAC3C,MAAM,UAAU,CAAC,GAAG,MAAM,IAAI,QAAQ,CAAC,CACpC,QAAQ,UAAU,CAAC,gBAAgB,IAAI,MAAM,OAAO,CAAC,CACrD,KAAK,UAAU,MAAM,KAAK;AAC7B,QAAM,IAAI,iBACR,sCAAsC,QAAQ,WAAW,IAAI,KAAK,IAAI,IAAI,QAAQ,KAAK,KAAK,GAC7F;;;;AAKL,SAAS,gBAAgB,KAA0C;AACjE,MAAK,MAAM,SAAS,IAAI,QAAQ,EAAE;EAChC,MAAM,UAAU,QAAQ,QAAQ,MAAM,OAAO;EAC7C,MAAM,yBACJ,QAAQ,WAAW,OAAO,KAAK,MAAM,MAAM,CAAC,UAC5C,QAAQ,OACL,QAAQ,OAAO,QAAQ,YAAY,OAAO,OAAO,MAAM,OAAO,IAAI,CACpE;EACH,MAAM,uBAAuB,oBAAoB,MAAM,QACrD,OAAO,OAAO,MAAM,QAAQ,IAAI,CACjC;AACD,MACE,OAAO,eAAe,MAAM,OAAO,KAAK,QACxC,CAAC,OAAO,aAAa,MAAM,OAAO,IAClC,CAAC,0BACD,qBAEA,OAAM,IAAI,iBACR,iBAAiB,MAAM,KAAK,0CAC7B;;;;AAMP,SAAS,oBAAoB,KAA0C;AACrE,MAAK,MAAM,SAAS,IAAI,QAAQ,CAI9B,KAHoB,OAAO,OAAO,MAAM,MAAM,CAAC,QAC5C,SAAS,KAAK,WAChB,CACe,SAAS,EACvB,OAAM,IAAI,iBACR,UAAU,MAAM,KAAK,mFACtB;;;AAMP,SAAS,qBACP,KACA,WACM;AACN,MAAK,MAAM,CAAC,MAAM,mBAAmB,UACnC,KAAI,eAAe,SAAS,KAAK,IAAI,IAAI,GAAG,KAAK,WAAW,CAC1D,OAAM,IAAI,iBACR,UAAU,KAAK,4DAA4D,KAAK,GACjF;;;AAMP,SAAS,cACP,KACA,YACA,WACsC;CACtC,MAAM,QAAQ,kBAAkB,IAAI,QAAQ,EAAE,WAAW;CACzD,MAAM,6BAAa,IAAI,KAAsC;AAC7D,MAAK,MAAM,SAAS,IAAI,QAAQ,EAAE;EAChC,MAAM,aAAa,MAAM,IAAI,MAAM,KAAK;AACxC,MAAI,CAAC,WACH,OAAM,IAAI,iBACR,iBAAiB,MAAM,KAAK,uBAC7B;EAEH,MAAM,UAAU,YAAwB;AACxC,OAAK,MAAM,CAAC,KAAK,SAAS,OAAO,QAAQ,WAAW,QAAQ,CAC1D,SAAQ,OAAO,eAAe,KAAK;AAErC,SAAO,OAAO,QAAQ;EACtB,MAAM,iBAAiB,kBAAkB,UAAU,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;EACzE,MAAM,iBAAiB,EAAE,UAAU,SAAS;AAC5C,aAAW,IAAI,MAAM,MAAM;GACzB;GACA,QAAQ,WAAW;GACnB,WAAW;GACX,cAAc,mBAAmB,eAAe;GAChD,cAAc,mBAAmB,eAAe;GACjD,CAAC;;AAEJ,QAAO;;;AAIT,SAAS,cACP,KACA,YACA,YACQ;CACR,MAAM,eAAe,CAAC,GAAG,IAAI,QAAQ,CAAC,CAAC,KAAK,UAAU;EACpD,MAAM,YAAY,WAAW,IAAI,MAAM,KAAK;AAC5C,MAAI,CAAC,UACH,OAAM,IAAI,iBACR,UAAU,MAAM,KAAK,qCACtB;AAEH,SAAO;GAAE;GAAO;GAAW;GAC3B;CAEF,MAAM,SAAS,YAAyB;CACxC,MAAM,SAAS,YAAuC;AACtD,MAAK,MAAM,EAAE,OAAO,eAAe,cAAc;AAC/C,SAAO,iBAAiB,MAAM,QAAQ;GACpC,OAAO,EAAE,OAAO,MAAM,MAAM;GAC5B,aAAa,EAAE,OAAO,YAAY;GAClC,UAAU,EAAE,OAAO,UAAU,SAAS;GACtC,SAAS,EAAE,OAAO,UAAU,QAAQ;GACpC,YAAY,EAAE,OAAO,UAAU,WAAW;GAC1C,eAAe,EAAE,OAAO,UAAU,cAAc;GAChD,eAAe,EAAE,OAAO,UAAU,cAAc;GACjD,CAAC;EACF,MAAM,YAAY,OAAO,OAAO,MAAM,OAAO;AAC7C,SAAO,MAAM,QAAQ;AACrB,SAAO,MAAM,QAAQ,UAAU;;CAGjC,MAAM,SAAS;EACb,aAAa;EACb,SAAS,OAAO,OAAO,OAAO;EAC9B,SAAS,OAAO,OAAO,OAAO;EAC/B;AACD,QAAO,eAAe,QAAQ,kBAAkB,EAAE,OAAO,MAAM,CAAC;AAChE,QAAO,OAAO,OAAO,OAAO;;;AAI9B,SAAgB,sBAAsB,OAAyC;AAC7E,KACE,UAAU,QACV,OAAO,UAAU,YAChB,MAAuC,sBAAsB,KAE9D,OAAM,IAAI,UAAU,8CAA8C;;;;;;AAQtE,SAAgB,aACd,SACA,SACwC;CACxC,MAAM,aAAa,SAAS,cAAc;AAC1C,KAAI,CAAC,WAAY,OAAM,IAAI,iBAAiB,8BAA8B;CAE1E,MAAM,QAA0B;EAC9B,qBAAK,IAAI,KAAK;EACd,6BAAa,IAAI,KAAK;EACvB;AAGD,wBAFiB,QAAQ,qBAAqB,MAAM,CAAC,EAEpB,MAAM;AACvC,iBAAgB,MAAM,IAAI;AAC1B,qBAAoB,MAAM,IAAI;AAC9B,oBAAmB,MAAM,IAAI;AAE7B,yBAAwB,MAAM,IAAI,QAAQ,CAAC;CAE3C,MAAM,YAAY,eAAe,MAAM,IAAI;AAC3C,sBAAqB,MAAM,KAAK,UAAU;CAC1C,MAAM,aAAa,cAAc,MAAM,KAAK,YAAY,UAAU;AAClE,QAAO,cAAc,MAAM,KAAK,YAAY,WAAW"}

package/dist/errors/database-validation.d.ts

@@ -0,0 +1,23 @@
+import { AppKitError } from "./base.js";
+
+//#region src/errors/database-validation.d.ts
+/** One rejected field; `path` names public columns, never their values. */
+interface DatabaseValidationIssue {
+  readonly path: readonly string[];
+  readonly message: string;
+}
+/**
+ * Deliberate validation failure raised by a database mutation hook. Generated
+ * routes answer `422` and echo only the issues naming a public column; every
+ * other failure raised inside a hook stays an opaque server error.
+ */
+declare class DatabaseValidationError extends AppKitError {
+  readonly code = "DATABASE_VALIDATION_ERROR";
+  readonly statusCode = 422;
+  readonly isRetryable = false;
+  readonly issues: readonly DatabaseValidationIssue[];
+  constructor(message: string, issues?: readonly DatabaseValidationIssue[]);
+}
+//#endregion
+export { DatabaseValidationError, DatabaseValidationIssue };
+//# sourceMappingURL=database-validation.d.ts.map

package/dist/errors/database-validation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"database-validation.d.ts","names":[],"sources":["../../src/errors/database-validation.ts"],"mappings":";;;;UAMiB,uBAAA;EAAA,SACN,IAAA;EAAA,SACA,OAAA;AAAA;;;AAQX;;;cAAa,uBAAA,SAAgC,WAAA;EAAA,SAClC,IAAA;EAAA,SACA,UAAA;EAAA,SACA,WAAA;EAAA,SACA,MAAA,WAAiB,uBAAA;cAGxB,OAAA,UACA,MAAA,YAAiB,uBAAA;AAAA"}

package/dist/errors/database-validation.js

@@ -0,0 +1,24 @@
+import { AppKitError } from "./base.js";
+
+//#region src/errors/database-validation.ts
+/** Cap how many issues one rejection can put on the wire. */
+const MAX_ISSUES = 50;
+/**
+* Deliberate validation failure raised by a database mutation hook. Generated
+* routes answer `422` and echo only the issues naming a public column; every
+* other failure raised inside a hook stays an opaque server error.
+*/
+var DatabaseValidationError = class extends AppKitError {
+	code = "DATABASE_VALIDATION_ERROR";
+	statusCode = 422;
+	isRetryable = false;
+	issues;
+	constructor(message, issues = []) {
+		super(message, { clientMessage: "Database request failed validation" });
+		this.issues = issues.slice(0, MAX_ISSUES);
+	}
+};
+
+//#endregion
+export { DatabaseValidationError };
+//# sourceMappingURL=database-validation.js.map

package/dist/errors/database-validation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"database-validation.js","names":[],"sources":["../../src/errors/database-validation.ts"],"sourcesContent":["import { AppKitError } from \"./base\";\n\n/** Cap how many issues one rejection can put on the wire. */\nconst MAX_ISSUES = 50;\n\n/** One rejected field; `path` names public columns, never their values. */\nexport interface DatabaseValidationIssue {\n  readonly path: readonly string[];\n  readonly message: string;\n}\n\n/**\n * Deliberate validation failure raised by a database mutation hook. Generated\n * routes answer `422` and echo only the issues naming a public column; every\n * other failure raised inside a hook stays an opaque server error.\n */\nexport class DatabaseValidationError extends AppKitError {\n  readonly code = \"DATABASE_VALIDATION_ERROR\";\n  readonly statusCode = 422;\n  readonly isRetryable = false;\n  readonly issues: readonly DatabaseValidationIssue[];\n\n  constructor(\n    message: string,\n    issues: readonly DatabaseValidationIssue[] = [],\n  ) {\n    super(message, { clientMessage: \"Database request failed validation\" });\n    this.issues = issues.slice(0, MAX_ISSUES);\n  }\n}\n"],"mappings":";;;;AAGA,MAAM,aAAa;;;;;;AAanB,IAAa,0BAAb,cAA6C,YAAY;CACvD,AAAS,OAAO;CAChB,AAAS,aAAa;CACtB,AAAS,cAAc;CACvB,AAAS;CAET,YACE,SACA,SAA6C,EAAE,EAC/C;AACA,QAAM,SAAS,EAAE,eAAe,sCAAsC,CAAC;AACvE,OAAK,SAAS,OAAO,MAAM,GAAG,WAAW"}

package/dist/errors/index.js

@@ -2,6 +2,7 @@ import { AppKitError } from "./base.js";
 import { AuthenticationError } from "./authentication.js";
 import { ConfigurationError } from "./configuration.js";
 import { ConnectionError } from "./connection.js";
+import { DatabaseValidationError } from "./database-validation.js";
 import { ExecutionError } from "./execution.js";
 import { InitializationError } from "./initialization.js";
 import { ServerError } from "./server.js";

package/dist/evals/dataset.d.ts

@@ -0,0 +1,36 @@
+import { WorkspaceClient } from "../workspace-client/index.js";
+
+//#region src/evals/dataset.d.ts
+/**
+ * One row of a managed evaluation dataset. `inputs` are the kwargs passed to the
+ * agent for the turn; `expectations` (when present) is the row's ground truth /
+ * guidelines. Mirrors the `{inputs, expectations}` shape of `mlflow.genai`
+ * datasets and of the Unity Catalog table backing a managed eval dataset.
+ */
+interface DatasetRow {
+  inputs: Record<string, unknown>;
+  expectations?: Record<string, unknown>;
+}
+interface ReadEvalDatasetOptions {
+  /** Fully-qualified UC table: `catalog.schema.table`. */
+  table: string;
+  /** SQL warehouse id to run the read against. */
+  warehouseId: string;
+  /** Optional row cap. */
+  limit?: number;
+}
+/**
+ * Read a Databricks managed evaluation dataset (a Unity Catalog table with
+ * `inputs`/`expectations` columns) into rows, over the public SQL Statement
+ * Execution API. Reuses {@link SQLWarehouseConnector} for submit/poll/transform
+ * — its result transform already JSON-parses string columns into objects, so
+ * `inputs`/`expectations` come back as records whether the table stores them as
+ * JSON strings or structs.
+ *
+ * The Python `mlflow.genai.datasets` API needs a Spark session (no TS
+ * equivalent), so we read the backing table directly.
+ */
+declare function readEvalDataset(client: WorkspaceClient, options: ReadEvalDatasetOptions): Promise<DatasetRow[]>;
+//#endregion
+export { DatasetRow, ReadEvalDatasetOptions, readEvalDataset };
+//# sourceMappingURL=dataset.d.ts.map

package/dist/evals/dataset.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dataset.d.ts","names":[],"sources":["../../src/evals/dataset.ts"],"mappings":";;;;;AASA;;;;UAAiB,UAAA;EACf,MAAA,EAAQ,MAAA;EACR,YAAA,GAAe,MAAA;AAAA;AAAA,UAGA,sBAAA;EAHM;EAKrB,KAAA;EAFqC;EAIrC,WAAA;EAJqC;EAMrC,KAAA;AAAA;;;;AAyBF;;;;;;;;iBAAsB,eAAA,CACpB,MAAA,EAAQ,eAAA,EACR,OAAA,EAAS,sBAAA,GACR,OAAA,CAAQ,UAAA"}

package/dist/evals/dataset.js

@@ -0,0 +1,36 @@
+import { SQLWarehouseConnector } from "../connectors/sql-warehouse/client.js";
+import "../connectors/index.js";
+
+//#region src/evals/dataset.ts
+/** A managed eval dataset is a UC table; only 3-level names are valid. */
+const UC_TABLE = /^[A-Za-z0-9_]+\.[A-Za-z0-9_]+\.[A-Za-z0-9_]+$/;
+/** Coerce a cell (already JSON-parsed by the connector for JSON columns) to a record. */
+function toRecord(value) {
+	if (value && typeof value === "object" && !Array.isArray(value)) return value;
+}
+/**
+* Read a Databricks managed evaluation dataset (a Unity Catalog table with
+* `inputs`/`expectations` columns) into rows, over the public SQL Statement
+* Execution API. Reuses {@link SQLWarehouseConnector} for submit/poll/transform
+* — its result transform already JSON-parses string columns into objects, so
+* `inputs`/`expectations` come back as records whether the table stores them as
+* JSON strings or structs.
+*
+* The Python `mlflow.genai.datasets` API needs a Spark session (no TS
+* equivalent), so we read the backing table directly.
+*/
+async function readEvalDataset(client, options) {
+	if (!UC_TABLE.test(options.table)) throw new Error(`Invalid dataset table "${options.table}" — expected catalog.schema.table`);
+	const limit = typeof options.limit === "number" && options.limit > 0 ? ` LIMIT ${Math.floor(options.limit)}` : "";
+	return ((await new SQLWarehouseConnector({}).executeStatement(client, {
+		warehouse_id: options.warehouseId,
+		statement: `SELECT inputs, expectations FROM ${options.table}${limit}`
+	})).result?.data ?? []).map((row) => ({
+		inputs: toRecord(row.inputs) ?? {},
+		expectations: toRecord(row.expectations)
+	}));
+}
+
+//#endregion
+export { readEvalDataset };
+//# sourceMappingURL=dataset.js.map

package/dist/evals/dataset.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dataset.js","names":[],"sources":["../../src/evals/dataset.ts"],"sourcesContent":["import { SQLWarehouseConnector } from \"../connectors\";\nimport type { WorkspaceClient } from \"../workspace-client\";\n\n/**\n * One row of a managed evaluation dataset. `inputs` are the kwargs passed to the\n * agent for the turn; `expectations` (when present) is the row's ground truth /\n * guidelines. Mirrors the `{inputs, expectations}` shape of `mlflow.genai`\n * datasets and of the Unity Catalog table backing a managed eval dataset.\n */\nexport interface DatasetRow {\n  inputs: Record<string, unknown>;\n  expectations?: Record<string, unknown>;\n}\n\nexport interface ReadEvalDatasetOptions {\n  /** Fully-qualified UC table: `catalog.schema.table`. */\n  table: string;\n  /** SQL warehouse id to run the read against. */\n  warehouseId: string;\n  /** Optional row cap. */\n  limit?: number;\n}\n\n/** A managed eval dataset is a UC table; only 3-level names are valid. */\nconst UC_TABLE = /^[A-Za-z0-9_]+\\.[A-Za-z0-9_]+\\.[A-Za-z0-9_]+$/;\n\n/** Coerce a cell (already JSON-parsed by the connector for JSON columns) to a record. */\nfunction toRecord(value: unknown): Record<string, unknown> | undefined {\n  if (value && typeof value === \"object\" && !Array.isArray(value)) {\n    return value as Record<string, unknown>;\n  }\n  return undefined;\n}\n\n/**\n * Read a Databricks managed evaluation dataset (a Unity Catalog table with\n * `inputs`/`expectations` columns) into rows, over the public SQL Statement\n * Execution API. Reuses {@link SQLWarehouseConnector} for submit/poll/transform\n * — its result transform already JSON-parses string columns into objects, so\n * `inputs`/`expectations` come back as records whether the table stores them as\n * JSON strings or structs.\n *\n * The Python `mlflow.genai.datasets` API needs a Spark session (no TS\n * equivalent), so we read the backing table directly.\n */\nexport async function readEvalDataset(\n  client: WorkspaceClient,\n  options: ReadEvalDatasetOptions,\n): Promise<DatasetRow[]> {\n  if (!UC_TABLE.test(options.table)) {\n    throw new Error(\n      `Invalid dataset table \"${options.table}\" — expected catalog.schema.table`,\n    );\n  }\n\n  const limit =\n    typeof options.limit === \"number\" && options.limit > 0\n      ? ` LIMIT ${Math.floor(options.limit)}`\n      : \"\";\n  const connector = new SQLWarehouseConnector({});\n  const response = await connector.executeStatement(client, {\n    warehouse_id: options.warehouseId,\n    statement: `SELECT inputs, expectations FROM ${options.table}${limit}`,\n  });\n\n  const rows =\n    (response.result as { data?: Array<Record<string, unknown>> } | undefined)\n      ?.data ?? [];\n\n  return rows.map((row) => ({\n    inputs: toRecord(row.inputs) ?? {},\n    expectations: toRecord(row.expectations),\n  }));\n}\n"],"mappings":";;;;;AAwBA,MAAM,WAAW;;AAGjB,SAAS,SAAS,OAAqD;AACrE,KAAI,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM,CAC7D,QAAO;;;;;;;;;;;;;AAgBX,eAAsB,gBACpB,QACA,SACuB;AACvB,KAAI,CAAC,SAAS,KAAK,QAAQ,MAAM,CAC/B,OAAM,IAAI,MACR,0BAA0B,QAAQ,MAAM,mCACzC;CAGH,MAAM,QACJ,OAAO,QAAQ,UAAU,YAAY,QAAQ,QAAQ,IACjD,UAAU,KAAK,MAAM,QAAQ,MAAM,KACnC;AAWN,UATiB,MADC,IAAI,sBAAsB,EAAE,CAAC,CACd,iBAAiB,QAAQ;EACxD,cAAc,QAAQ;EACtB,WAAW,oCAAoC,QAAQ,QAAQ;EAChE,CAAC,EAGU,QACN,QAAQ,EAAE,EAEJ,KAAK,SAAS;EACxB,QAAQ,SAAS,IAAI,OAAO,IAAI,EAAE;EAClC,cAAc,SAAS,IAAI,aAAa;EACzC,EAAE"}

package/dist/evals/http-driver.js

@@ -55,62 +55,67 @@ function createHttpDriver(options) {
 	const chatPath = options.path ?? "/api/agents/chat";
 	const timeoutMs = options.timeoutMs ?? 12e4;
 	let threadId;
-	return { async send(message) {
-		const signal = AbortSignal.timeout(timeoutMs);
-		let res;
-		try {
-			res = await fetch(`${options.baseUrl}${chatPath}`, {
-				method: "POST",
-				headers: {
-					"content-type": "application/json",
-					...options.headers
-				},
-				body: buildRequestBody(message, options, threadId),
-				redirect: "manual",
-				signal
-			});
-		} catch {
-			return {
+	return {
+		reset() {
+			threadId = void 0;
+		},
+		async send(message) {
+			const signal = AbortSignal.timeout(timeoutMs);
+			let res;
+			try {
+				res = await fetch(`${options.baseUrl}${chatPath}`, {
+					method: "POST",
+					headers: {
+						"content-type": "application/json",
+						...options.headers
+					},
+					body: buildRequestBody(message, options, threadId),
+					redirect: "manual",
+					signal
+				});
+			} catch {
+				return {
+					reply: "",
+					toolCalls: [],
+					succeeded: false
+				};
+			}
+			if (!res.ok || !res.body) return {
 				reply: "",
 				toolCalls: [],
-				succeeded: false
+				succeeded: false,
+				sessionId: threadId
 			};
-		}
-		if (!res.ok || !res.body) return {
-			reply: "",
-			toolCalls: [],
-			succeeded: false,
-			sessionId: threadId
-		};
-		const state = {
-			reply: "",
-			toolCalls: [],
-			seen: /* @__PURE__ */ new Set(),
-			ok: true
-		};
-		const setThread = (id) => {
-			threadId = id;
-		};
-		try {
-			for await (const { event, data } of readSseEvents(res.body, signal)) {
-				if (data === "[DONE]") continue;
-				if (event === "error") state.ok = false;
-				try {
-					applyEvent(JSON.parse(data), state, setThread);
-				} catch {}
+			const state = {
+				reply: "",
+				toolCalls: [],
+				seen: /* @__PURE__ */ new Set(),
+				ok: true
+			};
+			const setThread = (id) => {
+				threadId = id;
+			};
+			try {
+				for await (const { event, data } of readSseEvents(res.body, signal)) {
+					if (data === "[DONE]") continue;
+					if (event === "error") state.ok = false;
+					try {
+						applyEvent(JSON.parse(data), state, setThread);
+					} catch {}
+				}
+			} catch {
+				state.ok = false;
 			}
-		} catch {
-			state.ok = false;
+			if (signal.aborted) state.ok = false;
+			return {
+				reply: state.reply,
+				toolCalls: state.toolCalls,
+				succeeded: state.ok,
+				sessionId: threadId,
+				traceId: state.traceId
+			};
 		}
-		if (signal.aborted) state.ok = false;
-		return {
-			reply: state.reply,
-			toolCalls: state.toolCalls,
-			succeeded: state.ok,
-			sessionId: threadId,
-			traceId: state.traceId
-		};
-	} };
+	};
 }
 
 //#endregion

package/dist/evals/http-driver.js.map

@@ -1 +1 @@
-{"version":3,"file":"http-driver.js","names":[],"sources":["../../src/evals/http-driver.ts"],"sourcesContent":["import { readSseEvents } from \"../stream/sse-reader\";\nimport type { DriveResult, EvalDriver } from \"./types\";\n\nexport interface HttpDriverOptions {\n  /** Base URL of the running app, e.g. `http://localhost:3000`. */\n  baseUrl: string;\n  /** Agent alias to target. Omit to use the app's default agent. */\n  agent?: string;\n  /** Extra request headers (e.g. auth for a deployed app). */\n  headers?: Record<string, string>;\n  /** Chat endpoint path. Defaults to `/api/agents/chat`. */\n  path?: string;\n  /** MLflow run id to link each turn's trace to (for evaluation runs). */\n  mlflowRunId?: string;\n  /**\n   * Max wall-clock time for a single turn before it is abandoned as a failed\n   * turn (`succeeded: false`). Without this a hung agent — a blocked tool, a\n   * stalled model — never ends the SSE stream (heartbeats keep it alive), so\n   * the read loop spins forever and wedges the whole sequential suite.\n   * Defaults to 120s.\n   */\n  // ponytail: total per-turn cap; switch to an idle timeout if long legit turns get killed.\n  timeoutMs?: number;\n}\n\n/** Mutable running totals accumulated while draining one turn's SSE stream. */\ninterface DriveState {\n  reply: string;\n  toolCalls: string[];\n  seen: Set<string>;\n  ok: boolean;\n  traceId?: string;\n}\n\n/** Record a `function_call` output item once per call id (deduped). */\nfunction recordToolCall(\n  item: { type?: string; name?: string; call_id?: string } | undefined,\n  state: DriveState,\n): void {\n  if (item?.type !== \"function_call\" || !item.name) return;\n  const key = item.call_id ?? item.name;\n  if (state.seen.has(key)) return;\n  state.seen.add(key);\n  state.toolCalls.push(item.name);\n}\n\n/** Apply an `appkit.metadata` event's thread/trace ids. */\nfunction applyMetadata(\n  data: { threadId?: string; mlflowTraceId?: string } | undefined,\n  state: DriveState,\n  setThread: (id: string) => void,\n): void {\n  if (data?.threadId) setThread(data.threadId);\n  if (data?.mlflowTraceId) state.traceId = data.mlflowTraceId;\n}\n\n/** Parse a single Responses-API SSE `data:` payload into the running totals. */\nfunction applyEvent(\n  event: Record<string, unknown>,\n  state: DriveState,\n  setThread: (id: string) => void,\n): void {\n  const type = event.type;\n  if (type === \"response.output_text.delta\") {\n    if (typeof event.delta === \"string\") state.reply += event.delta;\n    return;\n  }\n  if (\n    type === \"response.output_item.added\" ||\n    type === \"response.output_item.done\"\n  ) {\n    const item = event.item as {\n      type?: string;\n      name?: string;\n      call_id?: string;\n      content?: Array<{ text?: string }>;\n    };\n    recordToolCall(item, state);\n    // A terminal `message` item carries the full reply text and *replaces*\n    // the accumulated deltas — the server emits no delta for a full message\n    // (event-translator `handleFullMessage`), so a non-streaming adapter's\n    // whole reply arrives only here. Guard on non-empty so a streaming turn's\n    // trailing done event can't clobber the deltas with \"\".\n    if (type === \"response.output_item.done\" && item.type === \"message\") {\n      const text = (item.content ?? []).map((c) => c.text ?? \"\").join(\"\");\n      if (text) state.reply = text;\n    }\n    return;\n  }\n  if (type === \"error\" || type === \"response.failed\") {\n    state.ok = false;\n    return;\n  }\n  if (type === \"appkit.metadata\") {\n    applyMetadata(\n      event.data as { threadId?: string; mlflowTraceId?: string },\n      state,\n      setThread,\n    );\n  }\n}\n\n/** Build the chat request payload, including only the optional fields that are set. */\nfunction buildRequestBody(\n  message: string,\n  options: HttpDriverOptions,\n  threadId: string | undefined,\n): string {\n  return JSON.stringify({\n    message,\n    ...(options.agent ? { agent: options.agent } : {}),\n    ...(threadId ? { threadId } : {}),\n    ...(options.mlflowRunId ? { mlflowRunId: options.mlflowRunId } : {}),\n  });\n}\n\n/**\n * Drives an agent by POSTing to a running app's chat endpoint and parsing the\n * SSE response. Keeps the thread id across `send`s so multi-turn evals share a\n * conversation. Agent/stream errors surface as `succeeded: false` rather than\n * throwing, so `t.succeeded()` can assert on them.\n */\nexport function createHttpDriver(options: HttpDriverOptions): EvalDriver {\n  const chatPath = options.path ?? \"/api/agents/chat\";\n  const timeoutMs = options.timeoutMs ?? 120_000;\n  let threadId: string | undefined;\n\n  return {\n    async send(message: string): Promise<DriveResult> {\n      // Bounds connect + the entire read below. Passed to both the fetch and\n      // the SSE reader: on expiry the reader is cancelled and the turn fails.\n      const signal = AbortSignal.timeout(timeoutMs);\n      let res: Response;\n      try {\n        res = await fetch(`${options.baseUrl}${chatPath}`, {\n          method: \"POST\",\n          headers: { \"content-type\": \"application/json\", ...options.headers },\n          body: buildRequestBody(message, options, threadId),\n          // A chat endpoint never needs a redirect; following one would replay\n          // custom auth headers (`options.headers`) to the redirect target.\n          // A 3xx becomes an opaque response (res.ok === false), handled as a\n          // failed turn by the !res.ok guard below.\n          redirect: \"manual\",\n          signal,\n        });\n      } catch {\n        return { reply: \"\", toolCalls: [], succeeded: false };\n      }\n\n      if (!res.ok || !res.body) {\n        return {\n          reply: \"\",\n          toolCalls: [],\n          succeeded: false,\n          sessionId: threadId,\n        };\n      }\n\n      const state: DriveState = {\n        reply: \"\",\n        toolCalls: [],\n        seen: new Set<string>(),\n        ok: true,\n      };\n      const setThread = (id: string) => {\n        threadId = id;\n      };\n      try {\n        for await (const { event, data } of readSseEvents(res.body, signal)) {\n          if (data === \"[DONE]\") continue;\n          // A stream-level error frame (a thrown exception in the generator)\n          // is framed as `event: error` with a payload that carries no `type`,\n          // so the SSE event name — not the payload — is the real signal.\n          if (event === \"error\") state.ok = false;\n          try {\n            applyEvent(\n              JSON.parse(data) as Record<string, unknown>,\n              state,\n              setThread,\n            );\n          } catch {\n            // skip malformed event payloads\n          }\n        }\n      } catch {\n        // mid-stream transport error or DoS-guard throw: a broken turn.\n        state.ok = false;\n      }\n      // A timeout ends the iterator cleanly (reader cancel) rather than\n      // throwing, so mark an aborted turn failed explicitly.\n      if (signal.aborted) state.ok = false;\n\n      return {\n        reply: state.reply,\n        toolCalls: state.toolCalls,\n        succeeded: state.ok,\n        sessionId: threadId,\n        traceId: state.traceId,\n      };\n    },\n  };\n}\n"],"mappings":";;;;AAmCA,SAAS,eACP,MACA,OACM;AACN,KAAI,MAAM,SAAS,mBAAmB,CAAC,KAAK,KAAM;CAClD,MAAM,MAAM,KAAK,WAAW,KAAK;AACjC,KAAI,MAAM,KAAK,IAAI,IAAI,CAAE;AACzB,OAAM,KAAK,IAAI,IAAI;AACnB,OAAM,UAAU,KAAK,KAAK,KAAK;;;AAIjC,SAAS,cACP,MACA,OACA,WACM;AACN,KAAI,MAAM,SAAU,WAAU,KAAK,SAAS;AAC5C,KAAI,MAAM,cAAe,OAAM,UAAU,KAAK;;;AAIhD,SAAS,WACP,OACA,OACA,WACM;CACN,MAAM,OAAO,MAAM;AACnB,KAAI,SAAS,8BAA8B;AACzC,MAAI,OAAO,MAAM,UAAU,SAAU,OAAM,SAAS,MAAM;AAC1D;;AAEF,KACE,SAAS,gCACT,SAAS,6BACT;EACA,MAAM,OAAO,MAAM;AAMnB,iBAAe,MAAM,MAAM;AAM3B,MAAI,SAAS,+BAA+B,KAAK,SAAS,WAAW;GACnE,MAAM,QAAQ,KAAK,WAAW,EAAE,EAAE,KAAK,MAAM,EAAE,QAAQ,GAAG,CAAC,KAAK,GAAG;AACnE,OAAI,KAAM,OAAM,QAAQ;;AAE1B;;AAEF,KAAI,SAAS,WAAW,SAAS,mBAAmB;AAClD,QAAM,KAAK;AACX;;AAEF,KAAI,SAAS,kBACX,eACE,MAAM,MACN,OACA,UACD;;;AAKL,SAAS,iBACP,SACA,SACA,UACQ;AACR,QAAO,KAAK,UAAU;EACpB;EACA,GAAI,QAAQ,QAAQ,EAAE,OAAO,QAAQ,OAAO,GAAG,EAAE;EACjD,GAAI,WAAW,EAAE,UAAU,GAAG,EAAE;EAChC,GAAI,QAAQ,cAAc,EAAE,aAAa,QAAQ,aAAa,GAAG,EAAE;EACpE,CAAC;;;;;;;;AASJ,SAAgB,iBAAiB,SAAwC;CACvE,MAAM,WAAW,QAAQ,QAAQ;CACjC,MAAM,YAAY,QAAQ,aAAa;CACvC,IAAI;AAEJ,QAAO,EACL,MAAM,KAAK,SAAuC;EAGhD,MAAM,SAAS,YAAY,QAAQ,UAAU;EAC7C,IAAI;AACJ,MAAI;AACF,SAAM,MAAM,MAAM,GAAG,QAAQ,UAAU,YAAY;IACjD,QAAQ;IACR,SAAS;KAAE,gBAAgB;KAAoB,GAAG,QAAQ;KAAS;IACnE,MAAM,iBAAiB,SAAS,SAAS,SAAS;IAKlD,UAAU;IACV;IACD,CAAC;UACI;AACN,UAAO;IAAE,OAAO;IAAI,WAAW,EAAE;IAAE,WAAW;IAAO;;AAGvD,MAAI,CAAC,IAAI,MAAM,CAAC,IAAI,KAClB,QAAO;GACL,OAAO;GACP,WAAW,EAAE;GACb,WAAW;GACX,WAAW;GACZ;EAGH,MAAM,QAAoB;GACxB,OAAO;GACP,WAAW,EAAE;GACb,sBAAM,IAAI,KAAa;GACvB,IAAI;GACL;EACD,MAAM,aAAa,OAAe;AAChC,cAAW;;AAEb,MAAI;AACF,cAAW,MAAM,EAAE,OAAO,UAAU,cAAc,IAAI,MAAM,OAAO,EAAE;AACnE,QAAI,SAAS,SAAU;AAIvB,QAAI,UAAU,QAAS,OAAM,KAAK;AAClC,QAAI;AACF,gBACE,KAAK,MAAM,KAAK,EAChB,OACA,UACD;YACK;;UAIJ;AAEN,SAAM,KAAK;;AAIb,MAAI,OAAO,QAAS,OAAM,KAAK;AAE/B,SAAO;GACL,OAAO,MAAM;GACb,WAAW,MAAM;GACjB,WAAW,MAAM;GACjB,WAAW;GACX,SAAS,MAAM;GAChB;IAEJ"}
+{"version":3,"file":"http-driver.js","names":[],"sources":["../../src/evals/http-driver.ts"],"sourcesContent":["import { readSseEvents } from \"../stream/sse-reader\";\nimport type { DriveResult, EvalDriver } from \"./types\";\n\nexport interface HttpDriverOptions {\n  /** Base URL of the running app, e.g. `http://localhost:3000`. */\n  baseUrl: string;\n  /** Agent alias to target. Omit to use the app's default agent. */\n  agent?: string;\n  /** Extra request headers (e.g. auth for a deployed app). */\n  headers?: Record<string, string>;\n  /** Chat endpoint path. Defaults to `/api/agents/chat`. */\n  path?: string;\n  /** MLflow run id to link each turn's trace to (for evaluation runs). */\n  mlflowRunId?: string;\n  /**\n   * Max wall-clock time for a single turn before it is abandoned as a failed\n   * turn (`succeeded: false`). Without this a hung agent — a blocked tool, a\n   * stalled model — never ends the SSE stream (heartbeats keep it alive), so\n   * the read loop spins forever and wedges the whole sequential suite.\n   * Defaults to 120s.\n   */\n  // ponytail: total per-turn cap; switch to an idle timeout if long legit turns get killed.\n  timeoutMs?: number;\n}\n\n/** Mutable running totals accumulated while draining one turn's SSE stream. */\ninterface DriveState {\n  reply: string;\n  toolCalls: string[];\n  seen: Set<string>;\n  ok: boolean;\n  traceId?: string;\n}\n\n/** Record a `function_call` output item once per call id (deduped). */\nfunction recordToolCall(\n  item: { type?: string; name?: string; call_id?: string } | undefined,\n  state: DriveState,\n): void {\n  if (item?.type !== \"function_call\" || !item.name) return;\n  const key = item.call_id ?? item.name;\n  if (state.seen.has(key)) return;\n  state.seen.add(key);\n  state.toolCalls.push(item.name);\n}\n\n/** Apply an `appkit.metadata` event's thread/trace ids. */\nfunction applyMetadata(\n  data: { threadId?: string; mlflowTraceId?: string } | undefined,\n  state: DriveState,\n  setThread: (id: string) => void,\n): void {\n  if (data?.threadId) setThread(data.threadId);\n  if (data?.mlflowTraceId) state.traceId = data.mlflowTraceId;\n}\n\n/** Parse a single Responses-API SSE `data:` payload into the running totals. */\nfunction applyEvent(\n  event: Record<string, unknown>,\n  state: DriveState,\n  setThread: (id: string) => void,\n): void {\n  const type = event.type;\n  if (type === \"response.output_text.delta\") {\n    if (typeof event.delta === \"string\") state.reply += event.delta;\n    return;\n  }\n  if (\n    type === \"response.output_item.added\" ||\n    type === \"response.output_item.done\"\n  ) {\n    const item = event.item as {\n      type?: string;\n      name?: string;\n      call_id?: string;\n      content?: Array<{ text?: string }>;\n    };\n    recordToolCall(item, state);\n    // A terminal `message` item carries the full reply text and *replaces*\n    // the accumulated deltas — the server emits no delta for a full message\n    // (event-translator `handleFullMessage`), so a non-streaming adapter's\n    // whole reply arrives only here. Guard on non-empty so a streaming turn's\n    // trailing done event can't clobber the deltas with \"\".\n    if (type === \"response.output_item.done\" && item.type === \"message\") {\n      const text = (item.content ?? []).map((c) => c.text ?? \"\").join(\"\");\n      if (text) state.reply = text;\n    }\n    return;\n  }\n  if (type === \"error\" || type === \"response.failed\") {\n    state.ok = false;\n    return;\n  }\n  if (type === \"appkit.metadata\") {\n    applyMetadata(\n      event.data as { threadId?: string; mlflowTraceId?: string },\n      state,\n      setThread,\n    );\n  }\n}\n\n/** Build the chat request payload, including only the optional fields that are set. */\nfunction buildRequestBody(\n  message: string,\n  options: HttpDriverOptions,\n  threadId: string | undefined,\n): string {\n  return JSON.stringify({\n    message,\n    ...(options.agent ? { agent: options.agent } : {}),\n    ...(threadId ? { threadId } : {}),\n    ...(options.mlflowRunId ? { mlflowRunId: options.mlflowRunId } : {}),\n  });\n}\n\n/**\n * Drives an agent by POSTing to a running app's chat endpoint and parsing the\n * SSE response. Keeps the thread id across `send`s so multi-turn evals share a\n * conversation. Agent/stream errors surface as `succeeded: false` rather than\n * throwing, so `t.succeeded()` can assert on them.\n */\nexport function createHttpDriver(options: HttpDriverOptions): EvalDriver {\n  const chatPath = options.path ?? \"/api/agents/chat\";\n  const timeoutMs = options.timeoutMs ?? 120_000;\n  let threadId: string | undefined;\n\n  return {\n    reset(): void {\n      threadId = undefined;\n    },\n    async send(message: string): Promise<DriveResult> {\n      // Bounds connect + the entire read below. Passed to both the fetch and\n      // the SSE reader: on expiry the reader is cancelled and the turn fails.\n      const signal = AbortSignal.timeout(timeoutMs);\n      let res: Response;\n      try {\n        res = await fetch(`${options.baseUrl}${chatPath}`, {\n          method: \"POST\",\n          headers: { \"content-type\": \"application/json\", ...options.headers },\n          body: buildRequestBody(message, options, threadId),\n          // A chat endpoint never needs a redirect; following one would replay\n          // custom auth headers (`options.headers`) to the redirect target.\n          // A 3xx becomes an opaque response (res.ok === false), handled as a\n          // failed turn by the !res.ok guard below.\n          redirect: \"manual\",\n          signal,\n        });\n      } catch {\n        return { reply: \"\", toolCalls: [], succeeded: false };\n      }\n\n      if (!res.ok || !res.body) {\n        return {\n          reply: \"\",\n          toolCalls: [],\n          succeeded: false,\n          sessionId: threadId,\n        };\n      }\n\n      const state: DriveState = {\n        reply: \"\",\n        toolCalls: [],\n        seen: new Set<string>(),\n        ok: true,\n      };\n      const setThread = (id: string) => {\n        threadId = id;\n      };\n      try {\n        for await (const { event, data } of readSseEvents(res.body, signal)) {\n          if (data === \"[DONE]\") continue;\n          // A stream-level error frame (a thrown exception in the generator)\n          // is framed as `event: error` with a payload that carries no `type`,\n          // so the SSE event name — not the payload — is the real signal.\n          if (event === \"error\") state.ok = false;\n          try {\n            applyEvent(\n              JSON.parse(data) as Record<string, unknown>,\n              state,\n              setThread,\n            );\n          } catch {\n            // skip malformed event payloads\n          }\n        }\n      } catch {\n        // mid-stream transport error or DoS-guard throw: a broken turn.\n        state.ok = false;\n      }\n      // A timeout ends the iterator cleanly (reader cancel) rather than\n      // throwing, so mark an aborted turn failed explicitly.\n      if (signal.aborted) state.ok = false;\n\n      return {\n        reply: state.reply,\n        toolCalls: state.toolCalls,\n        succeeded: state.ok,\n        sessionId: threadId,\n        traceId: state.traceId,\n      };\n    },\n  };\n}\n"],"mappings":";;;;AAmCA,SAAS,eACP,MACA,OACM;AACN,KAAI,MAAM,SAAS,mBAAmB,CAAC,KAAK,KAAM;CAClD,MAAM,MAAM,KAAK,WAAW,KAAK;AACjC,KAAI,MAAM,KAAK,IAAI,IAAI,CAAE;AACzB,OAAM,KAAK,IAAI,IAAI;AACnB,OAAM,UAAU,KAAK,KAAK,KAAK;;;AAIjC,SAAS,cACP,MACA,OACA,WACM;AACN,KAAI,MAAM,SAAU,WAAU,KAAK,SAAS;AAC5C,KAAI,MAAM,cAAe,OAAM,UAAU,KAAK;;;AAIhD,SAAS,WACP,OACA,OACA,WACM;CACN,MAAM,OAAO,MAAM;AACnB,KAAI,SAAS,8BAA8B;AACzC,MAAI,OAAO,MAAM,UAAU,SAAU,OAAM,SAAS,MAAM;AAC1D;;AAEF,KACE,SAAS,gCACT,SAAS,6BACT;EACA,MAAM,OAAO,MAAM;AAMnB,iBAAe,MAAM,MAAM;AAM3B,MAAI,SAAS,+BAA+B,KAAK,SAAS,WAAW;GACnE,MAAM,QAAQ,KAAK,WAAW,EAAE,EAAE,KAAK,MAAM,EAAE,QAAQ,GAAG,CAAC,KAAK,GAAG;AACnE,OAAI,KAAM,OAAM,QAAQ;;AAE1B;;AAEF,KAAI,SAAS,WAAW,SAAS,mBAAmB;AAClD,QAAM,KAAK;AACX;;AAEF,KAAI,SAAS,kBACX,eACE,MAAM,MACN,OACA,UACD;;;AAKL,SAAS,iBACP,SACA,SACA,UACQ;AACR,QAAO,KAAK,UAAU;EACpB;EACA,GAAI,QAAQ,QAAQ,EAAE,OAAO,QAAQ,OAAO,GAAG,EAAE;EACjD,GAAI,WAAW,EAAE,UAAU,GAAG,EAAE;EAChC,GAAI,QAAQ,cAAc,EAAE,aAAa,QAAQ,aAAa,GAAG,EAAE;EACpE,CAAC;;;;;;;;AASJ,SAAgB,iBAAiB,SAAwC;CACvE,MAAM,WAAW,QAAQ,QAAQ;CACjC,MAAM,YAAY,QAAQ,aAAa;CACvC,IAAI;AAEJ,QAAO;EACL,QAAc;AACZ,cAAW;;EAEb,MAAM,KAAK,SAAuC;GAGhD,MAAM,SAAS,YAAY,QAAQ,UAAU;GAC7C,IAAI;AACJ,OAAI;AACF,UAAM,MAAM,MAAM,GAAG,QAAQ,UAAU,YAAY;KACjD,QAAQ;KACR,SAAS;MAAE,gBAAgB;MAAoB,GAAG,QAAQ;MAAS;KACnE,MAAM,iBAAiB,SAAS,SAAS,SAAS;KAKlD,UAAU;KACV;KACD,CAAC;WACI;AACN,WAAO;KAAE,OAAO;KAAI,WAAW,EAAE;KAAE,WAAW;KAAO;;AAGvD,OAAI,CAAC,IAAI,MAAM,CAAC,IAAI,KAClB,QAAO;IACL,OAAO;IACP,WAAW,EAAE;IACb,WAAW;IACX,WAAW;IACZ;GAGH,MAAM,QAAoB;IACxB,OAAO;IACP,WAAW,EAAE;IACb,sBAAM,IAAI,KAAa;IACvB,IAAI;IACL;GACD,MAAM,aAAa,OAAe;AAChC,eAAW;;AAEb,OAAI;AACF,eAAW,MAAM,EAAE,OAAO,UAAU,cAAc,IAAI,MAAM,OAAO,EAAE;AACnE,SAAI,SAAS,SAAU;AAIvB,SAAI,UAAU,QAAS,OAAM,KAAK;AAClC,SAAI;AACF,iBACE,KAAK,MAAM,KAAK,EAChB,OACA,UACD;aACK;;WAIJ;AAEN,UAAM,KAAK;;AAIb,OAAI,OAAO,QAAS,OAAM,KAAK;AAE/B,UAAO;IACL,OAAO,MAAM;IACb,WAAW,MAAM;IACjB,WAAW,MAAM;IACjB,WAAW;IACX,SAAS,MAAM;IAChB;;EAEJ"}

package/dist/evals/index.d.ts

@@ -0,0 +1,14 @@
+import { DatabricksAuth, ResolveDatabricksAuthOptions, resolveDatabricksAuth, resolveWorkspaceClient } from "../connectors/mlflow/auth.js";
+import { MlflowClient, PostResult, normalizeHost } from "../connectors/mlflow/client.js";
+import "../connectors/mlflow/index.js";
+import { DatasetRow, ReadEvalDatasetOptions, readEvalDataset } from "./dataset.js";
+import { AssertionHandle, AssertionResult, CustomJudgeSpec, DriveResult, EvalDefinition, EvalDriver, EvalResult, MatchResult, Matcher, Severity, TestContext } from "./types.js";
+import { defineEval } from "./define-eval.js";
+import { DiscoveredEval, discoverEvalFiles } from "./discover.js";
+import { HttpDriverOptions, createHttpDriver } from "./http-driver.js";
+import { JudgeConfig, JudgeScore, configureJudge, isJudgeConfigured } from "./judge.js";
+import { equals, includes, matches } from "./matchers.js";
+import { Assessment, ReportOutcome, buildAssessments, reportToMlflow } from "./mlflow-report.js";
+import { EvalSummary, evalGlyph, formatEvalDetail, formatEvalHeadline, formatEvalResults, formatSummaryLine, summarize } from "./report.js";
+import { RunEvalOptions, runEval } from "./run-eval.js";
+import { EvalProgress, EvalRunSummary, RunEvalsOptions, runEvalsInDir } from "./run-evals.js";

package/dist/evals/index.js

@@ -1,5 +1,6 @@
-import { resolveDatabricksAuth } from "../connectors/mlflow/auth.js";
+import { resolveDatabricksAuth, resolveWorkspaceClient } from "../connectors/mlflow/auth.js";
 import { MlflowClient, normalizeHost } from "../connectors/mlflow/client.js";
+import { readEvalDataset } from "./dataset.js";
 import { defineEval } from "./define-eval.js";
 import { discoverEvalFiles } from "./discover.js";
 import { createHttpDriver } from "./http-driver.js";

package/dist/evals/judge.d.ts

@@ -1,4 +1,5 @@
 import { MlflowClient } from "../connectors/mlflow/client.js";
+import "../connectors/mlflow/index.js";
 
 //#region src/evals/judge.d.ts
 interface JudgeConfig {

package/dist/evals/judge.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"judge.d.ts","names":[],"sources":["../../src/evals/judge.ts"],"mappings":";;;UAuBiB,WAAA;;EAEf,MAAA,EAAQ,YAAA;EAFkB;EAI1B,KAAA;EAFoB;EAIpB,KAAA;AAAA;;UAIe,UAAA;EACf,KAAA;EACA,SAAA;AAAA;;;;;AAQF;iBAAsB,cAAA,CAAe,MAAA,EAAQ,WAAA,GAAc,OAAA;AAAA,iBAe3C,iBAAA,CAAA"}
+{"version":3,"file":"judge.d.ts","names":[],"sources":["../../src/evals/judge.ts"],"mappings":";;;;UAuBiB,WAAA;;EAEf,MAAA,EAAQ,YAAA;EAFO;EAIf,KAAA;;EAEA,KAAA;AAAA;;UAIe,UAAA;EACf,KAAA;EACA,SAAA;AAAA;AAFF;;;;;AAAA,iBAUsB,cAAA,CAAe,MAAA,EAAQ,WAAA,GAAc,OAAA;AAAA,iBAe3C,iBAAA,CAAA"}

package/dist/evals/mlflow-report.d.ts

@@ -1,4 +1,5 @@
 import { MlflowClient } from "../connectors/mlflow/client.js";
+import "../connectors/mlflow/index.js";
 import { EvalResult } from "./types.js";
 
 //#region src/evals/mlflow-report.d.ts

package/dist/evals/mlflow-report.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mlflow-report.d.ts","names":[],"sources":["../../src/evals/mlflow-report.ts"],"mappings":";;;;;UAqBiB,UAAA;EACf,QAAA;EACA,eAAA;EACA,MAAA;IAAU,WAAA;IAA6C,SAAA;EAAA;EACvD,QAAA;IAAY,KAAA;EAAA;EACZ,SAAA;EACA,QAAA,GAAW,MAAA;AAAA;AAAA,UAGI,aAAA;EACf,OAAA;EACA,OAAA;EACA,QAAA,EAAU,KAAA;IAAQ,OAAA;IAAiB,MAAA;IAAiB,KAAA;EAAA;AAAA;AAAA,iBA2DtC,gBAAA,CAAiB,MAAA,EAAQ,UAAA,GAAa,UAAA;;;;;iBA6FhC,cAAA,CACpB,MAAA,EAAQ,YAAA,EACR,OAAA,EAAS,UAAA,IACT,cAAA,YACC,OAAA,CAAQ,aAAA"}
+{"version":3,"file":"mlflow-report.d.ts","names":[],"sources":["../../src/evals/mlflow-report.ts"],"mappings":";;;;;;UAqBiB,UAAA;EACf,QAAA;EACA,eAAA;EACA,MAAA;IAAU,WAAA;IAA6C,SAAA;EAAA;EACvD,QAAA;IAAY,KAAA;EAAA;EACZ,SAAA;EACA,QAAA,GAAW,MAAA;AAAA;AAAA,UAGI,aAAA;EACf,OAAA;EACA,OAAA;EACA,QAAA,EAAU,KAAA;IAAQ,OAAA;IAAiB,MAAA;IAAiB,KAAA;EAAA;AAAA;AAAA,iBA2DtC,gBAAA,CAAiB,MAAA,EAAQ,UAAA,GAAa,UAAA;;;;;iBA6FhC,cAAA,CACpB,MAAA,EAAQ,YAAA,EACR,OAAA,EAAS,UAAA,IACT,cAAA,YACC,OAAA,CAAQ,aAAA"}

package/dist/evals/mlflow-run.d.ts

@@ -1,3 +1,5 @@
+import "../connectors/mlflow/index.js";
+
 //#region src/evals/mlflow-run.d.ts
 interface FinishOutcome {
   finished: boolean;

package/dist/evals/mlflow-run.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mlflow-run.d.ts","names":[],"sources":["../../src/evals/mlflow-run.ts"],"mappings":";UA6EiB,aAAA;EACf,QAAA;;EAEA,YAAA;;EAEA,WAAA;AAAA"}
+{"version":3,"file":"mlflow-run.d.ts","names":[],"sources":["../../src/evals/mlflow-run.ts"],"mappings":";;;UA6EiB,aAAA;EACf,QAAA;;EAEA,YAAA;;EAEA,WAAA;AAAA"}

package/dist/evals/run-eval.d.ts

@@ -1,3 +1,4 @@
+import { DatasetRow } from "./dataset.js";
 import { EvalDefinition, EvalDriver, EvalResult } from "./types.js";
 
 //#region src/evals/run-eval.d.ts
@@ -8,6 +9,8 @@ interface RunEvalOptions {
   driver: EvalDriver;
   /** When true, soft assertion failures also fail the eval. */
   strict?: boolean;
+  /** Dataset row bound to `t.input`/`t.expected` for dataset-driven evals. */
+  row?: DatasetRow;
 }
 /**
  * Run a single eval against a driver. Never throws for assertion or agent