Private
Public Access
1
0

Allow custom headers on OpenAI providers (to be added to more later)

This commit is contained in:
2024-08-12 17:14:53 +00:00
parent fe838f400f
commit 434fc4672b
3 changed files with 11 additions and 5 deletions

View File

@@ -16,6 +16,7 @@ import (
type OpenAIClient struct {
APIKey string
BaseURL string
Headers map[string]string
}
type ChatCompletionMessage struct {
@@ -198,6 +199,9 @@ func (c *OpenAIClient) sendRequest(ctx context.Context, r ChatCompletionRequest)
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer "+c.APIKey)
for header, val := range c.Headers {
req.Header.Set(header, val)
}
client := &http.Client{}
resp, err := client.Do(req)