modelcatalog

type Catalog struct {
	Version         int                       `yaml:"version" json:"version"`
	DefaultProvider string                    `yaml:"default_provider" json:"default_provider"`
	DefaultModel    string                    `yaml:"default_model" json:"default_model"`
	Defaults        map[string]HarnessDefault `yaml:"defaults" json:"defaults"`
	Providers       map[string]Provider       `yaml:"providers" json:"providers"`
}

type HarnessDefault struct {
	Provider string `yaml:"provider" json:"provider"`
	Model    string `yaml:"model" json:"model"`
}

type Model struct {
	ID        string                  `yaml:"id" json:"id"`
	Name      string                  `yaml:"name" json:"name"`
	Aliases   []string                `yaml:"aliases" json:"aliases"`
	Harnesses map[string]ModelHarness `yaml:"harnesses" json:"harnesses"`
}

type ModelHarness struct {
	ID       string         `yaml:"id" json:"id"`
	Disabled bool           `yaml:"disabled" json:"disabled"`
	Options  map[string]any `yaml:"options" json:"options"`
}

type OpenCodeProvider struct {
	ID        string
	Name      string
	BaseURL   string
	APIKeyEnv string
	Models    []string
}

type Provider struct {
	Name       string                     `yaml:"name" json:"name"`
	Kind       string                     `yaml:"kind" json:"kind"`
	BaseURL    string                     `yaml:"base_url" json:"base_url"`
	APIKeyEnv  string                     `yaml:"api_key_env" json:"api_key_env"`
	AwsProfile string                     `yaml:"aws_profile" json:"aws_profile"`
	AwsRegion  string                     `yaml:"aws_region" json:"aws_region"`
	Models     []Model                    `yaml:"models" json:"models"`
	Harnesses  map[string]ProviderHarness `yaml:"harnesses" json:"harnesses"`
}

type ProviderHarness struct {
	ID         string `yaml:"id" json:"id"`
	Name       string `yaml:"name" json:"name"`
	BaseURL    string `yaml:"base_url" json:"base_url"`
	APIKeyEnv  string `yaml:"api_key_env" json:"api_key_env"`
	AwsProfile string `yaml:"aws_profile" json:"aws_profile"`
	AwsRegion  string `yaml:"aws_region" json:"aws_region"`
	API        string `yaml:"api" json:"api"`
	AuthHeader bool   `yaml:"auth_header" json:"auth_header"`
	Disabled   bool   `yaml:"disabled" json:"disabled"`
}