add httpClient option to support proxy requests (#30)

Co-authored-by: huanghaifeng <[email protected]>

pkg/anthropic/client.go

@@ -12,7 +12,7 @@ type Client struct {
 }
 
 // NewClient initializes a new Anthropic API client with the required headers.
-func NewClient(apiKey string) (*Client, error) {
+func NewClient(apiKey string, options ...GenericOption[Client]) (*Client, error) {
 	if apiKey == "" {
 		return nil, ErrAnthropicApiKeyRequired
 	}
@@ -22,6 +22,9 @@ func NewClient(apiKey string) (*Client, error) {
 		apiKey:     apiKey,
 		baseURL:    "https://api.anthropic.com",
 	}
+	for _, opt := range options {
+		opt(client)
+	}
 
 	return client, nil
 }

pkg/anthropic/options.go

@@ -1,5 +1,7 @@
 package anthropic
 
+import "net/http"
+
 type CompletionOption func(*CompletionRequest)
 type MessageOption func(*MessageRequest)
 
@@ -124,3 +126,12 @@ func WithTopP[T any](topP float64) GenericOption[T] {
 		}
 	}
 }
+
+// WithHTTPClient sets a custom HTTP client for the Client.
+func WithHTTPClient[T any](httpClient *http.Client) GenericOption[T] {
+	return func(r *T) {
+		if v, ok := any(r).(*Client); ok {
+			v.httpClient = httpClient
+		}
+	}
+}