README.md

@@ -18,6 +18,7 @@ Maybe features:
 
 ```shell
 $ go install git.mlow.ca/mlow/lmcli@latest
+
 ```
 
 ## Usage

main.go

@@ -8,8 +8,9 @@ import (
 )
 
 func main() {
-	if err := cli.Execute(); err != nil {
+	cmd := cli.NewRootCmd()
-		fmt.Fprint(os.Stderr, err)
+	if err := cmd.Execute(); err != nil {
+		fmt.Println(err)
 		os.Exit(1)
 	}
 }

pkg/cli/cmd.go

@@ -12,21 +12,6 @@ import (
 // TODO: allow setting with flag
 const MAX_TOKENS = 256
 
-func init() {
-	rootCmd.AddCommand(
-		lsCmd,
-		newCmd,
-		promptCmd,
-		replyCmd,
-		rmCmd,
-		viewCmd,
-	)
-}
-
-func Execute() error {
-	return rootCmd.Execute()
-}
-
 var rootCmd = &cobra.Command{
 	Use:   "lmcli",
 	Short: "Interact with Large Language Models",
@@ -402,3 +387,15 @@ var promptCmd = &cobra.Command{
 		fmt.Println()
 	},
 }
+
+func NewRootCmd() *cobra.Command {
+	rootCmd.AddCommand(
+		lsCmd,
+		newCmd,
+		promptCmd,
+		replyCmd,
+		rmCmd,
+		viewCmd,
+	)
+	return rootCmd
+}