@llmops/core 0.2.11 → 0.2.12-beta.2

package/dist/{bun-sqlite-dialect-t1xHHJDz.cjs → bun-sqlite-dialect-Bxz4B97L.cjs}

@@ -1,4 +1,4 @@
-const require_db = require('./db-CqDD6Ydv.cjs');
+const require_db = require('./db-BOe6mM51.cjs');
 let kysely = require("kysely");
 
 //#region src/db/bun-sqlite-dialect.ts

package/dist/db/index.cjs

@@ -1,13 +1,16 @@
-const require_db = require('../db-CqDD6Ydv.cjs');
+const require_db = require('../db-BOe6mM51.cjs');
+const require_neon_dialect = require('../neon-dialect-BQey5lUw.cjs');
 
 exports.SCHEMA_METADATA = require_db.SCHEMA_METADATA;
 exports.configVariantsSchema = require_db.configVariantsSchema;
 exports.configsSchema = require_db.configsSchema;
 exports.createDatabase = require_db.createDatabase;
 exports.createDatabaseFromConnection = require_db.createDatabaseFromConnection;
+exports.createNeonDialect = require_neon_dialect.createNeonDialect;
 exports.detectDatabaseType = require_db.detectDatabaseType;
 exports.environmentSecretsSchema = require_db.environmentSecretsSchema;
 exports.environmentsSchema = require_db.environmentsSchema;
+exports.executeWithSchema = require_neon_dialect.executeWithSchema;
 exports.getMigrations = require_db.getMigrations;
 exports.llmRequestsSchema = require_db.llmRequestsSchema;
 exports.matchType = require_db.matchType;

package/dist/db/index.d.cts

@@ -1,2 +1,2 @@
-import { A as SCHEMA_METADATA, B as WorkspaceSettings, C as EnvironmentSecretsTable, D as LLMRequestsTable, E as LLMRequest, F as Updateable, G as environmentsSchema, H as configVariantsSchema, I as Variant, J as schemas, K as llmRequestsSchema, L as VariantVersion, M as TableName, N as TargetingRule, O as ProviderConfig, P as TargetingRulesTable, Q as workspaceSettingsSchema, R as VariantVersionsTable, S as EnvironmentSecret, T as Insertable, U as configsSchema, V as WorkspaceSettingsTable, W as environmentSecretsSchema, X as variantVersionsSchema, Y as targetingRulesSchema, Z as variantsSchema, _ as ConfigVariant, a as createDatabaseFromConnection, b as Database, c as MigrationResult, d as runAutoMigrations, f as parsePartialTableData, g as Config, h as validateTableData, i as createDatabase, j as Selectable, k as ProviderConfigsTable, l as getMigrations, m as validatePartialTableData, n as DatabaseOptions, o as detectDatabaseType, p as parseTableData, q as providerConfigsSchema, r as DatabaseType, s as MigrationOptions, t as DatabaseConnection, u as matchType, v as ConfigVariantsTable, w as EnvironmentsTable, x as Environment, y as ConfigsTable, z as VariantsTable } from "../index-BUlmWTh7.cjs";
-export { Config, ConfigVariant, ConfigVariantsTable, ConfigsTable, Database, DatabaseConnection, DatabaseOptions, DatabaseType, Environment, EnvironmentSecret, EnvironmentSecretsTable, EnvironmentsTable, Insertable, LLMRequest, LLMRequestsTable, MigrationOptions, MigrationResult, ProviderConfig, ProviderConfigsTable, SCHEMA_METADATA, Selectable, TableName, TargetingRule, TargetingRulesTable, Updateable, Variant, VariantVersion, VariantVersionsTable, VariantsTable, WorkspaceSettings, WorkspaceSettingsTable, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, detectDatabaseType, environmentSecretsSchema, environmentsSchema, getMigrations, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, providerConfigsSchema, runAutoMigrations, schemas, targetingRulesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
+import { $ as variantsSchema, A as ProviderConfig, B as VariantVersionsTable, C as Environment, D as Insertable, E as EnvironmentsTable, F as TargetingRule, G as configsSchema, H as WorkspaceSettings, I as TargetingRulesTable, J as llmRequestsSchema, K as environmentSecretsSchema, L as Updateable, M as SCHEMA_METADATA, N as Selectable, O as LLMRequest, P as TableName, Q as variantVersionsSchema, R as Variant, S as Database, T as EnvironmentSecretsTable, U as WorkspaceSettingsTable, V as VariantsTable, W as configVariantsSchema, X as schemas, Y as providerConfigsSchema, Z as targetingRulesSchema, _ as validateTableData, a as createDatabaseFromConnection, b as ConfigVariantsTable, c as executeWithSchema, d as getMigrations, et as workspaceSettingsSchema, f as matchType, g as validatePartialTableData, h as parseTableData, i as createDatabase, j as ProviderConfigsTable, k as LLMRequestsTable, l as MigrationOptions, m as parsePartialTableData, n as DatabaseOptions, o as detectDatabaseType, p as runAutoMigrations, q as environmentsSchema, r as DatabaseType, s as createNeonDialect, t as DatabaseConnection, u as MigrationResult, v as Config, w as EnvironmentSecret, x as ConfigsTable, y as ConfigVariant, z as VariantVersion } from "../index-DwZhaRvQ.cjs";
+export { Config, ConfigVariant, ConfigVariantsTable, ConfigsTable, Database, DatabaseConnection, DatabaseOptions, DatabaseType, Environment, EnvironmentSecret, EnvironmentSecretsTable, EnvironmentsTable, Insertable, LLMRequest, LLMRequestsTable, MigrationOptions, MigrationResult, ProviderConfig, ProviderConfigsTable, SCHEMA_METADATA, Selectable, TableName, TargetingRule, TargetingRulesTable, Updateable, Variant, VariantVersion, VariantVersionsTable, VariantsTable, WorkspaceSettings, WorkspaceSettingsTable, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, createNeonDialect, detectDatabaseType, environmentSecretsSchema, environmentsSchema, executeWithSchema, getMigrations, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, providerConfigsSchema, runAutoMigrations, schemas, targetingRulesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };

package/dist/db/index.d.mts

@@ -1,2 +1,2 @@
-import { A as SCHEMA_METADATA, B as WorkspaceSettings, C as EnvironmentSecretsTable, D as LLMRequestsTable, E as LLMRequest, F as Updateable, G as environmentsSchema, H as configVariantsSchema, I as Variant, J as schemas, K as llmRequestsSchema, L as VariantVersion, M as TableName, N as TargetingRule, O as ProviderConfig, P as TargetingRulesTable, Q as workspaceSettingsSchema, R as VariantVersionsTable, S as EnvironmentSecret, T as Insertable, U as configsSchema, V as WorkspaceSettingsTable, W as environmentSecretsSchema, X as variantVersionsSchema, Y as targetingRulesSchema, Z as variantsSchema, _ as ConfigVariant, a as createDatabaseFromConnection, b as Database, c as MigrationResult, d as runAutoMigrations, f as parsePartialTableData, g as Config, h as validateTableData, i as createDatabase, j as Selectable, k as ProviderConfigsTable, l as getMigrations, m as validatePartialTableData, n as DatabaseOptions, o as detectDatabaseType, p as parseTableData, q as providerConfigsSchema, r as DatabaseType, s as MigrationOptions, t as DatabaseConnection, u as matchType, v as ConfigVariantsTable, w as EnvironmentsTable, x as Environment, y as ConfigsTable, z as VariantsTable } from "../index-CkrtLH8W.mjs";
-export { Config, ConfigVariant, ConfigVariantsTable, ConfigsTable, Database, DatabaseConnection, DatabaseOptions, DatabaseType, Environment, EnvironmentSecret, EnvironmentSecretsTable, EnvironmentsTable, Insertable, LLMRequest, LLMRequestsTable, MigrationOptions, MigrationResult, ProviderConfig, ProviderConfigsTable, SCHEMA_METADATA, Selectable, TableName, TargetingRule, TargetingRulesTable, Updateable, Variant, VariantVersion, VariantVersionsTable, VariantsTable, WorkspaceSettings, WorkspaceSettingsTable, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, detectDatabaseType, environmentSecretsSchema, environmentsSchema, getMigrations, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, providerConfigsSchema, runAutoMigrations, schemas, targetingRulesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
+import { $ as variantsSchema, A as ProviderConfig, B as VariantVersionsTable, C as Environment, D as Insertable, E as EnvironmentsTable, F as TargetingRule, G as configsSchema, H as WorkspaceSettings, I as TargetingRulesTable, J as llmRequestsSchema, K as environmentSecretsSchema, L as Updateable, M as SCHEMA_METADATA, N as Selectable, O as LLMRequest, P as TableName, Q as variantVersionsSchema, R as Variant, S as Database, T as EnvironmentSecretsTable, U as WorkspaceSettingsTable, V as VariantsTable, W as configVariantsSchema, X as schemas, Y as providerConfigsSchema, Z as targetingRulesSchema, _ as validateTableData, a as createDatabaseFromConnection, b as ConfigVariantsTable, c as executeWithSchema, d as getMigrations, et as workspaceSettingsSchema, f as matchType, g as validatePartialTableData, h as parseTableData, i as createDatabase, j as ProviderConfigsTable, k as LLMRequestsTable, l as MigrationOptions, m as parsePartialTableData, n as DatabaseOptions, o as detectDatabaseType, p as runAutoMigrations, q as environmentsSchema, r as DatabaseType, s as createNeonDialect, t as DatabaseConnection, u as MigrationResult, v as Config, w as EnvironmentSecret, x as ConfigsTable, y as ConfigVariant, z as VariantVersion } from "../index-Bj69w0EU.mjs";
+export { Config, ConfigVariant, ConfigVariantsTable, ConfigsTable, Database, DatabaseConnection, DatabaseOptions, DatabaseType, Environment, EnvironmentSecret, EnvironmentSecretsTable, EnvironmentsTable, Insertable, LLMRequest, LLMRequestsTable, MigrationOptions, MigrationResult, ProviderConfig, ProviderConfigsTable, SCHEMA_METADATA, Selectable, TableName, TargetingRule, TargetingRulesTable, Updateable, Variant, VariantVersion, VariantVersionsTable, VariantsTable, WorkspaceSettings, WorkspaceSettingsTable, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, createNeonDialect, detectDatabaseType, environmentSecretsSchema, environmentsSchema, executeWithSchema, getMigrations, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, providerConfigsSchema, runAutoMigrations, schemas, targetingRulesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };

package/dist/db/index.mjs

@@ -1,3 +1,4 @@
-import { C as variantsSchema, S as variantVersionsSchema, _ as environmentsSchema, a as matchType, b as schemas, d as validatePartialTableData, f as validateTableData, g as environmentSecretsSchema, h as configsSchema, i as getMigrations, l as parsePartialTableData, m as configVariantsSchema, n as createDatabaseFromConnection, o as runAutoMigrations, p as SCHEMA_METADATA, r as detectDatabaseType, t as createDatabase, u as parseTableData, v as llmRequestsSchema, w as workspaceSettingsSchema, x as targetingRulesSchema, y as providerConfigsSchema } from "../db-BGDhJROC.mjs";
+import { C as variantsSchema, S as variantVersionsSchema, _ as environmentsSchema, a as matchType, b as schemas, d as validatePartialTableData, f as validateTableData, g as environmentSecretsSchema, h as configsSchema, i as getMigrations, l as parsePartialTableData, m as configVariantsSchema, n as createDatabaseFromConnection, o as runAutoMigrations, p as SCHEMA_METADATA, r as detectDatabaseType, t as createDatabase, u as parseTableData, v as llmRequestsSchema, w as workspaceSettingsSchema, x as targetingRulesSchema, y as providerConfigsSchema } from "../db-DohlAqJU.mjs";
+import { n as executeWithSchema, t as createNeonDialect } from "../neon-dialect-DySGBYUi.mjs";
 
-export { SCHEMA_METADATA, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, detectDatabaseType, environmentSecretsSchema, environmentsSchema, getMigrations, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, providerConfigsSchema, runAutoMigrations, schemas, targetingRulesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
+export { SCHEMA_METADATA, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, createNeonDialect, detectDatabaseType, environmentSecretsSchema, environmentsSchema, executeWithSchema, getMigrations, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, providerConfigsSchema, runAutoMigrations, schemas, targetingRulesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };

package/dist/{db-CqDD6Ydv.cjs → db-BOe6mM51.cjs}

@@ -13458,36 +13458,38 @@ const getAuthClientOptions = (options) => {
 
 //#endregion
 //#region src/db/migrations.ts
+const postgresMap = {
+	uuid: [
+		"character varying",
+		"varchar",
+		"text",
+		"uuid"
+	],
+	text: [
+		"character varying",
+		"varchar",
+		"text"
+	],
+	timestamp: [
+		"timestamptz",
+		"timestamp",
+		"date"
+	],
+	jsonb: ["json", "jsonb"],
+	integer: [
+		"integer",
+		"int4",
+		"int",
+		"smallint",
+		"bigint",
+		"int2",
+		"int8"
+	],
+	boolean: ["boolean", "bool"]
+};
 const typeMap = {
-	postgres: {
-		uuid: [
-			"character varying",
-			"varchar",
-			"text",
-			"uuid"
-		],
-		text: [
-			"character varying",
-			"varchar",
-			"text"
-		],
-		timestamp: [
-			"timestamptz",
-			"timestamp",
-			"date"
-		],
-		jsonb: ["json", "jsonb"],
-		integer: [
-			"integer",
-			"int4",
-			"int",
-			"smallint",
-			"bigint",
-			"int2",
-			"int8"
-		],
-		boolean: ["boolean", "bool"]
-	},
+	postgres: postgresMap,
+	neon: postgresMap,
 	mysql: {
 		text: ["varchar", "text"],
 		timestamp: [
@@ -13551,18 +13553,17 @@ async function ensurePostgresSchemaExists(db, schema) {
       ) as exists
     `.execute(db)).rows[0]?.exists) {
 			logger.info(`Creating PostgreSQL schema: ${schema}`);
-			await kysely.sql`CREATE SCHEMA IF NOT EXISTS ${kysely.sql.ref(schema)}`.execute(db);
+			await kysely.sql`CREATE SCHEMA IF NOT EXISTS ${kysely.sql.id(schema)}`.execute(db);
+			await kysely.sql`GRANT USAGE ON SCHEMA ${kysely.sql.id(schema)} TO PUBLIC`.execute(db);
+			await kysely.sql`GRANT CREATE ON SCHEMA ${kysely.sql.id(schema)} TO PUBLIC`.execute(db);
 		}
 	} catch (error$47) {
-		logger.warn(`Could not ensure schema exists: ${error$47}`);
-		try {
-			await kysely.sql`CREATE SCHEMA IF NOT EXISTS ${kysely.sql.ref(schema)}`.execute(db);
-		} catch {}
+		logger.warn({ error: error$47 }, "Failed to ensure PostgreSQL schema exists");
 	}
 }
 async function getMigrations(db, dbType, options) {
 	let currentSchema = "public";
-	if (dbType === "postgres") {
+	if (dbType === "postgres" || dbType === "neon") {
 		if (options?.schema) {
 			await ensurePostgresSchemaExists(db, options.schema);
 			currentSchema = options.schema;
@@ -13571,7 +13572,7 @@ async function getMigrations(db, dbType, options) {
 	}
 	const allTableMetadata = await db.introspection.getTables();
 	let tableMetadata = allTableMetadata;
-	if (dbType === "postgres") try {
+	if (dbType === "postgres" || dbType === "neon") try {
 		const tablesInSchema = await kysely.sql`
         SELECT table_name
         FROM information_schema.tables
@@ -13620,36 +13621,42 @@ async function getMigrations(db, dbType, options) {
 		return {
 			uuid: {
 				postgres: "uuid",
+				neon: "uuid",
 				mysql: "varchar(36)",
 				sqlite: "text",
 				mssql: "varchar(36)"
 			},
 			text: {
 				postgres: "text",
+				neon: "text",
 				mysql: fieldConfig.unique ? "varchar(255)" : "text",
 				sqlite: "text",
 				mssql: fieldConfig.unique ? "varchar(255)" : "varchar(8000)"
 			},
 			timestamp: {
 				postgres: "timestamptz",
+				neon: "timestamptz",
 				mysql: "timestamp(3)",
 				sqlite: "date",
 				mssql: kysely.sql`datetime2(3)`
 			},
 			jsonb: {
 				postgres: "jsonb",
+				neon: "jsonb",
 				mysql: "json",
 				sqlite: "text",
 				mssql: "varchar(8000)"
 			},
 			boolean: {
 				postgres: "boolean",
+				neon: "boolean",
 				mysql: "boolean",
 				sqlite: "integer",
 				mssql: kysely.sql`bit`
 			},
 			integer: {
 				postgres: "integer",
+				neon: "integer",
 				mysql: "integer",
 				sqlite: "integer",
 				mssql: "integer"
@@ -13657,7 +13664,8 @@ async function getMigrations(db, dbType, options) {
 		}[type]?.[dbType] || "text";
 	}
 	for (const table of toBeCreated) {
-		let builder = db.schema.createTable(table.table);
+		const qualifiedTableName = (dbType === "postgres" || dbType === "neon") && options?.schema ? `${options.schema}.${table.table}` : table.table;
+		let builder = db.schema.createTable(qualifiedTableName);
 		for (const [fieldName, fieldConfig] of Object.entries(table.fields)) {
 			const type = getColumnType(fieldConfig, fieldName);
 			builder = builder.addColumn(fieldName, type, (col) => {
@@ -13678,24 +13686,35 @@ async function getMigrations(db, dbType, options) {
 		}
 		migrations.push(builder);
 	}
-	for (const table of toBeAdded) for (const [fieldName, fieldConfig] of Object.entries(table.fields)) {
-		const type = getColumnType(fieldConfig, fieldName);
-		const builder = db.schema.alterTable(table.table).addColumn(fieldName, type, (col) => {
-			let c = col;
-			if (!fieldConfig.nullable) c = c.notNull();
-			if (fieldConfig.references) {
-				const refTable = fieldConfig.references.table;
-				const refColumn = fieldConfig.references.column;
-				c = c.references(`${refTable}.${refColumn}`).onDelete("cascade");
-			}
-			if (fieldConfig.unique) c = c.unique();
-			if (fieldConfig.default === "now()" && dbType !== "sqlite") if (dbType === "mysql") c = c.defaultTo(kysely.sql`CURRENT_TIMESTAMP(3)`);
-			else c = c.defaultTo(kysely.sql`CURRENT_TIMESTAMP`);
-			return c;
-		});
-		migrations.push(builder);
+	for (const table of toBeAdded) {
+		const qualifiedTableName = (dbType === "postgres" || dbType === "neon") && options?.schema ? `${options.schema}.${table.table}` : table.table;
+		for (const [fieldName, fieldConfig] of Object.entries(table.fields)) {
+			const type = getColumnType(fieldConfig, fieldName);
+			const builder = db.schema.alterTable(qualifiedTableName).addColumn(fieldName, type, (col) => {
+				let c = col;
+				if (!fieldConfig.nullable) c = c.notNull();
+				if (fieldConfig.references) {
+					const refTable = fieldConfig.references.table;
+					const refColumn = fieldConfig.references.column;
+					c = c.references(`${refTable}.${refColumn}`).onDelete("cascade");
+				}
+				if (fieldConfig.unique) c = c.unique();
+				if (fieldConfig.default === "now()" && dbType !== "sqlite") if (dbType === "mysql") c = c.defaultTo(kysely.sql`CURRENT_TIMESTAMP(3)`);
+				else c = c.defaultTo(kysely.sql`CURRENT_TIMESTAMP`);
+				return c;
+			});
+			migrations.push(builder);
+		}
 	}
-	const { toBeAdded: authChangesToBeAdded, toBeCreated: authChangesToBeCreated, runMigrations: runAuthMigrations } = await (0, better_auth_db.getMigrations)(getAuthClientOptions({ database: options?.rawConnection }));
+	if (dbType === "postgres") try {
+		await kysely.sql`SET search_path TO "${options?.schema ?? "llmops"}"`.execute(db);
+	} catch (error$47) {
+		logger.warn({ error: error$47 }, "Failed to set search_path for Better Auth migrations");
+	}
+	const { toBeAdded: authChangesToBeAdded, toBeCreated: authChangesToBeCreated, runMigrations: runAuthMigrations } = await (0, better_auth_db.getMigrations)(getAuthClientOptions({ database: {
+		db,
+		type: dbType === "neon" ? "postgres" : dbType
+	} }));
 	async function runMigrations() {
 		for (const migration of migrations) await migration.execute();
 		await runAuthMigrations();
@@ -13800,13 +13819,14 @@ function createDatabase(connection) {
 * Auto-detect database type from connection object
 */
 function detectDatabaseType(db) {
-	if (!db || typeof db !== "object") return null;
+	if (!db || typeof db !== "object" && typeof db !== "function") return null;
 	if ("createDriver" in db) {
 		if (db instanceof kysely.SqliteDialect) return "sqlite";
 		if (db instanceof kysely.MysqlDialect) return "mysql";
 		if (db instanceof kysely.PostgresDialect) return "postgres";
 		if (db instanceof kysely.MssqlDialect) return "mssql";
 	}
+	if (typeof db === "function" && db.name === "templateFn") return "neon";
 	if ("aggregate" in db) return "sqlite";
 	if ("getConnection" in db) return "mysql";
 	if ("connect" in db) return "postgres";
@@ -13829,12 +13849,12 @@ async function createDatabaseFromConnection(rawConnection, options) {
 		case "sqlite":
 			if ("aggregate" in rawConnection && !("createSession" in rawConnection)) dialect = new kysely.SqliteDialect({ database: rawConnection });
 			else if ("fileControl" in rawConnection) {
-				const { BunSqliteDialect } = await Promise.resolve().then(() => require("./bun-sqlite-dialect-t1xHHJDz.cjs"));
+				const { BunSqliteDialect } = await Promise.resolve().then(() => require("./bun-sqlite-dialect-Bxz4B97L.cjs"));
 				dialect = new BunSqliteDialect({ database: rawConnection });
 			} else if ("createSession" in rawConnection && typeof window === "undefined") try {
 				const { DatabaseSync } = await import("node:sqlite");
 				if (rawConnection instanceof DatabaseSync) {
-					const { NodeSqliteDialect } = await Promise.resolve().then(() => require("./node-sqlite-dialect-D4V5e49r.cjs"));
+					const { NodeSqliteDialect } = await Promise.resolve().then(() => require("./node-sqlite-dialect-N8j5UsT-.cjs"));
 					dialect = new NodeSqliteDialect({ database: rawConnection });
 				}
 			} catch {}
@@ -13850,6 +13870,23 @@ async function createDatabaseFromConnection(rawConnection, options) {
 				}
 			});
 			break;
+		case "neon":
+			if (schema && schema !== "public") {
+				const { Pool, neonConfig } = await import("@neondatabase/serverless");
+				const { default: ws } = await import("ws");
+				neonConfig.webSocketConstructor = ws;
+				const connectionString = typeof rawConnection === "string" ? rawConnection : process.env.NEON_PG_URL || "";
+				dialect = new kysely.PostgresDialect({
+					pool: new Pool({ connectionString: connectionString.includes("currentSchema") ? connectionString.split("currentSchema=")[1].split("&")[0] === "public" ? connectionString.replace(/currentSchema=[^&]*&?/, "") : connectionString : connectionString }),
+					onCreateConnection: async (connection) => {
+						await connection.executeQuery(kysely.CompiledQuery.raw(`SET search_path TO "${schema}"`));
+					}
+				});
+			} else {
+				const { createNeonDialect } = await Promise.resolve().then(() => require("./neon-dialect-C0GZuGot.cjs"));
+				dialect = createNeonDialect(rawConnection);
+			}
+			break;
 		case "mssql":
 			if ("createDriver" in rawConnection) dialect = rawConnection;
 			break;

package/dist/{db-BGDhJROC.mjs → db-DohlAqJU.mjs}

@@ -13434,36 +13434,38 @@ const getAuthClientOptions = (options) => {
 
 //#endregion
 //#region src/db/migrations.ts
+const postgresMap = {
+	uuid: [
+		"character varying",
+		"varchar",
+		"text",
+		"uuid"
+	],
+	text: [
+		"character varying",
+		"varchar",
+		"text"
+	],
+	timestamp: [
+		"timestamptz",
+		"timestamp",
+		"date"
+	],
+	jsonb: ["json", "jsonb"],
+	integer: [
+		"integer",
+		"int4",
+		"int",
+		"smallint",
+		"bigint",
+		"int2",
+		"int8"
+	],
+	boolean: ["boolean", "bool"]
+};
 const typeMap = {
-	postgres: {
-		uuid: [
-			"character varying",
-			"varchar",
-			"text",
-			"uuid"
-		],
-		text: [
-			"character varying",
-			"varchar",
-			"text"
-		],
-		timestamp: [
-			"timestamptz",
-			"timestamp",
-			"date"
-		],
-		jsonb: ["json", "jsonb"],
-		integer: [
-			"integer",
-			"int4",
-			"int",
-			"smallint",
-			"bigint",
-			"int2",
-			"int8"
-		],
-		boolean: ["boolean", "bool"]
-	},
+	postgres: postgresMap,
+	neon: postgresMap,
 	mysql: {
 		text: ["varchar", "text"],
 		timestamp: [
@@ -13527,18 +13529,17 @@ async function ensurePostgresSchemaExists(db, schema) {
       ) as exists
     `.execute(db)).rows[0]?.exists) {
 			logger.info(`Creating PostgreSQL schema: ${schema}`);
-			await sql`CREATE SCHEMA IF NOT EXISTS ${sql.ref(schema)}`.execute(db);
+			await sql`CREATE SCHEMA IF NOT EXISTS ${sql.id(schema)}`.execute(db);
+			await sql`GRANT USAGE ON SCHEMA ${sql.id(schema)} TO PUBLIC`.execute(db);
+			await sql`GRANT CREATE ON SCHEMA ${sql.id(schema)} TO PUBLIC`.execute(db);
 		}
 	} catch (error$47) {
-		logger.warn(`Could not ensure schema exists: ${error$47}`);
-		try {
-			await sql`CREATE SCHEMA IF NOT EXISTS ${sql.ref(schema)}`.execute(db);
-		} catch {}
+		logger.warn({ error: error$47 }, "Failed to ensure PostgreSQL schema exists");
 	}
 }
 async function getMigrations$1(db, dbType, options) {
 	let currentSchema = "public";
-	if (dbType === "postgres") {
+	if (dbType === "postgres" || dbType === "neon") {
 		if (options?.schema) {
 			await ensurePostgresSchemaExists(db, options.schema);
 			currentSchema = options.schema;
@@ -13547,7 +13548,7 @@ async function getMigrations$1(db, dbType, options) {
 	}
 	const allTableMetadata = await db.introspection.getTables();
 	let tableMetadata = allTableMetadata;
-	if (dbType === "postgres") try {
+	if (dbType === "postgres" || dbType === "neon") try {
 		const tablesInSchema = await sql`
         SELECT table_name
         FROM information_schema.tables
@@ -13596,36 +13597,42 @@ async function getMigrations$1(db, dbType, options) {
 		return {
 			uuid: {
 				postgres: "uuid",
+				neon: "uuid",
 				mysql: "varchar(36)",
 				sqlite: "text",
 				mssql: "varchar(36)"
 			},
 			text: {
 				postgres: "text",
+				neon: "text",
 				mysql: fieldConfig.unique ? "varchar(255)" : "text",
 				sqlite: "text",
 				mssql: fieldConfig.unique ? "varchar(255)" : "varchar(8000)"
 			},
 			timestamp: {
 				postgres: "timestamptz",
+				neon: "timestamptz",
 				mysql: "timestamp(3)",
 				sqlite: "date",
 				mssql: sql`datetime2(3)`
 			},
 			jsonb: {
 				postgres: "jsonb",
+				neon: "jsonb",
 				mysql: "json",
 				sqlite: "text",
 				mssql: "varchar(8000)"
 			},
 			boolean: {
 				postgres: "boolean",
+				neon: "boolean",
 				mysql: "boolean",
 				sqlite: "integer",
 				mssql: sql`bit`
 			},
 			integer: {
 				postgres: "integer",
+				neon: "integer",
 				mysql: "integer",
 				sqlite: "integer",
 				mssql: "integer"
@@ -13633,7 +13640,8 @@ async function getMigrations$1(db, dbType, options) {
 		}[type]?.[dbType] || "text";
 	}
 	for (const table of toBeCreated) {
-		let builder = db.schema.createTable(table.table);
+		const qualifiedTableName = (dbType === "postgres" || dbType === "neon") && options?.schema ? `${options.schema}.${table.table}` : table.table;
+		let builder = db.schema.createTable(qualifiedTableName);
 		for (const [fieldName, fieldConfig] of Object.entries(table.fields)) {
 			const type = getColumnType(fieldConfig, fieldName);
 			builder = builder.addColumn(fieldName, type, (col) => {
@@ -13654,24 +13662,35 @@ async function getMigrations$1(db, dbType, options) {
 		}
 		migrations.push(builder);
 	}
-	for (const table of toBeAdded) for (const [fieldName, fieldConfig] of Object.entries(table.fields)) {
-		const type = getColumnType(fieldConfig, fieldName);
-		const builder = db.schema.alterTable(table.table).addColumn(fieldName, type, (col) => {
-			let c = col;
-			if (!fieldConfig.nullable) c = c.notNull();
-			if (fieldConfig.references) {
-				const refTable = fieldConfig.references.table;
-				const refColumn = fieldConfig.references.column;
-				c = c.references(`${refTable}.${refColumn}`).onDelete("cascade");
-			}
-			if (fieldConfig.unique) c = c.unique();
-			if (fieldConfig.default === "now()" && dbType !== "sqlite") if (dbType === "mysql") c = c.defaultTo(sql`CURRENT_TIMESTAMP(3)`);
-			else c = c.defaultTo(sql`CURRENT_TIMESTAMP`);
-			return c;
-		});
-		migrations.push(builder);
+	for (const table of toBeAdded) {
+		const qualifiedTableName = (dbType === "postgres" || dbType === "neon") && options?.schema ? `${options.schema}.${table.table}` : table.table;
+		for (const [fieldName, fieldConfig] of Object.entries(table.fields)) {
+			const type = getColumnType(fieldConfig, fieldName);
+			const builder = db.schema.alterTable(qualifiedTableName).addColumn(fieldName, type, (col) => {
+				let c = col;
+				if (!fieldConfig.nullable) c = c.notNull();
+				if (fieldConfig.references) {
+					const refTable = fieldConfig.references.table;
+					const refColumn = fieldConfig.references.column;
+					c = c.references(`${refTable}.${refColumn}`).onDelete("cascade");
+				}
+				if (fieldConfig.unique) c = c.unique();
+				if (fieldConfig.default === "now()" && dbType !== "sqlite") if (dbType === "mysql") c = c.defaultTo(sql`CURRENT_TIMESTAMP(3)`);
+				else c = c.defaultTo(sql`CURRENT_TIMESTAMP`);
+				return c;
+			});
+			migrations.push(builder);
+		}
 	}
-	const { toBeAdded: authChangesToBeAdded, toBeCreated: authChangesToBeCreated, runMigrations: runAuthMigrations } = await getMigrations(getAuthClientOptions({ database: options?.rawConnection }));
+	if (dbType === "postgres") try {
+		await sql`SET search_path TO "${options?.schema ?? "llmops"}"`.execute(db);
+	} catch (error$47) {
+		logger.warn({ error: error$47 }, "Failed to set search_path for Better Auth migrations");
+	}
+	const { toBeAdded: authChangesToBeAdded, toBeCreated: authChangesToBeCreated, runMigrations: runAuthMigrations } = await getMigrations(getAuthClientOptions({ database: {
+		db,
+		type: dbType === "neon" ? "postgres" : dbType
+	} }));
 	async function runMigrations() {
 		for (const migration of migrations) await migration.execute();
 		await runAuthMigrations();
@@ -13776,13 +13795,14 @@ function createDatabase(connection) {
 * Auto-detect database type from connection object
 */
 function detectDatabaseType(db) {
-	if (!db || typeof db !== "object") return null;
+	if (!db || typeof db !== "object" && typeof db !== "function") return null;
 	if ("createDriver" in db) {
 		if (db instanceof SqliteDialect) return "sqlite";
 		if (db instanceof MysqlDialect) return "mysql";
 		if (db instanceof PostgresDialect) return "postgres";
 		if (db instanceof MssqlDialect) return "mssql";
 	}
+	if (typeof db === "function" && db.name === "templateFn") return "neon";
 	if ("aggregate" in db) return "sqlite";
 	if ("getConnection" in db) return "mysql";
 	if ("connect" in db) return "postgres";
@@ -13805,12 +13825,12 @@ async function createDatabaseFromConnection(rawConnection, options) {
 		case "sqlite":
 			if ("aggregate" in rawConnection && !("createSession" in rawConnection)) dialect = new SqliteDialect({ database: rawConnection });
 			else if ("fileControl" in rawConnection) {
-				const { BunSqliteDialect } = await import("./bun-sqlite-dialect-YVmcpOaa.mjs");
+				const { BunSqliteDialect } = await import("./bun-sqlite-dialect-CHpTFgAo.mjs");
 				dialect = new BunSqliteDialect({ database: rawConnection });
 			} else if ("createSession" in rawConnection && typeof window === "undefined") try {
 				const { DatabaseSync } = await import("node:sqlite");
 				if (rawConnection instanceof DatabaseSync) {
-					const { NodeSqliteDialect } = await import("./node-sqlite-dialect-rrsskSJp.mjs");
+					const { NodeSqliteDialect } = await import("./node-sqlite-dialect-B5oKhOm1.mjs");
 					dialect = new NodeSqliteDialect({ database: rawConnection });
 				}
 			} catch {}
@@ -13826,6 +13846,23 @@ async function createDatabaseFromConnection(rawConnection, options) {
 				}
 			});
 			break;
+		case "neon":
+			if (schema && schema !== "public") {
+				const { Pool, neonConfig } = await import("@neondatabase/serverless");
+				const { default: ws } = await import("ws");
+				neonConfig.webSocketConstructor = ws;
+				const connectionString = typeof rawConnection === "string" ? rawConnection : process.env.NEON_PG_URL || "";
+				dialect = new PostgresDialect({
+					pool: new Pool({ connectionString: connectionString.includes("currentSchema") ? connectionString.split("currentSchema=")[1].split("&")[0] === "public" ? connectionString.replace(/currentSchema=[^&]*&?/, "") : connectionString : connectionString }),
+					onCreateConnection: async (connection) => {
+						await connection.executeQuery(CompiledQuery.raw(`SET search_path TO "${schema}"`));
+					}
+				});
+			} else {
+				const { createNeonDialect } = await import("./neon-dialect-Hmo08nUq.mjs");
+				dialect = createNeonDialect(rawConnection);
+			}
+			break;
 		case "mssql":
 			if ("createDriver" in rawConnection) dialect = rawConnection;
 			break;