Переглянути джерело

要素提取补充AI提取

luojiehua 2 місяців тому
батько
коміт
fb8424e8bd
1 змінених файлів з 5 додано та 2 видалено
  1. 5 2
      BaseDataMaintenance/maintenance/dataflow_mq.py

+ 5 - 2
BaseDataMaintenance/maintenance/dataflow_mq.py

@@ -1218,7 +1218,7 @@ class Dataflow_ActivteMQ_extract(Dataflow_extract):
                 ackMsg(conn,message_id,subscription)
                 if to_ai:
                     #sent to ai
-                    item[document_extract2_extract_json] = _extract.getProperties().get(document_extract2_extract_json,"{}")
+                    item[document_extract2_extract_json] = json.loads(_extract.getProperties().get(document_extract2_extract_json,"{}"))
                     send_msg_toacmq(self.pool_mq,json.dumps(item,ensure_ascii=False),self.mq_extract_ai)
             else:
                 item["extract_times"] -= 1
@@ -1370,7 +1370,10 @@ class Dataflow_ActivteMQ_extract(Dataflow_extract):
         _extract = {}
         if extract_json is not None:
             try:
-                _extract = json.loads(extract_json)
+                if isinstance(extract_json,str):
+                    _extract = json.loads(extract_json)
+                else:
+                    _extract = extract_json
             except Exception as e:
                 pass
         _extract_ai = {}