优化案子处理逻辑

services/paddle_services/ie_cost.py

@@ -4,12 +4,14 @@ import logging.config
 from flask import Flask, request
 from paddlenlp import Taskflow
 
-from __init__ import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES
+from __init__ import IE_KEY
 from log import LOGGING_CONFIG
 from utils import process_request
 
 app = Flask(__name__)
-COST_LIST_SCHEMA = PATIENT_NAME + ADMISSION_DATE + DISCHARGE_DATE + MEDICAL_EXPENSES
+COST_LIST_SCHEMA = tuple(IE_KEY[key] for key in [
+    'name', 'admission_date', 'discharge_date', 'medical_expenses'
+])
 COST = Taskflow('information_extraction', schema=COST_LIST_SCHEMA, model='uie-x-base',
                 task_path='model/cost_list_model', layout_analysis=False, precision='fp16')