@axiom-lattice/pg-stores 3.3.11 → 3.3.13

package/dist/index.mjs

@@ -9478,19 +9478,18 @@ var PGVectorStoreProvider = class {
   }
   async create(params) {
     const { tableName, embeddingKey } = params;
-    let embeddings;
-    try {
-      embeddings = embeddingsLatticeManager.getEmbeddingsClient(embeddingKey);
-    } catch {
-      try {
-        embeddings = embeddingsLatticeManager.getEmbeddingsClient("default");
-      } catch {
-        throw new Error(
-          `No embedding model registered. Register one first:
-  import { registerEmbeddingsLattice } from "@axiom-lattice/core";
-  registerEmbeddingsLattice("default", new SomeEmbeddings(...));`
-        );
-      }
+    const tenantId = params.tenantId ?? "default";
+    const embeddings = await embeddingsLatticeManager.getEmbeddingsClientWithTenant(
+      tenantId,
+      embeddingKey
+    ) ?? 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.`
+      );
     }
     return PGVectorStore.initialize(embeddings, {
       postgresConnectionOptions: { connectionString: this.connectionString },
@@ -10350,6 +10349,21 @@ var addModelProviderModelDefault = {
     `);
   }
 };
+var addModelProviderModelKind = {
+  version: 193,
+  name: "add_model_provider_model_kind",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_model_provider_models
+      ADD COLUMN IF NOT EXISTS kind VARCHAR(16) NOT NULL DEFAULT 'chat'
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_model_provider_models DROP COLUMN IF EXISTS kind
+    `);
+  }
+};
 
 // src/stores/PostgreSQLCapabilityBundleStore.ts
 import { randomUUID as randomUUID3 } from "crypto";
@@ -11343,6 +11357,7 @@ var PostgreSQLModelProviderStore = class {
     this.migrationManager.register(addModelProviderModelStale);
     this.migrationManager.register(addModelProviderModelExtra);
     this.migrationManager.register(addModelProviderModelDefault);
+    this.migrationManager.register(addModelProviderModelKind);
     if (options.autoMigrate !== false) {
       this.initialize().catch((error) => {
         console.error("Failed to initialize PostgreSQLModelProviderStore:", error);
@@ -11589,7 +11604,7 @@ var PostgreSQLModelProviderStore = class {
     const result = providerId ? await this.pool.query(
       `
           SELECT id, tenant_id, provider_id, model_id, display_name, owned_by,
-                 upstream_created, enabled, stale, extra, is_default, discovered_at
+                 upstream_created, enabled, stale, extra, kind, is_default, discovered_at
           FROM lattice_model_provider_models
           WHERE tenant_id = $1 AND provider_id = $2
           ORDER BY model_id ASC
@@ -11598,7 +11613,7 @@ var PostgreSQLModelProviderStore = class {
     ) : await this.pool.query(
       `
           SELECT id, tenant_id, provider_id, model_id, display_name, owned_by,
-                 upstream_created, enabled, stale, extra, is_default, discovered_at
+                 upstream_created, enabled, stale, extra, kind, is_default, discovered_at
           FROM lattice_model_provider_models
           WHERE tenant_id = $1
           ORDER BY model_id ASC
@@ -11617,7 +11632,7 @@ var PostgreSQLModelProviderStore = class {
     const result = await this.pool.query(
       `
       SELECT id, tenant_id, provider_id, model_id, display_name, owned_by,
-             upstream_created, enabled, stale, extra, is_default, discovered_at
+             upstream_created, enabled, stale, extra, kind, is_default, discovered_at
       FROM lattice_model_provider_models
       WHERE tenant_id = $1 AND provider_id = $2 AND model_id = $3
     `,
@@ -11650,7 +11665,7 @@ var PostgreSQLModelProviderStore = class {
       await client.query("BEGIN");
       const current = await client.query(
         `
-        SELECT model_id, enabled, extra, is_default FROM lattice_model_provider_models
+        SELECT model_id, enabled, extra, kind, is_default FROM lattice_model_provider_models
         WHERE tenant_id = $1 AND provider_id = $2
       `,
         [tenantId, providerId]
@@ -11661,6 +11676,9 @@ var PostgreSQLModelProviderStore = class {
       const currentExtra = new Map(
         current.rows.map((row) => [row.model_id, row.extra])
       );
+      const currentKind = new Map(
+        current.rows.map((row) => [row.model_id, row.kind])
+      );
       const currentIsDefault = new Map(
         current.rows.map((row) => [row.model_id, row.is_default])
       );
@@ -11675,6 +11693,9 @@ var PostgreSQLModelProviderStore = class {
       const effectiveExtra = rows.map(
         (model) => model.extra ?? currentExtra.get(model.id) ?? {}
       );
+      const effectiveKind = rows.map(
+        (model) => model.kind ?? currentKind.get(model.id) ?? "chat"
+      );
       const namesDefault = rows.some((model) => model.isDefault === true);
       if (namesDefault) {
         await client.query(
@@ -11711,15 +11732,15 @@ var PostgreSQLModelProviderStore = class {
           `
           INSERT INTO lattice_model_provider_models (
             id, tenant_id, provider_id, model_id, display_name, owned_by,
-            upstream_created, raw, enabled, stale, extra, is_default, discovered_at
+            upstream_created, raw, enabled, stale, extra, kind, is_default, discovered_at
           )
           SELECT * FROM unnest(
             $1::text[], $2::text[], $3::text[], $4::text[], $5::text[],
             $6::text[], $7::timestamp[], $8::jsonb[], $9::boolean[], $10::boolean[],
-            $11::jsonb[], $12::boolean[], $13::timestamp[]
+            $11::jsonb[], $12::text[], $13::boolean[], $14::timestamp[]
           ) AS t(
             id, tenant_id, provider_id, model_id, display_name, owned_by,
-            upstream_created, raw, enabled, stale, extra, is_default, discovered_at
+            upstream_created, raw, enabled, stale, extra, kind, is_default, discovered_at
           )
           ON CONFLICT (tenant_id, provider_id, model_id) DO UPDATE SET
             display_name = EXCLUDED.display_name,
@@ -11729,6 +11750,7 @@ var PostgreSQLModelProviderStore = class {
             enabled = EXCLUDED.enabled,
             stale = EXCLUDED.stale,
             extra = EXCLUDED.extra,
+            kind = EXCLUDED.kind,
             is_default = EXCLUDED.is_default,
             discovered_at = EXCLUDED.discovered_at
         `,
@@ -11748,6 +11770,7 @@ var PostgreSQLModelProviderStore = class {
             effectiveEnabled,
             rows.map((model) => model.stale ?? false),
             effectiveExtra.map((extra) => JSON.stringify(extra)),
+            effectiveKind,
             effectiveIsDefault,
             rows.map(() => (/* @__PURE__ */ new Date()).toISOString())
           ]
@@ -11854,6 +11877,7 @@ var PostgreSQLModelProviderStore = class {
       enabled: row.enabled,
       stale: row.stale,
       extra: extraOrUndefined(row.extra),
+      kind: row.kind || "chat",
       isDefault: row.is_default,
       discoveredAt: row.discovered_at
     };
@@ -12284,6 +12308,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(dropPluginDeploymentAuthToken);
   mm.register(addModelProviderModelExtra);
   mm.register(addModelProviderModelDefault);
+  mm.register(addModelProviderModelKind);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {