doubaospeech

type ASRChunk = ASRV2Result

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

type ASRV2Config struct {
	Format     AudioFormat `json:"format" yaml:"format"`
	SampleRate SampleRate  `json:"sample_rate" yaml:"sample_rate"`
	Channel    int         `json:"channel,omitempty" yaml:"channel,omitempty"`
	Channels   int         `json:"channels,omitempty" yaml:"channels,omitempty"` // Backward-compatible alias field.
	Bits       int         `json:"bits,omitempty" yaml:"bits,omitempty"`
	Language   Language    `json:"language,omitempty" yaml:"language,omitempty"`

	EnableITN         bool     `json:"enable_itn,omitempty" yaml:"enable_itn,omitempty"`
	EnablePunc        bool     `json:"enable_punc,omitempty" yaml:"enable_punc,omitempty"`
	EnableDiarization bool     `json:"enable_diarization,omitempty" yaml:"enable_diarization,omitempty"`
	SpeakerNum        int      `json:"speaker_num,omitempty" yaml:"speaker_num,omitempty"`
	Hotwords          []string `json:"hotwords,omitempty" yaml:"hotwords,omitempty"`
	ResultType        string   `json:"result_type,omitempty" yaml:"result_type,omitempty"` // single/full

	ResourceID string `json:"resource_id,omitempty" yaml:"resource_id,omitempty"`
}

type ASRV2Result struct {
	Text       string           `json:"text"`
	Utterances []ASRV2Utterance `json:"utterances,omitempty"`
	IsFinal    bool             `json:"is_final"`
	Duration   int              `json:"duration,omitempty"`
	ReqID      string           `json:"reqid,omitempty"`
	TraceID    string           `json:"trace_id,omitempty"`
	LogID      string           `json:"log_id,omitempty"`
	ConnectID  string           `json:"connect_id,omitempty"`
}

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

type ASRV2Utterance struct {
	Text       string      `json:"text"`
	StartTime  int         `json:"start_time"`
	EndTime    int         `json:"end_time"`
	Definite   bool        `json:"definite"`
	SpeakerID  string      `json:"speaker_id,omitempty"`
	Words      []ASRV2Word `json:"words,omitempty"`
	Confidence float64     `json:"confidence,omitempty"`
}

type ASRV2Word struct {
	Text      string  `json:"text"`
	StartTime int     `json:"start_time"`
	EndTime   int     `json:"end_time"`
	Conf      float64 `json:"conf,omitempty"`
}

type ASTAudioConfig struct {
	Format  AudioFormat `json:"format" yaml:"format"`
	Codec   string      `json:"codec,omitempty" yaml:"codec,omitempty"`
	Rate    SampleRate  `json:"rate" yaml:"rate"`
	Bits    int         `json:"bits" yaml:"bits"`
	Channel int         `json:"channel" yaml:"channel"`
}

type ASTTargetAudioConfig struct {
	Format  AudioFormat `json:"format,omitempty" yaml:"format,omitempty"`
	Rate    SampleRate  `json:"rate,omitempty" yaml:"rate,omitempty"`
	Bits    int         `json:"bits,omitempty" yaml:"bits,omitempty"`
	Channel int         `json:"channel,omitempty" yaml:"channel,omitempty"`
}

type ASTTranslateBillingItem struct {
	Unit     string  `json:"unit"`
	Quantity float32 `json:"quantity"`
}

type ASTTranslateConfig struct {
	SessionID  string           `json:"session_id,omitempty" yaml:"session_id,omitempty"`
	ResourceID string           `json:"resource_id,omitempty" yaml:"resource_id,omitempty"`
	Mode       ASTTranslateMode `json:"mode" yaml:"mode"`

	SourceLanguage string `json:"source_language" yaml:"source_language"`
	TargetLanguage string `json:"target_language" yaml:"target_language"`

	SourceAudio ASTAudioConfig       `json:"source_audio" yaml:"source_audio"`
	TargetAudio ASTTargetAudioConfig `json:"target_audio,omitempty" yaml:"target_audio,omitempty"`

	SpeakerID                  string `json:"speaker_id,omitempty" yaml:"speaker_id,omitempty"`
	IsCustomSpeaker            bool   `json:"is_custom_speaker,omitempty" yaml:"is_custom_speaker,omitempty"`
	TTSResourceID              string `json:"tts_resource_id,omitempty" yaml:"tts_resource_id,omitempty"`
	SpeechRate                 int    `json:"speech_rate,omitempty" yaml:"speech_rate,omitempty"`
	EnableSourceLanguageDetect bool   `json:"enable_source_language_detect,omitempty" yaml:"enable_source_language_detect,omitempty"`
	Denoise                    *bool  `json:"denoise,omitempty" yaml:"denoise,omitempty"`

	Corpus *ASTTranslateCorpus `json:"corpus,omitempty" yaml:"corpus,omitempty"`
	User   ASTUser             `json:"user,omitempty" yaml:"user,omitempty"`

	EventBuffer         int           `json:"-" yaml:"-"`
	BackpressureTimeout time.Duration `json:"-" yaml:"-"`
}

type ASTTranslateCorpus struct {
	HotWords              []string          `json:"hot_words_list,omitempty" yaml:"hot_words_list,omitempty"`
	BoostingTableID       string            `json:"boosting_table_id,omitempty" yaml:"boosting_table_id,omitempty"`
	BoostingTableName     string            `json:"boosting_table_name,omitempty" yaml:"boosting_table_name,omitempty"`
	CorrectWords          map[string]string `json:"correct_words,omitempty" yaml:"correct_words,omitempty"`
	RegexCorrectTableID   string            `json:"regex_correct_table_id,omitempty" yaml:"regex_correct_table_id,omitempty"`
	RegexCorrectTableName string            `json:"regex_correct_table_name,omitempty" yaml:"regex_correct_table_name,omitempty"`
	Glossary              map[string]string `json:"glossary_list,omitempty" yaml:"glossary_list,omitempty"`
	GlossaryTableID       string            `json:"glossary_table_id,omitempty" yaml:"glossary_table_id,omitempty"`
	GlossaryTableName     string            `json:"glossary_table_name,omitempty" yaml:"glossary_table_name,omitempty"`
}

type ASTTranslateEvent struct {
	Type      ASTTranslateEventType `json:"type"`
	SessionID string                `json:"session_id,omitempty"`

	Text             string `json:"text,omitempty"`
	Audio            []byte `json:"-"`
	StartTimeMS      int    `json:"start_time_ms,omitempty"`
	EndTimeMS        int    `json:"end_time_ms,omitempty"`
	SpeakerChanged   bool   `json:"speaker_changed,omitempty"`
	DetectedLanguage string `json:"detected_language,omitempty"`
	MutedDurationMS  int    `json:"muted_duration_ms,omitempty"`

	Usage   *ASTTranslateUsage `json:"usage,omitempty"`
	Error   *Error             `json:"error,omitempty"`
	IsFinal bool               `json:"is_final,omitempty"`

	Payload []byte `json:"payload,omitempty"`
	ReqID   string `json:"reqid,omitempty"`
	TraceID string `json:"trace_id,omitempty"`
	LogID   string `json:"log_id,omitempty"`
}

type ASTTranslateEventType int32

type ASTTranslateMode string

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

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

type ASTTranslateUpdate struct {
	Corpus *ASTTranslateCorpus `json:"corpus,omitempty" yaml:"corpus,omitempty"`
}

type ASTTranslateUsage struct {
	Items      []ASTTranslateBillingItem `json:"items,omitempty"`
	DurationMS int64                     `json:"duration_ms,omitempty"`
	WordCount  int64                     `json:"word_count,omitempty"`
}

type ASTUser struct {
	UID        string `json:"uid,omitempty" yaml:"uid,omitempty"`
	DID        string `json:"did,omitempty" yaml:"did,omitempty"`
	Platform   string `json:"platform,omitempty" yaml:"platform,omitempty"`
	SDKVersion string `json:"sdk_version,omitempty" yaml:"sdk_version,omitempty"`
	AppVersion string `json:"app_version,omitempty" yaml:"app_version,omitempty"`
}

type AudioFormat string

type Client struct {
	// ASR V2 streaming recognition.
	ASR   *ASRServiceV2
	ASRV2 *ASRServiceV2

	// Voice cloning.
	VoiceClone *VoiceCloneService

	// Realtime dialogue.
	Realtime *RealtimeService

	// AST realtime translation.
	ASTTranslate *ASTTranslateService
	AST          *ASTTranslateService

	// TTS V2 WebSocket synthesis.
	TTS   *TTSServiceV2
	TTSV2 *TTSServiceV2
	// contains filtered or unexported fields
}

type Error struct {
	Code       int    `json:"code"`
	Message    string `json:"message"`
	TraceID    string `json:"trace_id,omitempty"`
	LogID      string `json:"log_id,omitempty"`
	ConnectID  string `json:"connect_id,omitempty"`
	HTTPStatus int    `json:"-"`
	ReqID      string `json:"reqid,omitempty"`
}

type Language string

type MultipartFile struct {
	FieldName   string
	FileName    string
	ContentType string
	Data        []byte
}

type Option func(*clientConfig)

type RealtimeASRConfig struct {
	Language Language       `json:"language,omitempty" yaml:"language,omitempty"`
	Extra    map[string]any `json:"extra,omitempty" yaml:"extra,omitempty"`
}

type RealtimeASRResult struct {
	Text      string `json:"text"`
	IsInterim bool   `json:"is_interim"`
}

type RealtimeAudioConfig struct {
	Channel      int         `json:"channel" yaml:"channel"`
	Format       AudioFormat `json:"format" yaml:"format"`
	SampleRate   SampleRate  `json:"sample_rate" yaml:"sample_rate"`
	Bits         int         `json:"bits,omitempty" yaml:"bits,omitempty"`
	SpeechRate   int         `json:"speech_rate,omitempty" yaml:"speech_rate,omitempty"`
	LoudnessRate int         `json:"loudness_rate,omitempty" yaml:"loudness_rate,omitempty"`
}

type RealtimeConfig struct {
	ASR    RealtimeASRConfig    `json:"asr" yaml:"asr"`
	TTS    RealtimeTTSConfig    `json:"tts" yaml:"tts"`
	Dialog RealtimeDialogConfig `json:"dialog" yaml:"dialog"`

	Prompt  RealtimePromptConfig          `json:"prompt" yaml:"prompt,omitempty"`
	Props   RealtimeGenerationProps       `json:"props" yaml:"props,omitempty"`
	History []RealtimeConversationMessage `json:"history,omitempty" yaml:"history,omitempty"`

	InputMode RealtimeInputMode    `json:"input_mode,omitempty" yaml:"input_mode,omitempty"`
	Model     RealtimeModelVersion `json:"model,omitempty" yaml:"model,omitempty"`

	ResourceID string `json:"resource_id,omitempty" yaml:"resource_id,omitempty"`

	// Local runtime controls (not sent to server).
	EventBuffer         int           `json:"-" yaml:"-"`
	BackpressureTimeout time.Duration `json:"-" yaml:"-"`
}

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

type RealtimeConversationMessage struct {
	Role    string `json:"role" yaml:"role"`
	Content string `json:"content" yaml:"content"`
}

type RealtimeDialogConfig struct {
	BotName           string         `json:"bot_name,omitempty" yaml:"bot_name,omitempty"`
	SystemRole        string         `json:"system_role,omitempty" yaml:"system_role,omitempty"`
	SpeakingStyle     string         `json:"speaking_style,omitempty" yaml:"speaking_style,omitempty"`
	CharacterManifest string         `json:"character_manifest,omitempty" yaml:"character_manifest,omitempty"`
	Extra             map[string]any `json:"extra,omitempty" yaml:"extra,omitempty"`
}

type RealtimeEvent struct {
	Type      RealtimeEventType `json:"type"`
	SessionID string            `json:"session_id,omitempty"`
	ConnectID string            `json:"connect_id,omitempty"`
	Sequence  int32             `json:"sequence,omitempty"`

	Text    string `json:"text,omitempty"`
	Audio   []byte `json:"audio,omitempty"`
	Payload []byte `json:"payload,omitempty"`

	QuestionID string              `json:"question_id,omitempty"`
	ReplyID    string              `json:"reply_id,omitempty"`
	TTSType    string              `json:"tts_type,omitempty"`
	StatusCode string              `json:"status_code,omitempty"`
	Results    []RealtimeASRResult `json:"results,omitempty"`
	Usage      *RealtimeUsage      `json:"usage,omitempty"`

	Error   *Error `json:"error,omitempty"`
	IsFinal bool   `json:"is_final,omitempty"`

	ReqID   string `json:"reqid,omitempty"`
	TraceID string `json:"trace_id,omitempty"`
	LogID   string `json:"log_id,omitempty"`
}

type RealtimeEventType int32

type RealtimeGenerationProps struct {
	Temperature      float64        `json:"temperature,omitempty" yaml:"temperature,omitempty"`
	TopP             float64        `json:"top_p,omitempty" yaml:"top_p,omitempty"`
	MaxTokens        int            `json:"max_tokens,omitempty" yaml:"max_tokens,omitempty"`
	PresencePenalty  float64        `json:"presence_penalty,omitempty" yaml:"presence_penalty,omitempty"`
	FrequencyPenalty float64        `json:"frequency_penalty,omitempty" yaml:"frequency_penalty,omitempty"`
	Extra            map[string]any `json:"extra,omitempty" yaml:"extra,omitempty"`
}

type RealtimeInputMode string

type RealtimeModelVersion string

type RealtimePromptConfig struct {
	System    string            `json:"system,omitempty" yaml:"system,omitempty"`
	Variables map[string]string `json:"variables,omitempty" yaml:"variables,omitempty"`
}

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

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

type RealtimeTTSConfig struct {
	Speaker     string              `json:"speaker" yaml:"speaker"`
	AudioConfig RealtimeAudioConfig `json:"audio_config" yaml:"audio_config"`
	Extra       map[string]any      `json:"extra,omitempty" yaml:"extra,omitempty"`
}

type RealtimeUsage struct {
	InputTextTokens   int `json:"input_text_tokens,omitempty"`
	InputAudioTokens  int `json:"input_audio_tokens,omitempty"`
	CachedTextTokens  int `json:"cached_text_tokens,omitempty"`
	CachedAudioTokens int `json:"cached_audio_tokens,omitempty"`
	OutputTextTokens  int `json:"output_text_tokens,omitempty"`
	OutputAudioTokens int `json:"output_audio_tokens,omitempty"`
}

type SampleRate int

type StreamASRConfig = ASRV2Config

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

type TTSV2Chunk struct {
	Audio   []byte `json:"-"`
	IsLast  bool   `json:"is_last"`
	ReqID   string `json:"reqid,omitempty"`
	TraceID string `json:"trace_id,omitempty"`
	LogID   string `json:"log_id,omitempty"`
	Code    int    `json:"code"`
	Message string `json:"message,omitempty"`
}

type TTSV2MixSpeaker struct {
	Speakers []TTSV2MixSpeakerSource `json:"speakers,omitempty" yaml:"speakers,omitempty"`
}

type TTSV2MixSpeakerSource struct {
	SourceSpeaker string  `json:"source_speaker" yaml:"source_speaker"`
	MixFactor     float64 `json:"mix_factor" yaml:"mix_factor"`
}

type TTSV2Request struct {
	Text    string `json:"text" yaml:"text"`
	Speaker string `json:"speaker" yaml:"speaker"`

	Format     AudioFormat `json:"format,omitempty" yaml:"format,omitempty"`
	SampleRate SampleRate  `json:"sample_rate,omitempty" yaml:"sample_rate,omitempty"`
	BitRate    int         `json:"bit_rate,omitempty" yaml:"bit_rate,omitempty"`

	SpeechRate int `json:"speech_rate,omitempty" yaml:"speech_rate,omitempty"`
	PitchRate  int `json:"pitch_rate,omitempty" yaml:"pitch_rate,omitempty"`
	VolumeRate int `json:"volume_rate,omitempty" yaml:"volume_rate,omitempty"`

	Emotion  string `json:"emotion,omitempty" yaml:"emotion,omitempty"`
	Language string `json:"language,omitempty" yaml:"language,omitempty"`

	ResourceID string           `json:"resource_id,omitempty" yaml:"resource_id,omitempty"`
	MixSpeaker *TTSV2MixSpeaker `json:"mix_speaker,omitempty" yaml:"mix_speaker,omitempty"`
}

type TTSV2WSChunk struct {
	Audio     []byte `json:"-"`
	IsFinal   bool   `json:"is_final"`
	Event     int32  `json:"event"`
	ReqID     string `json:"reqid,omitempty"`
	TraceID   string `json:"trace_id,omitempty"`
	LogID     string `json:"log_id,omitempty"`
	ConnectID string `json:"connect_id,omitempty"`
}

type TTSV2WSConfig struct {
	Speaker    string      `json:"speaker" yaml:"speaker"`
	Format     AudioFormat `json:"format,omitempty" yaml:"format,omitempty"`
	SampleRate SampleRate  `json:"sample_rate,omitempty" yaml:"sample_rate,omitempty"`

	// ResourceID defaults to seed-tts-2.0 when empty.
	ResourceID string `json:"resource_id,omitempty" yaml:"resource_id,omitempty"`
}

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

type Task[T any] struct {
	ID string
	// contains filtered or unexported fields
}

type TaskFailureMapper[T any] func(status TaskStatus, result *T) error

type TaskPoller[T any] func(ctx context.Context, taskID string) (status TaskStatus, result *T, err error)

type TaskStatus string

type TaskStatusMapper func(status TaskStatus) TaskStatus

type Utterance = ASRV2Utterance

type VoiceCloneRequest struct {
	// VoiceID is a custom voice identifier.
	VoiceID string `json:"voice_id,omitempty" yaml:"voice_id,omitempty"`
	// SpeakerID is an alias of VoiceID for compatibility with official docs.
	SpeakerID string `json:"speaker_id,omitempty" yaml:"speaker_id,omitempty"`

	Audio []byte `json:"-" yaml:"-"`

	AudioFileName    string `json:"audio_file_name,omitempty" yaml:"audio_file_name,omitempty"`
	AudioContentType string `json:"audio_content_type,omitempty" yaml:"audio_content_type,omitempty"`
	AudioFormat      string `json:"audio_format,omitempty" yaml:"audio_format,omitempty"`

	Text      string `json:"text,omitempty" yaml:"text,omitempty"`
	Language  int    `json:"language,omitempty" yaml:"language,omitempty"`
	ModelType int    `json:"model_type,omitempty" yaml:"model_type,omitempty"`
	Source    int    `json:"source,omitempty" yaml:"source,omitempty"`

	ResourceID   string        `json:"resource_id,omitempty" yaml:"resource_id,omitempty"`
	PollInterval time.Duration `json:"-" yaml:"-"`
}

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

type VoiceCloneStatus struct {
	TaskID    string `json:"task_id,omitempty"`
	SpeakerID string `json:"speaker_id,omitempty"`
	VoiceID   string `json:"voice_id,omitempty"`

	Status        TaskStatus `json:"status"`
	RawStatus     string     `json:"raw_status,omitempty"`
	RawStatusCode int        `json:"raw_status_code,omitempty"`

	StatusCode    int    `json:"status_code,omitempty"`
	StatusMessage string `json:"status_message,omitempty"`

	Version    string `json:"version,omitempty"`
	DemoAudio  string `json:"demo_audio,omitempty"`
	CreateTime int64  `json:"create_time,omitempty"`

	ReqID   string `json:"reqid,omitempty"`
	TraceID string `json:"trace_id,omitempty"`
	LogID   string `json:"log_id,omitempty"`
}

type Word = ASRV2Word