voiceprint

type AudioInfo struct {
	Format        string        `json:"format"`
	SampleRate    int           `json:"sample_rate"`
	Channels      int           `json:"channels"`
	BitsPerSample int           `json:"bits_per_sample"`
	Duration      time.Duration `json:"duration"`
	FileSize      int64         `json:"file_size"`
}

type AudioResource struct {
	Encoding   string `json:"encoding"`    // lame
	SampleRate int    `json:"sample_rate"` // 16000
	Channels   int    `json:"channels"`    // 1
	BitDepth   int    `json:"bit_depth"`   // 16
	Status     int    `json:"status"`      // 3
	Audio      string `json:"audio"`       // base64编码的音频数据
}

type BatchIdentifyRequest struct {
	Requests []IdentifyRequest `json:"requests" validate:"required,min=1"`
}

type BatchIdentifyResponse struct {
	Success   int                `json:"success"`
	Failed    int                `json:"failed"`
	Total     int                `json:"total"`
	Results   []IdentifyResponse `json:"results"`
	Timestamp time.Time          `json:"timestamp"`
}

type BatchRegisterRequest struct {
	Speakers []RegisterRequest `json:"speakers" validate:"required,min=1"`
}

type BatchRegisterResponse struct {
	Success   int                `json:"success"`
	Failed    int                `json:"failed"`
	Total     int                `json:"total"`
	Results   []RegisterResponse `json:"results"`
	Timestamp time.Time          `json:"timestamp"`
}

type Cache interface {
	Get(ctx context.Context, key string) (interface{}, bool)
	Set(ctx context.Context, key string, value interface{}, ttl time.Duration) error
	Exists(ctx context.Context, key string) bool
	Delete(ctx context.Context, key string) error
}

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

type Config struct {
	Enabled             bool          `env:"VOICEPRINT_ENABLED" mapstructure:"enabled"`
	BaseURL             string        `env:"VOICEPRINT_BASE_URL" mapstructure:"base_url"`
	APIKey              string        `env:"VOICEPRINT_API_KEY" mapstructure:"api_key"`
	Timeout             time.Duration `env:"VOICEPRINT_TIMEOUT" mapstructure:"timeout"`
	ConnectTimeout      time.Duration `env:"VOICEPRINT_CONNECT_TIMEOUT" mapstructure:"connect_timeout"`
	MaxRetries          int           `env:"VOICEPRINT_MAX_RETRIES" mapstructure:"max_retries"`
	RetryInterval       time.Duration `env:"VOICEPRINT_RETRY_INTERVAL" mapstructure:"retry_interval"`
	SimilarityThreshold float64       `env:"VOICEPRINT_SIMILARITY_THRESHOLD" mapstructure:"similarity_threshold"`
	MaxCandidates       int           `env:"VOICEPRINT_MAX_CANDIDATES" mapstructure:"max_candidates"`
	CacheEnabled        bool          `env:"VOICEPRINT_CACHE_ENABLED" mapstructure:"cache_enabled"`
	CacheTTL            time.Duration `env:"VOICEPRINT_CACHE_TTL" mapstructure:"cache_ttl"`
	LogEnabled          bool          `env:"VOICEPRINT_LOG_ENABLED" mapstructure:"log_enabled"`
	LogLevel            string        `env:"VOICEPRINT_LOG_LEVEL" mapstructure:"log_level"`
}

type CreateFeatureResult struct {
	FeatureID string `json:"featureId"`
}

type CreateGroupResult struct {
	GroupID   string `json:"groupId"`
	GroupName string `json:"groupName"`
	GroupInfo string `json:"groupInfo"`
}

type DeleteFeatureResult struct {
	Message string `json:"msg"`
}

type DeleteGroupResult struct {
	Message string `json:"msg"`
}

type DeleteRequest struct {
	SpeakerID string `json:"speaker_id" validate:"required"`
	AgentID   string `json:"agent_id,omitempty"`
}

type DeleteResponse struct {
	Success   bool      `json:"success"`
	Message   string    `json:"msg"`
	SpeakerID string    `json:"speaker_id,omitempty"`
	Timestamp time.Time `json:"timestamp,omitempty"`
}

type Factory struct{}

type FeatureItem struct {
	FeatureID   string `json:"featureId"`
	FeatureInfo string `json:"featureInfo"`
}

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

type HealthResponse struct {
	Status           string    `json:"status"`
	TotalVoiceprints int       `json:"total_voiceprints"`
	Timestamp        time.Time `json:"timestamp,omitempty"`
}

type IdentifyRequest struct {
	CandidateIDs []string `json:"candidate_ids" validate:"required,min=1"`
	AgentID      string   `json:"agent_id" validate:"required"`
	AudioData    []byte   `json:"-"`
	AudioFormat  string   `json:"audio_format,omitempty"`
	Threshold    float64  `json:"threshold,omitempty"`
	MaxResults   int      `json:"max_results,omitempty"`
}

type IdentifyResponse struct {
	SpeakerID  string    `json:"speaker_id"`
	Score      float64   `json:"score"`
	Timestamp  time.Time `json:"timestamp,omitempty"`
	Confidence string    `json:"confidence,omitempty"`
}

type IdentifyResult struct {
	SpeakerID   string        `json:"speaker_id"`
	Score       float64       `json:"score"`
	Confidence  string        `json:"confidence"`
	Threshold   float64       `json:"threshold"`
	IsMatch     bool          `json:"is_match"`
	ProcessTime time.Duration `json:"process_time"`
	Timestamp   time.Time     `json:"timestamp"`
}

type Provider string

type ProviderConfig struct {
	Provider Provider               `json:"provider"`
	Options  map[string]interface{} `json:"options"`
}

type QueryFeatureListResult struct {
	Features []FeatureItem `json:"features"`
}

type RegisterRequest struct {
	SpeakerID   string                 `json:"speaker_id" validate:"required"`
	AgentID     string                 `json:"agent_id" validate:"required"`
	AudioData   []byte                 `json:"-"`
	AudioFormat string                 `json:"audio_format,omitempty"`
	Metadata    map[string]interface{} `json:"metadata,omitempty"`
}

type RegisterResponse struct {
	Success   bool      `json:"success"`
	Message   string    `json:"msg"`
	SpeakerID string    `json:"speaker_id,omitempty"`
	Timestamp time.Time `json:"timestamp,omitempty"`
}

type ResponseFormat struct {
	Encoding string `json:"encoding"` // utf8
	Compress string `json:"compress"` // raw
	Format   string `json:"format"`   // json
}

type SearchFeaResult struct {
	ScoreList []SearchScoreFeaResult `json:"scoreList"`
}

type SearchScoreFeaResult struct {
	Score       float64 `json:"score"`
	FeatureID   string  `json:"featureId"`
	FeatureInfo string  `json:"featureInfo"`
}

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

type SpeakerInfo struct {
	SpeakerID    string                 `json:"speaker_id"`
	RegisterTime time.Time              `json:"register_time"`
	UpdateTime   time.Time              `json:"update_time"`
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
}

type Statistics struct {
	TotalSpeakers        int       `json:"total_speakers"`
	TotalIdentifications int       `json:"total_identifications"`
	SuccessRate          float64   `json:"success_rate"`
	AverageScore         float64   `json:"average_score"`
	LastActivity         time.Time `json:"last_activity"`
}

type TextResponse struct {
	Status string `json:"status,omitempty"`
	Text   string `json:"text"`
}

type UpdateFeatureResult struct {
	Message string `json:"msg"`
}

type VoiceprintError struct {
	Code    string `json:"code"`
	Message string `json:"message"`
	Details string `json:"details,omitempty"`
}

type VoiceprintProvider interface {
	// 特征库管理
	CreateGroup(ctx context.Context, groupID, groupName, groupInfo string) (*CreateGroupResult, error)
	DeleteGroup(ctx context.Context, groupID string) error

	// 特征管理
	CreateFeature(ctx context.Context, groupID, featureID, featureInfo string, audioData []byte) (*CreateFeatureResult, error)
	UpdateFeature(ctx context.Context, groupID, featureID, featureInfo string, audioData []byte, cover bool) (*UpdateFeatureResult, error)
	DeleteFeature(ctx context.Context, groupID, featureID string) error
	QueryFeatureList(ctx context.Context, groupID string) (*QueryFeatureListResult, error)

	// 特征比对
	SearchScoreFea(ctx context.Context, groupID, dstFeatureID string, audioData []byte) (*SearchScoreFeaResult, error)
	SearchFea(ctx context.Context, groupID string, topK int, audioData []byte) (*SearchFeaResult, error)

	// 健康检查
	HealthCheck(ctx context.Context) error

	// 关闭
	Close() error
}

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

type VolcengineConfig struct {
	AccessKey string        `json:"access_key"`
	SecretKey string        `json:"secret_key"`
	Region    string        `json:"region"`
	BaseURL   string        `json:"base_url"`
	Timeout   time.Duration `json:"timeout"`
}

type VolcengineDeleteRequest struct {
	UUID string `json:"UUID"` // 要删除的声纹唯一标识符
}

type VolcengineDeleteResponse struct {
	ResponseMetadata VolcengineResponseMetadata `json:"ResponseMetadata"`
	Result           interface{}                `json:"Result"`
}

type VolcengineError struct {
	ResponseMetadata VolcengineResponseMetadata `json:"ResponseMetadata"`
	Error            VolcengineErrorDetail      `json:"Error"`
}

type VolcengineErrorDetail struct {
	Code    string `json:"Code"`    // 错误码
	Message string `json:"Message"` // 错误信息
}

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

type VolcengineQueryRequest struct {
	UUIDs    []string `json:"UUIDs,omitempty"`    // 声纹ID列表，可选
	Limit    int      `json:"Limit,omitempty"`    // 每页返回的最大结果数量，默认50
	Iterator string   `json:"Iterator,omitempty"` // 分页标识
}

type VolcengineQueryResponse struct {
	ResponseMetadata VolcengineResponseMetadata `json:"ResponseMetadata"`
	Result           VolcengineQueryResult      `json:"Result"`
}

type VolcengineQueryResult struct {
	VoicePrints  []VolcengineVoicePrint `json:"VoicePrints,omitempty"`  // 声纹对象数组
	NextIterator string                 `json:"NextIterator,omitempty"` // 下一页的分页标识
}

type VolcengineRegisterRequest struct {
	Audio     string `json:"Audio"`     // Base64编码的WAV音频数据
	MetaInfo  string `json:"MetaInfo"`  // 声纹元信息，可选
	AudioName string `json:"AudioName"` // 音频样本名称
}

type VolcengineRegisterResponse struct {
	ResponseMetadata VolcengineResponseMetadata `json:"ResponseMetadata"`
	Result           VolcengineRegisterResult   `json:"Result"`
}

type VolcengineRegisterResult struct {
	UUID string `json:"UUID"` // 注册成功的声纹唯一标识符
}

type VolcengineResponseMetadata struct {
	Action    string `json:"Action"`    // 接口名称
	Region    string `json:"Region"`    // 区域
	Service   string `json:"Service"`   // 服务名称
	Version   string `json:"Version"`   // 接口版本
	RequestID string `json:"RequestId"` // 请求ID
}

type VolcengineUpdateRequest struct {
	UUID      string `json:"UUID"`                // 要更新的声纹唯一标识符
	Audio     string `json:"Audio,omitempty"`     // Base64编码的WAV音频数据，可选
	MetaInfo  string `json:"MetaInfo,omitempty"`  // 声纹元信息，可选
	AudioName string `json:"AudioName,omitempty"` // 音频样本名称，可选
}

type VolcengineUpdateResponse struct {
	ResponseMetadata VolcengineResponseMetadata `json:"ResponseMetadata"`
	Result           interface{}                `json:"Result"`
}

type VolcengineVoicePrint struct {
	UUID       string `json:"UUID,omitempty"`       // 声纹唯一标识符
	MetaInfo   string `json:"MetaInfo,omitempty"`   // 声纹元信息
	AccountID  string `json:"AccountID,omitempty"`  // 火山引擎账户ID
	AudioName  string `json:"AudioName,omitempty"`  // 音频样本名称
	CreatedAt  int64  `json:"CreatedAt,omitempty"`  // 创建时间，Unix时间戳（毫秒）
	UpdatedAt  int64  `json:"UpdatedAt,omitempty"`  // 最后更新时间，Unix时间戳（毫秒）
	VoicePrint string `json:"VoicePrint,omitempty"` // 声纹特征数据
}

type VolcengineVoicePrintInfo struct {
	UUID       string    // 声纹唯一标识符
	MetaInfo   string    // 声纹元信息
	AudioName  string    // 音频样本名称
	CreatedAt  time.Time // 创建时间
	UpdatedAt  time.Time // 最后更新时间
	VoicePrint string    // 声纹特征数据
}

type XunfeiAuth struct {
	APIKey    string
	APISecret string
	Host      string
}

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

type XunfeiConfig struct {
	APIKey    string        `json:"api_key"`
	APISecret string        `json:"api_secret"`
	AppID     string        `json:"app_id"`
	BaseURL   string        `json:"base_url"`
	Timeout   time.Duration `json:"timeout"`
}

type XunfeiHeader struct {
	AppID  string `json:"app_id"`
	Status int    `json:"status"` // 3 = 一次传完
}

type XunfeiParameter struct {
	S782b4996 XunfeiServiceParam `json:"s782b4996"`
}

type XunfeiPayload struct {
	Resource *AudioResource `json:"resource,omitempty"`
}

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

type XunfeiRequest struct {
	Header    XunfeiHeader    `json:"header"`
	Parameter XunfeiParameter `json:"parameter"`
	Payload   *XunfeiPayload  `json:"payload,omitempty"`
}

type XunfeiResponse struct {
	Header  XunfeiResponseHeader  `json:"header"`
	Payload XunfeiResponsePayload `json:"payload"`
}

type XunfeiResponseHeader struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	SID     string `json:"sid"`
}

type XunfeiResponsePayload struct {
	CreateGroupRes      *TextResponse `json:"createGroupRes,omitempty"`
	CreateFeatureRes    *TextResponse `json:"createFeatureRes,omitempty"`
	UpdateFeatureRes    *TextResponse `json:"updateFeatureRes,omitempty"`
	QueryFeatureListRes *TextResponse `json:"queryFeatureListRes,omitempty"`
	SearchScoreFeaRes   *TextResponse `json:"searchScoreFeaRes,omitempty"`
	SearchFeaRes        *TextResponse `json:"searchFeaRes,omitempty"`
	DeleteFeatureRes    *TextResponse `json:"deleteFeatureRes,omitempty"`
	DeleteGroupRes      *TextResponse `json:"deleteGroupRes,omitempty"`
}

type XunfeiServiceParam struct {
	Func string `json:"func"`

	// 创建特征库参数
	GroupID   string `json:"groupId,omitempty"`
	GroupName string `json:"groupName,omitempty"`
	GroupInfo string `json:"groupInfo,omitempty"`

	// 特征相关参数
	FeatureID    string `json:"featureId,omitempty"`
	FeatureInfo  string `json:"featureInfo,omitempty"`
	DstFeatureID string `json:"dstFeatureId,omitempty"`

	// 查询参数
	TopK int `json:"topK,omitempty"`

	// 更新参数
	Cover bool `json:"cover,omitempty"`

	// 响应格式配置
	CreateGroupRes      *ResponseFormat `json:"createGroupRes,omitempty"`
	CreateFeatureRes    *ResponseFormat `json:"createFeatureRes,omitempty"`
	UpdateFeatureRes    *ResponseFormat `json:"updateFeatureRes,omitempty"`
	QueryFeatureListRes *ResponseFormat `json:"queryFeatureListRes,omitempty"`
	SearchScoreFeaRes   *ResponseFormat `json:"searchScoreFeaRes,omitempty"`
	SearchFeaRes        *ResponseFormat `json:"searchFeaRes,omitempty"`
	DeleteFeatureRes    *ResponseFormat `json:"deleteFeatureRes,omitempty"`
	DeleteGroupRes      *ResponseFormat `json:"deleteGroupRes,omitempty"`
}