test.py 1.1 KB

123456789101112131415161718192021222324252627
  1. from agent.cdss.capbility import CDSSCapability
  2. from agent.cdss.models.schemas import CDSSInput, CDSSOutput, CDSSInt, CDSSText
  3. capability = CDSSCapability()
  4. record = CDSSInput(
  5. pat_age=CDSSInt(type="month", value=24),
  6. pat_sex=CDSSText(type="sex", value="男"),
  7. chief_complaint=["右下腹痛", "恶心", "呕吐"],
  8. #chief_complaint=["呕血", "黑便", "头晕", "心悸"],
  9. #chief_complaint=["流鼻涕"],
  10. department=CDSSText(type='department',value="急诊医学科")
  11. )
  12. if __name__ == "__main__":
  13. output = capability.process(input=record)
  14. #output = capability.process(input=record)
  15. print(output.departments.value)
  16. for item in output.diagnosis.value:
  17. print(f"DIAG {item} {output.diagnosis.value[item]} ")
  18. # for item in output.count_diagnosis.value:
  19. # print(f"count DIAG {item} {output.count_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]} ")