test.py 1022 B

12345678910111213141516171819202122232425262728
  1. from cdss.capbility import CDSSCapability
  2. from cdss.models.schemas import CDSSInput, CDSSOutput, CDSSInt
  3. import logging
  4. logging.basicConfig(level=logging.INFO)
  5. handler = logging.FileHandler('cdss.log', mode='w',encoding="utf-8")
  6. handler.setLevel(logging.INFO)
  7. formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
  8. handler.setFormatter(formatter)
  9. logging.getLogger().addHandler(handler)
  10. capability = CDSSCapability()
  11. record = CDSSInput(
  12. pat_age=CDSSInt(type="month", value=21),
  13. pat_sex=CDSSInt(type="sex", value=1),
  14. chief_complaint=["嗓子疼", "流鼻涕", "打喷嚏", "低烧", "全身没力气"]
  15. #chief_complaint=["腹痛", "发热", "腹泻"],
  16. )
  17. output = capability.process(input=record)
  18. for item in output.diagnosis.value:
  19. print(f"DIAG {item} {output.diagnosis.value[item]} ")
  20. for item in output.checks.value:
  21. print(f"CHECK {item} {output.checks.value[item]} ")
  22. for item in output.drugs.value:
  23. print(f"DRUG {item} {output.drugs.value[item]} ")