1 files changed, 4 insertions, 3 deletions
diff --git a/types.go b/types.go
index 3b7f3ff..98738bd 100644
--- a/types.go
+++ b/types.go
@@ -17,9 +17,10 @@ type LogModel struct {
 }
 
 type CustomCommand struct {
-	SQL    string `toml:"sql"`
-	Limit  int    `toml:"limit"`
-	Prompt string `toml:"prompt"`
+	SQL     string   `toml:"sql"`
+	Limit   int      `toml:"limit"`
+	Context []string `toml:"context"`
+	Prompt  string   `toml:"prompt"`
 }
 
 type LuaLstates struct {