Minor refactor

- Use init() function to set up commands
- Expose an Execute() function instead of the root command

main.go

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

pkg/cli/cmd.go

@@ -12,6 +12,21 @@ 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",
@@ -387,15 +402,3 @@ var promptCmd = &cobra.Command{
 		fmt.Println()
 	},
 }
-
-func NewRootCmd() *cobra.Command {
-	rootCmd.AddCommand(
-		lsCmd,
-		newCmd,
-		promptCmd,
-		replyCmd,
-		rmCmd,
-		viewCmd,
-	)
-	return rootCmd
-}