Support Request Level Custom Http Headers..

sashabaranov · Dec 3, 2024 · dc6bca5 · dc6bca5
1 parent c203ca0
commit dc6bca5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/chat.go b/chat.go
@@ -393,12 +393,12 @@ func (c *Client) CreateChatCompletion(
 	if err = validateRequestForO1Models(request); err != nil {
 		return
 	}
-
 	req, err := c.newRequest(
 		ctx,
 		http.MethodPost,
 		c.fullURL(urlSuffix, withModel(request.Model)),
 		withBody(request),
+		withCustomHeaders(request.Metadata),
 	)
 	if err != nil {
 		return

diff --git a/client.go b/client.go
@@ -90,6 +90,14 @@ func withContentType(contentType string) requestOption {
 	}
 }
 
+func withCustomHeaders(headers map[string]string) requestOption {
+	return func(args *requestOptions) {
+		for key, value := range headers {
+			args.header.Set(key, value)
+		}
+	}
+}
+
 func withBetaAssistantVersion(version string) requestOption {
 	return func(args *requestOptions) {
 		args.header.Set("OpenAI-Beta", fmt.Sprintf("assistants=%s", version))