test_api.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. import requests
  2. import json
  3. # 测试预览 API
  4. url = "http://localhost:8000/api/templates/preview"
  5. # 测试1: 简单模板
  6. payload1 = {
  7. "template_body": "{% set greeting = ['嗨', '哈喽'] %}{{ greeting | random }},我是小优!",
  8. "parameters": {}
  9. }
  10. r1 = requests.post(url, json=payload1)
  11. print(f"测试1: {r1.status_code}")
  12. print(json.dumps(r1.json(), ensure_ascii=False, indent=2))
  13. # 测试2: 从真实模板获取
  14. url_templates = "http://localhost:8000/api/templates?machine_model=x11&device_type=wash&template_type=common"
  15. r_templates = requests.get(url_templates)
  16. templates = r_templates.json()
  17. if templates:
  18. tpl = templates[0]
  19. print(f"\n测试真实模板: {tpl['name']}")
  20. payload2 = {
  21. "template_body": tpl['template_body'],
  22. "parameters": {
  23. "program_id": "15",
  24. "program_name": "速洗",
  25. "rerank_reason": "PIN_CONSECUTIVE"
  26. }
  27. }
  28. r2 = requests.post(url, json=payload2)
  29. print(f"状态: {r2.status_code}")
  30. result = r2.json()
  31. print(f"结果长度: {len(result.get('rendered_text', ''))}")
  32. print(f"结果: {result.get('rendered_text', '')[:300]}")