From cd1e0f85e42c6a8475abe05ab30f085dd931de5f Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Fri, 16 Jun 2023 08:57:53 +0100
Subject: [PATCH] Fix schema display, refs #34

---
 docs/logging.md | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/docs/logging.md b/docs/logging.md
index 66c86389..65e6da00 100644
--- a/docs/logging.md
+++ b/docs/logging.md
@@ -61,13 +61,14 @@ Here's the SQL schema used by the `log.db` database:
 import cog
 from llm.migrations import migrate
 import sqlite_utils
+import re
 db = sqlite_utils.Database(memory=True)
 migrate(db)
 schema = db["log"].schema
 
 def cleanup_sql(sql):
     first_line = sql.split('(')[0]
-    inner = sql.split('(')[1].split(')')[0];
+    inner = re.search(r'\((.*)\)', sql, re.DOTALL).group(1)
     columns = [l.strip() for l in inner.split(',')]
     return first_line + '(\n  ' + ',\n  '.join(columns) + '\n);'
 
@@ -83,7 +84,9 @@ CREATE TABLE "log" (
   [prompt] TEXT,
   [system] TEXT,
   [response] TEXT,
-  [chat_id] INTEGER REFERENCES [log]
+  [chat_id] INTEGER REFERENCES [log]([id]),
+  [debug] TEXT,
+  [duration_ms] INTEGER
 );
 ```
-<!-- [[[end]]] -->
\ No newline at end of file
+<!-- [[[end]]] -->