1 files changed, 15 insertions, 0 deletions
diff --git a/Omni/Agent/Memory.hs b/Omni/Agent/Memory.hs
index d59104c..c869b26 100644
--- a/Omni/Agent/Memory.hs
+++ b/Omni/Agent/Memory.hs
@@ -795,6 +795,21 @@ initMemoryDb conn = do
   SQL.execute_
     conn
     "CREATE INDEX IF NOT EXISTS idx_chat_history_time ON chat_history(created_at)"
+  SQL.execute_
+    conn
+    "CREATE TABLE IF NOT EXISTS tool_traces (\
+    \  id TEXT PRIMARY KEY,\
+    \  created_at TEXT NOT NULL,\
+    \  tool_name TEXT NOT NULL,\
+    \  input TEXT NOT NULL,\
+    \  output TEXT NOT NULL,\
+    \  duration_ms INTEGER NOT NULL,\
+    \  user_id TEXT,\
+    \  chat_id TEXT\
+    \)"
+  SQL.execute_
+    conn
+    "CREATE INDEX IF NOT EXISTS idx_traces_created ON tool_traces(created_at)"
 
 -- | Migrate conversation_messages to add sender_name and thread_id columns.
 migrateConversationMessages :: SQL.Connection -> IO ()