options.go

package goai

import (
	"time"
)

// Options contains configuration options for LLM providers
type Options struct {
	APIKey      string
	Endpoint    string
	Timeout     time.Duration
	MaxTokens   int
	Temperature float64
}

// Option is a functional option for configuring a Client
type Option func(*Options)

// defaultOptions returns default options
func defaultOptions() Options {
	return Options{
		Timeout:     60 * time.Second,
		MaxTokens:   500,
		Temperature: 0.7,
	}
}

// WithAPIKey sets the API key for the provider
//
// Example:
//   client := goai.New("openai", "gpt-4o",
//       goai.WithAPIKey(os.Getenv("OPENAI_API_KEY")),
//   )
func WithAPIKey(key string) Option {
	return func(o *Options) {
		o.APIKey = key
	}
}

// WithEndpoint sets a custom API endpoint
//
// Example:
//   client := goai.New("openai", "gpt-4o",
//       goai.WithAPIKey(apiKey),
//       goai.WithEndpoint("https://api.openai.com/v1/chat/completions"),
//   )
func WithEndpoint(url string) Option {
	return func(o *Options) {
		o.Endpoint = url
	}
}

// WithTimeout sets the timeout for API requests
//
// Example:
//   client := goai.New("openai", "gpt-4o",
//       goai.WithAPIKey(apiKey),
//       goai.WithTimeout(30*time.Second),
//   )
func WithTimeout(d time.Duration) Option {
	return func(o *Options) {
		o.Timeout = d
	}
}

// WithMaxTokens sets the maximum number of tokens in the response
//
// Example:
//   client := goai.New("openai", "gpt-4o",
//       goai.WithAPIKey(apiKey),
//       goai.WithMaxTokens(1000),
//   )
func WithMaxTokens(n int) Option {
	return func(o *Options) {
		o.MaxTokens = n
	}
}

// WithTemperature sets the temperature for response generation
//
// Temperature controls randomness: 0.0 is deterministic, higher values are more random.
//
// Example:
//   client := goai.New("openai", "gpt-4o",
//       goai.WithAPIKey(apiKey),
//       goai.WithTemperature(0.9),
//   )
func WithTemperature(t float64) Option {
	return func(o *Options) {
		o.Temperature = t
	}
}