add 真实well数据覆盖

2025-08-19 18:09:14 +08:00
parent fbd90237ec
commit 7be36f694e
6 changed files with 54 additions and 608 deletions
@@ -369,6 +369,12 @@ class DmsCrudScenarioStage(BaseAPIStage):
                        # 确保所有主键字段正确设置
                        for pk_field, pk_val in all_pk_values.items():
                            create_payload[pk_field] = pk_val
+
+                        # 🔑 关键修复：LLM生成的数据也需要井数据增强
+                        if hasattr(self, 'orchestrator') and hasattr(self.orchestrator, 'well_data_manager') and self.orchestrator.well_data_manager:
+                            create_payload = self.orchestrator.well_data_manager.enhance_data_with_well_values(create_payload)
+                            self.logger.info(f"LLM生成数据已通过井数据管理器增强")
+
                        self.logger.info(f"LLM成功生成智能测试数据: {create_payload}")
                    else:
                        self.logger.warning("LLM生成的数据格式不符合预期，回退到传统数据生成")