@axiom-lattice/pg-stores 3.0.0 → 3.0.2

package/dist/index.js

@@ -35,8 +35,9 @@ __export(index_exports, {
   MenuStore: () => MenuStore,
   MigrationManager: () => MigrationManager,
   PGVectorStoreProvider: () => PGVectorStoreProvider,
-  Pool: () => import_pg26.Pool,
+  Pool: () => import_pg27.Pool,
   PostgreSQLA2AApiKeyStore: () => PostgreSQLA2AApiKeyStore,
+  PostgreSQLAgentWebAppStore: () => PostgreSQLAgentWebAppStore,
   PostgreSQLAssistantStore: () => PostgreSQLAssistantStore,
   PostgreSQLChannelInstallationStore: () => PostgreSQLChannelInstallationStore,
   PostgreSQLCollectionStore: () => PostgreSQLCollectionStore,
@@ -74,6 +75,7 @@ __export(index_exports, {
   changeSkillPrimaryKey: () => changeSkillPrimaryKey,
   changeThreadPrimaryKey: () => changeThreadPrimaryKey,
   createA2AApiKeysTable: () => createA2AApiKeysTable,
+  createAgentWebAppsTable: () => createAgentWebAppsTable,
   createAssistantsTable: () => createAssistantsTable,
   createChannelBindingsTable: () => createChannelBindingsTable,
   createChannelIdentityMappingTables: () => createChannelIdentityMappingTables,
@@ -107,10 +109,10 @@ __export(index_exports, {
   safeParse: () => safeParse
 });
 module.exports = __toCommonJS(index_exports);
-var import_pg26 = require("pg");
+var import_pg27 = require("pg");
 
 // src/createPgStoreConfig.ts
-var import_pg23 = require("pg");
+var import_pg24 = require("pg");
 
 // src/migrations/migration.ts
 var MigrationManager = class {
@@ -6068,8 +6070,232 @@ var PostgreSQLA2AApiKeyStore = class {
   }
 };
 
-// src/stores/PostgreSQLScheduleStorage.ts
+// src/stores/PostgreSQLAgentWebAppStore.ts
+var import_crypto3 = require("crypto");
 var import_pg17 = require("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 import_pg17.Pool({ connectionString: options.poolConfig }) : options.poolConfig ? new import_pg17.Pool(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_${(0, import_crypto3.randomUUID)().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
+var import_pg18 = require("pg");
 var import_protocols = require("@axiom-lattice/protocols");
 
 // src/migrations/schedule_migrations.ts
@@ -6244,9 +6470,9 @@ var PostgreSQLScheduleStorage = class {
       return;
     }
     if (typeof options.poolConfig === "string") {
-      this.pool = new import_pg17.Pool({ connectionString: options.poolConfig });
+      this.pool = new import_pg18.Pool({ connectionString: options.poolConfig });
     } else if (options.poolConfig) {
-      this.pool = new import_pg17.Pool(options.poolConfig);
+      this.pool = new import_pg18.Pool(options.poolConfig);
     } else {
       throw new Error("Either pool or poolConfig must be provided");
     }
@@ -6684,7 +6910,7 @@ var PostgreSQLScheduleStorage = class {
 };
 
 // src/stores/PostgreSQLTaskStore.ts
-var import_pg18 = require("pg");
+var import_pg19 = require("pg");
 
 // src/migrations/task_migration.ts
 var createTasksTable = {
@@ -6874,9 +7100,9 @@ var PostgreSQLTaskStore = class {
       return;
     }
     if (typeof options.poolConfig === "string") {
-      this.pool = new import_pg18.Pool({ connectionString: options.poolConfig });
+      this.pool = new import_pg19.Pool({ connectionString: options.poolConfig });
     } else if (options.poolConfig) {
-      this.pool = new import_pg18.Pool(options.poolConfig);
+      this.pool = new import_pg19.Pool(options.poolConfig);
     } else {
       throw new Error("Either pool or poolConfig must be provided");
     }
@@ -7622,7 +7848,7 @@ var PostgreSQLTaskWorkItemStore = class {
 };
 
 // src/stores/MenuStore.ts
-var import_pg19 = require("pg");
+var import_pg20 = require("pg");
 
 // src/migrations/menu_items_migration.ts
 var createMenuItemsTable = {
@@ -7699,7 +7925,7 @@ var MenuStore = class {
       this.initialized = true;
       return;
     }
-    this.pool = typeof options.poolConfig === "string" ? new import_pg19.Pool({ connectionString: options.poolConfig }) : options.poolConfig ? new import_pg19.Pool(options.poolConfig) : (() => {
+    this.pool = typeof options.poolConfig === "string" ? new import_pg20.Pool({ connectionString: options.poolConfig }) : options.poolConfig ? new import_pg20.Pool(options.poolConfig) : (() => {
       throw new Error("Either pool or poolConfig must be provided");
     })();
     this.migrationManager = new MigrationManager(this.pool);
@@ -7854,7 +8080,7 @@ var MenuStore = class {
 };
 
 // src/stores/PostgresSharedResourceStore.ts
-var import_pg20 = require("pg");
+var import_pg21 = require("pg");
 
 // src/migrations/shared_resources_migration.ts
 var createSharedResourcesTable = {
@@ -7908,7 +8134,7 @@ var PostgresSharedResourceStore = class {
       this.initialized = true;
       return;
     }
-    this.pool = typeof options.poolConfig === "string" ? new import_pg20.Pool({ connectionString: options.poolConfig }) : options.poolConfig ? new import_pg20.Pool(options.poolConfig) : (() => {
+    this.pool = typeof options.poolConfig === "string" ? new import_pg21.Pool({ connectionString: options.poolConfig }) : options.poolConfig ? new import_pg21.Pool(options.poolConfig) : (() => {
       throw new Error("Either pool or poolConfig must be provided");
     })();
     this.migrationManager = new MigrationManager(this.pool);
@@ -8077,7 +8303,7 @@ var PostgresSharedResourceStore = class {
 };
 
 // src/stores/PostgreSQLCollectionStore.ts
-var import_pg21 = require("pg");
+var import_pg22 = require("pg");
 var PostgreSQLCollectionStore = class {
   constructor(options) {
     this.initialized = false;
@@ -8090,9 +8316,9 @@ var PostgreSQLCollectionStore = class {
       return;
     }
     if (typeof options.poolConfig === "string") {
-      this.pool = new import_pg21.Pool({ connectionString: options.poolConfig });
+      this.pool = new import_pg22.Pool({ connectionString: options.poolConfig });
     } else if (options.poolConfig) {
-      this.pool = new import_pg21.Pool(options.poolConfig);
+      this.pool = new import_pg22.Pool(options.poolConfig);
     } else {
       throw new Error("Either pool or poolConfig must be provided");
     }
@@ -8207,7 +8433,7 @@ var PostgreSQLCollectionStore = class {
 var import_langgraph_checkpoint_postgres = require("@langchain/langgraph-checkpoint-postgres");
 
 // src/PGVectorStoreProvider.ts
-var import_pg22 = require("pg");
+var import_pg23 = require("pg");
 var import_pgvector = require("@langchain/community/vectorstores/pgvector");
 var import_documents = require("@langchain/core/documents");
 var import_core6 = require("@axiom-lattice/core");
@@ -8281,7 +8507,7 @@ var PGVectorStoreProvider = class {
   }
 };
 function createPGVectorStoreProvider(connectionString) {
-  const pool = new import_pg22.Pool({ connectionString });
+  const pool = new import_pg23.Pool({ connectionString });
   return {
     provider: new PGVectorStoreProvider(pool, connectionString),
     pool
@@ -8632,7 +8858,7 @@ var addTaskWorkItemEventKeyMigration = {
 
 // src/createPgStoreConfig.ts
 async function createPgStoreConfig(connectionString) {
-  const pool = new import_pg23.Pool({ connectionString });
+  const pool = new import_pg24.Pool({ connectionString });
   const mm = new MigrationManager(pool);
   mm.register(createThreadsTable);
   mm.register(createScheduledTasksTable);
@@ -8686,6 +8912,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(addProjectKindColumn);
   mm.register(addA2AKeyAssistantIds);
   mm.register(addTaskWorkItemEventKeyMigration);
+  mm.register(createAgentWebAppsTable);
   await mm.migrate();
   const checkpoint = import_langgraph_checkpoint_postgres.PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {
@@ -8713,6 +8940,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),
@@ -8723,7 +8951,7 @@ async function createPgStoreConfig(connectionString) {
 }
 
 // src/stores/PostgreSQLSkillStore.ts
-var import_pg24 = require("pg");
+var import_pg25 = require("pg");
 var PostgreSQLSkillStore = class {
   constructor(options) {
     this.initialized = false;
@@ -8736,9 +8964,9 @@ var PostgreSQLSkillStore = class {
       return;
     }
     if (typeof options.poolConfig === "string") {
-      this.pool = new import_pg24.Pool({ connectionString: options.poolConfig });
+      this.pool = new import_pg25.Pool({ connectionString: options.poolConfig });
     } else if (options.poolConfig) {
-      this.pool = new import_pg24.Pool(options.poolConfig);
+      this.pool = new import_pg25.Pool(options.poolConfig);
     } else {
       throw new Error("Either pool or poolConfig must be provided");
     }
@@ -9039,7 +9267,7 @@ var PostgreSQLSkillStore = class {
 };
 
 // src/stores/ChannelIdentityMappingStore.ts
-var import_pg25 = require("pg");
+var import_pg26 = require("pg");
 var ChannelIdentityMappingStore = class {
   constructor(options) {
     this.initialized = false;
@@ -9051,7 +9279,7 @@ var ChannelIdentityMappingStore = class {
       this.initialized = true;
       return;
     }
-    this.pool = typeof options.poolConfig === "string" ? new import_pg25.Pool({ connectionString: options.poolConfig }) : options.poolConfig ? new import_pg25.Pool(options.poolConfig) : (() => {
+    this.pool = typeof options.poolConfig === "string" ? new import_pg26.Pool({ connectionString: options.poolConfig }) : options.poolConfig ? new import_pg26.Pool(options.poolConfig) : (() => {
       throw new Error("Either pool or poolConfig must be provided");
     })();
     this.migrationManager = new MigrationManager(this.pool);
@@ -9278,6 +9506,7 @@ function mapRowToChannelIdentityMapping(row) {
   PGVectorStoreProvider,
   Pool,
   PostgreSQLA2AApiKeyStore,
+  PostgreSQLAgentWebAppStore,
   PostgreSQLAssistantStore,
   PostgreSQLChannelInstallationStore,
   PostgreSQLCollectionStore,
@@ -9315,6 +9544,7 @@ function mapRowToChannelIdentityMapping(row) {
   changeSkillPrimaryKey,
   changeThreadPrimaryKey,
   createA2AApiKeysTable,
+  createAgentWebAppsTable,
   createAssistantsTable,
   createChannelBindingsTable,
   createChannelIdentityMappingTables,