runtime

type AuthConfig struct {
	Type            AuthType `json:"type,omitempty"`
	APIKeyEnv       string   `json:"api_key_env,omitempty"`
	APIKey          string   `json:"api_key,omitempty"`
	AuthorizeURL    string   `json:"authorize_url,omitempty"`
	TokenURL        string   `json:"token_url,omitempty"`
	ClientID        string   `json:"client_id,omitempty"`
	IDP             string   `json:"idp,omitempty"`
	TokenAuthMethod string   `json:"token_auth_method,omitempty"`
}

type AuthResolver interface {
	// Resolve returns a credential for the given provider configuration.
	Resolve(ctx context.Context, cfg ProviderConfig) (AuthResult, error)
}

type AuthResolverFunc func(ctx context.Context, cfg ProviderConfig) (AuthResult, error)

type AuthResult struct {
	Token   string
	Headers map[string]string
}

type AuthType string

type Capability string

type Catalog struct {
	// contains filtered or unexported fields
}

type CatalogModel struct {
	ID          string `json:"id"`
	Name        string `json:"name,omitzero"`
	Family      string `json:"family,omitzero"`
	Attachment  bool   `json:"attachment,omitzero"`
	Reasoning   bool   `json:"reasoning,omitzero"`
	ToolCall    bool   `json:"tool_call,omitzero"`
	Structured  bool   `json:"structured_output,omitzero"`
	Temperature bool   `json:"temperature,omitzero"`
	Modalities  struct {
		Input  []string `json:"input,omitzero"`
		Output []string `json:"output,omitzero"`
	} `json:"modalities"`
	Limit struct {
		Context int `json:"context,omitzero"`
		Output  int `json:"output,omitzero"`
	} `json:"limit"`
	Cost struct {
		Input      float64 `json:"input,omitzero"`
		Output     float64 `json:"output,omitzero"`
		CacheRead  float64 `json:"cache_read,omitzero"`
		CacheWrite float64 `json:"cache_write,omitzero"`
	} `json:"cost"`
	ReasoningOptions []ReasoningOption `json:"reasoning_options,omitzero"`
}

type CatalogOptions struct {
	// URL is the endpoint to fetch. Defaults to DefaultCatalogURL.
	URL string

	// CachePath is a file path used to store the fetched JSON. Empty
	// disables disk caching.
	CachePath string

	// TTL is the maximum age of a cached file before a network refresh.
	// Zero or negative disables freshness checks (always fetch).
	TTL time.Duration

	// HTTPClient is used for network requests. Nil means a default
	// client with a 30s timeout.
	HTTPClient *http.Client
}

type CatalogProvider struct {
	ID     string                  `json:"id"`
	NPM    string                  `json:"npm,omitzero"`
	API    string                  `json:"api,omitzero"`
	Env    []string                `json:"env,omitzero"`
	Models map[string]CatalogModel `json:"models,omitzero"`
}

type Config struct {
	// DefaultProvider is the provider ID used when a model reference
	// does not include a provider prefix.
	DefaultProvider string `json:"default_provider,omitempty"`

	// Providers are the configured provider instances. The map key is
	// the local provider ID and must match ProviderConfig.ID.
	Providers map[string]ProviderConfig `json:"providers,omitempty"`

	// CatalogURL overrides the default models.dev endpoint. Empty means
	// use DefaultCatalogURL.
	CatalogURL string `json:"catalog_url,omitempty"`

	// CatalogCachePath enables on-disk caching of the fetched catalog.
	CatalogCachePath string `json:"catalog_cache_path,omitempty"`
}

type CostConfig struct {
	Input      float64 `json:"input,omitempty"`
	Output     float64 `json:"output,omitempty"`
	CacheRead  float64 `json:"cache_read,omitempty"`
	CacheWrite float64 `json:"cache_write,omitempty"`
}

type ModelConfig struct {
	ID               string         `json:"id"`
	Name             string         `json:"name,omitempty"`
	URL              string         `json:"url,omitempty"`
	ContextWindow    int            `json:"context_window,omitempty"`
	MaxOutputTokens  int            `json:"max_output_tokens,omitempty"`
	Reasoning        bool           `json:"reasoning,omitempty"`
	ToolCall         bool           `json:"tool_call,omitempty"`
	StructuredOutput bool           `json:"structured_output,omitempty"`
	Temperature      bool           `json:"temperature,omitempty"`
	Cost             CostConfig     `json:"cost"`
	Extra            map[string]any `json:"extra,omitempty"`
}

type ModelInfo struct {
	ID               string
	ProviderID       string
	Name             string
	URL              string
	ContextWindow    int
	MaxOutputTokens  int
	Reasoning        bool
	ReasoningOptions []ReasoningOption
	ToolCall         bool
	StructuredOutput bool
	Temperature      bool
	Cost             CostConfig
	Extra            map[string]any
}

type ModelRef struct {
	ProviderID string
	ModelID    string
}

type OAuthPKCEResolver struct {
	// CacheDir overrides the directory used to store access tokens.
	// Defaults to os.UserConfigDir()/ai-sdk/oauth.
	CacheDir string

	// OpenBrowser wraps browser.OpenURL. Tests can override it.
	OpenBrowser func(url string) error
}

type ProviderClass interface {
	// Name returns the stable class identifier used in ProviderConfig.Class.
	Name() string

	// Supports reports whether this class can satisfy cap.
	Supports(cap Capability) bool

	// New builds a ProviderSet from cfg for the given model. ctx may be
	// used for short-lived setup requests (e.g. discovery or token
	// exchange), but must not be stored.
	New(ctx context.Context, cfg ProviderConfig, model ModelInfo) (ProviderSet, error)
}

type ProviderConfig struct {
	// ID is the local identifier for this provider instance (e.g. "openai",
	// "my-maas"). It must be unique within a Runtime.
	ID string

	// Class selects the registered ProviderClass factory. Examples:
	// "openai-compatible", "anthropic-messages".
	Class string

	// BaseURL is the provider's API base URL. Individual models may
	// override this via their own URL field.
	BaseURL string

	// Auth describes how to obtain credentials for this provider.
	Auth AuthConfig

	// Headers are extra HTTP headers merged into every request.
	Headers map[string]string

	// Insecure disables TLS certificate verification.
	Insecure bool

	// Timeout for provider HTTP requests in milliseconds. Zero means the
	// class default.
	Timeout int

	// Options carries class-specific options as a generic map. Classes
	// read only the keys they understand.
	Options map[string]any

	// Models allows per-provider model overrides/advertisements. These
	// augment or override the catalog when the runtime resolves model
	// references.
	Models []ModelConfig
}

type ProviderSet struct {
	Chat  chat.Provider
	Embed embed.Provider
}

type ReasoningOption struct {
	Type   string   `json:"type"`
	Values []string `json:"values,omitzero"`
}

type Runtime struct {
	// contains filtered or unexported fields
}