matt/ollama37
1
0
Fork 0
mirror of https://github.com/dogkeeper886/ollama37.git synced 2025-12-11 00:07:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

clean up cli flags

Browse Source
This commit is contained in:
Jeffrey Morgan
2023-08-10 09:27:03 -07:00
parent 5b5cc9c9f1
commit 040a5b9750
3 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cmd/cmd.go
View File

@@ -538,7 +538,7 @@ func getRunServerParams(cmd *cobra.Command) (host, port string, extraOrigins []s
if portFlag.Changed || port == "" {
port = portFlag.Value.String()
}
extraOrigins, err = cmd.Flags().GetStringSlice("allowed-origins")
extraOrigins, err = cmd.Flags().GetStringSlice("origins")
if err != nil {
return "", "", nil, err
}
@@ -546,7 +546,7 @@ func getRunServerParams(cmd *cobra.Command) (host, port string, extraOrigins []s
}
func RunServer(cmd *cobra.Command, _ []string) error {
host, port, extraOrigins, err := getRunServerParams(cmd)
host, port, origins, err := getRunServerParams(cmd)
if err != nil {
return err
}
@@ -556,7 +556,7 @@ func RunServer(cmd *cobra.Command, _ []string) error {
return err
}
return server.Serve(ln, extraOrigins)
return server.Serve(ln, origins)
}
func startMacApp(client *api.Client) error {
@@ -650,7 +650,7 @@ func NewCLI() *cobra.Command {
serveCmd.Flags().String("port", "11434", "Port to listen on, may also use OLLAMA_PORT environment variable")
serveCmd.Flags().String("host", "127.0.0.1", "Host listen address, may also use OLLAMA_HOST environment variable")
serveCmd.Flags().StringSlice("allowed-origins", []string{}, "Additional allowed CORS origins (outside of localhost), specify as comma-separated list")
serveCmd.Flags().StringSlice("origins", nil, "Additional allowed CORS origins as comma-separated list")
pullCmd := &cobra.Command{
Use: "pull MODEL",

2
cmd/cmd_test.go
View File

@@ -84,7 +84,7 @@ func TestGetRunServerParams(t *testing.T) {
// now set command flags
serveCmd.Flags().Set("host", "localhost")
serveCmd.Flags().Set("port", "8888")
serveCmd.Flags().Set("allowed-origins", "http://foo.example.com,http://192.168.1.1")
serveCmd.Flags().Set("origins", "http://foo.example.com,http://192.168.1.1")
host, port, extraOrigins, err := getRunServerParams(serveCmd)
if err != nil {