@axiom-lattice/pg-stores 4.0.0 → 4.0.2

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @axiom-lattice/pg-stores@4.0.0 build /home/runner/work/agentic/agentic/packages/pg-stores
+> @axiom-lattice/pg-stores@4.0.2 build /home/runner/work/agentic/agentic/packages/pg-stores
 > tsup src/index.ts --format cjs,esm --dts --sourcemap
 
 CLI Building entry: src/index.ts
@@ -8,13 +8,13 @@
 CLI Target: es2020
 CJS Build start
 ESM Build start
-CJS dist/index.js     528.09 KB
-CJS dist/index.js.map 960.67 KB
-CJS ⚡️ Build success in 969ms
-ESM dist/index.mjs     518.65 KB
-ESM dist/index.mjs.map 960.39 KB
-ESM ⚡️ Build success in 1032ms
+CJS dist/index.js     529.56 KB
+CJS dist/index.js.map 963.70 KB
+CJS ⚡️ Build success in 847ms
+ESM dist/index.mjs     520.06 KB
+ESM dist/index.mjs.map 963.41 KB
+ESM ⚡️ Build success in 848ms
 DTS Build start
-DTS ⚡️ Build success in 11614ms
+DTS ⚡️ Build success in 10223ms
 DTS dist/index.d.ts  110.68 KB
 DTS dist/index.d.mts 110.68 KB

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # @axiom-lattice/pg-stores
 
+## 4.0.2
+
+### Patch Changes
+
+- Updated dependencies [a75acf3]
+  - @axiom-lattice/protocols@6.1.0
+  - @axiom-lattice/core@10.0.2
+
+## 4.0.1
+
+### Patch Changes
+
+- Updated dependencies [11f6d6a]
+  - @axiom-lattice/core@10.0.1
+
 ## 4.0.0
 
 ### Major Changes

package/dist/index.js

@@ -9692,6 +9692,7 @@ var import_pg25 = require("pg");
 var import_pgvector = require("@langchain/community/vectorstores/pgvector");
 var import_documents = require("@langchain/core/documents");
 var import_core7 = require("@axiom-lattice/core");
+var import_protocols6 = require("@axiom-lattice/protocols");
 function safeTableName(name) {
   if (!/^[a-z][a-z0-9_-]*$/.test(name)) {
     throw new Error(`Invalid table name: ${name}`);
@@ -9706,16 +9707,20 @@ var PGVectorStoreProvider = class {
   async create(params) {
     const { tableName, embeddingKey } = params;
     const tenantId = params.tenantId ?? "default";
+    const requestedKey = embeddingKey || import_protocols6.EMBEDDING_DEFAULT_KEY;
     const embeddings = await import_core7.embeddingsLatticeManager.getEmbeddingsClientWithTenant(
       tenantId,
-      embeddingKey
-    ) ?? await import_core7.embeddingsLatticeManager.getEmbeddingsClientWithTenant(
+      requestedKey
+    ) ?? (requestedKey === import_protocols6.EMBEDDING_DEFAULT_KEY ? void 0 : await import_core7.embeddingsLatticeManager.getEmbeddingsClientWithTenant(
+      tenantId,
+      import_protocols6.EMBEDDING_DEFAULT_KEY
+    )) ?? await import_core7.embeddingsLatticeManager.getEmbeddingsClientWithTenant(
       tenantId,
       "default"
     );
     if (!embeddings) {
       throw new Error(
-        `No embedding model "${embeddingKey}" registered for tenant "${tenantId}", and no deployment default. Configure one in the model provider panel.`
+        `No embedding model "${embeddingKey}" registered for tenant "${tenantId}", and no tenant or deployment default. Configure one in the model service panel.`
       );
     }
     return import_pgvector.PGVectorStore.initialize(embeddings, {
@@ -10884,6 +10889,28 @@ var addModelProviderModelKind = {
     `);
   }
 };
+var addModelProviderKindDefault = {
+  version: 207,
+  name: "add_model_provider_kind_default",
+  up: async (client) => {
+    await client.query(`
+      DROP INDEX IF EXISTS uk_lattice_model_provider_models_tenant_default
+    `);
+    await client.query(`
+      CREATE UNIQUE INDEX IF NOT EXISTS uk_lattice_model_provider_models_tenant_kind_default
+      ON lattice_model_provider_models (tenant_id, kind) WHERE is_default
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      DROP INDEX IF EXISTS uk_lattice_model_provider_models_tenant_kind_default
+    `);
+    await client.query(`
+      CREATE UNIQUE INDEX IF NOT EXISTS uk_lattice_model_provider_models_tenant_default
+      ON lattice_model_provider_models (tenant_id) WHERE is_default
+    `);
+  }
+};
 
 // src/stores/PostgreSQLCapabilityBundleStore.ts
 var import_crypto4 = require("crypto");
@@ -11878,6 +11905,7 @@ var PostgreSQLModelProviderStore = class {
     this.migrationManager.register(addModelProviderModelExtra);
     this.migrationManager.register(addModelProviderModelDefault);
     this.migrationManager.register(addModelProviderModelKind);
+    this.migrationManager.register(addModelProviderKindDefault);
     if (options.autoMigrate !== false) {
       this.initialize().catch((error) => {
         console.error("Failed to initialize PostgreSQLModelProviderStore:", error);
@@ -12216,15 +12244,22 @@ var PostgreSQLModelProviderStore = class {
       const effectiveKind = rows.map(
         (model) => model.kind ?? currentKind.get(model.id) ?? "chat"
       );
-      const namesDefault = rows.some((model) => model.isDefault === true);
-      if (namesDefault) {
+      const namedDefaultKinds = /* @__PURE__ */ new Set();
+      rows.forEach((model, index) => {
+        if (model.isDefault === true) {
+          namedDefaultKinds.add(effectiveKind[index]);
+        }
+      });
+      for (const kind of namedDefaultKinds) {
         await client.query(
-          `UPDATE lattice_model_provider_models SET is_default = FALSE WHERE tenant_id = $1`,
-          [tenantId]
+          `UPDATE lattice_model_provider_models
+              SET is_default = FALSE
+            WHERE tenant_id = $1 AND kind = $2 AND is_default`,
+          [tenantId, kind]
         );
       }
       const effectiveIsDefault = rows.map(
-        (model) => model.isDefault ?? (namesDefault ? false : currentIsDefault.get(model.id)) ?? false
+        (model, index) => model.isDefault ?? (namedDefaultKinds.has(effectiveKind[index]) ? false : currentIsDefault.get(model.id)) ?? false
       );
       await client.query(
         `
@@ -13697,7 +13732,7 @@ function isActiveRunViolation(error) {
 
 // src/stores/PostgreSQLChannelReplyJobStore.ts
 var import_pg31 = require("pg");
-var import_protocols6 = require("@axiom-lattice/protocols");
+var import_protocols7 = require("@axiom-lattice/protocols");
 var DB_CLOCK_MS2 = "FLOOR(EXTRACT(EPOCH FROM clock_timestamp()) * 1000)::bigint";
 var COLUMNS = `run_id, tenant_id, assistant_id, thread_id, adapter_channel,
   channel_installation_id, reply_target, input_message_id, stream_cursor, status,
@@ -13788,7 +13823,7 @@ var PostgreSQLChannelReplyJobStore = class {
   }
   /** Creates a pending job with database-clock created/updated timestamps. */
   async create(input) {
-    (0, import_protocols6.assertValidCreateChannelReplyJobInput)(input);
+    (0, import_protocols7.assertValidCreateChannelReplyJobInput)(input);
     const result = await this.pool.query(
       `WITH clock AS (SELECT ${DB_CLOCK_MS2} AS now_ms)
        INSERT INTO lattice_channel_reply_jobs (
@@ -13839,7 +13874,7 @@ var PostgreSQLChannelReplyJobStore = class {
           AND (lease_owner IS NULL OR lease_expires_at IS NULL OR lease_expires_at <= $1)
         ORDER BY available_at ASC, created_at ASC, run_id ASC
         LIMIT $2`,
-      [now, (0, import_protocols6.normalizeChannelReplyJobListLimit)(limit)]
+      [now, (0, import_protocols7.normalizeChannelReplyJobListLimit)(limit)]
     );
     return result.rows.map(mapRow9);
   }
@@ -13965,7 +14000,7 @@ var PostgreSQLChannelReplyJobStore = class {
            ORDER BY updated_at ASC, run_id ASC
            LIMIT $2
         )`,
-      [before, (0, import_protocols6.normalizeChannelReplyJobListLimit)(limit)]
+      [before, (0, import_protocols7.normalizeChannelReplyJobListLimit)(limit)]
     );
     return result.rowCount ?? 0;
   }
@@ -14419,7 +14454,7 @@ var PostgreSQLCheckpointStorage = class {
 };
 
 // src/stores/PostgreSQLStore.ts
-var import_protocols7 = require("@axiom-lattice/protocols");
+var import_protocols8 = require("@axiom-lattice/protocols");
 function mapRow10(row) {
   return {
     namespace: row.namespace,
@@ -14438,9 +14473,9 @@ var PostgreSQLStore = class {
     this.pool = pool;
   }
   async put(tenantId, input) {
-    (0, import_protocols7.assertStoreIdentity)(input.namespace, input.key);
+    (0, import_protocols8.assertStoreIdentity)(input.namespace, input.key);
     const now = /* @__PURE__ */ new Date();
-    const expiresAt = (0, import_protocols7.storeExpiresAt)(input.ttlMinutes, now) ?? null;
+    const expiresAt = (0, import_protocols8.storeExpiresAt)(input.ttlMinutes, now) ?? null;
     const result = await this.pool.query(
       `INSERT INTO lattice_store_items
         (tenant_id, namespace, key, value, created_at, updated_at, expires_at)
@@ -14455,7 +14490,7 @@ var PostgreSQLStore = class {
     return mapRow10(result.rows[0]);
   }
   async get(tenantId, namespace, key) {
-    (0, import_protocols7.assertStoreIdentity)(namespace, key);
+    (0, import_protocols8.assertStoreIdentity)(namespace, key);
     const result = await this.pool.query(
       `SELECT tenant_id, namespace, key, value, created_at, updated_at, expires_at
        FROM lattice_store_items
@@ -14466,7 +14501,7 @@ var PostgreSQLStore = class {
     return result.rows[0] ? mapRow10(result.rows[0]) : null;
   }
   async delete(tenantId, namespace, key) {
-    (0, import_protocols7.assertStoreIdentity)(namespace, key);
+    (0, import_protocols8.assertStoreIdentity)(namespace, key);
     const result = await this.pool.query(
       `DELETE FROM lattice_store_items
        WHERE tenant_id = $1 AND namespace = $2 AND key = $3`,
@@ -14481,7 +14516,7 @@ var PostgreSQLStore = class {
     ];
     const params = [tenantId, /* @__PURE__ */ new Date()];
     if (input.namespacePrefix) {
-      (0, import_protocols7.assertStoreNamespacePrefix)(input.namespacePrefix);
+      (0, import_protocols8.assertStoreNamespacePrefix)(input.namespacePrefix);
     }
     if (input.namespacePrefix && input.namespacePrefix.length > 0) {
       params.push(input.namespacePrefix);
@@ -14495,9 +14530,9 @@ var PostgreSQLStore = class {
       params.push(input.query.replace(/[\\%_]/g, "\\$&"));
       conditions.push(`value::text ILIKE '%' || $${params.length} || '%' ESCAPE '\\'`);
     }
-    params.push((0, import_protocols7.storePaginationValue)(input.limit, 10));
+    params.push((0, import_protocols8.storePaginationValue)(input.limit, 10));
     const limitParam = params.length;
-    params.push((0, import_protocols7.storePaginationValue)(input.offset, 0));
+    params.push((0, import_protocols8.storePaginationValue)(input.offset, 0));
     const offsetParam = params.length;
     const result = await this.pool.query(
       `SELECT tenant_id, namespace, key, value, created_at, updated_at, expires_at
@@ -14510,7 +14545,7 @@ var PostgreSQLStore = class {
     return result.rows.map(mapRow10);
   }
   async listNamespaces(tenantId, prefix = [], limit = 100) {
-    (0, import_protocols7.assertStoreNamespacePrefix)(prefix);
+    (0, import_protocols8.assertStoreNamespacePrefix)(prefix);
     const conditions = [
       "tenant_id = $1",
       "(expires_at IS NULL OR expires_at > $2)"
@@ -14520,7 +14555,7 @@ var PostgreSQLStore = class {
       params.push(prefix);
       conditions.push(`namespace[1:array_length($${params.length}::text[], 1)] = $${params.length}`);
     }
-    params.push((0, import_protocols7.storePaginationValue)(limit, 100));
+    params.push((0, import_protocols8.storePaginationValue)(limit, 100));
     const result = await this.pool.query(
       `SELECT DISTINCT namespace
        FROM lattice_store_items
@@ -14532,7 +14567,7 @@ var PostgreSQLStore = class {
     return result.rows.map((row) => row.namespace);
   }
   async deleteExpired(now, limit) {
-    const bounded = (0, import_protocols7.storePaginationValue)(limit, 0);
+    const bounded = (0, import_protocols8.storePaginationValue)(limit, 0);
     const result = await this.pool.query(
       `WITH expired AS (
          SELECT ctid FROM lattice_store_items
@@ -14676,6 +14711,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(createAgentRunRetentionMigration);
   mm.register(createAgentRunProjectTaskMigration);
   mm.register(agentRunProjectTaskRunningIndexMigration);
+  mm.register(addModelProviderKindDefault);
   await mm.migrate();
   const checkpoint = import_langgraph_checkpoint_postgres.PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {