baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

cmd: set ExactArgs

Browse source
This commit is contained in:
Michael Yang 2023-10-18 11:57:22 -07:00
parent 235e43d7f6
commit 36c88cb9db
1 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
cmd/cmd.go
View file

@ -932,7 +932,7 @@ func NewCLI() *cobra.Command {
createCmd := &cobra.Command{ createCmd := &cobra.Command{
Use: "create MODEL", Use: "create MODEL",
Short: "Create a model from a Modelfile", Short: "Create a model from a Modelfile",
Args: cobra.MinimumNArgs(1), Args: cobra.ExactArgs(1),
PreRunE: checkServerHeartbeat, PreRunE: checkServerHeartbeat,
RunE: CreateHandler, RunE: CreateHandler,
} }
@ -942,7 +942,7 @@ func NewCLI() *cobra.Command {
showCmd := &cobra.Command{ showCmd := &cobra.Command{
Use: "show MODEL", Use: "show MODEL",
Short: "Show information for a model", Short: "Show information for a model",
Args: cobra.MinimumNArgs(1), Args: cobra.ExactArgs(1),
PreRunE: checkServerHeartbeat, PreRunE: checkServerHeartbeat,
RunE: ShowHandler, RunE: ShowHandler,
} }
@ -969,13 +969,14 @@ func NewCLI() *cobra.Command {
Use: "serve", Use: "serve",
Aliases: []string{"start"}, Aliases: []string{"start"},
Short: "Start ollama", Short: "Start ollama",
Args: cobra.ExactArgs(0),
RunE: RunServer, RunE: RunServer,
} }
pullCmd := &cobra.Command{ pullCmd := &cobra.Command{
Use: "pull MODEL", Use: "pull MODEL",
Short: "Pull a model from a registry", Short: "Pull a model from a registry",
Args: cobra.MinimumNArgs(1), Args: cobra.ExactArgs(1),
PreRunE: checkServerHeartbeat, PreRunE: checkServerHeartbeat,
RunE: PullHandler, RunE: PullHandler,
} }
@ -985,7 +986,7 @@ func NewCLI() *cobra.Command {
pushCmd := &cobra.Command{ pushCmd := &cobra.Command{
Use: "push MODEL", Use: "push MODEL",
Short: "Push a model to a registry", Short: "Push a model to a registry",
Args: cobra.MinimumNArgs(1), Args: cobra.ExactArgs(1),
PreRunE: checkServerHeartbeat, PreRunE: checkServerHeartbeat,
RunE: PushHandler, RunE: PushHandler,
} }
@ -1001,15 +1002,15 @@ func NewCLI() *cobra.Command {
} }
copyCmd := &cobra.Command{ copyCmd := &cobra.Command{
Use: "cp", Use: "cp SOURCE TARGET",
Short: "Copy a model", Short: "Copy a model",
Args: cobra.MinimumNArgs(2), Args: cobra.ExactArgs(2),
PreRunE: checkServerHeartbeat, PreRunE: checkServerHeartbeat,
RunE: CopyHandler, RunE: CopyHandler,
} }
deleteCmd := &cobra.Command{ deleteCmd := &cobra.Command{
Use: "rm", Use: "rm MODEL [MODEL...]",
Short: "Remove a model", Short: "Remove a model",
Args: cobra.MinimumNArgs(1), Args: cobra.MinimumNArgs(1),
PreRunE: checkServerHeartbeat, PreRunE: checkServerHeartbeat,