diff --git a/scripts/init_database/recreate_table.sql b/scripts/init_database/recreate_table.sql
index fa83535..7be7447 100644
--- a/scripts/init_database/recreate_table.sql
+++ b/scripts/init_database/recreate_table.sql
@@ -1,5 +1,6 @@
 -- Drop the index first (if it exists)
 DROP INDEX IF EXISTS idx_messages_conversation;
+DROP INDEX IF EXISTS idx_messages_pipeline;
 
 -- Drop the messages table (if it exists)
 DROP TABLE IF EXISTS messages;
@@ -9,6 +10,7 @@ DROP TABLE IF EXISTS messages;
 CREATE TABLE messages (
     id BIGSERIAL PRIMARY KEY,
     conversation_id TEXT NOT NULL,
+    pipeline_id TEXT,
     message_type VARCHAR(10) NOT NULL CHECK (message_type IN ('human', 'ai', 'tool')),
     content TEXT NOT NULL,
     sequence_number INTEGER NOT NULL CHECK (sequence_number >= 0),
@@ -18,3 +20,6 @@ CREATE TABLE messages (
 -- Recreate the index for fast retrieval of all messages in a conversation (in order)
 CREATE INDEX idx_messages_conversation ON messages (conversation_id, sequence_number);
 
+-- Index for fast lookup by pipeline_id
+CREATE INDEX idx_messages_pipeline ON messages (pipeline_id);
+