feat: add proxy and SOCKS5 support to HTTP Anthropic client (#213)

- Add support for proxy URL, SOCKS URL, and skip verify options in the Anthropic client configuration
- Introduce `DefaultHeaderTransport` to add custom headers to HTTP requests
- Add new HTTP transport configuration for handling proxy and SOCKS URLs
- Update the Anthropic client initialization to use the new HTTP client with custom transport
- Add `WithProxyURL`, `WithSocksURL`, and `WithSkipVerify` options for configuring the Anthropic client

Signed-off-by: appleboy <[email protected]>

Author: appleboy

cmd/provider.go

@@ -62,6 +62,9 @@ func NewAnthropic(ctx context.Context) (*anthropic.Client, error) {
 		anthropic.WithMaxTokens(viper.GetInt("openai.max_tokens")),
 		anthropic.WithTemperature(float32(viper.GetFloat64("openai.temperature"))),
 		anthropic.WithTopP(float32(viper.GetFloat64("openai.top_p"))),
+		anthropic.WithProxyURL(viper.GetString("openai.proxy")),
+		anthropic.WithSocksURL(viper.GetString("openai.socks")),
+		anthropic.WithSkipVerify(viper.GetBool("openai.skip_verify")),
 	)
 }
 

provider/anthropic/anthropic.go

@@ -2,11 +2,15 @@ package anthropic
 
 import (
 	"context"
+	"crypto/tls"
 	"encoding/json"
 	"errors"
 	"fmt"
+	"net/http"
+	"net/url"
 
 	"github.com/appleboy/CodeGPT/core"
+	"golang.org/x/net/proxy"
 
 	"github.com/appleboy/com/convert"
 	"github.com/liushuangls/go-anthropic/v2"
@@ -96,6 +100,22 @@ func (c *Client) GetSummaryPrefix(ctx context.Context, content string) (*core.Re
 	}, nil
 }
 
+// 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)
+}
+
 func New(opts ...Option) (c *Client, err error) {
 	// Create a new config object with the given options.
 	cfg := newConfig(opts...)
@@ -105,9 +125,40 @@ func New(opts ...Option) (c *Client, err error) {
 		return nil, err
 	}
 
+	// Create a new HTTP transport.
+	tr := &http.Transport{}
+	if cfg.skipVerify {
+		tr.TLSClientConfig = &tls.Config{InsecureSkipVerify: true} //nolint:gosec
+	}
+
+	// Create a new HTTP client with the specified timeout and proxy, if any.
+	httpClient := http.DefaultClient
+
+	if cfg.proxyURL != "" {
+		proxyURL, err := url.Parse(cfg.proxyURL)
+		if err != nil {
+			return nil, fmt.Errorf("can't parse the proxy URL: %s", err)
+		}
+		tr.Proxy = http.ProxyURL(proxyURL)
+	} else if cfg.socksURL != "" {
+		dialer, err := proxy.SOCKS5("tcp", cfg.socksURL, nil, proxy.Direct)
+		if err != nil {
+			return nil, fmt.Errorf("can't connect to the proxy: %s", err)
+		}
+		tr.DialContext = dialer.(proxy.ContextDialer).DialContext
+	}
+
+	// Set the HTTP client to use the default header transport with the specified headers.
+	httpClient.Transport = &DefaultHeaderTransport{
+		Origin: tr,
+	}
+
 	// Create a new client instance with the necessary fields.
 	engine := &Client{
-		client:      anthropic.NewClient(cfg.apiKey),
+		client: anthropic.NewClient(
+			cfg.apiKey,
+			anthropic.WithHTTPClient(httpClient),
+		),
 		model:       cfg.model,
 		maxTokens:   cfg.maxTokens,
 		temperature: cfg.temperature,

provider/anthropic/options.go

@@ -81,13 +81,37 @@ func WithTopP(val float32) Option {
 	})
 }
 
+// WithProxyURL is a function that returns an Option, which sets the proxyURL field of the config struct.
+func WithProxyURL(val string) Option {
+	return optionFunc(func(c *config) {
+		c.proxyURL = val
+	})
+}
+
+// WithSocksURL is a function that returns an Option, which sets the socksURL field of the config struct.
+func WithSocksURL(val string) Option {
+	return optionFunc(func(c *config) {
+		c.socksURL = val
+	})
+}
+
+// WithSkipVerify returns a new Option that sets the skipVerify for the client configuration.
+func WithSkipVerify(val bool) Option {
+	return optionFunc(func(c *config) {
+		c.skipVerify = val
+	})
+}
+
 // config is a struct that stores configuration options for the instrumentation.
 type config struct {
 	apiKey      string
 	model       anthropic.Model
 	maxTokens   int
 	temperature float32
 	topP        float32
+	proxyURL    string
+	socksURL    string
+	skipVerify  bool
 }
 
 // valid checks whether a config object is valid, returning an error if it is not.