优化案子处理逻辑

services/paddle_services/ie_discharge.py

@@ -4,14 +4,14 @@ import logging.config
 from flask import Flask, request
 from paddlenlp import Taskflow
 
-from __init__ import HOSPITAL, DEPARTMENT, PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, DOCTOR, ADMISSION_ID, AGE
+from __init__ import IE_KEY
 from log import LOGGING_CONFIG
 from utils import process_request
 
 app = Flask(__name__)
-DISCHARGE_RECORD_SCHEMA = (
-        HOSPITAL + DEPARTMENT + PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + DOCTOR + ADMISSION_ID + AGE
-)
+DISCHARGE_RECORD_SCHEMA = tuple(IE_KEY[key] for key in [
+    'hospital', 'department', 'name', 'admission_date', 'discharge_date', 'doctor', 'admission_id', 'age'
+])
 DISCHARGE = Taskflow('information_extraction', schema=DISCHARGE_RECORD_SCHEMA, model='uie-x-base',
                      task_path='model/discharge_record_model', layout_analysis=False, precision='fp16')