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

pkg/api/provider/openai/openai.go

@@ -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)

pkg/lmcli/config.go

@@ -36,6 +36,7 @@ type Config struct {
 		BaseURL string            `yaml:"baseUrl,omitempty"`
 		APIKey  string            `yaml:"apiKey,omitempty"`
 		Models  []string          `yaml:"models"`
+		Headers map[string]string `yaml:"headers"`
 	} `yaml:"providers"`
 }
 

pkg/lmcli/lmcli.go

@@ -179,6 +179,7 @@ func (c *Context) GetModelProvider(model string) (string, api.ChatCompletionProv
 					return model, &openai.OpenAIClient{
 						BaseURL: url,
 						APIKey:  p.APIKey,
+						Headers: p.Headers,
 					}, nil
 				default:
 					return "", nil, fmt.Errorf("unknown provider kind: %s", p.Kind)