diff --git a/internal-packages/database/prisma/migrations/20241028133512_add_attempt_id_index_to_checkpoint/migration.sql b/internal-packages/database/prisma/migrations/20241028133512_add_attempt_id_index_to_checkpoint/migration.sql
new file mode 100644
index 000000000..950f44d02
--- /dev/null
+++ b/internal-packages/database/prisma/migrations/20241028133512_add_attempt_id_index_to_checkpoint/migration.sql
@@ -0,0 +1,2 @@
+-- CreateIndex
+CREATE INDEX CONCURRENTLY IF NOT EXISTS "Checkpoint_attemptId_idx" ON "Checkpoint"("attemptId");
\ No newline at end of file
diff --git a/internal-packages/database/prisma/migrations/20241028133719_add_run_id_index_to_checkpoint/migration.sql b/internal-packages/database/prisma/migrations/20241028133719_add_run_id_index_to_checkpoint/migration.sql
new file mode 100644
index 000000000..57039cec8
--- /dev/null
+++ b/internal-packages/database/prisma/migrations/20241028133719_add_run_id_index_to_checkpoint/migration.sql
@@ -0,0 +1,2 @@
+-- CreateIndex
+CREATE INDEX CONCURRENTLY IF NOT EXISTS "Checkpoint_runId_idx" ON "Checkpoint"("runId");
\ No newline at end of file
diff --git a/internal-packages/database/prisma/schema.prisma b/internal-packages/database/prisma/schema.prisma
index 8e040191d..cceb5572f 100644
--- a/internal-packages/database/prisma/schema.prisma
+++ b/internal-packages/database/prisma/schema.prisma
@@ -2246,6 +2246,9 @@ model Checkpoint {
 
   createdAt DateTime @default(now())
   updatedAt DateTime @updatedAt
+
+  @@index([attemptId])
+  @@index([runId])
 }
 
 enum CheckpointType {