Also add the provider and model with using the gateway

Signed-off-by: David Gageot <david.gageot@docker.com>

Author: dgageot

pkg/httpclient/client.go

@@ -54,6 +54,18 @@ func WithProxiedBaseURL(value string) Opt {
 	}
 }
 
+func WithProvider(provider string) Opt {
+	return func(o *HTTPOptions) {
+		o.Header.Set("X-Cagent-Provider", provider)
+	}
+}
+
+func WithModel(model string) Opt {
+	return func(o *HTTPOptions) {
+		o.Header.Set("X-Cagent-Model", model)
+	}
+}
+
 type userAgentTransport struct {
 	httpOptions HTTPOptions
 	rt          http.RoundTripper

pkg/model/provider/anthropic/client.go

@@ -112,6 +112,8 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
 				option.WithBaseURL(gateway),
 				option.WithHTTPClient(httpclient.NewHTTPClient(
 					httpclient.WithProxiedBaseURL(defaultsTo(cfg.BaseURL, "https://api.anthropic.com/")),
+					httpclient.WithProvider(cfg.Provider),
+					httpclient.WithModel(cfg.Model),
 				)),
 			), nil
 		}

pkg/model/provider/gemini/client.go

@@ -84,6 +84,8 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
 				Backend: genai.BackendGeminiAPI,
 				HTTPClient: httpclient.NewHTTPClient(
 					httpclient.WithProxiedBaseURL(defaultsTo(cfg.BaseURL, "https://generativelanguage.googleapis.com/")),
+					httpclient.WithProvider(cfg.Provider),
+					httpclient.WithModel(cfg.Model),
 				),
 				HTTPOptions: genai.HTTPOptions{
 					BaseURL: gateway,

pkg/model/provider/openai/client.go

@@ -111,6 +111,8 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
 			openaiConfig.BaseURL = gateway + "/v1"
 			openaiConfig.HTTPClient = httpclient.NewHTTPClient(
 				httpclient.WithProxiedBaseURL(defaultsTo(cfg.BaseURL, "https://api.openai.com/v1")),
+				httpclient.WithProvider(cfg.Provider),
+				httpclient.WithModel(cfg.Model),
 			)
 
 			return openai.NewClientWithConfig(openaiConfig), nil