0fd2f875
 tangwang
translate
|
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
"""Manual smoke test for the translator service."""
from __future__ import annotations
import argparse
import json
from typing import Optional
from translation import create_translation_client
def main() -> None:
parser = argparse.ArgumentParser(description="Smoke test the translator service")
parser.add_argument("--text", default="蓝牙耳机", help="Text to translate")
parser.add_argument("--source-lang", default="zh", help="Source language")
parser.add_argument("--target-lang", default="en", help="Target language")
parser.add_argument("--model", default=None, help="Enabled translation capability name")
parser.add_argument("--scene", default="sku_name", help="Translation scene")
args = parser.parse_args()
client = create_translation_client()
result: Optional[str] = client.translate(
text=args.text,
target_lang=args.target_lang,
source_lang=args.source_lang,
model=args.model,
scene=args.scene,
)
payload = {
"text": args.text,
"source_lang": args.source_lang,
"target_lang": args.target_lang,
"model": args.model or client.default_model,
"scene": args.scene,
"translated_text": result,
}
print(json.dumps(payload, ensure_ascii=False, indent=2))
if __name__ == "__main__":
|