Add optFns

hupe1980 · Jun 25, 2023 · 57d46c6 · 57d46c6
1 parent 39034d9
commit 57d46c6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/huggingface.go b/huggingface.go
@@ -35,12 +35,16 @@ type InferenceClient struct {
 }
 
 // NewInferenceClient creates a new InferenceClient instance with the specified token.
-func NewInferenceClient(token string) *InferenceClient {
+func NewInferenceClient(token string, optFns ...func(o *InferenceClientOptions)) *InferenceClient {
 	opts := InferenceClientOptions{
 		Endpoint:          "https://huggingface.co",
 		InferenceEndpoint: "https://api-inference.huggingface.co",
 	}
 
+	for _, fn := range optFns {
+		fn(&opts)
+	}
+
 	return &InferenceClient{
 		httpClient: http.DefaultClient,
 		token:      token,