Compare commits

..

2 Commits

Author SHA1 Message Date
cac2a1e80c Removed blank line 2023-11-14 17:04:44 +00:00
6599af042b Minor refactor
- Use init() function to set up commands
- Expose an Execute() function instead of the root command
2023-11-14 17:04:12 +00:00
3 changed files with 19 additions and 18 deletions

View File

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

View File

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

View File

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