调整数据打标接口入参名称data->transactions

llmops/agents/data_classify_agent.py

@@ -36,7 +36,7 @@ class DataClassifyAgent:
             json_data = {
                 "id": self.DATA_CLASSIFY_ENGINE_PARAM_MAPPER.get(industry),
                 "input": {
-                    "data": data_set
+                    "transactions": data_set
                 }
             }
 
@@ -47,7 +47,7 @@ class DataClassifyAgent:
                 data_set_classified = response.json()
                 if isinstance(data_set_classified, dict):
                     # 取出打标数据集
-                    ds = data_set_classified.get('resultTag', 0)
+                    ds = data_set_classified.get('transactions', [])
                     print(f"✅ 成功分类打标数量： {len(ds)}")
                     # 将分类好的数据写入数据目录中
                     self.save_classified_data(ds, file_name)
@@ -63,6 +63,8 @@ class DataClassifyAgent:
 
         except Exception as e:
             print(f"❌ 调用数据分类打标时发生错误: {str(e)}")
+            import traceback
+            traceback.print_exc()
             return []