@axiom-lattice/pg-stores 3.0.0 → 3.0.2

package/dist/index.mjs

@@ -1,8 +1,8 @@
 // src/index.ts
-import { Pool as Pool26 } from "pg";
+import { Pool as Pool27 } from "pg";
 
 // src/createPgStoreConfig.ts
-import { Pool as Pool23 } from "pg";
+import { Pool as Pool24 } from "pg";
 
 // src/migrations/migration.ts
 var MigrationManager = class {
@@ -5960,8 +5960,232 @@ var PostgreSQLA2AApiKeyStore = class {
   }
 };
 
-// src/stores/PostgreSQLScheduleStorage.ts
+// src/stores/PostgreSQLAgentWebAppStore.ts
+import { randomUUID as randomUUID2 } from "crypto";
 import { Pool as Pool17 } from "pg";
+
+// src/migrations/agent_web_apps_migration.ts
+var createAgentWebAppsTable = {
+  version: 169,
+  name: "create_agent_web_apps_table",
+  up: async (client) => {
+    await client.query(`
+      CREATE TABLE IF NOT EXISTS lattice_agent_web_apps (
+        id VARCHAR(255) PRIMARY KEY,
+        tenant_id VARCHAR(255) NOT NULL,
+        assistant_id VARCHAR(255) NOT NULL,
+        name VARCHAR(255) NOT NULL,
+        description TEXT,
+        status VARCHAR(32) NOT NULL DEFAULT 'draft'
+          CONSTRAINT chk_lattice_agent_web_apps_status CHECK (status IN ('draft', 'active', 'disabled')),
+        integration JSONB NOT NULL,
+        scope JSONB NOT NULL,
+        features JSONB NOT NULL,
+        appearance JSONB NOT NULL DEFAULT '{}'::jsonb,
+        created_at TIMESTAMPTZ NOT NULL DEFAULT date_trunc('milliseconds', NOW()),
+        updated_at TIMESTAMPTZ NOT NULL DEFAULT date_trunc('milliseconds', NOW())
+      )
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_agent_web_apps_tenant_assistant
+      ON lattice_agent_web_apps(tenant_id, assistant_id)
+    `);
+  },
+  down: async (client) => {
+    await client.query(
+      "DROP INDEX IF EXISTS idx_lattice_agent_web_apps_tenant_assistant"
+    );
+    await client.query("DROP TABLE IF EXISTS lattice_agent_web_apps");
+  }
+};
+
+// src/stores/PostgreSQLAgentWebAppStore.ts
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function isStringArray(value) {
+  return Array.isArray(value) && value.every((item) => typeof item === "string");
+}
+function isIntegration(value) {
+  return isRecord(value) && value.type === "react_sdk";
+}
+function isScope(value) {
+  return isRecord(value) && typeof value.defaultProjectId === "string" && isStringArray(value.allowedProjectIds) && (value.defaultModelKey === void 0 || typeof value.defaultModelKey === "string") && (value.allowedModelKeys === void 0 || isStringArray(value.allowedModelKeys));
+}
+function isFeatures(value) {
+  return isRecord(value) && typeof value.projectSelector === "boolean" && typeof value.modelSelector === "boolean" && typeof value.threadManagement === "boolean" && typeof value.attachments === "boolean" && typeof value.hitl === "boolean" && typeof value.genUI === "boolean";
+}
+function isAppearance(value) {
+  return isRecord(value) && (value.title === void 0 || typeof value.title === "string") && (value.welcomeMessage === void 0 || typeof value.welcomeMessage === "string") && (value.primaryColor === void 0 || typeof value.primaryColor === "string");
+}
+function isStatus(value) {
+  return value === "draft" || value === "active" || value === "disabled";
+}
+function mapRow(row) {
+  if (!isStatus(row.status) || !isIntegration(row.integration) || !isScope(row.scope) || !isFeatures(row.features) || !isAppearance(row.appearance)) {
+    throw new Error(`Invalid agent web app row: ${row.id}`);
+  }
+  return {
+    id: row.id,
+    tenantId: row.tenant_id,
+    assistantId: row.assistant_id,
+    name: row.name,
+    description: row.description ?? void 0,
+    status: row.status,
+    integration: row.integration,
+    scope: row.scope,
+    features: row.features,
+    appearance: row.appearance,
+    createdAt: row.created_at,
+    updatedAt: row.updated_at
+  };
+}
+var PostgreSQLAgentWebAppStore = class {
+  constructor(options) {
+    this.initialized = false;
+    this.ownsPool = true;
+    this.initPromise = null;
+    if (options.pool) {
+      this.pool = options.pool;
+      this.ownsPool = false;
+      this.initialized = true;
+      return;
+    }
+    this.pool = typeof options.poolConfig === "string" ? new Pool17({ connectionString: options.poolConfig }) : options.poolConfig ? new Pool17(options.poolConfig) : (() => {
+      throw new Error("Either pool or poolConfig must be provided");
+    })();
+    this.migrationManager = new MigrationManager(this.pool);
+    this.migrationManager.register(createAgentWebAppsTable);
+    if (options.autoMigrate !== false) {
+      this.startInitialization();
+    }
+  }
+  async initialize() {
+    if (this.initialized) return;
+    if (this.initPromise) return this.initPromise;
+    return this.startInitialization();
+  }
+  startInitialization() {
+    this.initPromise = this.migrationManager.migrate().then(() => {
+      this.initialized = true;
+    });
+    void this.initPromise.catch(() => void 0);
+    return this.initPromise;
+  }
+  async dispose() {
+    if (this.ownsPool) await this.pool.end();
+  }
+  async ensureInitialized() {
+    if (!this.initialized) await this.initialize();
+  }
+  async list(tenantId, assistantId) {
+    await this.ensureInitialized();
+    const result = assistantId === void 0 ? await this.pool.query(
+      `SELECT * FROM lattice_agent_web_apps
+         WHERE tenant_id = $1
+         ORDER BY created_at DESC, id DESC`,
+      [tenantId]
+    ) : await this.pool.query(
+      `SELECT * FROM lattice_agent_web_apps
+         WHERE tenant_id = $1 AND assistant_id = $2
+         ORDER BY created_at DESC, id DESC`,
+      [tenantId, assistantId]
+    );
+    return result.rows.map(mapRow);
+  }
+  async getById(tenantId, webAppId) {
+    await this.ensureInitialized();
+    const result = await this.pool.query(
+      `SELECT * FROM lattice_agent_web_apps
+       WHERE tenant_id = $1 AND id = $2`,
+      [tenantId, webAppId]
+    );
+    return result.rows[0] ? mapRow(result.rows[0]) : null;
+  }
+  async findById(webAppId) {
+    await this.ensureInitialized();
+    const result = await this.pool.query(
+      `SELECT * FROM lattice_agent_web_apps
+       WHERE id = $1`,
+      [webAppId]
+    );
+    return result.rows[0] ? mapRow(result.rows[0]) : null;
+  }
+  async create(tenantId, input) {
+    await this.ensureInitialized();
+    const id = `webapp_${randomUUID2().replace(/-/g, "")}`;
+    const result = await this.pool.query(
+      `INSERT INTO lattice_agent_web_apps
+       (id, tenant_id, assistant_id, name, description, status, integration, scope, features, appearance)
+       VALUES ($1, $2, $3, $4, $5, $6, $7::jsonb, $8::jsonb, $9::jsonb, $10::jsonb)
+       RETURNING *`,
+      [
+        id,
+        tenantId,
+        input.assistantId,
+        input.name,
+        input.description ?? null,
+        "draft",
+        JSON.stringify(input.integration),
+        JSON.stringify(input.scope),
+        JSON.stringify(input.features),
+        JSON.stringify(input.appearance)
+      ]
+    );
+    return mapRow(result.rows[0]);
+  }
+  async update(tenantId, webAppId, patch, options) {
+    await this.ensureInitialized();
+    const assignments = [];
+    const values = [];
+    const add = (column, value) => {
+      values.push(value);
+      assignments.push(`${column} = $${values.length}`);
+    };
+    if (patch.name !== void 0) add("name", patch.name);
+    if (Object.prototype.hasOwnProperty.call(patch, "description")) {
+      add("description", patch.description ?? null);
+    }
+    if (patch.scope !== void 0) {
+      add("scope", JSON.stringify(patch.scope));
+    }
+    if (patch.features !== void 0) {
+      add("features", JSON.stringify(patch.features));
+    }
+    if (patch.appearance !== void 0) {
+      add("appearance", JSON.stringify(patch.appearance));
+    }
+    if (patch.status !== void 0) add("status", patch.status);
+    if (assignments.length === 0) return this.getById(tenantId, webAppId);
+    values.push(tenantId, webAppId);
+    const tenantParam = values.length - 1;
+    const idParam = values.length;
+    const expectedUpdatedAtClause = options?.expectedUpdatedAt ? ` AND date_trunc('milliseconds', updated_at) = $${values.push(options.expectedUpdatedAt)}` : "";
+    const result = await this.pool.query(
+      `UPDATE lattice_agent_web_apps
+       SET ${assignments.join(", ")}, updated_at = GREATEST(
+         date_trunc('milliseconds', clock_timestamp()),
+         updated_at + interval '1 millisecond'
+       )
+       WHERE tenant_id = $${tenantParam} AND id = $${idParam}${expectedUpdatedAtClause}
+       RETURNING *`,
+      values
+    );
+    return result.rows[0] ? mapRow(result.rows[0]) : null;
+  }
+  async delete(tenantId, webAppId) {
+    await this.ensureInitialized();
+    const result = await this.pool.query(
+      `DELETE FROM lattice_agent_web_apps
+       WHERE tenant_id = $1 AND id = $2`,
+      [tenantId, webAppId]
+    );
+    return result.rowCount === 1;
+  }
+};
+
+// src/stores/PostgreSQLScheduleStorage.ts
+import { Pool as Pool18 } from "pg";
 import {
   ScheduledTaskStatus
 } from "@axiom-lattice/protocols";
@@ -6138,9 +6362,9 @@ var PostgreSQLScheduleStorage = class {
       return;
     }
     if (typeof options.poolConfig === "string") {
-      this.pool = new Pool17({ connectionString: options.poolConfig });
+      this.pool = new Pool18({ connectionString: options.poolConfig });
     } else if (options.poolConfig) {
-      this.pool = new Pool17(options.poolConfig);
+      this.pool = new Pool18(options.poolConfig);
     } else {
       throw new Error("Either pool or poolConfig must be provided");
     }
@@ -6578,7 +6802,7 @@ var PostgreSQLScheduleStorage = class {
 };
 
 // src/stores/PostgreSQLTaskStore.ts
-import { Pool as Pool18 } from "pg";
+import { Pool as Pool19 } from "pg";
 
 // src/migrations/task_migration.ts
 var createTasksTable = {
@@ -6768,9 +6992,9 @@ var PostgreSQLTaskStore = class {
       return;
     }
     if (typeof options.poolConfig === "string") {
-      this.pool = new Pool18({ connectionString: options.poolConfig });
+      this.pool = new Pool19({ connectionString: options.poolConfig });
     } else if (options.poolConfig) {
-      this.pool = new Pool18(options.poolConfig);
+      this.pool = new Pool19(options.poolConfig);
     } else {
       throw new Error("Either pool or poolConfig must be provided");
     }
@@ -7516,7 +7740,7 @@ var PostgreSQLTaskWorkItemStore = class {
 };
 
 // src/stores/MenuStore.ts
-import { Pool as Pool19 } from "pg";
+import { Pool as Pool20 } from "pg";
 
 // src/migrations/menu_items_migration.ts
 var createMenuItemsTable = {
@@ -7593,7 +7817,7 @@ var MenuStore = class {
       this.initialized = true;
       return;
     }
-    this.pool = typeof options.poolConfig === "string" ? new Pool19({ connectionString: options.poolConfig }) : options.poolConfig ? new Pool19(options.poolConfig) : (() => {
+    this.pool = typeof options.poolConfig === "string" ? new Pool20({ connectionString: options.poolConfig }) : options.poolConfig ? new Pool20(options.poolConfig) : (() => {
       throw new Error("Either pool or poolConfig must be provided");
     })();
     this.migrationManager = new MigrationManager(this.pool);
@@ -7748,7 +7972,7 @@ var MenuStore = class {
 };
 
 // src/stores/PostgresSharedResourceStore.ts
-import { Pool as Pool20 } from "pg";
+import { Pool as Pool21 } from "pg";
 
 // src/migrations/shared_resources_migration.ts
 var createSharedResourcesTable = {
@@ -7802,7 +8026,7 @@ var PostgresSharedResourceStore = class {
       this.initialized = true;
       return;
     }
-    this.pool = typeof options.poolConfig === "string" ? new Pool20({ connectionString: options.poolConfig }) : options.poolConfig ? new Pool20(options.poolConfig) : (() => {
+    this.pool = typeof options.poolConfig === "string" ? new Pool21({ connectionString: options.poolConfig }) : options.poolConfig ? new Pool21(options.poolConfig) : (() => {
       throw new Error("Either pool or poolConfig must be provided");
     })();
     this.migrationManager = new MigrationManager(this.pool);
@@ -7971,7 +8195,7 @@ var PostgresSharedResourceStore = class {
 };
 
 // src/stores/PostgreSQLCollectionStore.ts
-import { Pool as Pool21 } from "pg";
+import { Pool as Pool22 } from "pg";
 var PostgreSQLCollectionStore = class {
   constructor(options) {
     this.initialized = false;
@@ -7984,9 +8208,9 @@ var PostgreSQLCollectionStore = class {
       return;
     }
     if (typeof options.poolConfig === "string") {
-      this.pool = new Pool21({ connectionString: options.poolConfig });
+      this.pool = new Pool22({ connectionString: options.poolConfig });
     } else if (options.poolConfig) {
-      this.pool = new Pool21(options.poolConfig);
+      this.pool = new Pool22(options.poolConfig);
     } else {
       throw new Error("Either pool or poolConfig must be provided");
     }
@@ -8101,7 +8325,7 @@ var PostgreSQLCollectionStore = class {
 import { PostgresSaver } from "@langchain/langgraph-checkpoint-postgres";
 
 // src/PGVectorStoreProvider.ts
-import { Pool as Pool22 } from "pg";
+import { Pool as Pool23 } from "pg";
 import { PGVectorStore } from "@langchain/community/vectorstores/pgvector";
 import { Document } from "@langchain/core/documents";
 import { embeddingsLatticeManager } from "@axiom-lattice/core";
@@ -8175,7 +8399,7 @@ var PGVectorStoreProvider = class {
   }
 };
 function createPGVectorStoreProvider(connectionString) {
-  const pool = new Pool22({ connectionString });
+  const pool = new Pool23({ connectionString });
   return {
     provider: new PGVectorStoreProvider(pool, connectionString),
     pool
@@ -8526,7 +8750,7 @@ var addTaskWorkItemEventKeyMigration = {
 
 // src/createPgStoreConfig.ts
 async function createPgStoreConfig(connectionString) {
-  const pool = new Pool23({ connectionString });
+  const pool = new Pool24({ connectionString });
   const mm = new MigrationManager(pool);
   mm.register(createThreadsTable);
   mm.register(createScheduledTasksTable);
@@ -8580,6 +8804,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(addProjectKindColumn);
   mm.register(addA2AKeyAssistantIds);
   mm.register(addTaskWorkItemEventKeyMigration);
+  mm.register(createAgentWebAppsTable);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {
@@ -8607,6 +8832,7 @@ async function createPgStoreConfig(connectionString) {
     task: new PostgreSQLTaskStore(opts),
     taskWorkItem: taskWorkItemStore,
     a2aApiKey: new PostgreSQLA2AApiKeyStore(opts),
+    agentWebApp: new PostgreSQLAgentWebAppStore(opts),
     schedule: new PostgreSQLScheduleStorage(opts),
     menu: new MenuStore(opts),
     sharedResource: new PostgresSharedResourceStore(opts),
@@ -8617,7 +8843,7 @@ async function createPgStoreConfig(connectionString) {
 }
 
 // src/stores/PostgreSQLSkillStore.ts
-import { Pool as Pool24 } from "pg";
+import { Pool as Pool25 } from "pg";
 var PostgreSQLSkillStore = class {
   constructor(options) {
     this.initialized = false;
@@ -8630,9 +8856,9 @@ var PostgreSQLSkillStore = class {
       return;
     }
     if (typeof options.poolConfig === "string") {
-      this.pool = new Pool24({ connectionString: options.poolConfig });
+      this.pool = new Pool25({ connectionString: options.poolConfig });
     } else if (options.poolConfig) {
-      this.pool = new Pool24(options.poolConfig);
+      this.pool = new Pool25(options.poolConfig);
     } else {
       throw new Error("Either pool or poolConfig must be provided");
     }
@@ -8933,7 +9159,7 @@ var PostgreSQLSkillStore = class {
 };
 
 // src/stores/ChannelIdentityMappingStore.ts
-import { Pool as Pool25 } from "pg";
+import { Pool as Pool26 } from "pg";
 var ChannelIdentityMappingStore = class {
   constructor(options) {
     this.initialized = false;
@@ -8945,7 +9171,7 @@ var ChannelIdentityMappingStore = class {
       this.initialized = true;
       return;
     }
-    this.pool = typeof options.poolConfig === "string" ? new Pool25({ connectionString: options.poolConfig }) : options.poolConfig ? new Pool25(options.poolConfig) : (() => {
+    this.pool = typeof options.poolConfig === "string" ? new Pool26({ connectionString: options.poolConfig }) : options.poolConfig ? new Pool26(options.poolConfig) : (() => {
       throw new Error("Either pool or poolConfig must be provided");
     })();
     this.migrationManager = new MigrationManager(this.pool);
@@ -9169,8 +9395,9 @@ export {
   MenuStore,
   MigrationManager,
   PGVectorStoreProvider,
-  Pool26 as Pool,
+  Pool27 as Pool,
   PostgreSQLA2AApiKeyStore,
+  PostgreSQLAgentWebAppStore,
   PostgreSQLAssistantStore,
   PostgreSQLChannelInstallationStore,
   PostgreSQLCollectionStore,
@@ -9208,6 +9435,7 @@ export {
   changeSkillPrimaryKey,
   changeThreadPrimaryKey,
   createA2AApiKeysTable,
+  createAgentWebAppsTable,
   createAssistantsTable,
   createChannelBindingsTable,
   createChannelIdentityMappingTables,