@llmops/core 0.4.8-beta.5 → 0.4.8-beta.7

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,4 +1,4 @@
1
- const require_db = require('./db-CEt1ZD2V.cjs');
1
+ const require_db = require('./db-CIEpN_uB.cjs');
2
2
  let kysely = require("kysely");
3
3
 
4
4
  //#region src/db/bun-sqlite-dialect.ts
package/dist/db/index.cjs CHANGED
@@ -1,5 +1,5 @@
1
- const require_db = require('../db-CEt1ZD2V.cjs');
2
- const require_neon_dialect = require('../neon-dialect-sUVNvVYh.cjs');
1
+ const require_db = require('../db-CIEpN_uB.cjs');
2
+ const require_neon_dialect = require('../neon-dialect-NI1iA903.cjs');
3
3
 
4
4
  exports.SCHEMA_METADATA = require_db.SCHEMA_METADATA;
5
5
  exports.configVariantsSchema = require_db.configVariantsSchema;
@@ -1,2 +1,2 @@
1
- import { $ as ProviderConfigsTable, A as DatasetVersionRecord, At as schemas, B as GuardrailResult, C as ConfigVariantsTable, Ct as llmRequestsSchema, D as DatasetRecord, Dt as playgroundsSchema, E as Dataset, Et as playgroundRunsSchema, F as EnvironmentSecret, G as Playground, H as Insertable, I as EnvironmentSecretsTable, J as PlaygroundResultsTable, K as PlaygroundColumn, L as EnvironmentsTable, M as DatasetVersionsTable, Mt as variantVersionsSchema, N as DatasetsTable, Nt as variantsSchema, O as DatasetRecordsTable, Ot as providerConfigsSchema, P as Environment, Pt as workspaceSettingsSchema, Q as ProviderConfig, R as GuardrailConfig, S as ConfigVariant, St as guardrailConfigsSchema, T as Database, Tt as playgroundResultsSchema, U as LLMRequest, V as GuardrailResults, W as LLMRequestsTable, X as PlaygroundRunsTable, Y as PlaygroundRun, Z as PlaygroundsTable, _ as validateTableData, _t as datasetVersionRecordsSchema, a as createDatabaseFromConnection, at as TargetingRule, b as runSchemaSQL, bt as environmentSecretsSchema, c as executeWithSchema, ct as Variant, d as getMigrations, dt as VariantsTable, et as ProviderGuardrailOverride, f as matchType, ft as WorkspaceSettings, g as validatePartialTableData, gt as datasetRecordsSchema, h as parseTableData, ht as configsSchema, i as createDatabase, it as TableName, j as DatasetVersionRecordsTable, jt as targetingRulesSchema, k as DatasetVersion, kt as providerGuardrailOverridesSchema, l as MigrationOptions, lt as VariantVersion, m as parsePartialTableData, mt as configVariantsSchema, n as DatabaseOptions, nt as SCHEMA_METADATA, o as detectDatabaseType, ot as TargetingRulesTable, p as runAutoMigrations, pt as WorkspaceSettingsTable, q as PlaygroundResult, r as DatabaseType, rt as Selectable, s as createNeonDialect, st as Updateable, t as DatabaseConnection, tt as ProviderGuardrailOverridesTable, u as MigrationResult, ut as VariantVersionsTable, v as createNeonSqlFunction, vt as datasetVersionsSchema, w as ConfigsTable, wt as playgroundColumnSchema, x as Config, xt as environmentsSchema, y as generatePostgresSchemaSQL, yt as datasetsSchema, z as GuardrailConfigsTable } from "../index-Bk_flDhH.cjs";
1
+ import { $ as ProviderConfigsTable, A as DatasetVersionRecord, At as schemas, B as GuardrailResult, C as ConfigVariantsTable, Ct as llmRequestsSchema, D as DatasetRecord, Dt as playgroundsSchema, E as Dataset, Et as playgroundRunsSchema, F as EnvironmentSecret, G as Playground, H as Insertable, I as EnvironmentSecretsTable, J as PlaygroundResultsTable, K as PlaygroundColumn, L as EnvironmentsTable, M as DatasetVersionsTable, Mt as variantVersionsSchema, N as DatasetsTable, Nt as variantsSchema, O as DatasetRecordsTable, Ot as providerConfigsSchema, P as Environment, Pt as workspaceSettingsSchema, Q as ProviderConfig, R as GuardrailConfig, S as ConfigVariant, St as guardrailConfigsSchema, T as Database, Tt as playgroundResultsSchema, U as LLMRequest, V as GuardrailResults, W as LLMRequestsTable, X as PlaygroundRunsTable, Y as PlaygroundRun, Z as PlaygroundsTable, _ as validateTableData, _t as datasetVersionRecordsSchema, a as createDatabaseFromConnection, at as TargetingRule, b as runSchemaSQL, bt as environmentSecretsSchema, c as executeWithSchema, ct as Variant, d as getMigrations, dt as VariantsTable, et as ProviderGuardrailOverride, f as matchType, ft as WorkspaceSettings, g as validatePartialTableData, gt as datasetRecordsSchema, h as parseTableData, ht as configsSchema, i as createDatabase, it as TableName, j as DatasetVersionRecordsTable, jt as targetingRulesSchema, k as DatasetVersion, kt as providerGuardrailOverridesSchema, l as MigrationOptions, lt as VariantVersion, m as parsePartialTableData, mt as configVariantsSchema, n as DatabaseOptions, nt as SCHEMA_METADATA, o as detectDatabaseType, ot as TargetingRulesTable, p as runAutoMigrations, pt as WorkspaceSettingsTable, q as PlaygroundResult, r as DatabaseType, rt as Selectable, s as createNeonDialect, st as Updateable, t as DatabaseConnection, tt as ProviderGuardrailOverridesTable, u as MigrationResult, ut as VariantVersionsTable, v as createNeonSqlFunction, vt as datasetVersionsSchema, w as ConfigsTable, wt as playgroundColumnSchema, x as Config, xt as environmentsSchema, y as generatePostgresSchemaSQL, yt as datasetsSchema, z as GuardrailConfigsTable } from "../index-BYhMgUES.cjs";
2
2
  export { Config, ConfigVariant, ConfigVariantsTable, ConfigsTable, Database, DatabaseConnection, DatabaseOptions, DatabaseType, Dataset, DatasetRecord, DatasetRecordsTable, DatasetVersion, DatasetVersionRecord, DatasetVersionRecordsTable, DatasetVersionsTable, DatasetsTable, Environment, EnvironmentSecret, EnvironmentSecretsTable, EnvironmentsTable, GuardrailConfig, GuardrailConfigsTable, GuardrailResult, GuardrailResults, Insertable, LLMRequest, LLMRequestsTable, MigrationOptions, MigrationResult, Playground, PlaygroundColumn, PlaygroundResult, PlaygroundResultsTable, PlaygroundRun, PlaygroundRunsTable, PlaygroundsTable, ProviderConfig, ProviderConfigsTable, ProviderGuardrailOverride, ProviderGuardrailOverridesTable, SCHEMA_METADATA, Selectable, TableName, TargetingRule, TargetingRulesTable, Updateable, Variant, VariantVersion, VariantVersionsTable, VariantsTable, WorkspaceSettings, WorkspaceSettingsTable, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, createNeonDialect, createNeonSqlFunction, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, environmentSecretsSchema, environmentsSchema, executeWithSchema, generatePostgresSchemaSQL, getMigrations, guardrailConfigsSchema, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, playgroundColumnSchema, playgroundResultsSchema, playgroundRunsSchema, playgroundsSchema, providerConfigsSchema, providerGuardrailOverridesSchema, runAutoMigrations, runSchemaSQL, schemas, targetingRulesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
@@ -1,2 +1,2 @@
1
- import { $ as ProviderConfigsTable, A as DatasetVersionRecord, At as schemas, B as GuardrailResult, C as ConfigVariantsTable, Ct as llmRequestsSchema, D as DatasetRecord, Dt as playgroundsSchema, E as Dataset, Et as playgroundRunsSchema, F as EnvironmentSecret, G as Playground, H as Insertable, I as EnvironmentSecretsTable, J as PlaygroundResultsTable, K as PlaygroundColumn, L as EnvironmentsTable, M as DatasetVersionsTable, Mt as variantVersionsSchema, N as DatasetsTable, Nt as variantsSchema, O as DatasetRecordsTable, Ot as providerConfigsSchema, P as Environment, Pt as workspaceSettingsSchema, Q as ProviderConfig, R as GuardrailConfig, S as ConfigVariant, St as guardrailConfigsSchema, T as Database, Tt as playgroundResultsSchema, U as LLMRequest, V as GuardrailResults, W as LLMRequestsTable, X as PlaygroundRunsTable, Y as PlaygroundRun, Z as PlaygroundsTable, _ as validateTableData, _t as datasetVersionRecordsSchema, a as createDatabaseFromConnection, at as TargetingRule, b as runSchemaSQL, bt as environmentSecretsSchema, c as executeWithSchema, ct as Variant, d as getMigrations, dt as VariantsTable, et as ProviderGuardrailOverride, f as matchType, ft as WorkspaceSettings, g as validatePartialTableData, gt as datasetRecordsSchema, h as parseTableData, ht as configsSchema, i as createDatabase, it as TableName, j as DatasetVersionRecordsTable, jt as targetingRulesSchema, k as DatasetVersion, kt as providerGuardrailOverridesSchema, l as MigrationOptions, lt as VariantVersion, m as parsePartialTableData, mt as configVariantsSchema, n as DatabaseOptions, nt as SCHEMA_METADATA, o as detectDatabaseType, ot as TargetingRulesTable, p as runAutoMigrations, pt as WorkspaceSettingsTable, q as PlaygroundResult, r as DatabaseType, rt as Selectable, s as createNeonDialect, st as Updateable, t as DatabaseConnection, tt as ProviderGuardrailOverridesTable, u as MigrationResult, ut as VariantVersionsTable, v as createNeonSqlFunction, vt as datasetVersionsSchema, w as ConfigsTable, wt as playgroundColumnSchema, x as Config, xt as environmentsSchema, y as generatePostgresSchemaSQL, yt as datasetsSchema, z as GuardrailConfigsTable } from "../index-BvrZuRon.mjs";
1
+ import { $ as ProviderConfigsTable, A as DatasetVersionRecord, At as schemas, B as GuardrailResult, C as ConfigVariantsTable, Ct as llmRequestsSchema, D as DatasetRecord, Dt as playgroundsSchema, E as Dataset, Et as playgroundRunsSchema, F as EnvironmentSecret, G as Playground, H as Insertable, I as EnvironmentSecretsTable, J as PlaygroundResultsTable, K as PlaygroundColumn, L as EnvironmentsTable, M as DatasetVersionsTable, Mt as variantVersionsSchema, N as DatasetsTable, Nt as variantsSchema, O as DatasetRecordsTable, Ot as providerConfigsSchema, P as Environment, Pt as workspaceSettingsSchema, Q as ProviderConfig, R as GuardrailConfig, S as ConfigVariant, St as guardrailConfigsSchema, T as Database, Tt as playgroundResultsSchema, U as LLMRequest, V as GuardrailResults, W as LLMRequestsTable, X as PlaygroundRunsTable, Y as PlaygroundRun, Z as PlaygroundsTable, _ as validateTableData, _t as datasetVersionRecordsSchema, a as createDatabaseFromConnection, at as TargetingRule, b as runSchemaSQL, bt as environmentSecretsSchema, c as executeWithSchema, ct as Variant, d as getMigrations, dt as VariantsTable, et as ProviderGuardrailOverride, f as matchType, ft as WorkspaceSettings, g as validatePartialTableData, gt as datasetRecordsSchema, h as parseTableData, ht as configsSchema, i as createDatabase, it as TableName, j as DatasetVersionRecordsTable, jt as targetingRulesSchema, k as DatasetVersion, kt as providerGuardrailOverridesSchema, l as MigrationOptions, lt as VariantVersion, m as parsePartialTableData, mt as configVariantsSchema, n as DatabaseOptions, nt as SCHEMA_METADATA, o as detectDatabaseType, ot as TargetingRulesTable, p as runAutoMigrations, pt as WorkspaceSettingsTable, q as PlaygroundResult, r as DatabaseType, rt as Selectable, s as createNeonDialect, st as Updateable, t as DatabaseConnection, tt as ProviderGuardrailOverridesTable, u as MigrationResult, ut as VariantVersionsTable, v as createNeonSqlFunction, vt as datasetVersionsSchema, w as ConfigsTable, wt as playgroundColumnSchema, x as Config, xt as environmentsSchema, y as generatePostgresSchemaSQL, yt as datasetsSchema, z as GuardrailConfigsTable } from "../index-BvU4Gv3C.mjs";
2
2
  export { Config, ConfigVariant, ConfigVariantsTable, ConfigsTable, Database, DatabaseConnection, DatabaseOptions, DatabaseType, Dataset, DatasetRecord, DatasetRecordsTable, DatasetVersion, DatasetVersionRecord, DatasetVersionRecordsTable, DatasetVersionsTable, DatasetsTable, Environment, EnvironmentSecret, EnvironmentSecretsTable, EnvironmentsTable, GuardrailConfig, GuardrailConfigsTable, GuardrailResult, GuardrailResults, Insertable, LLMRequest, LLMRequestsTable, MigrationOptions, MigrationResult, Playground, PlaygroundColumn, PlaygroundResult, PlaygroundResultsTable, PlaygroundRun, PlaygroundRunsTable, PlaygroundsTable, ProviderConfig, ProviderConfigsTable, ProviderGuardrailOverride, ProviderGuardrailOverridesTable, SCHEMA_METADATA, Selectable, TableName, TargetingRule, TargetingRulesTable, Updateable, Variant, VariantVersion, VariantVersionsTable, VariantsTable, WorkspaceSettings, WorkspaceSettingsTable, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, createNeonDialect, createNeonSqlFunction, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, environmentSecretsSchema, environmentsSchema, executeWithSchema, generatePostgresSchemaSQL, getMigrations, guardrailConfigsSchema, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, playgroundColumnSchema, playgroundResultsSchema, playgroundRunsSchema, playgroundsSchema, providerConfigsSchema, providerGuardrailOverridesSchema, runAutoMigrations, runSchemaSQL, schemas, targetingRulesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
package/dist/db/index.mjs CHANGED
@@ -1,4 +1,4 @@
1
- import { A as playgroundsSchema, C as environmentSecretsSchema, D as playgroundColumnSchema, E as llmRequestsSchema, F as variantVersionsSchema, I as variantsSchema, L as workspaceSettingsSchema, M as providerGuardrailOverridesSchema, N as schemas, O as playgroundResultsSchema, P as targetingRulesSchema, S as datasetsSchema, T as guardrailConfigsSchema, _ as configVariantsSchema, a as matchType, b as datasetVersionRecordsSchema, d as validatePartialTableData, f as validateTableData, g as SCHEMA_METADATA, h as runSchemaSQL, i as getMigrations, j as providerConfigsSchema, k as playgroundRunsSchema, l as parsePartialTableData, m as generatePostgresSchemaSQL, n as createDatabaseFromConnection, o as runAutoMigrations, p as createNeonSqlFunction, r as detectDatabaseType, t as createDatabase, u as parseTableData, v as configsSchema, w as environmentsSchema, x as datasetVersionsSchema, y as datasetRecordsSchema } from "../db-BN99c_b5.mjs";
1
+ import { A as playgroundsSchema, C as environmentSecretsSchema, D as playgroundColumnSchema, E as llmRequestsSchema, F as variantVersionsSchema, I as variantsSchema, L as workspaceSettingsSchema, M as providerGuardrailOverridesSchema, N as schemas, O as playgroundResultsSchema, P as targetingRulesSchema, S as datasetsSchema, T as guardrailConfigsSchema, _ as configVariantsSchema, a as matchType, b as datasetVersionRecordsSchema, d as validatePartialTableData, f as validateTableData, g as SCHEMA_METADATA, h as runSchemaSQL, i as getMigrations, j as providerConfigsSchema, k as playgroundRunsSchema, l as parsePartialTableData, m as generatePostgresSchemaSQL, n as createDatabaseFromConnection, o as runAutoMigrations, p as createNeonSqlFunction, r as detectDatabaseType, t as createDatabase, u as parseTableData, v as configsSchema, w as environmentsSchema, x as datasetVersionsSchema, y as datasetRecordsSchema } from "../db-DhXoqPZu.mjs";
2
2
  import { n as executeWithSchema, t as createNeonDialect } from "../neon-dialect-DySGBYUi.mjs";
3
3
 
4
4
  export { SCHEMA_METADATA, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, createNeonDialect, createNeonSqlFunction, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, environmentSecretsSchema, environmentsSchema, executeWithSchema, generatePostgresSchemaSQL, getMigrations, guardrailConfigsSchema, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, playgroundColumnSchema, playgroundResultsSchema, playgroundRunsSchema, playgroundsSchema, providerConfigsSchema, providerGuardrailOverridesSchema, runAutoMigrations, runSchemaSQL, schemas, targetingRulesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
@@ -14112,27 +14112,28 @@ async function runSchemaSQL(sql$1, schemaName) {
14112
14112
  }
14113
14113
  }
14114
14114
  /**
14115
- * Split SQL into individual statements, keeping DO blocks together
14115
+ * Split SQL into individual statements, keeping $$ blocks together
14116
+ * Handles both DO $$ blocks and CREATE FUNCTION ... AS $$ blocks
14116
14117
  */
14117
14118
  function splitSQLStatements(sql$1) {
14118
14119
  const statements = [];
14119
14120
  let current = "";
14120
- let inDoBlock = false;
14121
+ let inDollarBlock = false;
14121
14122
  const lines = sql$1.split("\n");
14122
14123
  for (const line of lines) {
14123
14124
  const trimmed = line.trim();
14124
14125
  if (!trimmed || trimmed.startsWith("--")) continue;
14125
- if (trimmed === "DO $$") {
14126
- inDoBlock = true;
14127
- current = line + "\n";
14126
+ if (!inDollarBlock && (trimmed === "DO $$" || trimmed.endsWith("AS $$"))) {
14127
+ inDollarBlock = true;
14128
+ current += line + "\n";
14128
14129
  continue;
14129
14130
  }
14130
- if (inDoBlock) {
14131
+ if (inDollarBlock) {
14131
14132
  current += line + "\n";
14132
- if (trimmed === "END $$;") {
14133
+ if (trimmed.startsWith("$$") && trimmed.endsWith(";")) {
14133
14134
  statements.push(current.trim());
14134
14135
  current = "";
14135
- inDoBlock = false;
14136
+ inDollarBlock = false;
14136
14137
  }
14137
14138
  continue;
14138
14139
  }
@@ -14148,19 +14149,29 @@ function splitSQLStatements(sql$1) {
14148
14149
  /**
14149
14150
  * Create a Neon SQL function from various connection types
14150
14151
  *
14152
+ * The Neon serverless library returns a tagged template function from neon(),
14153
+ * but for conventional string queries we need to use sql.query() instead.
14154
+ * This helper wraps the neon instance to provide a simple query function.
14155
+ *
14151
14156
  * @param rawConnection - neon() function, connection string, or undefined (uses env vars)
14152
14157
  * @returns SQL function that can be used with runSchemaSQL, or null if unable to create
14153
14158
  */
14154
14159
  async function createNeonSqlFunction(rawConnection) {
14155
- if (typeof rawConnection === "function") return rawConnection;
14160
+ const wrapNeonSql = (sql$1) => {
14161
+ return (query) => sql$1.query(query);
14162
+ };
14163
+ if (typeof rawConnection === "function") {
14164
+ if ("query" in rawConnection && typeof rawConnection.query === "function") return wrapNeonSql(rawConnection);
14165
+ return rawConnection;
14166
+ }
14156
14167
  if (typeof rawConnection === "string" && rawConnection) {
14157
14168
  const { neon: neon$1 } = await import("@neondatabase/serverless");
14158
- return neon$1(rawConnection);
14169
+ return wrapNeonSql(neon$1(rawConnection));
14159
14170
  }
14160
14171
  const connectionString = process.env.NEON_CONNECTION_STRING || process.env.NEON_PG_URL || process.env.DATABASE_URL || process.env.POSTGRES_URL || "";
14161
14172
  if (!connectionString) return null;
14162
14173
  const { neon } = await import("@neondatabase/serverless");
14163
- return neon(connectionString);
14174
+ return wrapNeonSql(neon(connectionString));
14164
14175
  }
14165
14176
 
14166
14177
  //#endregion
@@ -14621,12 +14632,12 @@ async function createDatabaseFromConnection(rawConnection, options) {
14621
14632
  case "sqlite":
14622
14633
  if ("aggregate" in rawConnection && !("createSession" in rawConnection)) dialect = new kysely.SqliteDialect({ database: rawConnection });
14623
14634
  else if ("fileControl" in rawConnection) {
14624
- const { BunSqliteDialect } = await Promise.resolve().then(() => require("./bun-sqlite-dialect-lNniX_lj.cjs"));
14635
+ const { BunSqliteDialect } = await Promise.resolve().then(() => require("./bun-sqlite-dialect-BynSKaRn.cjs"));
14625
14636
  dialect = new BunSqliteDialect({ database: rawConnection });
14626
14637
  } else if ("createSession" in rawConnection && typeof window === "undefined") try {
14627
14638
  const { DatabaseSync } = await import("node:sqlite");
14628
14639
  if (rawConnection instanceof DatabaseSync) {
14629
- const { NodeSqliteDialect } = await Promise.resolve().then(() => require("./node-sqlite-dialect-BPGejxOC.cjs"));
14640
+ const { NodeSqliteDialect } = await Promise.resolve().then(() => require("./node-sqlite-dialect-DQMW9PEY.cjs"));
14630
14641
  dialect = new NodeSqliteDialect({ database: rawConnection });
14631
14642
  }
14632
14643
  } catch {}
@@ -14644,7 +14655,7 @@ async function createDatabaseFromConnection(rawConnection, options) {
14644
14655
  break;
14645
14656
  case "neon": {
14646
14657
  if (typeof rawConnection === "function") {
14647
- const { createNeonDialect: createNeonDialect$1 } = await Promise.resolve().then(() => require("./neon-dialect-C9_0GrsB.cjs"));
14658
+ const { createNeonDialect: createNeonDialect$1 } = await Promise.resolve().then(() => require("./neon-dialect-CeFVHjjF.cjs"));
14648
14659
  dialect = createNeonDialect$1(rawConnection);
14649
14660
  break;
14650
14661
  }
@@ -14652,7 +14663,7 @@ async function createDatabaseFromConnection(rawConnection, options) {
14652
14663
  if (!connectionString) throw new Error("Neon connection string is required. Pass it directly as the database option or set one of: NEON_CONNECTION_STRING, NEON_PG_URL, DATABASE_URL, POSTGRES_URL");
14653
14664
  const { neon } = await import("@neondatabase/serverless");
14654
14665
  const sql$1 = neon(connectionString);
14655
- const { createNeonDialect } = await Promise.resolve().then(() => require("./neon-dialect-C9_0GrsB.cjs"));
14666
+ const { createNeonDialect } = await Promise.resolve().then(() => require("./neon-dialect-CeFVHjjF.cjs"));
14656
14667
  dialect = createNeonDialect(sql$1);
14657
14668
  break;
14658
14669
  }
@@ -14088,27 +14088,28 @@ async function runSchemaSQL(sql$1, schemaName) {
14088
14088
  }
14089
14089
  }
14090
14090
  /**
14091
- * Split SQL into individual statements, keeping DO blocks together
14091
+ * Split SQL into individual statements, keeping $$ blocks together
14092
+ * Handles both DO $$ blocks and CREATE FUNCTION ... AS $$ blocks
14092
14093
  */
14093
14094
  function splitSQLStatements(sql$1) {
14094
14095
  const statements = [];
14095
14096
  let current = "";
14096
- let inDoBlock = false;
14097
+ let inDollarBlock = false;
14097
14098
  const lines = sql$1.split("\n");
14098
14099
  for (const line of lines) {
14099
14100
  const trimmed = line.trim();
14100
14101
  if (!trimmed || trimmed.startsWith("--")) continue;
14101
- if (trimmed === "DO $$") {
14102
- inDoBlock = true;
14103
- current = line + "\n";
14102
+ if (!inDollarBlock && (trimmed === "DO $$" || trimmed.endsWith("AS $$"))) {
14103
+ inDollarBlock = true;
14104
+ current += line + "\n";
14104
14105
  continue;
14105
14106
  }
14106
- if (inDoBlock) {
14107
+ if (inDollarBlock) {
14107
14108
  current += line + "\n";
14108
- if (trimmed === "END $$;") {
14109
+ if (trimmed.startsWith("$$") && trimmed.endsWith(";")) {
14109
14110
  statements.push(current.trim());
14110
14111
  current = "";
14111
- inDoBlock = false;
14112
+ inDollarBlock = false;
14112
14113
  }
14113
14114
  continue;
14114
14115
  }
@@ -14124,19 +14125,29 @@ function splitSQLStatements(sql$1) {
14124
14125
  /**
14125
14126
  * Create a Neon SQL function from various connection types
14126
14127
  *
14128
+ * The Neon serverless library returns a tagged template function from neon(),
14129
+ * but for conventional string queries we need to use sql.query() instead.
14130
+ * This helper wraps the neon instance to provide a simple query function.
14131
+ *
14127
14132
  * @param rawConnection - neon() function, connection string, or undefined (uses env vars)
14128
14133
  * @returns SQL function that can be used with runSchemaSQL, or null if unable to create
14129
14134
  */
14130
14135
  async function createNeonSqlFunction(rawConnection) {
14131
- if (typeof rawConnection === "function") return rawConnection;
14136
+ const wrapNeonSql = (sql$1) => {
14137
+ return (query) => sql$1.query(query);
14138
+ };
14139
+ if (typeof rawConnection === "function") {
14140
+ if ("query" in rawConnection && typeof rawConnection.query === "function") return wrapNeonSql(rawConnection);
14141
+ return rawConnection;
14142
+ }
14132
14143
  if (typeof rawConnection === "string" && rawConnection) {
14133
14144
  const { neon: neon$1 } = await import("@neondatabase/serverless");
14134
- return neon$1(rawConnection);
14145
+ return wrapNeonSql(neon$1(rawConnection));
14135
14146
  }
14136
14147
  const connectionString = process.env.NEON_CONNECTION_STRING || process.env.NEON_PG_URL || process.env.DATABASE_URL || process.env.POSTGRES_URL || "";
14137
14148
  if (!connectionString) return null;
14138
14149
  const { neon } = await import("@neondatabase/serverless");
14139
- return neon(connectionString);
14150
+ return wrapNeonSql(neon(connectionString));
14140
14151
  }
14141
14152
 
14142
14153
  //#endregion
@@ -1,7 +1,7 @@
1
1
  import { ColumnType, Dialect, Generated, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
2
- import { NeonQueryFunction } from "@neondatabase/serverless";
3
2
  import * as zod0 from "zod";
4
3
  import { z } from "zod";
4
+ import { NeonQueryFunction } from "@neondatabase/serverless";
5
5
 
6
6
  //#region src/db/schema.d.ts
7
7
  declare const configsSchema: z.ZodObject<{
@@ -1927,6 +1927,10 @@ declare function runSchemaSQL(sql: (query: string) => Promise<unknown>, schemaNa
1927
1927
  /**
1928
1928
  * Create a Neon SQL function from various connection types
1929
1929
  *
1930
+ * The Neon serverless library returns a tagged template function from neon(),
1931
+ * but for conventional string queries we need to use sql.query() instead.
1932
+ * This helper wraps the neon instance to provide a simple query function.
1933
+ *
1930
1934
  * @param rawConnection - neon() function, connection string, or undefined (uses env vars)
1931
1935
  * @returns SQL function that can be used with runSchemaSQL, or null if unable to create
1932
1936
  */
@@ -2115,22 +2119,6 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
2115
2119
  id: string;
2116
2120
  createdAt: Date;
2117
2121
  updatedAt: Date;
2118
- }> | zod0.ZodSafeParseSuccess<{
2119
- providerConfigId: string;
2120
- guardrailConfigId: string;
2121
- enabled: boolean;
2122
- id: string;
2123
- createdAt: Date;
2124
- updatedAt: Date;
2125
- parameters?: Record<string, unknown> | null | undefined;
2126
- }> | zod0.ZodSafeParseError<{
2127
- providerConfigId: string;
2128
- guardrailConfigId: string;
2129
- enabled: boolean;
2130
- id: string;
2131
- createdAt: Date;
2132
- updatedAt: Date;
2133
- parameters?: Record<string, unknown> | null | undefined;
2134
2122
  }> | zod0.ZodSafeParseSuccess<{
2135
2123
  datasetId: string;
2136
2124
  versionNumber: number;
@@ -2181,6 +2169,22 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
2181
2169
  id: string;
2182
2170
  createdAt: Date;
2183
2171
  updatedAt: Date;
2172
+ }> | zod0.ZodSafeParseSuccess<{
2173
+ providerConfigId: string;
2174
+ guardrailConfigId: string;
2175
+ enabled: boolean;
2176
+ id: string;
2177
+ createdAt: Date;
2178
+ updatedAt: Date;
2179
+ parameters?: Record<string, unknown> | null | undefined;
2180
+ }> | zod0.ZodSafeParseError<{
2181
+ providerConfigId: string;
2182
+ guardrailConfigId: string;
2183
+ enabled: boolean;
2184
+ id: string;
2185
+ createdAt: Date;
2186
+ updatedAt: Date;
2187
+ parameters?: Record<string, unknown> | null | undefined;
2184
2188
  }> | zod0.ZodSafeParseSuccess<{
2185
2189
  requestId: string;
2186
2190
  provider: string;
@@ -2421,22 +2425,6 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
2421
2425
  id?: string | undefined;
2422
2426
  createdAt?: Date | undefined;
2423
2427
  updatedAt?: Date | undefined;
2424
- }> | zod0.ZodSafeParseSuccess<{
2425
- providerConfigId?: string | undefined;
2426
- guardrailConfigId?: string | undefined;
2427
- enabled?: boolean | undefined;
2428
- parameters?: Record<string, unknown> | null | undefined;
2429
- id?: string | undefined;
2430
- createdAt?: Date | undefined;
2431
- updatedAt?: Date | undefined;
2432
- }> | zod0.ZodSafeParseError<{
2433
- providerConfigId?: string | undefined;
2434
- guardrailConfigId?: string | undefined;
2435
- enabled?: boolean | undefined;
2436
- parameters?: Record<string, unknown> | null | undefined;
2437
- id?: string | undefined;
2438
- createdAt?: Date | undefined;
2439
- updatedAt?: Date | undefined;
2440
2428
  }> | zod0.ZodSafeParseSuccess<{
2441
2429
  datasetId?: string | undefined;
2442
2430
  versionNumber?: number | undefined;
@@ -2487,6 +2475,22 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
2487
2475
  id?: string | undefined;
2488
2476
  createdAt?: Date | undefined;
2489
2477
  updatedAt?: Date | undefined;
2478
+ }> | zod0.ZodSafeParseSuccess<{
2479
+ providerConfigId?: string | undefined;
2480
+ guardrailConfigId?: string | undefined;
2481
+ enabled?: boolean | undefined;
2482
+ parameters?: Record<string, unknown> | null | undefined;
2483
+ id?: string | undefined;
2484
+ createdAt?: Date | undefined;
2485
+ updatedAt?: Date | undefined;
2486
+ }> | zod0.ZodSafeParseError<{
2487
+ providerConfigId?: string | undefined;
2488
+ guardrailConfigId?: string | undefined;
2489
+ enabled?: boolean | undefined;
2490
+ parameters?: Record<string, unknown> | null | undefined;
2491
+ id?: string | undefined;
2492
+ createdAt?: Date | undefined;
2493
+ updatedAt?: Date | undefined;
2490
2494
  }> | zod0.ZodSafeParseSuccess<{
2491
2495
  requestId?: string | undefined;
2492
2496
  configId?: string | null | undefined;
@@ -2650,14 +2654,6 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
2650
2654
  id: string;
2651
2655
  createdAt: Date;
2652
2656
  updatedAt: Date;
2653
- } | {
2654
- providerConfigId: string;
2655
- guardrailConfigId: string;
2656
- enabled: boolean;
2657
- id: string;
2658
- createdAt: Date;
2659
- updatedAt: Date;
2660
- parameters?: Record<string, unknown> | null | undefined;
2661
2657
  } | {
2662
2658
  datasetId: string;
2663
2659
  versionNumber: number;
@@ -2683,6 +2679,14 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
2683
2679
  id: string;
2684
2680
  createdAt: Date;
2685
2681
  updatedAt: Date;
2682
+ } | {
2683
+ providerConfigId: string;
2684
+ guardrailConfigId: string;
2685
+ enabled: boolean;
2686
+ id: string;
2687
+ createdAt: Date;
2688
+ updatedAt: Date;
2689
+ parameters?: Record<string, unknown> | null | undefined;
2686
2690
  } | {
2687
2691
  requestId: string;
2688
2692
  provider: string;
@@ -2805,14 +2809,6 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
2805
2809
  id?: string | undefined;
2806
2810
  createdAt?: Date | undefined;
2807
2811
  updatedAt?: Date | undefined;
2808
- } | {
2809
- providerConfigId?: string | undefined;
2810
- guardrailConfigId?: string | undefined;
2811
- enabled?: boolean | undefined;
2812
- parameters?: Record<string, unknown> | null | undefined;
2813
- id?: string | undefined;
2814
- createdAt?: Date | undefined;
2815
- updatedAt?: Date | undefined;
2816
2812
  } | {
2817
2813
  datasetId?: string | undefined;
2818
2814
  versionNumber?: number | undefined;
@@ -2838,6 +2834,14 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
2838
2834
  id?: string | undefined;
2839
2835
  createdAt?: Date | undefined;
2840
2836
  updatedAt?: Date | undefined;
2837
+ } | {
2838
+ providerConfigId?: string | undefined;
2839
+ guardrailConfigId?: string | undefined;
2840
+ enabled?: boolean | undefined;
2841
+ parameters?: Record<string, unknown> | null | undefined;
2842
+ id?: string | undefined;
2843
+ createdAt?: Date | undefined;
2844
+ updatedAt?: Date | undefined;
2841
2845
  } | {
2842
2846
  requestId?: string | undefined;
2843
2847
  configId?: string | null | undefined;
@@ -1,7 +1,7 @@
1
1
  import { ColumnType, Dialect, Generated, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
2
+ import { NeonQueryFunction } from "@neondatabase/serverless";
2
3
  import * as zod0 from "zod";
3
4
  import { z } from "zod";
4
- import { NeonQueryFunction } from "@neondatabase/serverless";
5
5
 
6
6
  //#region src/db/schema.d.ts
7
7
  declare const configsSchema: z.ZodObject<{
@@ -1927,6 +1927,10 @@ declare function runSchemaSQL(sql: (query: string) => Promise<unknown>, schemaNa
1927
1927
  /**
1928
1928
  * Create a Neon SQL function from various connection types
1929
1929
  *
1930
+ * The Neon serverless library returns a tagged template function from neon(),
1931
+ * but for conventional string queries we need to use sql.query() instead.
1932
+ * This helper wraps the neon instance to provide a simple query function.
1933
+ *
1930
1934
  * @param rawConnection - neon() function, connection string, or undefined (uses env vars)
1931
1935
  * @returns SQL function that can be used with runSchemaSQL, or null if unable to create
1932
1936
  */
@@ -2115,22 +2119,6 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
2115
2119
  id: string;
2116
2120
  createdAt: Date;
2117
2121
  updatedAt: Date;
2118
- }> | zod0.ZodSafeParseSuccess<{
2119
- providerConfigId: string;
2120
- guardrailConfigId: string;
2121
- enabled: boolean;
2122
- id: string;
2123
- createdAt: Date;
2124
- updatedAt: Date;
2125
- parameters?: Record<string, unknown> | null | undefined;
2126
- }> | zod0.ZodSafeParseError<{
2127
- providerConfigId: string;
2128
- guardrailConfigId: string;
2129
- enabled: boolean;
2130
- id: string;
2131
- createdAt: Date;
2132
- updatedAt: Date;
2133
- parameters?: Record<string, unknown> | null | undefined;
2134
2122
  }> | zod0.ZodSafeParseSuccess<{
2135
2123
  datasetId: string;
2136
2124
  versionNumber: number;
@@ -2181,6 +2169,22 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
2181
2169
  id: string;
2182
2170
  createdAt: Date;
2183
2171
  updatedAt: Date;
2172
+ }> | zod0.ZodSafeParseSuccess<{
2173
+ providerConfigId: string;
2174
+ guardrailConfigId: string;
2175
+ enabled: boolean;
2176
+ id: string;
2177
+ createdAt: Date;
2178
+ updatedAt: Date;
2179
+ parameters?: Record<string, unknown> | null | undefined;
2180
+ }> | zod0.ZodSafeParseError<{
2181
+ providerConfigId: string;
2182
+ guardrailConfigId: string;
2183
+ enabled: boolean;
2184
+ id: string;
2185
+ createdAt: Date;
2186
+ updatedAt: Date;
2187
+ parameters?: Record<string, unknown> | null | undefined;
2184
2188
  }> | zod0.ZodSafeParseSuccess<{
2185
2189
  requestId: string;
2186
2190
  provider: string;
@@ -2421,22 +2425,6 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
2421
2425
  id?: string | undefined;
2422
2426
  createdAt?: Date | undefined;
2423
2427
  updatedAt?: Date | undefined;
2424
- }> | zod0.ZodSafeParseSuccess<{
2425
- providerConfigId?: string | undefined;
2426
- guardrailConfigId?: string | undefined;
2427
- enabled?: boolean | undefined;
2428
- parameters?: Record<string, unknown> | null | undefined;
2429
- id?: string | undefined;
2430
- createdAt?: Date | undefined;
2431
- updatedAt?: Date | undefined;
2432
- }> | zod0.ZodSafeParseError<{
2433
- providerConfigId?: string | undefined;
2434
- guardrailConfigId?: string | undefined;
2435
- enabled?: boolean | undefined;
2436
- parameters?: Record<string, unknown> | null | undefined;
2437
- id?: string | undefined;
2438
- createdAt?: Date | undefined;
2439
- updatedAt?: Date | undefined;
2440
2428
  }> | zod0.ZodSafeParseSuccess<{
2441
2429
  datasetId?: string | undefined;
2442
2430
  versionNumber?: number | undefined;
@@ -2487,6 +2475,22 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
2487
2475
  id?: string | undefined;
2488
2476
  createdAt?: Date | undefined;
2489
2477
  updatedAt?: Date | undefined;
2478
+ }> | zod0.ZodSafeParseSuccess<{
2479
+ providerConfigId?: string | undefined;
2480
+ guardrailConfigId?: string | undefined;
2481
+ enabled?: boolean | undefined;
2482
+ parameters?: Record<string, unknown> | null | undefined;
2483
+ id?: string | undefined;
2484
+ createdAt?: Date | undefined;
2485
+ updatedAt?: Date | undefined;
2486
+ }> | zod0.ZodSafeParseError<{
2487
+ providerConfigId?: string | undefined;
2488
+ guardrailConfigId?: string | undefined;
2489
+ enabled?: boolean | undefined;
2490
+ parameters?: Record<string, unknown> | null | undefined;
2491
+ id?: string | undefined;
2492
+ createdAt?: Date | undefined;
2493
+ updatedAt?: Date | undefined;
2490
2494
  }> | zod0.ZodSafeParseSuccess<{
2491
2495
  requestId?: string | undefined;
2492
2496
  configId?: string | null | undefined;
@@ -2650,14 +2654,6 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
2650
2654
  id: string;
2651
2655
  createdAt: Date;
2652
2656
  updatedAt: Date;
2653
- } | {
2654
- providerConfigId: string;
2655
- guardrailConfigId: string;
2656
- enabled: boolean;
2657
- id: string;
2658
- createdAt: Date;
2659
- updatedAt: Date;
2660
- parameters?: Record<string, unknown> | null | undefined;
2661
2657
  } | {
2662
2658
  datasetId: string;
2663
2659
  versionNumber: number;
@@ -2683,6 +2679,14 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
2683
2679
  id: string;
2684
2680
  createdAt: Date;
2685
2681
  updatedAt: Date;
2682
+ } | {
2683
+ providerConfigId: string;
2684
+ guardrailConfigId: string;
2685
+ enabled: boolean;
2686
+ id: string;
2687
+ createdAt: Date;
2688
+ updatedAt: Date;
2689
+ parameters?: Record<string, unknown> | null | undefined;
2686
2690
  } | {
2687
2691
  requestId: string;
2688
2692
  provider: string;
@@ -2805,14 +2809,6 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
2805
2809
  id?: string | undefined;
2806
2810
  createdAt?: Date | undefined;
2807
2811
  updatedAt?: Date | undefined;
2808
- } | {
2809
- providerConfigId?: string | undefined;
2810
- guardrailConfigId?: string | undefined;
2811
- enabled?: boolean | undefined;
2812
- parameters?: Record<string, unknown> | null | undefined;
2813
- id?: string | undefined;
2814
- createdAt?: Date | undefined;
2815
- updatedAt?: Date | undefined;
2816
2812
  } | {
2817
2813
  datasetId?: string | undefined;
2818
2814
  versionNumber?: number | undefined;
@@ -2838,6 +2834,14 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
2838
2834
  id?: string | undefined;
2839
2835
  createdAt?: Date | undefined;
2840
2836
  updatedAt?: Date | undefined;
2837
+ } | {
2838
+ providerConfigId?: string | undefined;
2839
+ guardrailConfigId?: string | undefined;
2840
+ enabled?: boolean | undefined;
2841
+ parameters?: Record<string, unknown> | null | undefined;
2842
+ id?: string | undefined;
2843
+ createdAt?: Date | undefined;
2844
+ updatedAt?: Date | undefined;
2841
2845
  } | {
2842
2846
  requestId?: string | undefined;
2843
2847
  configId?: string | null | undefined;
package/dist/index.cjs CHANGED
@@ -1,5 +1,5 @@
1
- const require_db = require('./db-CEt1ZD2V.cjs');
2
- const require_neon_dialect = require('./neon-dialect-sUVNvVYh.cjs');
1
+ const require_db = require('./db-CIEpN_uB.cjs');
2
+ const require_neon_dialect = require('./neon-dialect-NI1iA903.cjs');
3
3
  let __llmops_gateway = require("@llmops/gateway");
4
4
  __llmops_gateway = require_db.__toESM(__llmops_gateway);
5
5
  let kysely = require("kysely");