diff --git a/src/lead/core/multi_objective/evolution.py b/src/lead/core/multi_objective/evolution.py index abba534..f7b4eb7 100644 --- a/src/lead/core/multi_objective/evolution.py +++ b/src/lead/core/multi_objective/evolution.py @@ -47,6 +47,7 @@ class MultiObjectiveEvolutionEngine: # 基于每个思路生成具体实现 population = [] for i, idea in enumerate(ideas): + print(f"生成第 {i+1}/{size} 个个体,思路:{idea}") code = self.initialize_operator.generate_initial_code( problem_config["description"], problem_config["function_name"], diff --git a/src/lead/core/operators/initialize_operator.py b/src/lead/core/operators/initialize_operator.py index ab28c80..8048184 100644 --- a/src/lead/core/operators/initialize_operator.py +++ b/src/lead/core/operators/initialize_operator.py @@ -35,8 +35,9 @@ class InitializeOperator: def generate_initial_code(self, problem_desc: str, function_name: str, input_fmt: str, output_fmt: str, idea: str = None) -> str: """基于特定思路生成初始代码""" - base_prompt = f"""请用Python实现以下算法思路: -思路:{idea or '默认算法'} + base_prompt = f"""请用Python为所描述的问题通过以下算法思路给出代码实现: +问题描述:{problem_desc} +思路:{idea or '无特定思路'} 具体要求: 1. 函数名:{function_name}