config

type Budget struct {
	Tokens  int     `yaml:"tokens"`
	Dollars float64 `yaml:"dollars"`
}

type File struct {
	Models      []ModelManifest              `yaml:"models"`
	Tools       []ToolManifest               `yaml:"tools"`
	Include     []string                     `yaml:"include"` // Add include support for role files
	Memory      string                       `yaml:"memory"`
	Store       string                       `yaml:"store"`
	Vector      VectorManifest               `yaml:"vector_store"`
	Theme       string                       `yaml:"theme"`
	Themes      map[string]string            `yaml:"themes"`
	Keybinds    map[string]string            `yaml:"keybinds"`
	Credentials map[string]map[string]string `yaml:"credentials"`
	MCPServers  map[string]string            `yaml:"mcp_servers"`
	Collector   string                       `yaml:"collector"`
	Port        string                       `yaml:"port"`
	Sandbox     Sandbox                      `yaml:"sandbox"`
	Permissions Permissions                  `yaml:"permissions"`
	Budget      Budget                       `yaml:"budget"`
}

type ModelManifest struct {
	Name     string            `yaml:"name"`
	Provider string            `yaml:"provider"`
	Options  map[string]string `yaml:"options,omitempty"`
}

type Permissions struct {
	Tools []string `yaml:"tools"`
}

type Sandbox struct {
	Engine string `yaml:"engine"`
}

type ToolManifest struct {
	Name        string          `yaml:"name"`
	Description string          `yaml:"description"`
	Type        string          `yaml:"type,omitempty"`
	Command     string          `yaml:"command,omitempty"`
	HTTP        string          `yaml:"http,omitempty"`
	Args        map[string]any  `yaml:"args,omitempty"`
	Privileged  bool            `yaml:"privileged,omitempty"`
	Net         string          `yaml:"net,omitempty"`
	CPULimit    string          `yaml:"cpu_limit,omitempty"`
	MemLimit    string          `yaml:"mem_limit,omitempty"`
	Engine      string          `yaml:"engine,omitempty"`
	Permissions ToolPermissions `yaml:"permissions,omitempty"`
}

type ToolPermissions struct {
	Allow *bool `yaml:"allow"`
}

type VectorManifest struct {
	Type       string `yaml:"type"`
	URL        string `yaml:"url"`
	Collection string `yaml:"collection,omitempty"`
}