translator

type BaiduTranslateResponse struct {
	From        string `json:"from"`
	To          string `json:"to"`
	TransResult []struct {
		Src string `json:"src"`
		Dst string `json:"dst"`
	} `json:"trans_result"`
	ErrorCode string `json:"error_code,omitempty"`
	ErrorMsg  string `json:"error_msg,omitempty"`
}

type BaiduTranslator struct {
	AppID     string
	AppSecret string
	Client    *resty.Client
}

type DeepLTranslateRequest struct {
	Text               []string `json:"text"`
	TargetLang         string   `json:"target_lang"`
	SourceLang         string   `json:"source_lang,omitempty"`
	SplitSentences     string   `json:"split_sentences,omitempty"`
	PreserveFormatting bool     `json:"preserve_formatting,omitempty"`
	Formality          string   `json:"formality,omitempty"`
}

type DeepLTranslateResponse struct {
	Translations []struct {
		DetectedSourceLanguage string `json:"detected_source_language"`
		Text                   string `json:"text"`
	} `json:"translations"`
}

type DeepLTranslator struct {
	APIKey string
	IsFree bool
	Client *resty.Client
}

type GlossaryConfig struct {
	Glossary string `json:"glossary"`
}

type GoogleTranslateRequest struct {
	Contents       []string        `json:"contents"`
	SourceLanguage string          `json:"sourceLanguageCode,omitempty"`
	TargetLanguage string          `json:"targetLanguageCode"`
	MimeType       string          `json:"mimeType,omitempty"`
	Model          string          `json:"model,omitempty"`
	GlossaryConfig *GlossaryConfig `json:"glossaryConfig,omitempty"`
}

type GoogleTranslateResponse struct {
	Translations []struct {
		TranslatedText         string `json:"translatedText"`
		DetectedSourceLanguage string `json:"detectedSourceLanguage,omitempty"`
	} `json:"translations"`
}

type GoogleTranslator struct {
	APIKey string
	Client *resty.Client
}

type OpenAIChatRequest struct {
	Model    string `json:"model"`
	Messages []struct {
		Role    string `json:"role"`
		Content string `json:"content"`
	} `json:"messages"`
	Temperature      float64              `json:"temperature,omitempty"`
	MaxTokens        int                  `json:"max_tokens,omitempty"`
	TopP             float64              `json:"top_p,omitempty"`
	FrequencyPenalty float64              `json:"frequency_penalty,omitempty"`
	PresencePenalty  float64              `json:"presence_penalty,omitempty"`
	Stream           bool                 `json:"stream"`
	StreamOptions    *OpenAIStreamOptions `json:"stream_options,omitempty"`
}

type OpenAIChatResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	Model   string `json:"model"`
	Choices []struct {
		Message struct {
			Role    string          `json:"role"`
			Content json.RawMessage `json:"content"`
		} `json:"message"`
		FinishReason string `json:"finish_reason"`
		Index        int    `json:"index"`
	} `json:"choices"`
	Usage struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
	Error *struct {
		Message string `json:"message"`
		Type    string `json:"type"`
		Param   string `json:"param,omitempty"`
		Code    string `json:"code,omitempty"`
	} `json:"error,omitempty"`
}

type OpenAIStreamOptions struct {
	IncludeUsage bool `json:"include_usage"`
}

type OpenAITranslator struct {
	APIKey      string
	APIBaseURL  string
	Model       string
	Client      *resty.Client
	Temperature float64
	MaxTokens   int
}

type ProgressReporter func(done, total int, last model.TranslationResponse)

type TranslationService struct {
	Provider    model.TranslationProvider
	Concurrency int
	Timeout     time.Duration
}