fix:dms api

2025-08-19 17:38:43 +08:00
parent 75d42b7aab
commit a9ca927ba8
3 changed files with 188 additions and 3 deletions
@@ -82,6 +82,9 @@ class DataGenerator:
                    llm_data = self._generate_with_llm(schema, llm_service, context_name, operation_id)
                    if llm_data is not None:
                        self.logger.debug(f"{log_prefix}LLM successfully generated data for{context_log}")
+                        # 🔑 关键修复：LLM生成的数据也需要井数据增强
+                        if self.well_data_manager and isinstance(llm_data, dict):
+                            llm_data = self.well_data_manager.enhance_data_with_well_values(llm_data)
                        return llm_data
                except Exception as e:
                    self.logger.debug(f"{log_prefix}LLM generation failed for{context_log}: {e}, falling back to traditional generation")