test.py 672 B

12345678910111213141516171819
  1. from cdss.capbility import CDSSCapability
  2. from cdss.models.schemas import CDSSInput, CDSSOutput, CDSSInt
  3. capability = CDSSCapability()
  4. record = CDSSInput(
  5. pat_age=CDSSInt(type="month", value=21),
  6. pat_sex=CDSSInt(type="sex", value=1),
  7. chief_complaint=["腹痛", "发热", "腹泻"],
  8. )
  9. if __name__ == "__main__":
  10. output = capability.process(input=record)
  11. for item in output.diagnosis.value:
  12. print(f"DIAG {item} {output.diagnosis.value[item]} ")
  13. for item in output.checks.value:
  14. print(f"CHECK {item} {output.checks.value[item]} ")
  15. for item in output.drugs.value:
  16. print(f"DRUG {item} {output.drugs.value[item]} ")