@llmops/core 0.2.11 → 0.2.12-beta.1

package/dist/{index-BUlmWTh7.d.cts → index-BO5Rse5J.d.mts}

@@ -1,4 +1,5 @@
-import { ColumnType, Generated, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
+import { ColumnType, Dialect, Generated, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
+import { NeonQueryFunction } from "@neondatabase/serverless";
 import * as zod0 from "zod";
 import { z } from "zod";
 
@@ -1358,7 +1359,6 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
 };
 //#endregion
 //#region src/db/migrations.d.ts
-type DatabaseType$1 = 'postgres' | 'mysql' | 'sqlite' | 'mssql';
 /**
  * Options for migration operations
  */
@@ -1370,7 +1370,7 @@ interface MigrationOptions {
   schema?: string;
   rawConnection?: unknown;
 }
-declare function matchType(columnDataType: string, fieldType: string, dbType: DatabaseType$1): boolean;
+declare function matchType(columnDataType: string, fieldType: string, dbType: DatabaseType): boolean;
 interface MigrationResult {
   toBeCreated: Array<{
     table: string;
@@ -1387,7 +1387,7 @@ interface MigrationResult {
   migrations: any[];
   needsMigration: boolean;
 }
-declare function getMigrations(db: Kysely<Database>, dbType: DatabaseType$1, options?: MigrationOptions): Promise<MigrationResult>;
+declare function getMigrations(db: Kysely<Database>, dbType: DatabaseType, options?: MigrationOptions): Promise<MigrationResult>;
 /**
  * Run migrations if needed
  * @param db - Kysely database instance
@@ -1395,17 +1395,35 @@ declare function getMigrations(db: Kysely<Database>, dbType: DatabaseType$1, opt
  * @param options - Migration options (schema, etc.)
  * @returns true if migrations were run, false otherwise
  */
-declare function runAutoMigrations(db: Kysely<Database>, dbType: DatabaseType$1, options?: MigrationOptions): Promise<{
+declare function runAutoMigrations(db: Kysely<Database>, dbType: DatabaseType, options?: MigrationOptions): Promise<{
   ran: boolean;
   tables: string[];
   fields: string[];
 }>;
 //#endregion
+//#region src/db/neon-dialect.d.ts
+/**
+ * Create a Neon serverless dialect for Kysely
+ *
+ * @param neonInstance - Neon database query function
+ * @returns Kysely dialect configured for Neon serverless
+ */
+declare function createNeonDialect(neonInstance: NeonQueryFunction<false, false>): any;
+/**
+ * Execute a query with search_path set for Neon serverless
+ *
+ * @param neonInstance - Neon database query function
+ * @param query - SQL query to execute
+ * @param schema - Schema to set (optional)
+ * @returns Query result
+ */
+declare function executeWithSchema(neonInstance: NeonQueryFunction<false, false>, query: string, schema?: string): Promise<any>;
+//#endregion
 //#region src/db/index.d.ts
 /**
  * Supported database types
  */
-type DatabaseType = 'postgres' | 'mysql' | 'sqlite' | 'mssql';
+type DatabaseType = 'postgres' | 'mysql' | 'sqlite' | 'mssql' | 'neon';
 /**
  * Options for creating a database connection
  */
@@ -1431,6 +1449,9 @@ type DatabaseConnection = {
 } | {
   type: 'mssql';
   dialect: MssqlDialect;
+} | {
+  type: 'neon';
+  dialect: Dialect;
 } | {
   type: DatabaseType;
   kysely: Kysely<Database>;
@@ -1451,4 +1472,4 @@ declare function detectDatabaseType(db: unknown): DatabaseType | null;
  */
 declare function createDatabaseFromConnection(rawConnection: any, options?: DatabaseOptions): Promise<Kysely<Database> | null>;
 //#endregion
-export { SCHEMA_METADATA as A, WorkspaceSettings as B, EnvironmentSecretsTable as C, LLMRequestsTable as D, LLMRequest as E, Updateable as F, environmentsSchema as G, configVariantsSchema as H, Variant as I, schemas as J, llmRequestsSchema as K, VariantVersion as L, TableName as M, TargetingRule as N, ProviderConfig as O, TargetingRulesTable as P, workspaceSettingsSchema as Q, VariantVersionsTable as R, EnvironmentSecret as S, Insertable as T, configsSchema as U, WorkspaceSettingsTable as V, environmentSecretsSchema as W, variantVersionsSchema as X, targetingRulesSchema as Y, variantsSchema as Z, ConfigVariant as _, createDatabaseFromConnection as a, Database as b, MigrationResult as c, runAutoMigrations as d, parsePartialTableData as f, Config as g, validateTableData as h, createDatabase as i, Selectable as j, ProviderConfigsTable as k, getMigrations as l, validatePartialTableData as m, DatabaseOptions as n, detectDatabaseType as o, parseTableData as p, providerConfigsSchema as q, DatabaseType as r, MigrationOptions as s, DatabaseConnection as t, matchType as u, ConfigVariantsTable as v, EnvironmentsTable as w, Environment as x, ConfigsTable as y, VariantsTable as z };
+export { variantsSchema as $, ProviderConfig as A, VariantVersionsTable as B, Environment as C, Insertable as D, EnvironmentsTable as E, TargetingRule as F, configsSchema as G, WorkspaceSettings as H, TargetingRulesTable as I, llmRequestsSchema as J, environmentSecretsSchema as K, Updateable as L, SCHEMA_METADATA as M, Selectable as N, LLMRequest as O, TableName as P, variantVersionsSchema as Q, Variant as R, Database as S, EnvironmentSecretsTable as T, WorkspaceSettingsTable as U, VariantsTable as V, configVariantsSchema as W, schemas as X, providerConfigsSchema as Y, targetingRulesSchema as Z, validateTableData as _, createDatabaseFromConnection as a, ConfigVariantsTable as b, executeWithSchema as c, getMigrations as d, workspaceSettingsSchema as et, matchType as f, validatePartialTableData as g, parseTableData as h, createDatabase as i, ProviderConfigsTable as j, LLMRequestsTable as k, MigrationOptions as l, parsePartialTableData as m, DatabaseOptions as n, detectDatabaseType as o, runAutoMigrations as p, environmentsSchema as q, DatabaseType as r, createNeonDialect as s, DatabaseConnection as t, MigrationResult as u, Config as v, EnvironmentSecret as w, ConfigsTable as x, ConfigVariant as y, VariantVersion as z };

package/dist/{index-CkrtLH8W.d.mts → index-DwZhaRvQ.d.cts}

@@ -1,6 +1,7 @@
-import { ColumnType, Generated, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
+import { ColumnType, Dialect, Generated, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
 import * as zod497 from "zod";
 import { z } from "zod";
+import { NeonQueryFunction } from "@neondatabase/serverless";
 
 //#region src/db/schema.d.ts
 declare const configsSchema: z.ZodObject<{
@@ -1358,7 +1359,6 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
 };
 //#endregion
 //#region src/db/migrations.d.ts
-type DatabaseType$1 = 'postgres' | 'mysql' | 'sqlite' | 'mssql';
 /**
  * Options for migration operations
  */
@@ -1370,7 +1370,7 @@ interface MigrationOptions {
   schema?: string;
   rawConnection?: unknown;
 }
-declare function matchType(columnDataType: string, fieldType: string, dbType: DatabaseType$1): boolean;
+declare function matchType(columnDataType: string, fieldType: string, dbType: DatabaseType): boolean;
 interface MigrationResult {
   toBeCreated: Array<{
     table: string;
@@ -1387,7 +1387,7 @@ interface MigrationResult {
   migrations: any[];
   needsMigration: boolean;
 }
-declare function getMigrations(db: Kysely<Database>, dbType: DatabaseType$1, options?: MigrationOptions): Promise<MigrationResult>;
+declare function getMigrations(db: Kysely<Database>, dbType: DatabaseType, options?: MigrationOptions): Promise<MigrationResult>;
 /**
  * Run migrations if needed
  * @param db - Kysely database instance
@@ -1395,17 +1395,35 @@ declare function getMigrations(db: Kysely<Database>, dbType: DatabaseType$1, opt
  * @param options - Migration options (schema, etc.)
  * @returns true if migrations were run, false otherwise
  */
-declare function runAutoMigrations(db: Kysely<Database>, dbType: DatabaseType$1, options?: MigrationOptions): Promise<{
+declare function runAutoMigrations(db: Kysely<Database>, dbType: DatabaseType, options?: MigrationOptions): Promise<{
   ran: boolean;
   tables: string[];
   fields: string[];
 }>;
 //#endregion
+//#region src/db/neon-dialect.d.ts
+/**
+ * Create a Neon serverless dialect for Kysely
+ *
+ * @param neonInstance - Neon database query function
+ * @returns Kysely dialect configured for Neon serverless
+ */
+declare function createNeonDialect(neonInstance: NeonQueryFunction<false, false>): any;
+/**
+ * Execute a query with search_path set for Neon serverless
+ *
+ * @param neonInstance - Neon database query function
+ * @param query - SQL query to execute
+ * @param schema - Schema to set (optional)
+ * @returns Query result
+ */
+declare function executeWithSchema(neonInstance: NeonQueryFunction<false, false>, query: string, schema?: string): Promise<any>;
+//#endregion
 //#region src/db/index.d.ts
 /**
  * Supported database types
  */
-type DatabaseType = 'postgres' | 'mysql' | 'sqlite' | 'mssql';
+type DatabaseType = 'postgres' | 'mysql' | 'sqlite' | 'mssql' | 'neon';
 /**
  * Options for creating a database connection
  */
@@ -1431,6 +1449,9 @@ type DatabaseConnection = {
 } | {
   type: 'mssql';
   dialect: MssqlDialect;
+} | {
+  type: 'neon';
+  dialect: Dialect;
 } | {
   type: DatabaseType;
   kysely: Kysely<Database>;
@@ -1451,4 +1472,4 @@ declare function detectDatabaseType(db: unknown): DatabaseType | null;
  */
 declare function createDatabaseFromConnection(rawConnection: any, options?: DatabaseOptions): Promise<Kysely<Database> | null>;
 //#endregion
-export { SCHEMA_METADATA as A, WorkspaceSettings as B, EnvironmentSecretsTable as C, LLMRequestsTable as D, LLMRequest as E, Updateable as F, environmentsSchema as G, configVariantsSchema as H, Variant as I, schemas as J, llmRequestsSchema as K, VariantVersion as L, TableName as M, TargetingRule as N, ProviderConfig as O, TargetingRulesTable as P, workspaceSettingsSchema as Q, VariantVersionsTable as R, EnvironmentSecret as S, Insertable as T, configsSchema as U, WorkspaceSettingsTable as V, environmentSecretsSchema as W, variantVersionsSchema as X, targetingRulesSchema as Y, variantsSchema as Z, ConfigVariant as _, createDatabaseFromConnection as a, Database as b, MigrationResult as c, runAutoMigrations as d, parsePartialTableData as f, Config as g, validateTableData as h, createDatabase as i, Selectable as j, ProviderConfigsTable as k, getMigrations as l, validatePartialTableData as m, DatabaseOptions as n, detectDatabaseType as o, parseTableData as p, providerConfigsSchema as q, DatabaseType as r, MigrationOptions as s, DatabaseConnection as t, matchType as u, ConfigVariantsTable as v, EnvironmentsTable as w, Environment as x, ConfigsTable as y, VariantsTable as z };
+export { variantsSchema as $, ProviderConfig as A, VariantVersionsTable as B, Environment as C, Insertable as D, EnvironmentsTable as E, TargetingRule as F, configsSchema as G, WorkspaceSettings as H, TargetingRulesTable as I, llmRequestsSchema as J, environmentSecretsSchema as K, Updateable as L, SCHEMA_METADATA as M, Selectable as N, LLMRequest as O, TableName as P, variantVersionsSchema as Q, Variant as R, Database as S, EnvironmentSecretsTable as T, WorkspaceSettingsTable as U, VariantsTable as V, configVariantsSchema as W, schemas as X, providerConfigsSchema as Y, targetingRulesSchema as Z, validateTableData as _, createDatabaseFromConnection as a, ConfigVariantsTable as b, executeWithSchema as c, getMigrations as d, workspaceSettingsSchema as et, matchType as f, validatePartialTableData as g, parseTableData as h, createDatabase as i, ProviderConfigsTable as j, LLMRequestsTable as k, MigrationOptions as l, parsePartialTableData as m, DatabaseOptions as n, detectDatabaseType as o, runAutoMigrations as p, environmentsSchema as q, DatabaseType as r, createNeonDialect as s, DatabaseConnection as t, MigrationResult as u, Config as v, EnvironmentSecret as w, ConfigsTable as x, ConfigVariant as y, VariantVersion as z };

package/dist/index.cjs

@@ -1,4 +1,5 @@
-const require_db = require('./db-CqDD6Ydv.cjs');
+const require_db = require('./db-BOe6mM51.cjs');
+const require_neon_dialect = require('./neon-dialect-BQey5lUw.cjs');
 let __llmops_gateway = require("@llmops/gateway");
 __llmops_gateway = require_db.__toESM(__llmops_gateway);
 let kysely = require("kysely");
@@ -2595,12 +2596,14 @@ exports.createDataLayer = createDataLayer;
 exports.createDatabase = require_db.createDatabase;
 exports.createDatabaseFromConnection = require_db.createDatabaseFromConnection;
 exports.createLLMRequestsDataLayer = createLLMRequestsDataLayer;
+exports.createNeonDialect = require_neon_dialect.createNeonDialect;
 exports.createProviderConfigsDataLayer = createProviderConfigsDataLayer;
 exports.createWorkspaceSettingsDataLayer = createWorkspaceSettingsDataLayer;
 exports.detectDatabaseType = require_db.detectDatabaseType;
 exports.dollarsToMicroDollars = dollarsToMicroDollars;
 exports.environmentSecretsSchema = require_db.environmentSecretsSchema;
 exports.environmentsSchema = require_db.environmentsSchema;
+exports.executeWithSchema = require_neon_dialect.executeWithSchema;
 exports.formatCost = formatCost;
 Object.defineProperty(exports, 'gateway', {
   enumerable: true,