Blame view

providers/translation.py 4.64 KB
d4cadc13   tangwang   翻译重构
1
  """Translation provider factory and HTTP provider implementation."""
a0a173ae   tangwang   last
2
3
4
  from __future__ import annotations
  
  import logging
d4cadc13   tangwang   翻译重构
5
  from typing import Any, Dict, Optional
a0a173ae   tangwang   last
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
  import requests
  
  from config.services_config import get_translation_config, get_translation_base_url
  
  logger = logging.getLogger(__name__)
  
  
  class HttpTranslationProvider:
      """Translation via HTTP service."""
  
      def __init__(
          self,
          base_url: str,
          model: str = "qwen",
          timeout_sec: float = 10.0,
a0a173ae   tangwang   last
21
22
23
24
      ):
          self.base_url = (base_url or "").rstrip("/")
          self.model = model or "qwen"
          self.timeout_sec = float(timeout_sec or 10.0)
a0a173ae   tangwang   last
25
26
27
28
29
30
  
      def _translate_once(
          self,
          text: str,
          target_lang: str,
          source_lang: Optional[str] = None,
d4cadc13   tangwang   翻译重构
31
32
          context: Optional[str] = None,
          prompt: Optional[str] = None,
a0a173ae   tangwang   last
33
34
35
36
37
38
39
40
41
42
43
      ) -> Optional[str]:
          if not text or not str(text).strip():
              return text
          try:
              url = f"{self.base_url}/translate"
              payload = {
                  "text": text,
                  "target_lang": target_lang,
                  "source_lang": source_lang or "auto",
                  "model": self.model,
              }
d4cadc13   tangwang   翻译重构
44
45
46
47
              if context:
                  payload["context"] = context
              if prompt:
                  payload["prompt"] = prompt
a0a173ae   tangwang   last
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
              response = requests.post(url, json=payload, timeout=self.timeout_sec)
              if response.status_code != 200:
                  logger.warning(
                      "HTTP translator failed: status=%s body=%s",
                      response.status_code,
                      (response.text or "")[:200],
                  )
                  return None
              data = response.json()
              translated = data.get("translated_text")
              return translated if translated is not None else None
          except Exception as exc:
              logger.warning("HTTP translator request failed: %s", exc, exc_info=True)
              return None
  
      def translate(
          self,
          text: str,
          target_lang: str,
          source_lang: Optional[str] = None,
          context: Optional[str] = None,
          prompt: Optional[str] = None,
      ) -> Optional[str]:
d4cadc13   tangwang   翻译重构
71
72
73
74
75
76
77
          return self._translate_once(
              text=text,
              target_lang=target_lang,
              source_lang=source_lang,
              context=context,
              prompt=prompt,
          )
a0a173ae   tangwang   last
78
79
80
81
82
83
84
85
86
87
88
89
  
  
  def create_translation_provider(query_config: Any = None) -> Any:
      """
      Create translation provider from services config.
  
      query_config: optional, for api_key/glossary_id/context (used by direct provider).
      """
      cfg = get_translation_config()
      provider = cfg.provider
      pc = cfg.get_provider_cfg()
  
d4cadc13   tangwang   翻译重构
90
      if provider in ("qwen-mt", "direct", "local", "inprocess"):
a0a173ae   tangwang   last
91
          from query.qwen_mt_translate import Translator
d4cadc13   tangwang   翻译重构
92
          model = pc.get("model") or "qwen-mt-flash"
a0a173ae   tangwang   last
93
94
95
96
97
98
99
100
101
          qc = query_config or _empty_query_config()
          return Translator(
              model=model,
              api_key=getattr(qc, "translation_api_key", None),
              use_cache=True,
              glossary_id=getattr(qc, "translation_glossary_id", None),
              translation_context=getattr(qc, "translation_context", "e-commerce product search"),
          )
  
d4cadc13   tangwang   翻译重构
102
      elif provider in ("http", "service"):
a0a173ae   tangwang   last
103
104
105
106
107
108
109
110
          base_url = get_translation_base_url()
          model = pc.get("model") or "qwen"
          timeout = pc.get("timeout_sec", 10.0)
          qc = query_config or _empty_query_config()
          return HttpTranslationProvider(
              base_url=base_url,
              model=model,
              timeout_sec=float(timeout),
d4cadc13   tangwang   翻译重构
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
          )
  
      elif provider == "llm":
          from query.llm_translate import LLMTranslatorProvider
          model = pc.get("model")
          timeout = float(pc.get("timeout_sec", 30.0))
          base_url = (pc.get("base_url") or "").strip() or None
          return LLMTranslatorProvider(
              model=model,
              timeout_sec=timeout,
              base_url=base_url,
          )
  
      elif provider == "deepl":
          from query.deepl_provider import DeepLProvider
          qc = query_config or _empty_query_config()
          return DeepLProvider(
              api_key=getattr(qc, "translation_api_key", None),
              timeout=float(pc.get("timeout_sec", 10.0)),
              glossary_id=pc.get("glossary_id") or getattr(qc, "translation_glossary_id", None),
a0a173ae   tangwang   last
131
132
133
134
135
136
137
138
139
140
141
142
          )
  
      raise ValueError(f"Unsupported translation provider: {provider}")
  
  
  def _empty_query_config() -> Any:
      """Minimal object with default translation attrs."""
      class _QC:
          translation_api_key = None
          translation_glossary_id = None
          translation_context = "e-commerce product search"
      return _QC()