@axiom-lattice/pg-stores 3.1.4 → 3.1.6

package/dist/index.mjs

@@ -6320,6 +6320,22 @@ var createAgentWebAppsTable = {
   }
 };
 
+// src/migrations/add_agent_web_apps_identity_migration.ts
+var addAgentWebAppsIdentity = {
+  version: 176,
+  name: "add_agent_web_apps_identity",
+  up: async (client) => {
+    await client.query(
+      "ALTER TABLE lattice_agent_web_apps ADD COLUMN IF NOT EXISTS identity JSONB"
+    );
+  },
+  down: async (client) => {
+    await client.query(
+      "ALTER TABLE lattice_agent_web_apps DROP COLUMN IF EXISTS identity"
+    );
+  }
+};
+
 // src/stores/PostgreSQLAgentWebAppStore.ts
 function isRecord(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
@@ -6334,16 +6350,23 @@ 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";
+  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" && (value.files === void 0 || typeof value.files === "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 isIdentity(value) {
+  if (value === null || value === void 0) return true;
+  if (!isRecord(value)) return false;
+  const policyOk = value.policy === "unverified" || value.policy === "verified" || value.policy === "both";
+  const issuersOk = Array.isArray(value.issuers) && value.issuers.every((issuer) => isRecord(issuer) && typeof issuer.iss === "string" && typeof issuer.aud === "string" && (issuer.alg === "HS256" || issuer.alg === "RS256"));
+  return policyOk && issuersOk;
+}
 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)) {
+  if (!isStatus(row.status) || !isIntegration(row.integration) || !isScope(row.scope) || !isFeatures(row.features) || !isAppearance(row.appearance) || !isIdentity(row.identity)) {
     throw new Error(`Invalid agent web app row: ${row.id}`);
   }
   return {
@@ -6357,6 +6380,7 @@ function mapRow(row) {
     scope: row.scope,
     features: row.features,
     appearance: row.appearance,
+    identity: row.identity === null ? void 0 : row.identity,
     createdAt: row.created_at,
     updatedAt: row.updated_at
   };
@@ -6377,6 +6401,7 @@ var PostgreSQLAgentWebAppStore = class {
     })();
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createAgentWebAppsTable);
+    this.migrationManager.register(addAgentWebAppsIdentity);
     if (options.autoMigrate !== false) {
       this.startInitialization();
     }
@@ -6437,8 +6462,8 @@ var PostgreSQLAgentWebAppStore = class {
     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)
+       (id, tenant_id, assistant_id, name, description, status, integration, scope, features, appearance, identity)
+       VALUES ($1, $2, $3, $4, $5, $6, $7::jsonb, $8::jsonb, $9::jsonb, $10::jsonb, $11::jsonb)
        RETURNING *`,
       [
         id,
@@ -6450,7 +6475,8 @@ var PostgreSQLAgentWebAppStore = class {
         JSON.stringify(input.integration),
         JSON.stringify(input.scope),
         JSON.stringify(input.features),
-        JSON.stringify(input.appearance)
+        JSON.stringify(input.appearance),
+        input.identity ? JSON.stringify(input.identity) : null
       ]
     );
     return mapRow(result.rows[0]);
@@ -6476,6 +6502,9 @@ var PostgreSQLAgentWebAppStore = class {
     if (patch.appearance !== void 0) {
       add("appearance", JSON.stringify(patch.appearance));
     }
+    if (patch.identity !== void 0) {
+      add("identity", JSON.stringify(patch.identity));
+    }
     if (patch.status !== void 0) add("status", patch.status);
     if (assignments.length === 0) return this.getById(tenantId, webAppId);
     values.push(tenantId, webAppId);
@@ -10424,6 +10453,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(createProjectRoomTables);
   mm.register(addTrustedRunContextColumn);
   mm.register(addProjectLifecycleEventIndex);
+  mm.register(addAgentWebAppsIdentity);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {