Luis Pater
e3e60f914b
- Added `disable-image-generation` configuration flag to disable the `image_generation` tool globally. - Updated payload handling to remove `image_generation` tools from request payload arrays when the flag is enabled. - Modified OpenAI image handlers (`ImagesGenerations`, `ImagesEdits`) to return 404 when the feature is disabled. - Enhanced configuration diff logging to track changes for the `disable-image-generation` flag. - Added accompanying unit tests for the new feature in payload helpers and image handler logic.
56 lines
3.0 KiB
Go
56 lines
3.0 KiB
Go
// Package config provides configuration management for the CLI Proxy API server.
|
|
// It handles loading and parsing YAML configuration files, and provides structured
|
|
// access to application settings including server port, authentication directory,
|
|
// debug settings, proxy configuration, and API keys.
|
|
package config
|
|
|
|
// SDKConfig represents the application's configuration, loaded from a YAML file.
|
|
type SDKConfig struct {
|
|
// ProxyURL is the URL of an optional proxy server to use for outbound requests.
|
|
ProxyURL string `yaml:"proxy-url" json:"proxy-url"`
|
|
|
|
// DisableImageGeneration disables the built-in image_generation tool when true.
|
|
// When enabled, the server will avoid injecting image_generation into request payloads,
|
|
// will remove any existing image_generation tool entries from tools arrays, and will
|
|
// return 404 for /v1/images/generations and /v1/images/edits.
|
|
DisableImageGeneration bool `yaml:"disable-image-generation" json:"disable-image-generation"`
|
|
|
|
// EnableGeminiCLIEndpoint controls whether Gemini CLI internal endpoints (/v1internal:*) are enabled.
|
|
// Default is false for safety; when false, /v1internal:* requests are rejected.
|
|
EnableGeminiCLIEndpoint bool `yaml:"enable-gemini-cli-endpoint" json:"enable-gemini-cli-endpoint"`
|
|
|
|
// ForceModelPrefix requires explicit model prefixes (e.g., "teamA/gemini-3-pro-preview")
|
|
// to target prefixed credentials. When false, unprefixed model requests may use prefixed
|
|
// credentials as well.
|
|
ForceModelPrefix bool `yaml:"force-model-prefix" json:"force-model-prefix"`
|
|
|
|
// RequestLog enables or disables detailed request logging functionality.
|
|
RequestLog bool `yaml:"request-log" json:"request-log"`
|
|
|
|
// APIKeys is a list of keys for authenticating clients to this proxy server.
|
|
APIKeys []string `yaml:"api-keys" json:"api-keys"`
|
|
|
|
// PassthroughHeaders controls whether upstream response headers are forwarded to downstream clients.
|
|
// Default is false (disabled).
|
|
PassthroughHeaders bool `yaml:"passthrough-headers" json:"passthrough-headers"`
|
|
|
|
// Streaming configures server-side streaming behavior (keep-alives and safe bootstrap retries).
|
|
Streaming StreamingConfig `yaml:"streaming" json:"streaming"`
|
|
|
|
// NonStreamKeepAliveInterval controls how often blank lines are emitted for non-streaming responses.
|
|
// <= 0 disables keep-alives. Value is in seconds.
|
|
NonStreamKeepAliveInterval int `yaml:"nonstream-keepalive-interval,omitempty" json:"nonstream-keepalive-interval,omitempty"`
|
|
}
|
|
|
|
// StreamingConfig holds server streaming behavior configuration.
|
|
type StreamingConfig struct {
|
|
// KeepAliveSeconds controls how often the server emits SSE heartbeats (": keep-alive\n\n").
|
|
// <= 0 disables keep-alives. Default is 0.
|
|
KeepAliveSeconds int `yaml:"keepalive-seconds,omitempty" json:"keepalive-seconds,omitempty"`
|
|
|
|
// BootstrapRetries controls how many times the server may retry a streaming request before any bytes are sent,
|
|
// to allow auth rotation / transient recovery.
|
|
// <= 0 disables bootstrap retries. Default is 0.
|
|
BootstrapRetries int `yaml:"bootstrap-retries,omitempty" json:"bootstrap-retries,omitempty"`
|
|
}
|