pass flags to serve to allow setting allowed-origins + host and port

* resolves: https://github.com/jmorganca/ollama/issues/300 and
https://github.com/jmorganca/ollama/issues/282

* example usage:
```
ollama serve --port 9999 --allowed-origins "http://foo.example.com,http://192.0.0.1"
```
This commit is contained in:
cmiller01
2023-08-07 03:34:37 +00:00
parent 06fc48ad66
commit fb593b7bfc
2 changed files with 26 additions and 10 deletions

View File

@@ -301,11 +301,11 @@ func CopyModelHandler(c *gin.Context) {
}
}
func Serve(ln net.Listener) error {
func Serve(ln net.Listener, extraOrigins []string) error {
config := cors.DefaultConfig()
config.AllowWildcard = true
// only allow http/https from localhost
config.AllowOrigins = []string{
allowedOrigins := []string{
"http://localhost",
"http://localhost:*",
"https://localhost",
@@ -315,6 +315,8 @@ func Serve(ln net.Listener) error {
"https://127.0.0.1",
"https://127.0.0.1:*",
}
allowedOrigins = append(allowedOrigins, extraOrigins...)
config.AllowOrigins = allowedOrigins
r := gin.Default()
r.Use(cors.New(config))