123456789101112131415161718192021222324252627 |
- from agent.cdss.capbility import CDSSCapability
- from agent.cdss.models.schemas import CDSSInput, CDSSOutput, CDSSInt, CDSSText
- capability = CDSSCapability()
- record = CDSSInput(
- pat_age=CDSSInt(type="month", value=24),
- pat_sex=CDSSText(type="sex", value="男"),
- chief_complaint=["右下腹痛", "恶心", "呕吐"],
- #chief_complaint=["呕血", "黑便", "头晕", "心悸"],
- #chief_complaint=["流鼻涕"],
- department=CDSSText(type='department',value="急诊医学科")
- )
- if __name__ == "__main__":
- output = capability.process(input=record)
- #output = capability.process(input=record)
- print(output.departments.value)
- for item in output.diagnosis.value:
- print(f"DIAG {item} {output.diagnosis.value[item]} ")
- # for item in output.count_diagnosis.value:
- # print(f"count DIAG {item} {output.count_diagnosis.value[item]} ")
- # for item in output.checks.value:
- # print(f"CHECK {item} {output.checks.value[item]} ")
- # for item in output.drugs.value:
- # print(f"DRUG {item} {output.drugs.value[item]} ")
|