-- ================================================================================================
-- Migration 026: Add account_id column to ensemble_predictions table
-- Fixes schema drift - column expected by code but missing from table
-- ================================================================================================

-- Add account_id column (nullable for backward compatibility)
ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS account_id VARCHAR(64);

-- Add index for account_id queries
CREATE INDEX IF NOT EXISTS idx_ensemble_predictions_account_id
ON ensemble_predictions(account_id)
WHERE account_id IS NOT NULL;

-- Add missing columns that may have been skipped
ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS ab_variant VARCHAR(50);

ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS dqn_checkpoint_id VARCHAR(255);

ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS ppo_checkpoint_id VARCHAR(255);

ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS mamba2_checkpoint_id VARCHAR(255);

ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS tft_checkpoint_id VARCHAR(255);

ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS node_id VARCHAR(50);

ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS user_id VARCHAR(64);

ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS session_id UUID;

ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS request_id UUID;

ALTER TABLE ensemble_predictions
ADD COLUMN IF NOT EXISTS strategy_id VARCHAR(100);

-- Add comment
COMMENT ON COLUMN ensemble_predictions.account_id IS 'Trading account identifier for multi-account tracking';

-- ================================================================================================
-- END MIGRATION 026
-- ================================================================================================