Can add custom headers?

[zjvn]

I use a third party proxy for openai and they require some custom headers to verify my identity.
Some third parties that process OpenAI may require additional headers to perform some verification on the basis of complying with OpenAI rules. For example, if I preset some tools on a third-party platform, I need to specify certain headers to tell him where the tools I need are.

[sashabaranov]

@zjvn you can customize it withHTTPClient field in ClientConfig

[savishy]

I had the same problem. The following should be a quick and dirty example for how to do this.

type myTransport struct{}

// custom transport to add header
func (t *myTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	req.Header.Add("X-My-Header", "my-header-value")
	return http.DefaultTransport.RoundTrip(req)
}

func main() {
	// create custom http client using myTransport.
	httpClient := &http.Client{Transport: &myTransport{}}
	config := openai.DefaultAzureConfig("API-Token", "URL")
	config.HTTPClient = httpClient
        ....

[appleboy]

Try the following code:

package openai

import (
	"net/http"
	"strings"
)

// DefaultHeaderTransport is an http.RoundTripper that adds the given headers to
type DefaultHeaderTransport struct {
	Origin http.RoundTripper
	Header http.Header
}

// RoundTrip implements the http.RoundTripper interface.
func (t *DefaultHeaderTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	for key, values := range t.Header {
		for _, value := range values {
			req.Header.Add(key, value)
		}
	}
	return t.Origin.RoundTrip(req)
}

// NewHeaders creates a new http.Header from the given slice of headers.
func NewHeaders(headers []string) http.Header {
	h := make(http.Header)
	for _, header := range headers {
		// split header into key and value with = as delimiter
		vals := strings.Split(header, "=")
		if len(vals) != 2 {
			continue
		}
		h.Add(vals[0], vals[1])
	}
	return h
}

See https://github.com/appleboy/CodeGPT/blob/66d623dc4bf4c98c0d90088e602f1d5730a2bb5c/openai/client.go#L8-L36