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

@@ -31,11 +31,12 @@ type Config struct {
 		Tools        []string `yaml:"tools"`
 	} `yaml:"agents"`
 	Providers []*struct {
-		Name    string   `yaml:"name,omitempty"`
-		Kind    string   `yaml:"kind"`
-		BaseURL string   `yaml:"baseUrl,omitempty"`
-		APIKey  string   `yaml:"apiKey,omitempty"`
-		Models  []string `yaml:"models"`
+		Name    string            `yaml:"name,omitempty"`
+		Kind    string            `yaml:"kind"`
+		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)