From 17c93615599b9e197848749cae60a33e9b11b2c7 Mon Sep 17 00:00:00 2001
From: Patrick Dawkins <patrick@platform.sh>
Date: Mon, 22 Sep 2025 14:57:29 +0100
Subject: [PATCH] Add --context option to the init command

---
 commands/init.go        |  2 ++
 commands/list_models.go | 14 ++++++++++++++
 2 files changed, 16 insertions(+)

diff --git a/commands/init.go b/commands/init.go
index e8445e7..616c3f2 100644
--- a/commands/init.go
+++ b/commands/init.go
@@ -47,6 +47,8 @@ func newInitCommand(cnf *config.Config, assets *vendorization.VendorAssets) *cob
 	}
 
 	cmd.Flags().BoolVar(&useAI, "ai", false, "Use AI configuration")
+	cmd.Flags().StringVar(&initOptions.ExtraContext, "context", "",
+		"Add extra context for AI configuration")
 	cmd.Flags().BoolVar(&initOptions.OnlyShowDigest, "digest", false,
 		"Only show the repository digest (the AI configuration input), without sending it")
 
diff --git a/commands/list_models.go b/commands/list_models.go
index 1c27795..4b39867 100644
--- a/commands/list_models.go
+++ b/commands/list_models.go
@@ -41,6 +41,10 @@ func innerProjectInitCommand(cnf *config.Config) Command {
 				Commandline: "--ai=false",
 				Description: "Disable AI mode",
 			},
+			{
+				Commandline: "--ai --context='Use PostgreSQL for the database'",
+				Description: "Add context for AI configuration",
+			},
 		},
 		Definition: Definition{
 			Arguments: &orderedmap.OrderedMap[string, Argument]{},
@@ -54,6 +58,16 @@ func innerProjectInitCommand(cnf *config.Config) Command {
 						Default:     Any{any: false},
 					},
 				},
+				orderedmap.Pair[string, Option]{
+					Key: "context",
+					Value: Option{
+						Name:            "--context",
+						AcceptValue:     true,
+						IsValueRequired: true,
+						Description:     "Add extra context for AI configuration",
+						Default:         Any{any: ""},
+					},
+				},
 				orderedmap.Pair[string, Option]{
 					Key: "digest",
 					Value: Option{