From 307af2474cc9eb063aa3dc3ea6b66730968bbda4 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Mon, 3 Jul 2023 07:29:58 -0700
Subject: [PATCH] Fix column order in logs

---
 docs/logging.md   |  4 ++--
 llm/migrations.py | 21 +++++++++++++++++++++
 2 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/docs/logging.md b/docs/logging.md
index e79775d..b6382e1 100644
--- a/docs/logging.md
+++ b/docs/logging.md
@@ -84,8 +84,8 @@ CREATE TABLE "logs" (
   [options_json] TEXT,
   [response] TEXT,
   [response_json] TEXT,
-  [reply_to_id] INTEGER,
-  [chat_id] INTEGER REFERENCES "logs"([id]),
+  [reply_to_id] INTEGER REFERENCES [logs]([id]),
+  [chat_id] INTEGER REFERENCES [logs]([id]),
   [duration_ms] INTEGER,
   [datetime_utc] TEXT
 );
diff --git a/llm/migrations.py b/llm/migrations.py
index 2e8756b..ab47ac2 100644
--- a/llm/migrations.py
+++ b/llm/migrations.py
@@ -137,3 +137,24 @@ def m007_finish_logs_table(db):
 @migration
 def m008_reply_to_id_foreign_key(db):
     db["logs"].add_foreign_key("reply_to_id", "logs", "id")
+
+
+@migration
+def m008_fix_column_order_in_logs(db):
+    # reply_to_id ended up at the end after foreign key added
+    db["logs"].transform(
+        column_order=(
+            "id",
+            "model",
+            "prompt",
+            "system",
+            "prompt_json",
+            "options_json",
+            "response",
+            "response_json",
+            "reply_to_id",
+            "chat_id",
+            "duration_ms",
+            "timestamp_utc",
+        ),
+    )