Blame view

query/translator.py 36.9 KB
be52af70   tangwang   first commit
1
2
3
  """
  Translation service for multi-language query support.
  
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
4
5
6
  Supports multiple translation models:
  - Qwen (default): Alibaba Cloud DashScope API using qwen-mt-flash model
  - DeepL: DeepL API for high-quality translations
0064e946   tangwang   feat: 增量索引服务、租户配置...
7
  
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
8
  使用方法 (Usage):
0064e946   tangwang   feat: 增量索引服务、租户配置...
9
  
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
  ```python
  from query.translator import Translator
  
  # 使用默认的 qwen 模型(推荐)
  translator = Translator()  # 默认使用 qwen 模型
  
  # 或显式指定模型
  translator = Translator(model='qwen')  # 使用 qwen 模型
  translator = Translator(model='deepl')  # 使用 DeepL 模型
  
  # 翻译文本
  result = translator.translate(
      text="我看到这个视频后没有笑",
      target_lang="en",
      source_lang="auto"  # 自动检测源语言
  )
  ```
  
  配置说明 (Configuration):
  - Qwen 模型需要设置 DASHSCOPE_API_KEY 环境变量(在 .env 文件中)
  - DeepL 模型需要设置 DEEPL_AUTH_KEY 环境变量(在 .env 文件中)
0064e946   tangwang   feat: 增量索引服务、租户配置...
31
  
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
32
33
34
  Qwen 模型参考文档:
  - 官方文档:https://help.aliyun.com/zh/model-studio/get-api-key
  - 模型:qwen-mt-flash(快速翻译模型)
0064e946   tangwang   feat: 增量索引服务、租户配置...
35
  
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
36
37
  DeepL 官方文档:
  https://developers.deepl.com/api-reference/translate/request-translation
be52af70   tangwang   first commit
38
39
  """
  
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
40
  import os
be52af70   tangwang   first commit
41
  import requests
a5a6bab8   tangwang   多语言查询优化
42
  import re
453992a8   tangwang   需求:
43
  import redis
3ec5bfe6   tangwang   1. get_translatio...
44
  from concurrent.futures import ThreadPoolExecutor, Future
453992a8   tangwang   需求:
45
  from datetime import timedelta
3ec5bfe6   tangwang   1. get_translatio...
46
  from typing import Dict, List, Optional, Union
6e0e310c   tangwang   1. Translator 类增强
47
  import logging
d90e7428   tangwang   补充重排
48
  import time
6e0e310c   tangwang   1. Translator 类增强
49
50
  
  logger = logging.getLogger(__name__)
be52af70   tangwang   first commit
51
  
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
52
53
  from config.env_config import DEEPL_AUTH_KEY, DASHSCOPE_API_KEY, REDIS_CONFIG
  from openai import OpenAI
325eec03   tangwang   1. 日志、配置基础设施,使用优化
54
  
be52af70   tangwang   first commit
55
56
  
  class Translator:
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
57
58
59
60
61
      """
      Multi-language translator supporting Qwen and DeepL APIs.
      
      Default model is 'qwen' which uses Alibaba Cloud DashScope API.
      """
be52af70   tangwang   first commit
62
  
16c42787   tangwang   feat: implement r...
63
      DEEPL_API_URL = "https://api.deepl.com/v2/translate"  # Pro tier
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
64
      QWEN_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"  # 北京地域
3b84605d   tangwang   docs
65
      # QWEN_BASE_URL = "https://dashscope-intl.aliyuncs.com/compatible-mode/v1"  # 新加坡
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
66
67
      # 如果使用新加坡地域的模型,需要将base_url替换为:https://dashscope-intl.aliyuncs.com/compatible-mode/v1
      QWEN_MODEL = "qwen-mt-flash"  # 快速翻译模型
be52af70   tangwang   first commit
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
  
      # Language code mapping
      LANG_CODE_MAP = {
          'zh': 'ZH',
          'en': 'EN',
          'ru': 'RU',
          'ar': 'AR',
          'ja': 'JA',
          'es': 'ES',
          'de': 'DE',
          'fr': 'FR',
          'it': 'IT',
          'pt': 'PT',
      }
  
      def __init__(
          self,
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
85
          model: str = "qwen",
be52af70   tangwang   first commit
86
87
          api_key: Optional[str] = None,
          use_cache: bool = True,
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
88
89
90
          timeout: int = 10,
          glossary_id: Optional[str] = None,
          translation_context: Optional[str] = None
be52af70   tangwang   first commit
91
92
93
94
95
      ):
          """
          Initialize translator.
  
          Args:
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
96
97
              model: Translation model to use. Options: 'qwen' (default) or 'deepl'
              api_key: API key for the selected model (or None to use from config/env)
be52af70   tangwang   first commit
98
99
              use_cache: Whether to cache translations
              timeout: Request timeout in seconds
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
100
              glossary_id: DeepL glossary ID for custom terminology (optional, only for DeepL)
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
101
              translation_context: Context hint for translation (e.g., "e-commerce", "product search")
be52af70   tangwang   first commit
102
          """
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
103
104
105
106
          self.model = model.lower()
          if self.model not in ['qwen', 'deepl']:
              raise ValueError(f"Unsupported model: {model}. Supported models: 'qwen', 'deepl'")
          
d79810d5   tangwang   first commit
107
          # Get API key from config if not provided
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
108
109
110
111
112
          if api_key is None:
              if self.model == 'qwen':
                  api_key = DASHSCOPE_API_KEY or os.getenv("DASHSCOPE_API_KEY")
              else:  # deepl
                  api_key = DEEPL_AUTH_KEY or os.getenv("DEEPL_AUTH_KEY")
d79810d5   tangwang   first commit
113
  
be52af70   tangwang   first commit
114
115
116
          self.api_key = api_key
          self.timeout = timeout
          self.use_cache = use_cache
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
117
118
          self.glossary_id = glossary_id
          self.translation_context = translation_context or "e-commerce product search"
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
119
120
121
122
123
124
125
126
127
128
129
          
          # Initialize OpenAI client for Qwen if needed
          self.qwen_client = None
          if self.model == 'qwen':
              if not self.api_key:
                  logger.warning("DASHSCOPE_API_KEY not set. Qwen translation will not work.")
              else:
                  self.qwen_client = OpenAI(
                      api_key=self.api_key,
                      base_url=self.QWEN_BASE_URL,
                  )
be52af70   tangwang   first commit
130
  
453992a8   tangwang   需求:
131
          # Initialize Redis cache if enabled
be52af70   tangwang   first commit
132
          if use_cache:
453992a8   tangwang   需求:
133
134
135
136
137
138
139
140
141
142
143
144
145
              try:
                  self.redis_client = redis.Redis(
                      host=REDIS_CONFIG.get('host', 'localhost'),
                      port=REDIS_CONFIG.get('port', 6479),
                      password=REDIS_CONFIG.get('password'),
                      decode_responses=True,  # Return str instead of bytes
                      socket_timeout=REDIS_CONFIG.get('socket_timeout', 1),
                      socket_connect_timeout=REDIS_CONFIG.get('socket_connect_timeout', 1),
                      retry_on_timeout=REDIS_CONFIG.get('retry_on_timeout', False),
                      health_check_interval=10,  # 避免复用坏连接
                  )
                  # Test connection
                  self.redis_client.ping()
a866b688   tangwang   翻译接口
146
147
148
                  expire_days = REDIS_CONFIG.get('translation_cache_expire_days', 360)
                  self.expire_time = timedelta(days=expire_days)
                  self.expire_seconds = int(self.expire_time.total_seconds())  # Redis 需要秒数
453992a8   tangwang   需求:
149
150
151
152
153
154
                  self.cache_prefix = REDIS_CONFIG.get('translation_cache_prefix', 'trans')
                  logger.info("Redis cache initialized for translations")
              except Exception as e:
                  logger.warning(f"Failed to initialize Redis cache: {e}, falling back to no cache")
                  self.redis_client = None
                  self.cache = None
be52af70   tangwang   first commit
155
          else:
453992a8   tangwang   需求:
156
              self.redis_client = None
be52af70   tangwang   first commit
157
              self.cache = None
6e0e310c   tangwang   1. Translator 类增强
158
159
160
          
          # Thread pool for async translation
          self.executor = ThreadPoolExecutor(max_workers=2, thread_name_prefix="translator")
be52af70   tangwang   first commit
161
162
163
164
165
  
      def translate(
          self,
          text: str,
          target_lang: str,
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
166
          source_lang: Optional[str] = None,
0064e946   tangwang   feat: 增量索引服务、租户配置...
167
168
          context: Optional[str] = None,
          prompt: Optional[str] = None
be52af70   tangwang   first commit
169
170
      ) -> Optional[str]:
          """
0064e946   tangwang   feat: 增量索引服务、租户配置...
171
          Translate text to target language (synchronous mode).
be52af70   tangwang   first commit
172
173
174
175
  
          Args:
              text: Text to translate
              target_lang: Target language code ('zh', 'en', 'ru', etc.)
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
176
              source_lang: Source language code (option al, auto-detect if None)
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
177
              context: Additional context for translation (overrides default context)
0064e946   tangwang   feat: 增量索引服务、租户配置...
178
              prompt: Translation prompt/instruction (optional, for better translation quality)
be52af70   tangwang   first commit
179
180
181
182
183
184
185
186
187
188
189
190
  
          Returns:
              Translated text or None if translation fails
          """
          if not text or not text.strip():
              return text
  
          # Normalize language codes
          target_lang = target_lang.lower()
          if source_lang:
              source_lang = source_lang.lower()
  
a5a6bab8   tangwang   多语言查询优化
191
192
          # Optimization: Skip translation if not needed
          if target_lang == 'en' and self._is_english_text(text):
3652f85f   tangwang   trans for index
193
              logger.info(f"[Translator] Text is already English, skipping translation: '{text[:50]}...'")
a5a6bab8   tangwang   多语言查询优化
194
195
196
              return text
          
          if target_lang == 'zh' and (self._contains_chinese(text) or self._is_pure_number(text)):
70dab99f   tangwang   add logs
197
198
199
200
              logger.info(
                  f"[Translator] Translation request | Original text: '{text}' | Target language: {target_lang} | "
                  f"Source language: {source_lang or 'auto'} | Result: Skip translation (contains Chinese or pure number)"
              )
a5a6bab8   tangwang   多语言查询优化
201
202
              return text
  
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
203
204
          # Use provided context or default context
          translation_context = context or self.translation_context
0064e946   tangwang   feat: 增量索引服务、租户配置...
205
206
207
208
209
210
211
212
213
          
          # Build cache key (include prompt in cache key if provided)
          cache_key_parts = [source_lang or 'auto', target_lang, translation_context]
          if prompt:
              cache_key_parts.append(prompt)
          cache_key_parts.append(text)
          cache_key = ':'.join(cache_key_parts)
  
          # Check cache (include context and prompt in cache key for accuracy)
453992a8   tangwang   需求:
214
215
          if self.use_cache and self.redis_client:
              cached = self._get_cached_translation_redis(text, target_lang, source_lang, translation_context, prompt)
be52af70   tangwang   first commit
216
              if cached:
3652f85f   tangwang   trans for index
217
                  logger.info(
70dab99f   tangwang   add logs
218
219
                      f"[Translator] Translation request | Original text: '{text}' | Target language: {target_lang} | "
                      f"Source language: {source_lang or 'auto'} | Result: '{cached}' | Source: Cache hit"
3652f85f   tangwang   trans for index
220
                  )
be52af70   tangwang   first commit
221
222
223
224
                  return cached
  
          # If no API key, return mock translation (for testing)
          if not self.api_key:
70dab99f   tangwang   add logs
225
226
227
228
              logger.info(
                  f"[Translator] Translation request | Original text: '{text}' | Target language: {target_lang} | "
                  f"Source language: {source_lang or 'auto'} | Result: '{text}' | Source: Mock mode (no API key)"
              )
be52af70   tangwang   first commit
229
230
              return text
  
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
231
          # Translate using selected model
3652f85f   tangwang   trans for index
232
          logger.info(
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
233
              f"[Translator] Translation request | Model: {self.model} | Original text: '{text}' | Target language: {target_lang} | "
70dab99f   tangwang   add logs
234
235
              f"Source language: {source_lang or 'auto'} | Context: {translation_context} | "
              f"Prompt: {'yes' if prompt else 'no'} | Status: Starting translation"
3652f85f   tangwang   trans for index
236
          )
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
237
238
239
240
241
          
          if self.model == 'qwen':
              result = self._translate_qwen(text, target_lang, source_lang, translation_context, prompt)
          else:  # deepl
              result = self._translate_deepl(text, target_lang, source_lang, translation_context, prompt)
be52af70   tangwang   first commit
242
  
16c42787   tangwang   feat: implement r...
243
244
          # If still failed, return original text with warning
          if result is None:
70dab99f   tangwang   add logs
245
246
247
248
              logger.warning(
                  f"[Translator] Translation request | Original text: '{text}' | Target language: {target_lang} | "
                  f"Source language: {source_lang or 'auto'} | Result: '{text}' | Status: Translation failed, returning original"
              )
16c42787   tangwang   feat: implement r...
249
              result = text
70dab99f   tangwang   add logs
250
251
252
253
254
          else:
              logger.info(
                  f"[Translator] Translation request | Original text: '{text}' | Target language: {target_lang} | "
                  f"Source language: {source_lang or 'auto'} | Result: '{result}' | Status: Translation successful"
              )
3652f85f   tangwang   trans for index
255
  
be52af70   tangwang   first commit
256
          # Cache result
453992a8   tangwang   需求:
257
258
          if result and self.use_cache and self.redis_client:
              self._set_cached_translation_redis(text, target_lang, result, source_lang, translation_context, prompt)
be52af70   tangwang   first commit
259
260
261
  
          return result
  
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
      def _translate_qwen(
          self,
          text: str,
          target_lang: str,
          source_lang: Optional[str],
          context: Optional[str] = None,
          prompt: Optional[str] = None
      ) -> Optional[str]:
          """
          Translate using Qwen MT Flash model via Alibaba Cloud DashScope API.
          
          Args:
              text: Text to translate
              target_lang: Target language code ('zh', 'en', 'ru', etc.)
              source_lang: Source language code (optional, 'auto' if None)
              context: Context hint for translation (optional)
              prompt: Translation prompt/instruction (optional)
              
          Returns:
              Translated text or None if translation fails
          """
          if not self.qwen_client:
              logger.error("[Translator] Qwen client not initialized. Check DASHSCOPE_API_KEY.")
              return None
          
          # Qwen (qwen-mt-plus/flash/turbo) supported languages mapping
          # 标准来自:你提供的“语言 / 英文名 / 代码”表
          qwen_lang_map = {
              "en": "English",
              "zh": "Chinese",
              "zh_tw": "Traditional Chinese",
              "ru": "Russian",
              "ja": "Japanese",
              "ko": "Korean",
              "es": "Spanish",
              "fr": "French",
              "pt": "Portuguese",
              "de": "German",
              "it": "Italian",
              "th": "Thai",
              "vi": "Vietnamese",
              "id": "Indonesian",
              "ms": "Malay",
              "ar": "Arabic",
              "hi": "Hindi",
              "he": "Hebrew",
              "my": "Burmese",
              "ta": "Tamil",
              "ur": "Urdu",
              "bn": "Bengali",
              "pl": "Polish",
              "nl": "Dutch",
              "ro": "Romanian",
              "tr": "Turkish",
              "km": "Khmer",
              "lo": "Lao",
              "yue": "Cantonese",
              "cs": "Czech",
              "el": "Greek",
              "sv": "Swedish",
              "hu": "Hungarian",
              "da": "Danish",
              "fi": "Finnish",
              "uk": "Ukrainian",
              "bg": "Bulgarian",
          }
          
          # Convert target language
          target_lang_normalized = target_lang.lower()
          target_lang_qwen = qwen_lang_map.get(target_lang_normalized, target_lang.capitalize())
  
          # Convert source language
          source_lang_normalized = (source_lang or "").strip().lower()
          if not source_lang_normalized or source_lang_normalized == "auto":
              source_lang_qwen = "auto"
          else:
              source_lang_qwen = qwen_lang_map.get(source_lang_normalized, source_lang.capitalize())
          
          # Prepare translation options
          translation_options = {
              "source_lang": source_lang_qwen,
              "target_lang": target_lang_qwen,
          }
          
          # Prepare messages
          messages = [
              {
                  "role": "user",
                  "content": text
              }
          ]
          
d90e7428   tangwang   补充重排
354
          start_time = time.time()
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
355
356
357
358
359
360
361
362
363
364
          try:
              completion = self.qwen_client.chat.completions.create(
                  model=self.QWEN_MODEL,
                  messages=messages,
                  extra_body={
                      "translation_options": translation_options
                  }
              )
              
              translated_text = completion.choices[0].message.content.strip()
d90e7428   tangwang   补充重排
365
              duration_ms = (time.time() - start_time) * 1000
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
366
              
d90e7428   tangwang   补充重排
367
              logger.info(
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
368
                  f"[Translator] Qwen API response success | Original text: '{text}' | Target language: {target_lang_qwen} | "
d90e7428   tangwang   补充重排
369
                  f"Translation result: '{translated_text}' | Duration: {duration_ms:.2f} ms"
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
370
371
372
373
              )
              return translated_text
              
          except Exception as e:
d90e7428   tangwang   补充重排
374
              duration_ms = (time.time() - start_time) * 1000
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
375
376
              logger.error(
                  f"[Translator] Qwen API request exception | Original text: '{text}' | Target language: {target_lang_qwen} | "
d90e7428   tangwang   补充重排
377
                  f"Duration: {duration_ms:.2f} ms | Error: {e}", exc_info=True
3cd09b3b   tangwang   翻译接口改为调用qwen-mt-f...
378
379
380
              )
              return None
  
be52af70   tangwang   first commit
381
382
383
384
      def _translate_deepl(
          self,
          text: str,
          target_lang: str,
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
385
          source_lang: Optional[str],
0064e946   tangwang   feat: 增量索引服务、租户配置...
386
387
          context: Optional[str] = None,
          prompt: Optional[str] = None
be52af70   tangwang   first commit
388
      ) -> Optional[str]:
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
389
390
391
392
393
394
395
396
397
          """
          Translate using DeepL API with context and glossary support.
          
          Args:
              text: Text to translate
              target_lang: Target language code
              source_lang: Source language code (optional)
              context: Context hint for translation (e.g., "e-commerce product search")
          """
be52af70   tangwang   first commit
398
399
400
401
402
403
404
405
          # Map to DeepL language codes
          target_code = self.LANG_CODE_MAP.get(target_lang, target_lang.upper())
  
          headers = {
              "Authorization": f"DeepL-Auth-Key {self.api_key}",
              "Content-Type": "application/json",
          }
  
0064e946   tangwang   feat: 增量索引服务、租户配置...
406
407
408
409
410
          # Use prompt as context parameter for DeepL API (not as text prefix)
          # According to DeepL API: context is "Additional context that can influence a translation but is not translated itself"
          # If prompt is provided, use it as context; otherwise use the default context
          api_context = prompt if prompt else context
          
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
411
412
          # For e-commerce, add context words to help DeepL understand the domain
          # This is especially important for single-word ambiguous terms like "车" (car vs rook)
0064e946   tangwang   feat: 增量索引服务、租户配置...
413
          text_to_translate, needs_extraction = self._add_ecommerce_context(text, source_lang, api_context)
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
414
  
be52af70   tangwang   first commit
415
          payload = {
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
416
              "text": [text_to_translate],
be52af70   tangwang   first commit
417
418
419
420
421
422
423
              "target_lang": target_code,
          }
  
          if source_lang:
              source_code = self.LANG_CODE_MAP.get(source_lang, source_lang.upper())
              payload["source_lang"] = source_code
  
0064e946   tangwang   feat: 增量索引服务、租户配置...
424
425
426
427
428
          # Add context parameter (prompt or default context)
          # Context influences translation but is not translated itself
          if api_context:
              payload["context"] = api_context
  
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
429
430
431
432
          # Add glossary if configured
          if self.glossary_id:
              payload["glossary_id"] = self.glossary_id
  
0064e946   tangwang   feat: 增量索引服务、租户配置...
433
434
435
          # Note: DeepL API v2 supports "context" parameter for additional context
          # that influences translation but is not translated itself.
          # We use prompt as context parameter when provided.
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
436
  
be52af70   tangwang   first commit
437
438
439
440
441
442
443
444
445
446
447
          try:
              response = requests.post(
                  self.DEEPL_API_URL,
                  headers=headers,
                  json=payload,
                  timeout=self.timeout
              )
  
              if response.status_code == 200:
                  data = response.json()
                  if "translations" in data and len(data["translations"]) > 0:
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
448
449
450
451
452
453
                      translated_text = data["translations"][0]["text"]
                      # If we added context, extract just the term from the result
                      if needs_extraction:
                          translated_text = self._extract_term_from_translation(
                              translated_text, text, target_code
                          )
70dab99f   tangwang   add logs
454
455
456
457
                      logger.debug(
                          f"[Translator] DeepL API response success | Original text: '{text}' | Target language: {target_code} | "
                          f"Translation result: '{translated_text}'"
                      )
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
458
                      return translated_text
be52af70   tangwang   first commit
459
              else:
70dab99f   tangwang   add logs
460
461
462
463
                  logger.error(
                      f"[Translator] DeepL API error | Original text: '{text}' | Target language: {target_code} | "
                      f"Status code: {response.status_code} | Error message: {response.text}"
                  )
be52af70   tangwang   first commit
464
465
466
                  return None
  
          except requests.Timeout:
70dab99f   tangwang   add logs
467
468
469
470
              logger.warning(
                  f"[Translator] DeepL API request timeout | Original text: '{text}' | Target language: {target_code} | "
                  f"Timeout: {self.timeout}s"
              )
be52af70   tangwang   first commit
471
472
              return None
          except Exception as e:
70dab99f   tangwang   add logs
473
474
475
476
              logger.error(
                  f"[Translator] DeepL API request exception | Original text: '{text}' | Target language: {target_code} | "
                  f"Error: {e}", exc_info=True
              )
be52af70   tangwang   first commit
477
478
              return None
  
3652f85f   tangwang   trans for index
479
480
481
      # NOTE: _translate_deepl_free is intentionally not implemented.
      # We do not support automatic fallback to the free endpoint, to avoid
      # mixing Pro keys with https://api-free.deepl.com and related 403 errors.
16c42787   tangwang   feat: implement r...
482
  
be52af70   tangwang   first commit
483
484
485
486
      def translate_multi(
          self,
          text: str,
          target_langs: List[str],
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
487
          source_lang: Optional[str] = None,
6e0e310c   tangwang   1. Translator 类增强
488
          context: Optional[str] = None,
0064e946   tangwang   feat: 增量索引服务、租户配置...
489
490
          async_mode: bool = True,
          prompt: Optional[str] = None
be52af70   tangwang   first commit
491
492
493
      ) -> Dict[str, Optional[str]]:
          """
          Translate text to multiple target languages.
6e0e310c   tangwang   1. Translator 类增强
494
495
496
          
          In async_mode=True (default):
          - Returns cached translations immediately if available
a5a6bab8   tangwang   多语言查询优化
497
498
499
500
          - For translations that can be optimized (e.g., pure numbers, already in target language),
            returns result immediately via synchronous call
          - Launches async tasks for other missing translations (non-blocking)
          - Returns None for missing translations that require async processing
6e0e310c   tangwang   1. Translator 类增强
501
502
503
          
          In async_mode=False:
          - Waits for all translations to complete (blocking)
be52af70   tangwang   first commit
504
505
506
507
508
  
          Args:
              text: Text to translate
              target_langs: List of target language codes
              source_lang: Source language code (optional)
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
509
              context: Context hint for translation (optional)
6e0e310c   tangwang   1. Translator 类增强
510
              async_mode: If True, return cached results immediately and translate missing ones async
0064e946   tangwang   feat: 增量索引服务、租户配置...
511
              prompt: Translation prompt/instruction (optional)
be52af70   tangwang   first commit
512
513
  
          Returns:
6e0e310c   tangwang   1. Translator 类增强
514
              Dictionary mapping language code to translated text (only cached results in async mode)
be52af70   tangwang   first commit
515
516
          """
          results = {}
6e0e310c   tangwang   1. Translator 类增强
517
          missing_langs = []
a5a6bab8   tangwang   多语言查询优化
518
          async_langs = []
6e0e310c   tangwang   1. Translator 类增强
519
520
          
          # First, get cached translations
be52af70   tangwang   first commit
521
          for lang in target_langs:
0064e946   tangwang   feat: 增量索引服务、租户配置...
522
              cached = self._get_cached_translation(text, lang, source_lang, context, prompt)
6e0e310c   tangwang   1. Translator 类增强
523
524
525
526
527
              if cached is not None:
                  results[lang] = cached
              else:
                  missing_langs.append(lang)
          
a5a6bab8   tangwang   多语言查询优化
528
          # If async mode and there are missing translations
6e0e310c   tangwang   1. Translator 类增强
529
          if async_mode and missing_langs:
a5a6bab8   tangwang   多语言查询优化
530
              # Check if translation can be optimized (immediate return)
6e0e310c   tangwang   1. Translator 类增强
531
              for lang in missing_langs:
a5a6bab8   tangwang   多语言查询优化
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
                  target_lang = lang.lower()
                  # Check optimization conditions (same as in translate method)
                  can_optimize = False
                  if target_lang == 'en' and self._is_english_text(text):
                      can_optimize = True
                  elif target_lang == 'zh' and (self._contains_chinese(text) or self._is_pure_number(text)):
                      can_optimize = True
                  
                  if can_optimize:
                      # Can be optimized, call translate synchronously for immediate result
                      results[lang] = self.translate(text, lang, source_lang, context, prompt)
                  else:
                      # Requires actual translation, add to async list
                      async_langs.append(lang)
              
              # Launch async tasks for translations that require actual API calls
              if async_langs:
                  for lang in async_langs:
                      self._translate_async(text, lang, source_lang, context, prompt)
                  # Return None for async translations
                  for lang in async_langs:
                      results[lang] = None
6e0e310c   tangwang   1. Translator 类增强
554
555
556
          else:
              # Synchronous mode: wait for all translations
              for lang in missing_langs:
0064e946   tangwang   feat: 增量索引服务、租户配置...
557
                  results[lang] = self.translate(text, lang, source_lang, context, prompt)
6e0e310c   tangwang   1. Translator 类增强
558
          
be52af70   tangwang   first commit
559
          return results
6e0e310c   tangwang   1. Translator 类增强
560
      
3ec5bfe6   tangwang   1. get_translatio...
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
      def translate_multi_async(
          self,
          text: str,
          target_langs: List[str],
          source_lang: Optional[str] = None,
          context: Optional[str] = None,
          prompt: Optional[str] = None
      ) -> Dict[str, Union[str, Future]]:
          """
          Translate text to multiple target languages asynchronously, returning Futures that can be awaited.
          
          This method returns a dictionary where:
          - If translation is cached, the value is the translation string (immediate)
          - If translation needs to be done, the value is a Future object that can be awaited
          
          Args:
              text: Text to translate
              target_langs: List of target language codes
              source_lang: Source language code (optional)
              context: Context hint for translation (optional)
              prompt: Translation prompt/instruction (optional)
  
          Returns:
              Dictionary mapping language code to either translation string (cached) or Future object
          """
          results = {}
          missing_langs = []
          
          # First, get cached translations
          for lang in target_langs:
              cached = self._get_cached_translation(text, lang, source_lang, context, prompt)
              if cached is not None:
                  results[lang] = cached
              else:
                  missing_langs.append(lang)
          
          # For missing translations, submit async tasks and return Futures
          for lang in missing_langs:
              future = self.executor.submit(
                  self.translate,
                  text,
                  lang,
                  source_lang,
                  context,
                  prompt
              )
              results[lang] = future
          
          return results
      
6e0e310c   tangwang   1. Translator 类增强
611
612
613
614
615
      def _get_cached_translation(
          self,
          text: str,
          target_lang: str,
          source_lang: Optional[str] = None,
0064e946   tangwang   feat: 增量索引服务、租户配置...
616
617
          context: Optional[str] = None,
          prompt: Optional[str] = None
6e0e310c   tangwang   1. Translator 类增强
618
619
      ) -> Optional[str]:
          """Get translation from cache if available."""
453992a8   tangwang   需求:
620
621
622
623
624
625
626
627
628
629
630
631
          if not self.redis_client:
              return None
          return self._get_cached_translation_redis(text, target_lang, source_lang, context, prompt)
      
      def _get_cached_translation_redis(
          self,
          text: str,
          target_lang: str,
          source_lang: Optional[str] = None,
          context: Optional[str] = None,
          prompt: Optional[str] = None
      ) -> Optional[str]:
a866b688   tangwang   翻译接口
632
633
634
635
636
637
638
          """
          Get translation from Redis cache with sliding expiration.
          
          滑动过期机制:每次访问缓存时,重置过期时间为配置的过期时间(默认720天)。
          这样缓存会在最后一次访问后的720天才过期,而不是写入后的720天。
          这确保了常用的翻译缓存不会被过早删除。
          """
453992a8   tangwang   需求:
639
          if not self.redis_client:
6e0e310c   tangwang   1. Translator 类增强
640
641
              return None
          
453992a8   tangwang   需求:
642
643
644
645
646
647
648
649
          try:
              # Build cache key: prefix:target_lang:text
              # For simplicity, we use target_lang and text as key
              # Context and prompt are not included in key to maximize cache hits
              cache_key = f"{self.cache_prefix}:{target_lang.upper()}:{text}"
              value = self.redis_client.get(cache_key)
              if value:
                  # Sliding expiration: reset expiration time on access
a866b688   tangwang   翻译接口
650
651
652
653
654
655
656
657
658
                  # 每次读取缓存时,重置过期时间为配置的过期时间(最后一次访问后的N天才过期)
                  try:
                      self.redis_client.expire(cache_key, self.expire_seconds)
                  except Exception as expire_error:
                      # 即使 expire 失败,也返回缓存值(不影响功能)
                      logger.warning(
                          f"[Translator] Failed to update cache expiration for key {cache_key}: {expire_error}"
                      )
                  
70dab99f   tangwang   add logs
659
660
                  logger.debug(
                      f"[Translator] Redis cache hit | Original text: '{text}' | Target language: {target_lang} | "
a866b688   tangwang   翻译接口
661
                      f"Cache key: {cache_key} | Translation result: '{value}' | TTL reset to {self.expire_seconds}s"
3652f85f   tangwang   trans for index
662
                  )
453992a8   tangwang   需求:
663
                  return value
70dab99f   tangwang   add logs
664
665
666
667
              logger.debug(
                  f"[Translator] Redis cache miss | Original text: '{text}' | Target language: {target_lang} | "
                  f"Cache key: {cache_key}"
              )
453992a8   tangwang   需求:
668
669
              return None
          except Exception as e:
70dab99f   tangwang   add logs
670
              logger.error(f"[Translator] Redis error during get translation cache | Original text: '{text}' | Target language: {target_lang} | Error: {e}")
453992a8   tangwang   需求:
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
              return None
      
      def _set_cached_translation_redis(
          self,
          text: str,
          target_lang: str,
          translation: str,
          source_lang: Optional[str] = None,
          context: Optional[str] = None,
          prompt: Optional[str] = None
      ) -> None:
          """Store translation in Redis cache."""
          if not self.redis_client:
              return
          
          try:
              cache_key = f"{self.cache_prefix}:{target_lang.upper()}:{text}"
a866b688   tangwang   翻译接口
688
              self.redis_client.setex(cache_key, self.expire_seconds, translation)
153a592e   tangwang   redis统计脚本
689
              logger.info(
70dab99f   tangwang   add logs
690
691
                  f"[Translator] Redis cache write | Original text: '{text}' | Target language: {target_lang} | "
                  f"Cache key: {cache_key} | Translation result: '{translation}'"
3652f85f   tangwang   trans for index
692
              )
453992a8   tangwang   需求:
693
          except Exception as e:
70dab99f   tangwang   add logs
694
695
696
697
              logger.error(
                  f"[Translator] Redis cache write failed | Original text: '{text}' | Target language: {target_lang} | "
                  f"Error: {e}"
              )
6e0e310c   tangwang   1. Translator 类增强
698
699
700
701
702
703
      
      def _translate_async(
          self,
          text: str,
          target_lang: str,
          source_lang: Optional[str] = None,
0064e946   tangwang   feat: 增量索引服务、租户配置...
704
705
          context: Optional[str] = None,
          prompt: Optional[str] = None
6e0e310c   tangwang   1. Translator 类增强
706
707
708
709
      ):
          """Launch async translation task."""
          def _do_translate():
              try:
0064e946   tangwang   feat: 增量索引服务、租户配置...
710
                  result = self.translate(text, target_lang, source_lang, context, prompt)
6e0e310c   tangwang   1. Translator 类增强
711
712
713
714
715
716
                  if result:
                      logger.debug(f"Async translation completed: {text} -> {target_lang}: {result}")
              except Exception as e:
                  logger.warning(f"Async translation failed: {text} -> {target_lang}: {e}")
          
          self.executor.submit(_do_translate)
be52af70   tangwang   first commit
717
  
522a3964   tangwang   多语言搜索翻译的优化(deepL添...
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
      def _add_ecommerce_context(
          self,
          text: str,
          source_lang: Optional[str],
          context: Optional[str]
      ) -> tuple:
          """
          Add e-commerce context to text for better disambiguation.
          
          For single-word ambiguous Chinese terms, we add context words that help
          DeepL understand this is an e-commerce/product search context.
          
          Args:
              text: Original text to translate
              source_lang: Source language code
              context: Context hint
              
          Returns:
              Tuple of (text_with_context, needs_extraction)
              - text_with_context: Text to send to DeepL
              - needs_extraction: Whether we need to extract the term from the result
          """
          # Only apply for e-commerce context and Chinese source
          if not context or "e-commerce" not in context.lower():
              return text, False
              
          if not source_lang or source_lang.lower() != 'zh':
              return text, False
              
          # For single-word queries, add context to help disambiguation
          text_stripped = text.strip()
          if len(text_stripped.split()) == 1 and len(text_stripped) <= 2:
              # Common ambiguous Chinese e-commerce terms like "车" (car vs rook)
              # We add a context phrase: "购买 [term]" (buy [term]) or "商品 [term]" (product [term])
              # This helps DeepL understand the e-commerce context
              # We'll need to extract just the term from the translation result
              context_phrase = f"购买 {text_stripped}"
              return context_phrase, True
          
          # For multi-word queries, DeepL usually has enough context
          return text, False
  
      def _extract_term_from_translation(
          self,
          translated_text: str,
          original_text: str,
          target_lang_code: str
      ) -> str:
          """
          Extract the actual term from a translation that included context.
          
          For example, if we translated "购买 车" (buy car) and got "buy car",
          we want to extract just "car".
          
          Args:
              translated_text: Full translation result
              original_text: Original single-word query
              target_lang_code: Target language code (EN, ZH, etc.)
              
          Returns:
              Extracted term or original translation if extraction fails
          """
          # For English target, try to extract the last word (the actual term)
          if target_lang_code == "EN":
              words = translated_text.strip().split()
              if len(words) > 1:
                  # Usually the last word is the term we want
                  # But we need to be smart - if it's "buy car", we want "car"
                  # Common context words to skip: buy, purchase, product, item, etc.
                  context_words = {"buy", "purchase", "product", "item", "commodity", "goods"}
                  # Try to find the term (not a context word)
                  for word in reversed(words):
                      word_lower = word.lower().rstrip('.,!?;:')
                      if word_lower not in context_words:
                          return word_lower
                  # If all words are context words, return the last one
                  return words[-1].lower().rstrip('.,!?;:')
          
          # For other languages or if extraction fails, return as-is
          # The user can configure a glossary for better results
          return translated_text
  
038e4e2f   tangwang   refactor(i18n): t...
800
801
802
803
804
805
806
807
808
809
810
811
      def _shop_lang_matches(self, shop_lang_lower: str, lang_code: str) -> bool:
          """True if shop language matches index language (use source, no translate)."""
          if not shop_lang_lower or not lang_code:
              return False
          if shop_lang_lower == lang_code:
              return True
          if lang_code == "zh" and "zh" in shop_lang_lower:
              return True
          if lang_code == "en" and "en" in shop_lang_lower:
              return True
          return False
  
453992a8   tangwang   需求:
812
813
814
815
816
817
      def translate_for_indexing(
          self,
          text: str,
          shop_language: str,
          source_lang: Optional[str] = None,
          context: Optional[str] = None,
345d960b   tangwang   1. 删除全局 enable_tr...
818
          prompt: Optional[str] = None,
038e4e2f   tangwang   refactor(i18n): t...
819
          index_languages: Optional[List[str]] = None,
453992a8   tangwang   需求:
820
821
      ) -> Dict[str, Optional[str]]:
          """
038e4e2f   tangwang   refactor(i18n): t...
822
823
824
825
826
          Translate text for indexing based on shop language and tenant index_languages.
  
          For each language in index_languages: use source text if shop language matches,
          otherwise translate to that language.
  
453992a8   tangwang   需求:
827
828
          Args:
              text: Text to translate
038e4e2f   tangwang   refactor(i18n): t...
829
830
              shop_language: Shop primary language (e.g. 'zh', 'en', 'ru')
              source_lang: Source language code (optional)
453992a8   tangwang   需求:
831
              context: Additional context for translation (optional)
038e4e2f   tangwang   refactor(i18n): t...
832
833
834
              prompt: Translation prompt (optional)
              index_languages: Languages to index (from tenant_config). Default ["en", "zh"].
  
453992a8   tangwang   需求:
835
          Returns:
038e4e2f   tangwang   refactor(i18n): t...
836
              Dict keyed by each index_language with translated or source text (or None).
453992a8   tangwang   需求:
837
          """
038e4e2f   tangwang   refactor(i18n): t...
838
839
          langs = index_languages if index_languages else ["en", "zh"]
          results = {lang: None for lang in langs}
453992a8   tangwang   需求:
840
          if not text or not text.strip():
038e4e2f   tangwang   refactor(i18n): t...
841
              return results
453992a8   tangwang   需求:
842
843
          if re.match(r'^[\d\s_-]+$', text):
              logger.info(f"[Translator] Skip translation for symbol-only query: '{text}'")
453992a8   tangwang   需求:
844
              return results
038e4e2f   tangwang   refactor(i18n): t...
845
846
847
848
849
850
851
852
853
  
          shop_lang_lower = (shop_language or "").strip().lower()
          targets = []
          for lang in langs:
              if self._shop_lang_matches(shop_lang_lower, lang):
                  results[lang] = text
              else:
                  targets.append(lang)
  
453992a8   tangwang   需求:
854
          for target_lang in targets:
453992a8   tangwang   需求:
855
856
857
858
859
              cached = self._get_cached_translation_redis(text, target_lang, source_lang, context, prompt)
              if cached:
                  results[target_lang] = cached
                  logger.debug(f"[Translator] Cache hit for indexing: '{text}' -> {target_lang}: {cached}")
                  continue
453992a8   tangwang   需求:
860
861
862
863
864
              translated = self.translate(
                  text,
                  target_lang=target_lang,
                  source_lang=source_lang or shop_language,
                  context=context,
038e4e2f   tangwang   refactor(i18n): t...
865
                  prompt=prompt,
453992a8   tangwang   需求:
866
867
              )
              results[target_lang] = translated
453992a8   tangwang   需求:
868
869
          return results
  
be52af70   tangwang   first commit
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
      def get_translation_needs(
          self,
          detected_lang: str,
          supported_langs: List[str]
      ) -> List[str]:
          """
          Determine which languages need translation.
  
          Args:
              detected_lang: Detected query language
              supported_langs: List of supported languages
  
          Returns:
              List of language codes to translate to
          """
          # If detected language is in supported list, translate to others
          if detected_lang in supported_langs:
453992a8   tangwang   需求:
887
              return [lang for lang in supported_langs if detected_lang != lang]
be52af70   tangwang   first commit
888
889
890
  
          # Otherwise, translate to all supported languages
          return supported_langs
a5a6bab8   tangwang   多语言查询优化
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
      
      def _is_english_text(self, text: str) -> bool:
          """
          Check if text is primarily English (ASCII letters, numbers, common punctuation).
          
          Args:
              text: Text to check
              
          Returns:
              True if text appears to be English
          """
          if not text or not text.strip():
              return True
          
          # Remove whitespace and common punctuation
          text_clean = re.sub(r'[\s\.,!?;:\-\'\"\(\)\[\]{}]', '', text)
          if not text_clean:
              return True
          
          # Check if all remaining characters are ASCII (letters, numbers)
          # This is a simple heuristic: if most characters are ASCII, it's likely English
          ascii_count = sum(1 for c in text_clean if ord(c) < 128)
          ratio = ascii_count / len(text_clean) if text_clean else 0
          
          # If more than 80% are ASCII characters, consider it English
          return ratio > 0.8
      
      def _contains_chinese(self, text: str) -> bool:
          """
          Check if text contains Chinese characters (Han characters).
          
          Args:
              text: Text to check
              
          Returns:
              True if text contains Chinese characters
          """
          if not text:
              return False
          
          # Check for Chinese characters (Unicode range: \u4e00-\u9fff)
          chinese_pattern = re.compile(r'[\u4e00-\u9fff]')
          return bool(chinese_pattern.search(text))
      
      def _is_pure_number(self, text: str) -> bool:
          """
          Check if text is purely numeric (digits, possibly with spaces, dots, commas).
          
          Args:
              text: Text to check
              
          Returns:
              True if text is purely numeric
          """
          if not text or not text.strip():
              return False
          
          # Remove whitespace, dots, commas (common number separators)
          text_clean = re.sub(r'[\s\.,]', '', text.strip())
          if not text_clean:
              return False
          
          # Check if all remaining characters are digits
          return text_clean.isdigit()