commit 9d57dc75b612e4926009fb9025ba54d141174386 · julien.rbrt.fr/ollama-proxy

+17
proxy.go
···

       50
       50
        
       	}

     

       51
       51
        
       

     

       52
       52
        
       	proxy := httputil.NewSingleHostReverseProxy(target)

     

       53
       53
       +
       

     

       54
       54
       +
       	// Customize the request before sending to backend

     

       55
       55
       +
       	originalDirector := proxy.Director

     

       56
       56
       +
       	proxy.Director = func(req *http.Request) {

     

       57
       57
       +
       		originalDirector(req)

     

       58
       58
       +
       

     

       59
       59
       +
       		// Remove proxy headers to make it look like a local request

     

       60
       60
       +
       		req.Header.Del("X-Forwarded-For")

     

       61
       61
       +
       		req.Header.Del("X-Forwarded-Host")

     

       62
       62
       +
       		req.Header.Del("X-Forwarded-Proto")

     

       63
       63
       +
       		req.Header.Del("Via")

     

       64
       64
       +
       		req.Header.Del("Authorization")

     

       65
       65
       +
       

     

       66
       66
       +
       		// Set Host to the target host to make it appear local

     

       67
       67
       +
       		req.Host = target.Host

     

       68
       68
       +
       	}

     

       69
       69
       +
       

     

       53
       70
        
       	return proxy, nil

     

       54
       71
        
       }