feat: implement core architecture (v0.0.2)
- Implement Config class with YAML loading and environment variable support - Implement Provider interface and factory pattern - Implement SiliconFlow provider as example - Implement Translator core class with prompt management - Create CLI entry point - Add configuration template and unit tests - Update changelog and discussion records Version: 0.0.2
This commit is contained in:
177
internal/translator/translator.go
Normal file
177
internal/translator/translator.go
Normal file
@@ -0,0 +1,177 @@
|
||||
package translator
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/titor/fanyi/internal/config"
|
||||
"github.com/titor/fanyi/internal/provider"
|
||||
)
|
||||
|
||||
// Translator 核心翻译类
|
||||
type Translator struct {
|
||||
config *config.Config
|
||||
provider provider.Provider
|
||||
prompt *PromptManager
|
||||
}
|
||||
|
||||
// NewTranslator 创建翻译器实例
|
||||
func NewTranslator(config *config.Config, provider provider.Provider) *Translator {
|
||||
return &Translator{
|
||||
config: config,
|
||||
provider: provider,
|
||||
prompt: NewPromptManager(config.Prompts),
|
||||
}
|
||||
}
|
||||
|
||||
// Translate 执行翻译
|
||||
func (t *Translator) Translate(ctx context.Context, text string, options *TranslateOptions) (*TranslateResult, error) {
|
||||
// 设置超时
|
||||
timeoutCtx, cancel := context.WithTimeout(ctx, time.Duration(t.config.Timeout)*time.Second)
|
||||
defer cancel()
|
||||
|
||||
// 选择Prompt
|
||||
prompt := ""
|
||||
if options.PromptName != "" {
|
||||
prompt = t.prompt.GetPrompt(options.PromptName)
|
||||
}
|
||||
|
||||
// 构建请求
|
||||
req := &provider.TranslateRequest{
|
||||
Text: text,
|
||||
FromLang: options.FromLang,
|
||||
ToLang: options.ToLang,
|
||||
Prompt: prompt,
|
||||
Model: t.selectModel(options.Model),
|
||||
Options: options.ExtraOptions,
|
||||
}
|
||||
|
||||
// 调用厂商API
|
||||
resp, err := t.provider.Translate(timeoutCtx, req)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("翻译失败: %w", err)
|
||||
}
|
||||
|
||||
// 构建结果
|
||||
return &TranslateResult{
|
||||
Original: text,
|
||||
Translated: resp.Text,
|
||||
FromLang: resp.FromLang,
|
||||
ToLang: resp.ToLang,
|
||||
Model: resp.Model,
|
||||
Usage: resp.Usage,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// TranslateWithProvider 使用指定厂商执行翻译
|
||||
func (t *Translator) TranslateWithProvider(ctx context.Context, text string, providerName string, options *TranslateOptions) (*TranslateResult, error) {
|
||||
// 创建指定厂商实例
|
||||
providerConfig, err := t.config.GetProviderConfig(providerName)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取厂商配置失败: %w", err)
|
||||
}
|
||||
|
||||
// 创建厂商实例
|
||||
providerInstance, err := provider.CreateProvider(providerName, provider.ProviderConfig{
|
||||
APIHost: providerConfig.APIHost,
|
||||
APIKey: providerConfig.APIKey,
|
||||
Model: providerConfig.Model,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建厂商实例失败: %w", err)
|
||||
}
|
||||
|
||||
// 临时切换厂商
|
||||
originalProvider := t.provider
|
||||
t.provider = providerInstance
|
||||
defer func() {
|
||||
t.provider = originalProvider
|
||||
}()
|
||||
|
||||
// 执行翻译
|
||||
return t.Translate(ctx, text, options)
|
||||
}
|
||||
|
||||
// selectModel 选择模型
|
||||
func (t *Translator) selectModel(model string) string {
|
||||
if model != "" {
|
||||
return model
|
||||
}
|
||||
return t.config.DefaultModel
|
||||
}
|
||||
|
||||
// GetProvider 获取当前厂商
|
||||
func (t *Translator) GetProvider() provider.Provider {
|
||||
return t.provider
|
||||
}
|
||||
|
||||
// GetConfig 获取配置
|
||||
func (t *Translator) GetConfig() *config.Config {
|
||||
return t.config
|
||||
}
|
||||
|
||||
// GetPromptManager 获取Prompt管理器
|
||||
func (t *Translator) GetPromptManager() *PromptManager {
|
||||
return t.prompt
|
||||
}
|
||||
|
||||
// SetTimeout 设置超时时间
|
||||
func (t *Translator) SetTimeout(seconds int) {
|
||||
t.config.Timeout = seconds
|
||||
}
|
||||
|
||||
// TranslateOptions 翻译选项
|
||||
type TranslateOptions struct {
|
||||
FromLang string
|
||||
ToLang string
|
||||
PromptName string
|
||||
Model string
|
||||
Temperature float64
|
||||
ExtraOptions map[string]interface{}
|
||||
}
|
||||
|
||||
// TranslateResult 翻译结果
|
||||
type TranslateResult struct {
|
||||
Original string
|
||||
Translated string
|
||||
FromLang string
|
||||
ToLang string
|
||||
Model string
|
||||
Usage *provider.Usage
|
||||
}
|
||||
|
||||
// String 返回翻译结果的字符串表示
|
||||
func (r *TranslateResult) String() string {
|
||||
return r.Translated
|
||||
}
|
||||
|
||||
// TranslateResultWithInfo 带详细信息的翻译结果
|
||||
type TranslateResultWithInfo struct {
|
||||
Result *TranslateResult
|
||||
Duration time.Duration
|
||||
Provider string
|
||||
Timestamp time.Time
|
||||
}
|
||||
|
||||
// BatchTranslate 批量翻译结果
|
||||
type BatchTranslateRequest struct {
|
||||
Texts []string
|
||||
Options *TranslateOptions
|
||||
}
|
||||
|
||||
// BatchTranslateResult 批量翻译结果
|
||||
type BatchTranslateResult struct {
|
||||
Results []*TranslateResult
|
||||
Errors []error
|
||||
Summary BatchTranslateSummary
|
||||
}
|
||||
|
||||
// BatchTranslateSummary 批量翻译摘要
|
||||
type BatchTranslateSummary struct {
|
||||
Total int
|
||||
Success int
|
||||
Failed int
|
||||
Duration time.Duration
|
||||
AvgTokens int
|
||||
}
|
||||
Reference in New Issue
Block a user