ability

type Ability struct {
	Demo bool    `json:"demo"`
	LLM  LLMAblt `json:"llm"`
	TTS  TTSAblt `json:"tts"`
	STT  STTAblt `json:"stt"`
}

type ChatGPTAblt struct {
	Available bool     `json:"available"`
	Models    []string `json:"models"`
}

type ChatGPTOption struct {
	Model            string  `json:"model"`
	MaxTokens        int     `json:"maxTokens"`
	Temperature      float32 `json:"temperature"`
	TopP             float32 `json:"topP"`
	PresencePenalty  float32 `json:"presencePenalty"`
	FrequencyPenalty float32 `json:"frequencyPenalty"`
}

type ElevenlabsTTSAblt struct {
	Available bool         `json:"available"`
	Voices    []TaggedItem `json:"voices"`
}

type ElevenlabsTTSOption struct {
	VoiceId   string  `json:"voiceId"`
	Stability float32 `json:"stability"`
	Clarity   float32 `json:"clarity"`
}

type GoogleSTTAb struct {
	Available   bool         `json:"available"`
	Recognizers []TaggedItem `json:"recognizers"`
}

type GoogleSTTOption struct {
	Recognizer string `json:"recognizer"`
	Model      string `json:"model"`
	Language   string `json:"language"`
}

type GoogleTTSAblt struct {
	Available bool         `json:"available"`
	Voices    []TaggedItem `json:"voices"`
}

type GoogleTTSOption struct {
	// if VoiceId is provided, LanguageCode and Gender will not be used
	VoiceId      string                         `json:"voiceId"`
	LanguageCode string                         `json:"languageCode"`
	Gender       texttospeechpb.SsmlVoiceGender `json:"gender"`
	SpeakingRate float64                        `json:"speakingRate"`
	Pitch        float64                        `json:"pitch"`
	VolumeGainDb float64                        `json:"volumeGainDb"`
}

type LLMAblt struct {
	Available bool        `json:"available"`
	ChatGPT   ChatGPTAblt `json:"chatGPT"`
}

type LLMOption struct {
	ChatGPT *ChatGPTOption `json:"chatGPT"`
}

type STTAblt struct {
	Available bool         `json:"available"`
	Whisper   WhisperSTTAb `json:"whisper"`
	Google    GoogleSTTAb  `json:"google"`
}

type STTOption struct {
	Whisper *WhisperOption   `json:"whisper"`
	Google  *GoogleSTTOption `json:"google"`
}

type TTSAblt struct {
	Available  bool              `json:"available"`
	Google     GoogleTTSAblt     `json:"google"`
	Elevenlabs ElevenlabsTTSAblt `json:"elevenlabs"`
}

type TTSOption struct {
	Elevenlabs *ElevenlabsTTSOption `json:"elevenlabs"`
	Google     *GoogleTTSOption     `json:"google"`
}

type TaggedItem struct {
	Id   string   `json:"id"`   // used by TTSAblt client
	Name string   `json:"name"` // to display on UI
	Tags []string `json:"tags"` // gender, accent, age, etc
}

type WhisperOption struct {
	Model string `json:"model"`
}

type WhisperSTTAb struct {
	Available bool     `json:"available"`
	Models    []string `json:"models"`
}