@axiom-lattice/pg-stores 1.0.21 → 1.0.23

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/pg-stores",
-  "version": "1.0.21",
+  "version": "1.0.23",
   "description": "PG stores implementation for Axiom Lattice framework",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -21,8 +21,8 @@
   "license": "MIT",
   "dependencies": {
     "pg": "^8.16.3",
-    "@axiom-lattice/protocols": "2.1.18",
-    "@axiom-lattice/core": "2.1.31"
+    "@axiom-lattice/protocols": "2.1.20",
+    "@axiom-lattice/core": "2.1.33"
   },
   "devDependencies": {
     "@types/node": "^20.11.24",

package/src/index.ts

@@ -18,9 +18,16 @@ export * from "./stores/PostgreSQLTenantStore";
 export * from "./stores/PostgreSQLUserTenantLinkStore";
 export * from "./migrations/migration";
 export * from "./migrations/thread_migrations";
+export * from "./migrations/thread_tenant_migration";
+export * from "./migrations/thread_pk_migration";
 export * from "./migrations/assistant_migrations";
+export * from "./migrations/assistant_tenant_migration";
+export * from "./migrations/assistant_pk_migration";
 export * from "./migrations/schedule_migrations";
+export * from "./migrations/schedule_tenant_migration";
 export * from "./migrations/skill_migrations";
+export * from "./migrations/skill_tenant_migration";
+export * from "./migrations/skill_pk_migration";
 export * from "./migrations/database_config_migrations";
 export * from "./migrations/metrics_config_migrations";
 export * from "./migrations/mcp_server_config_migrations";

package/src/migrations/assistant_migrations.ts

@@ -14,12 +14,14 @@ export const createAssistantsTable: Migration = {
   up: async (client: PoolClient) => {
     await client.query(`
       CREATE TABLE IF NOT EXISTS lattice_assistants (
-        id VARCHAR(255) PRIMARY KEY,
+        id VARCHAR(255) NOT NULL,
+        tenant_id VARCHAR(255) NOT NULL DEFAULT 'default',
         name VARCHAR(255) NOT NULL,
         description TEXT,
         graph_definition JSONB NOT NULL,
         created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
-        updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+        updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+        PRIMARY KEY (tenant_id, id)
       )
     `);
 

package/src/migrations/assistant_pk_migration.ts

@@ -0,0 +1,101 @@
+/**
+ * Assistant table primary key migration
+ * Changes primary key from (id) to (tenant_id, id) for multi-tenancy
+ */
+
+import { PoolClient } from "pg";
+import { Migration } from "./migration";
+
+/**
+ * Migration: Change primary key to support multi-tenancy
+ */
+export const changeAssistantPrimaryKey: Migration = {
+  version: 15,
+  name: "change_assistant_primary_key",
+  up: async (client: PoolClient) => {
+    // Check if table exists and has the old primary key structure
+    const tableExists = await client.query(`
+      SELECT EXISTS (
+        SELECT FROM information_schema.tables 
+        WHERE table_name = 'lattice_assistants'
+      )
+    `);
+
+    if (!tableExists.rows[0].exists) {
+      // Table doesn't exist, it will be created by createAssistantsTable migration
+      // with the correct primary key (tenant_id, id)
+      return;
+    }
+
+    // Check current primary key
+    const pkResult = await client.query(`
+      SELECT a.attname as column_name
+      FROM pg_index i
+      JOIN pg_attribute a ON a.attrelid = i.indrelid AND a.attnum = ANY(i.indkey)
+      WHERE i.indrelid = 'lattice_assistants'::regclass
+      AND i.indisprimary
+    `);
+
+    const pkColumns = pkResult.rows.map(row => row.column_name);
+    
+    // If primary key is already (tenant_id, id), nothing to do
+    if (pkColumns.includes('tenant_id') && pkColumns.includes('id')) {
+      console.log('Primary key already correct (tenant_id, id)');
+      return;
+    }
+
+    // If primary key is just (id), we need to change it
+    if (pkColumns.length === 1 && pkColumns[0] === 'id') {
+      console.log('Changing primary key from (id) to (tenant_id, id)');
+      
+      // Check if tenant_id column exists
+      const tenantIdExists = await client.query(`
+        SELECT EXISTS (
+          SELECT FROM information_schema.columns 
+          WHERE table_name = 'lattice_assistants' 
+          AND column_name = 'tenant_id'
+        )
+      `);
+
+      if (!tenantIdExists.rows[0].exists) {
+        // Add tenant_id column if it doesn't exist
+        await client.query(`
+          ALTER TABLE lattice_assistants
+          ADD COLUMN tenant_id VARCHAR(255) DEFAULT 'default'
+        `);
+      }
+
+      // Update existing records to have tenant_id
+      await client.query(`
+        UPDATE lattice_assistants 
+        SET tenant_id = 'default' 
+        WHERE tenant_id IS NULL
+      `);
+
+      // Make tenant_id NOT NULL
+      await client.query(`
+        ALTER TABLE lattice_assistants
+        ALTER COLUMN tenant_id SET NOT NULL
+      `);
+
+      // Drop the old primary key
+      await client.query(`
+        ALTER TABLE lattice_assistants 
+        DROP CONSTRAINT IF EXISTS lattice_assistants_pkey
+      `);
+
+      // Add new composite primary key
+      await client.query(`
+        ALTER TABLE lattice_assistants 
+        ADD PRIMARY KEY (tenant_id, id)
+      `);
+
+      console.log('Primary key changed successfully to (tenant_id, id)');
+    }
+  },
+  down: async (client: PoolClient) => {
+    // Reverting this is complex and potentially dangerous
+    // as it would require ensuring no duplicate IDs across tenants
+    console.warn('Down migration for changeAssistantPrimaryKey is not supported');
+  },
+};

package/src/migrations/assistant_tenant_migration.ts

@@ -0,0 +1,29 @@
+/**
+ * Assistant table tenant isolation migration
+ * Adds tenant_id column to lattice_assistants table
+ */
+
+import { PoolClient } from "pg";
+import { Migration } from "./migration";
+
+/**
+ * Migration: Add tenant_id column to assistants table
+ */
+export const addAssistantTenantId: Migration = {
+  version: 14,
+  name: "add_assistant_tenant_id",
+  up: async (client: PoolClient) => {
+    // tenant_id column is now created in the initial migration
+    // This migration is kept for backward compatibility but does nothing
+    // Create index for tenant-based queries if not exists
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_assistants_tenant_id 
+      ON lattice_assistants(tenant_id)
+    `);
+  },
+  down: async (client: PoolClient) => {
+    await client.query(
+      "DROP INDEX IF EXISTS idx_lattice_assistants_tenant_id"
+    );
+  },
+};

package/src/migrations/schedule_tenant_migration.ts

@@ -0,0 +1,75 @@
+/**
+ * Schedule table tenant isolation migration
+ * Adds tenant_id column to lattice_scheduled_tasks table
+ */
+
+import { PoolClient } from "pg";
+import { Migration } from "./migration";
+
+/**
+ * Migration: Add tenant_id column to scheduled tasks table
+ */
+export const addScheduleTenantId: Migration = {
+  version: 18,
+  name: "add_schedule_tenant_id",
+  up: async (client: PoolClient) => {
+    // Add tenant_id column
+    await client.query(`
+      ALTER TABLE lattice_scheduled_tasks
+      ADD COLUMN IF NOT EXISTS tenant_id VARCHAR(255) DEFAULT 'default'
+    `);
+
+    // Update existing records to have tenant_id
+    await client.query(`
+      UPDATE lattice_scheduled_tasks 
+      SET tenant_id = 'default' 
+      WHERE tenant_id IS NULL
+    `);
+
+    // Make tenant_id NOT NULL after setting defaults
+    await client.query(`
+      ALTER TABLE lattice_scheduled_tasks
+      ALTER COLUMN tenant_id SET NOT NULL
+    `);
+
+    // Drop the default constraint
+    await client.query(`
+      ALTER TABLE lattice_scheduled_tasks
+      ALTER COLUMN tenant_id DROP DEFAULT
+    `);
+
+    // Create index for tenant-based queries
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_scheduled_tasks_tenant_id 
+      ON lattice_scheduled_tasks(tenant_id)
+    `);
+
+    // Create composite index for common tenant+assistant queries
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_scheduled_tasks_tenant_assistant 
+      ON lattice_scheduled_tasks(tenant_id, assistant_id)
+      WHERE assistant_id IS NOT NULL
+    `);
+
+    // Create composite index for common tenant+thread queries
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_scheduled_tasks_tenant_thread 
+      ON lattice_scheduled_tasks(tenant_id, thread_id)
+      WHERE thread_id IS NOT NULL
+    `);
+  },
+  down: async (client: PoolClient) => {
+    await client.query(
+      "DROP INDEX IF EXISTS idx_scheduled_tasks_tenant_thread"
+    );
+    await client.query(
+      "DROP INDEX IF EXISTS idx_scheduled_tasks_tenant_assistant"
+    );
+    await client.query(
+      "DROP INDEX IF EXISTS idx_scheduled_tasks_tenant_id"
+    );
+    await client.query(
+      "ALTER TABLE lattice_scheduled_tasks DROP COLUMN IF EXISTS tenant_id"
+    );
+  },
+};

package/src/migrations/skill_pk_migration.ts

@@ -0,0 +1,97 @@
+/**
+ * Skill table primary key migration
+ * Changes primary key from (id) to (tenant_id, id) for multi-tenancy
+ */
+
+import { PoolClient } from "pg";
+import { Migration } from "./migration";
+
+/**
+ * Migration: Change skill table primary key to support multi-tenancy
+ */
+export const changeSkillPrimaryKey: Migration = {
+  version: 17,
+  name: "change_skill_primary_key",
+  up: async (client: PoolClient) => {
+    // Check if table exists
+    const tableExists = await client.query(`
+      SELECT EXISTS (
+        SELECT FROM information_schema.tables 
+        WHERE table_name = 'lattice_skills'
+      )
+    `);
+
+    if (!tableExists.rows[0].exists) {
+      return;
+    }
+
+    // Check current primary key
+    const pkResult = await client.query(`
+      SELECT a.attname as column_name
+      FROM pg_index i
+      JOIN pg_attribute a ON a.attrelid = i.indrelid AND a.attnum = ANY(i.indkey)
+      WHERE i.indrelid = 'lattice_skills'::regclass
+      AND i.indisprimary
+    `);
+
+    const pkColumns = pkResult.rows.map(row => row.column_name);
+    
+    // If primary key is already (tenant_id, id), nothing to do
+    if (pkColumns.includes('tenant_id') && pkColumns.includes('id')) {
+      console.log('Skill table primary key already correct (tenant_id, id)');
+      return;
+    }
+
+    // If primary key is just (id), we need to change it
+    if (pkColumns.length === 1 && pkColumns[0] === 'id') {
+      console.log('Changing skill table primary key from (id) to (tenant_id, id)');
+      
+      // Check if tenant_id column exists
+      const tenantIdExists = await client.query(`
+        SELECT EXISTS (
+          SELECT FROM information_schema.columns 
+          WHERE table_name = 'lattice_skills' 
+          AND column_name = 'tenant_id'
+        )
+      `);
+
+      if (!tenantIdExists.rows[0].exists) {
+        // Add tenant_id column if it doesn't exist
+        await client.query(`
+          ALTER TABLE lattice_skills
+          ADD COLUMN tenant_id VARCHAR(255) DEFAULT 'default'
+        `);
+      }
+
+      // Update existing records to have tenant_id
+      await client.query(`
+        UPDATE lattice_skills 
+        SET tenant_id = 'default' 
+        WHERE tenant_id IS NULL
+      `);
+
+      // Make tenant_id NOT NULL
+      await client.query(`
+        ALTER TABLE lattice_skills
+        ALTER COLUMN tenant_id SET NOT NULL
+      `);
+
+      // Drop the old primary key
+      await client.query(`
+        ALTER TABLE lattice_skills 
+        DROP CONSTRAINT IF EXISTS lattice_skills_pkey
+      `);
+
+      // Add new composite primary key
+      await client.query(`
+        ALTER TABLE lattice_skills 
+        ADD PRIMARY KEY (tenant_id, id)
+      `);
+
+      console.log('Skill table primary key changed successfully to (tenant_id, id)');
+    }
+  },
+  down: async (client: PoolClient) => {
+    console.warn('Down migration for changeSkillPrimaryKey is not supported');
+  },
+};

package/src/migrations/skill_tenant_migration.ts

@@ -0,0 +1,55 @@
+/**
+ * Skill table tenant isolation migration
+ * Adds tenant_id column to lattice_skills table
+ */
+
+import { PoolClient } from "pg";
+import { Migration } from "./migration";
+
+/**
+ * Migration: Add tenant_id column to skills table
+ */
+export const addSkillTenantId: Migration = {
+  version: 16,
+  name: "add_skill_tenant_id",
+  up: async (client: PoolClient) => {
+    // Add tenant_id column
+    await client.query(`
+      ALTER TABLE lattice_skills
+      ADD COLUMN IF NOT EXISTS tenant_id VARCHAR(255) DEFAULT 'default'
+    `);
+
+    // Update existing records to have tenant_id
+    await client.query(`
+      UPDATE lattice_skills 
+      SET tenant_id = 'default' 
+      WHERE tenant_id IS NULL
+    `);
+
+    // Make tenant_id NOT NULL after setting defaults
+    await client.query(`
+      ALTER TABLE lattice_skills
+      ALTER COLUMN tenant_id SET NOT NULL
+    `);
+
+    // Drop the default constraint
+    await client.query(`
+      ALTER TABLE lattice_skills
+      ALTER COLUMN tenant_id DROP DEFAULT
+    `);
+
+    // Create index for tenant-based queries
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_skills_tenant_id 
+      ON lattice_skills(tenant_id)
+    `);
+  },
+  down: async (client: PoolClient) => {
+    await client.query(
+      "DROP INDEX IF EXISTS idx_lattice_skills_tenant_id"
+    );
+    await client.query(
+      "ALTER TABLE lattice_skills DROP COLUMN IF EXISTS tenant_id"
+    );
+  },
+};

package/src/migrations/thread_pk_migration.ts

@@ -0,0 +1,99 @@
+/**
+ * Thread table primary key migration
+ * Changes primary key from (id, assistant_id) to (id, tenant_id) for multi-tenancy
+ */
+
+import { PoolClient } from "pg";
+import { Migration } from "./migration";
+
+/**
+ * Migration: Change primary key to support multi-tenancy
+ */
+export const changeThreadPrimaryKey: Migration = {
+  version: 19,
+  name: "change_thread_primary_key",
+  up: async (client: PoolClient) => {
+    // Check if table exists
+    const tableExists = await client.query(`
+      SELECT EXISTS (
+        SELECT FROM information_schema.tables 
+        WHERE table_name = 'lattice_threads'
+      )
+    `);
+
+    if (!tableExists.rows[0].exists) {
+      // Table doesn't exist, it will be created with correct schema
+      return;
+    }
+
+    // Check current primary key
+    const pkResult = await client.query(`
+      SELECT a.attname as column_name
+      FROM pg_index i
+      JOIN pg_attribute a ON a.attrelid = i.indrelid AND a.attnum = ANY(i.indkey)
+      WHERE i.indrelid = 'lattice_threads'::regclass
+      AND i.indisprimary
+    `);
+
+    const pkColumns = pkResult.rows.map(row => row.column_name);
+    
+    // If primary key is already (id, tenant_id), nothing to do
+    if (pkColumns.includes('tenant_id') && pkColumns.includes('id') && pkColumns.length === 2) {
+      console.log('Primary key already correct (id, tenant_id)');
+      return;
+    }
+
+    // If primary key is (id, assistant_id), we need to change it
+    if (pkColumns.includes('id') && pkColumns.includes('assistant_id')) {
+      console.log('Changing primary key from (id, assistant_id) to (id, tenant_id)');
+      
+      // Check if tenant_id column exists
+      const tenantIdExists = await client.query(`
+        SELECT EXISTS (
+          SELECT FROM information_schema.columns 
+          WHERE table_name = 'lattice_threads' 
+          AND column_name = 'tenant_id'
+        )
+      `);
+
+      if (!tenantIdExists.rows[0].exists) {
+        // Add tenant_id column if it doesn't exist
+        await client.query(`
+          ALTER TABLE lattice_threads
+          ADD COLUMN tenant_id VARCHAR(255) DEFAULT 'default'
+        `);
+      }
+
+      // Update existing records to have tenant_id
+      await client.query(`
+        UPDATE lattice_threads 
+        SET tenant_id = 'default' 
+        WHERE tenant_id IS NULL
+      `);
+
+      // Make tenant_id NOT NULL
+      await client.query(`
+        ALTER TABLE lattice_threads
+        ALTER COLUMN tenant_id SET NOT NULL
+      `);
+
+      // Drop the old primary key
+      await client.query(`
+        ALTER TABLE lattice_threads 
+        DROP CONSTRAINT IF EXISTS lattice_threads_pkey
+      `);
+
+      // Add new composite primary key
+      await client.query(`
+        ALTER TABLE lattice_threads 
+        ADD PRIMARY KEY (id, tenant_id)
+      `);
+
+      console.log('Primary key changed successfully to (id, tenant_id)');
+    }
+  },
+  down: async (client: PoolClient) => {
+    // Reverting this is complex and potentially dangerous
+    console.warn('Down migration for changeThreadPrimaryKey is not supported');
+  },
+};

package/src/migrations/thread_tenant_migration.ts

@@ -0,0 +1,55 @@
+/**
+ * Thread table tenant isolation migration
+ * Adds tenant_id column to lattice_threads table
+ */
+
+import { PoolClient } from "pg";
+import { Migration } from "./migration";
+
+/**
+ * Migration: Add tenant_id column to threads table
+ */
+export const addThreadTenantId: Migration = {
+  version: 15,
+  name: "add_thread_tenant_id",
+  up: async (client: PoolClient) => {
+    // Add tenant_id column
+    await client.query(`
+      ALTER TABLE lattice_threads
+      ADD COLUMN IF NOT EXISTS tenant_id VARCHAR(255) DEFAULT 'default'
+    `);
+
+    // Update existing records to have tenant_id
+    await client.query(`
+      UPDATE lattice_threads 
+      SET tenant_id = 'default' 
+      WHERE tenant_id IS NULL
+    `);
+
+    // Make tenant_id NOT NULL after setting defaults
+    await client.query(`
+      ALTER TABLE lattice_threads
+      ALTER COLUMN tenant_id SET NOT NULL
+    `);
+
+    // Drop the default constraint
+    await client.query(`
+      ALTER TABLE lattice_threads
+      ALTER COLUMN tenant_id DROP DEFAULT
+    `);
+
+    // Create index for tenant-based queries
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_threads_tenant_id 
+      ON lattice_threads(tenant_id)
+    `);
+  },
+  down: async (client: PoolClient) => {
+    await client.query(
+      "DROP INDEX IF EXISTS idx_lattice_threads_tenant_id"
+    );
+    await client.query(
+      "ALTER TABLE lattice_threads DROP COLUMN IF EXISTS tenant_id"
+    );
+  },
+};