package agent

type ChatMessage struct {
	Role     string
	Contents []ChatMessageContent
}

type ChatMessageContent struct {
	Type    string // text, image, audio, video
	Content string
}

type ChatModelConfig struct {
	defaultConfig *ChatModelConfig
	Model         string
	Ratio         float64
	MaxTokens     int
	Temperature   float64
	TopP          float64
	Tools         map[string]any
}

func (cmc *ChatModelConfig) SetDefault(config *ChatModelConfig) {
	cmc.defaultConfig = config
}

func (cmc *ChatModelConfig) GetModel() string {
	if cmc.Model == "" && cmc.defaultConfig != nil {
		return cmc.defaultConfig.Model
	}
	return cmc.Model
}

func (cmc *ChatModelConfig) GetMaxTokens() int {
	if cmc.MaxTokens == 0 && cmc.defaultConfig != nil {
		return cmc.defaultConfig.MaxTokens
	}
	return cmc.MaxTokens
}

func (cmc *ChatModelConfig) GetTemperature() float64 {
	if cmc.Temperature == 0 && cmc.defaultConfig != nil {
		return cmc.defaultConfig.Temperature
	}
	return cmc.Temperature
}

func (cmc *ChatModelConfig) GetTopP() float64 {
	if cmc.TopP == 0 && cmc.defaultConfig != nil {
		return cmc.defaultConfig.TopP
	}
	return cmc.TopP
}

func (cmc *ChatModelConfig) GetTools() map[string]any {
	if cmc.Tools == nil && cmc.defaultConfig != nil {
		return cmc.defaultConfig.Tools
	}
	return cmc.Tools
}