@axiom-lattice/pg-stores 4.0.0 → 4.0.2

package/dist/index.mjs

@@ -9559,6 +9559,9 @@ import { Pool as Pool25 } from "pg";
 import { PGVectorStore } from "@langchain/community/vectorstores/pgvector";
 import { Document } from "@langchain/core/documents";
 import { embeddingsLatticeManager } from "@axiom-lattice/core";
+import {
+  EMBEDDING_DEFAULT_KEY
+} from "@axiom-lattice/protocols";
 function safeTableName(name) {
   if (!/^[a-z][a-z0-9_-]*$/.test(name)) {
     throw new Error(`Invalid table name: ${name}`);
@@ -9573,16 +9576,20 @@ var PGVectorStoreProvider = class {
   async create(params) {
     const { tableName, embeddingKey } = params;
     const tenantId = params.tenantId ?? "default";
+    const requestedKey = embeddingKey || EMBEDDING_DEFAULT_KEY;
     const embeddings = await embeddingsLatticeManager.getEmbeddingsClientWithTenant(
       tenantId,
-      embeddingKey
-    ) ?? await embeddingsLatticeManager.getEmbeddingsClientWithTenant(
+      requestedKey
+    ) ?? (requestedKey === EMBEDDING_DEFAULT_KEY ? void 0 : await embeddingsLatticeManager.getEmbeddingsClientWithTenant(
+      tenantId,
+      EMBEDDING_DEFAULT_KEY
+    )) ?? await 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 PGVectorStore.initialize(embeddings, {
@@ -10751,6 +10758,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
 import { randomUUID as randomUUID3 } from "crypto";
@@ -11745,6 +11774,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);
@@ -12083,15 +12113,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(
         `
@@ -14560,6 +14597,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(createAgentRunRetentionMigration);
   mm.register(createAgentRunProjectTaskMigration);
   mm.register(agentRunProjectTaskRunningIndexMigration);
+  mm.register(addModelProviderKindDefault);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {