yangxiaohui пре 1 месец
родитељ
комит
eea80bfa04
3 измењених фајлова са 9 додато и 3 уклоњено
  1. 3 1
      sug_v5_0.py
  2. 3 1
      sug_v5_0_with_eval_v2_yx.py
  3. 3 1
      sug_v5_3.py

+ 3 - 1
sug_v5_0.py

@@ -23,7 +23,8 @@ class RunContext(BaseModel):
     log_dir: str
     question_annotation: str | None = Field(default=None, description="问题的标注结果")
     operations_history: list[dict] = Field(default_factory=list, description="记录所有操作的历史")
-    final_output: str | None = Field(default=None, description="最终输出结果")
+    optimization_result: dict | None = Field(default=None, description="最终优化结果对象")
+    final_output: str | None = Field(default=None, description="最终输出结果(格式化文本)")
 
 
 # ============================================================================
@@ -697,6 +698,7 @@ async def main(input_dir: str, max_rounds: int = 20):
     print(final_output)
 
     # 保存结果
+    run_context.optimization_result = optimization_result
     run_context.final_output = final_output
 
     # 保存 RunContext 到 log_dir

+ 3 - 1
sug_v5_0_with_eval_v2_yx.py

@@ -23,7 +23,8 @@ class RunContext(BaseModel):
     log_dir: str
     question_annotation: str | None = Field(default=None, description="问题的标注结果")
     operations_history: list[dict] = Field(default_factory=list, description="记录所有操作的历史")
-    final_output: str | None = Field(default=None, description="最终输出结果")
+    optimization_result: dict | None = Field(default=None, description="最终优化结果对象")
+    final_output: str | None = Field(default=None, description="最终输出结果(格式化文本)")
 
 
 # ============================================================================
@@ -934,6 +935,7 @@ async def main(input_dir: str, max_rounds: int = 20):
     print(final_output)
 
     # 保存结果
+    run_context.optimization_result = optimization_result
     run_context.final_output = final_output
 
     # 保存 RunContext 到 log_dir

+ 3 - 1
sug_v5_3.py

@@ -23,7 +23,8 @@ class RunContext(BaseModel):
     log_dir: str
     question_annotation: str | None = Field(default=None, description="问题的标注结果")
     operations_history: list[dict] = Field(default_factory=list, description="记录所有操作的历史")
-    final_output: str | None = Field(default=None, description="最终输出结果")
+    optimization_result: dict | None = Field(default=None, description="最终优化结果对象")
+    final_output: str | None = Field(default=None, description="最终输出结果(格式化文本)")
 
 
 # ============================================================================
@@ -697,6 +698,7 @@ async def main(input_dir: str, max_rounds: int = 20):
     print(final_output)
 
     # 保存结果
+    run_context.optimization_result = optimization_result
     run_context.final_output = final_output
 
     # 保存 RunContext 到 log_dir