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", + ), + )