Merge pull request #894 from dgageot/generating-titles

Generating titles

Author: dgageot

pkg/model/provider/anthropic/client.go

@@ -115,18 +115,21 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
 			baseURL := fmt.Sprintf("%s://%s%s/", url.Scheme, url.Host, url.Path)
 
 			// Configure a custom HTTP client to inject headers and query params used by the Gateway.
-			httpClient := httpclient.NewHTTPClient(
+			httpOptions := []httpclient.Opt{
 				httpclient.WithProxiedBaseURL(defaultsTo(cfg.BaseURL, "https://api.anthropic.com/")),
 				httpclient.WithProvider(cfg.Provider),
 				httpclient.WithModel(cfg.Model),
 				httpclient.WithQuery(url.Query()),
-			)
+			}
+			if globalOptions.GeneratingTitle() {
+				httpOptions = append(httpOptions, httpclient.WithHeader("X-Cagent-GeneratingTitle", "1"))
+			}
 
 			client := anthropic.NewClient(
 				option.WithAuthToken(authToken),
 				option.WithAPIKey(authToken),
 				option.WithBaseURL(baseURL),
-				option.WithHTTPClient(httpClient),
+				option.WithHTTPClient(httpclient.NewHTTPClient(httpOptions...)),
 			)
 
 			return client, nil

pkg/model/provider/clone.go

@@ -16,6 +16,7 @@ func CloneWithOptions(ctx context.Context, base Provider, opts ...options.Opt) P
 	// Preserve existing options, then apply overrides. Later opts take precedence.
 	baseOpts := options.FromModelOptions(config.ModelOptions)
 	mergedOpts := append(baseOpts, opts...)
+	mergedOpts = append(mergedOpts, options.WithGeneratingTitle())
 
 	clone, err := New(ctx, &config.ModelConfig, config.Env, mergedOpts...)
 	if err != nil {

pkg/model/provider/gemini/client.go

@@ -87,15 +87,20 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
 			}
 			baseURL := fmt.Sprintf("%s://%s%s/", url.Scheme, url.Host, url.Path)
 
+			httpOptions := []httpclient.Opt{
+				httpclient.WithProxiedBaseURL(defaultsTo(cfg.BaseURL, "https://generativelanguage.googleapis.com/")),
+				httpclient.WithProvider(cfg.Provider),
+				httpclient.WithModel(cfg.Model),
+				httpclient.WithQuery(url.Query()),
+			}
+			if globalOptions.GeneratingTitle() {
+				httpOptions = append(httpOptions, httpclient.WithHeader("X-Cagent-GeneratingTitle", "1"))
+			}
+
 			return genai.NewClient(ctx, &genai.ClientConfig{
-				APIKey:  authToken,
-				Backend: genai.BackendGeminiAPI,
-				HTTPClient: httpclient.NewHTTPClient(
-					httpclient.WithProxiedBaseURL(defaultsTo(cfg.BaseURL, "https://generativelanguage.googleapis.com/")),
-					httpclient.WithProvider(cfg.Provider),
-					httpclient.WithModel(cfg.Model),
-					httpclient.WithQuery(url.Query()),
-				),
+				APIKey:     authToken,
+				Backend:    genai.BackendGeminiAPI,
+				HTTPClient: httpclient.NewHTTPClient(httpOptions...),
 				HTTPOptions: genai.HTTPOptions{
 					BaseURL: baseURL,
 					Headers: http.Header{

pkg/model/provider/openai/client.go

@@ -106,17 +106,20 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
 			baseURL := fmt.Sprintf("%s://%s%s/v1/", url.Scheme, url.Host, url.Path)
 
 			// Configure a custom HTTP client to inject headers and query params used by the Gateway.
-			httpClient := httpclient.NewHTTPClient(
+			httpOptions := []httpclient.Opt{
 				httpclient.WithProxiedBaseURL(defaultsTo(cfg.BaseURL, "https://api.openai.com/v1")),
 				httpclient.WithProvider(cfg.Provider),
 				httpclient.WithModel(cfg.Model),
 				httpclient.WithQuery(url.Query()),
-			)
+			}
+			if globalOptions.GeneratingTitle() {
+				httpOptions = append(httpOptions, httpclient.WithHeader("X-Cagent-GeneratingTitle", "1"))
+			}
 
 			client := openai.NewClient(
 				option.WithAPIKey(authToken),
 				option.WithBaseURL(baseURL),
-				option.WithHTTPClient(httpClient),
+				option.WithHTTPClient(httpclient.NewHTTPClient(httpOptions...)),
 			)
 
 			return &client, nil

pkg/model/provider/options/options.go

@@ -7,6 +7,7 @@ import (
 type ModelOptions struct {
 	gateway          string
 	structuredOutput *latest.StructuredOutput
+	generatingTitle  bool
 }
 
 func (c *ModelOptions) Gateway() string {
@@ -17,6 +18,10 @@ func (c *ModelOptions) StructuredOutput() *latest.StructuredOutput {
 	return c.structuredOutput
 }
 
+func (c *ModelOptions) GeneratingTitle() bool {
+	return c.generatingTitle
+}
+
 type Opt func(*ModelOptions)
 
 func WithGateway(gateway string) Opt {
@@ -31,6 +36,12 @@ func WithStructuredOutput(structuredOutput *latest.StructuredOutput) Opt {
 	}
 }
 
+func WithGeneratingTitle() Opt {
+	return func(cfg *ModelOptions) {
+		cfg.generatingTitle = true
+	}
+}
+
 // FromModelOptions converts a concrete ModelOptions value into a slice of
 // Opt configuration functions. Later Opts override earlier ones when applied.
 func FromModelOptions(m ModelOptions) []Opt {
@@ -41,5 +52,8 @@ func FromModelOptions(m ModelOptions) []Opt {
 	if m.structuredOutput != nil {
 		out = append(out, WithStructuredOutput(m.structuredOutput))
 	}
+	if m.generatingTitle {
+		out = append(out, WithGeneratingTitle())
+	}
 	return out
 }