Add --context option

Author: pjcdawkins

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")
 	cmd.Flags().BoolVar(&initOptions.SaveConfig, "save", false,

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{
@@ -224,8 +238,12 @@ func (c *Command) HelpPage(cnf *config.Config) string {
 		fmt.Fprintln(writer, color.YellowString("Examples:"))
 		for _, example := range c.Examples {
 			fmt.Fprintln(writer, indentLines(example.Description.String()+":", 1))
+			usage := cnf.Application.Executable + " " + c.Name.String()
+			if len(c.Usage) > 0 {
+				usage = c.Usage[0]
+			}
 			fmt.Fprintln(writer,
-				color.GreenString(fmt.Sprintf("   %s %s %s", cnf.Application.Executable, c.Name.String(), example.Commandline)))
+				color.GreenString(fmt.Sprintf("   %s %s", usage, example.Commandline)))
 			fmt.Fprintln(writer, "")
 		}
 	}