apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-25 00:36:55 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

modify spo logger show final result

Browse source
This commit is contained in:
xiangjinyu 2025-02-11 18:50:59 +08:00
parent af3862fc59
commit 3ca2c1756d
2 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
metagpt/ext/spo/app.py
View file

@ -81,14 +81,14 @@ def main():
eval_model = st.selectbox(
"Evaluation Model",
["claude-3-5-sonnet-20240620", "gpt-4o", "gpt-4o-mini", "deepseek-chat"],
["gpt-4o-mini", "claude-3-5-sonnet-20240620", "gpt-4o", "deepseek-chat"],
index=0
)
eval_temp = st.slider("Evaluation Temperature", 0.0, 1.0, 0.3)
exec_model = st.selectbox(
"Execution Model",
["claude-3-5-sonnet-20240620", "gpt-4o", "gpt-4o-mini", "deepseek-chat"],
["gpt-4o-mini", "claude-3-5-sonnet-20240620", "gpt-4o", "deepseek-chat"],
index=0
)
exec_temp = st.slider("Execution Temperature", 0.0, 1.0, 0.0)

14
metagpt/ext/spo/components/optimizer.py
View file

@ -45,6 +45,7 @@ class PromptOptimizer:
prompt = loop.run_until_complete(self._optimize_prompt())
self.round += 1
# logger.info(f"Prompt generated in round {self.round}: {prompt}")
self.show_final_result()
else:
loop = asyncio.new_event_loop()
@ -52,6 +53,16 @@ class PromptOptimizer:
prompt = loop.run_until_complete(self._test_prompt())
# logger.info(f"Prompt generated in round {self.round}: {prompt}")
def show_final_result(self):
best_round = self.data_utils.get_best_round()
logger.info("\n" + "=" * 50)
logger.info("\n🏆 OPTIMIZATION COMPLETED - FINAL RESULTS 🏆\n")
logger.info(f"\n📌 Best Performing Round: {best_round['round']}")
logger.info(f"\n🎯 Final Optimized Prompt:\n{best_round['prompt']}")
logger.info("\n" + "=" * 50 + "\n")
async def _optimize_prompt(self):
prompt_path = f"{self.root_path}/prompts"
load.set_file_name(self.template)
@ -130,9 +141,6 @@ class PromptOptimizer:
logger.info("\n🎯 OPTIMIZATION RESULT 🎯\n")
logger.info(f"\nRound {self.round + 1} Optimization: {'✅ SUCCESS' if success else '❌ FAILED'}\n")
# logger.info(f"now is {self.round + 1}")
return prompt
async def _test_prompt(self):