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

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

Browse Source
This commit is contained in:
Bruce MacDonald
2023-08-08 10:41:42 -04:00
committed by GitHub
parent 8713ac23a8 93492f1e18
commit 34a13a9d05
3 changed files with 140 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
server/routes.go
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))