/ai/options.go

package ai

import (
	"context"
)

// Options for model configuration
type Options struct {
	// Context for the model
	Context context.Context
	// Model name (e.g., "gpt-4o", "claude-sonnet-4-20250514")
	Model string
	// APIKey for authentication
	APIKey string
	// BaseURL for the API endpoint
	BaseURL string
	// ToolHandler handles tool calls (optional, for automatic tool execution)
	ToolHandler ToolHandler
}

// GenerateOptions for generate call
type GenerateOptions struct {
	// Context for this specific generate call
	Context context.Context
}

// Option is a function that modifies Options
type Option func(*Options)

// GenerateOption is a function that modifies GenerateOptions
type GenerateOption func(*GenerateOptions)

// NewOptions creates new Options with defaults
func NewOptions(opts ...Option) Options {
	options := Options{
		Context: context.Background(),
	}
	for _, o := range opts {
		o(&options)
	}
	return options
}

// WithModel sets the model name
func WithModel(m string) Option {
	return func(o *Options) {
		o.Model = m
	}
}

// WithAPIKey sets the API key
func WithAPIKey(key string) Option {
	return func(o *Options) {
		o.APIKey = key
	}
}

// WithBaseURL sets the base URL
func WithBaseURL(url string) Option {
	return func(o *Options) {
		o.BaseURL = url
	}
}

// WithContext sets the context
func WithContext(ctx context.Context) Option {
	return func(o *Options) {
		o.Context = ctx
	}
}

// WithToolHandler sets the tool handler
func WithToolHandler(handler ToolHandler) Option {
	return func(o *Options) {
		o.ToolHandler = handler
	}
}

// WithTools wires a Tools instance into the model, setting the tool
// handler so the model can execute discovered service endpoints. The
// tool list itself is passed per-request via Request.Tools.
//
//	tools := ai.NewTools(service.Registry())
//	list, _ := tools.Discover()
//	m := ai.New("anthropic", ai.WithAPIKey(key), ai.WithTools(tools))
//	resp, _ := m.Generate(ctx, &ai.Request{Prompt: input, Tools: list})
func WithTools(t *Tools) Option {
	return func(o *Options) {
		if t != nil {
			o.ToolHandler = t.Handler()
		}
	}
}