clean_agent

agents/clean_agent/tools.py

@@ -128,13 +128,13 @@ def batch_evaluate_content(contents: list, db: Session, request_id: str, query_w
         # 处理评估结果
         evaluation_results = []
         
-        for i, (parsing_id, score, reason, parsing_data) in enumerate(evaluation_results_raw):
+        for i, (parsing_id, score, score_reason, parsing_data) in enumerate(evaluation_results_raw):
             # 创建KnowledgeExtractionContent对象
             extraction_content = KnowledgeExtractionContent(
                 request_id=request_id,
                 parsing_id=parsing_id,
                 score=score,
-                reason=reason,
+                score_reason=score_reason,
                 create_at=datetime.now()
             )
             db.add(extraction_content)
@@ -142,7 +142,7 @@ def batch_evaluate_content(contents: list, db: Session, request_id: str, query_w
             evaluation_results.append({
                 "parsing_id": parsing_id,
                 "score": score,
-                "reason": reason,
+                "score_reason": score_reason,
                 "parsing_data": parsing_data,
                 "extraction_content": extraction_content
             })
@@ -245,9 +245,9 @@ def batch_call_llm_for_evaluation(contents: list, query_word: str) -> list:
             parsing_id = contents[i].id
             parsing_data = contents[i].parsing_data   
             score = result.get("score", -2)
-            reason = result.get("reason", "")
+            score_reason = result.get("score_reason", "")
             
-            evaluation_results.append((parsing_id, score, reason, parsing_data))
+            evaluation_results.append((parsing_id, score, score_reason, parsing_data))
         
         return evaluation_results
         

database/models.py

@@ -25,8 +25,9 @@ class KnowledgeExtractionContent(Base):
     request_id = Column(String(128), nullable=False)
     parsing_id = Column(BigInteger, nullable=False)
     score = Column(Integer, default=-1)
-    reason = Column(Text, comment='打分原因')
+    score_reason = Column(Text, comment='打分原因')
     data = Column(Text, comment='结构化数据')
+    clean_reason = Column(Text, comment='清洗原因')
     create_at = Column(DateTime, default=datetime.now)
     status = Column(Integer, default=0, comment='0: 未开始，1:处理中 2: 处理完成 3:处理失败')