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

reform spo optimizer

Browse source
This commit is contained in:
xiangjinyu 2025-02-12 11:52:01 +08:00
parent 71b02a0047
commit 4417d805de
1 changed files with 6 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -42,19 +42,17 @@ class PromptOptimizer:
for opt_round in range(self.max_rounds):
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
prompt = loop.run_until_complete(self._optimize_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()
asyncio.set_event_loop(loop)
prompt = loop.run_until_complete(self._test_prompt())
# logger.info(f"Prompt generated in round {self.round}: {prompt}")
loop.run_until_complete(self._test_prompt())
def show_final_result(self):
best_round = self.data_utils.get_best_round()
logger.info("\n" + "=" * 50)
@ -70,7 +68,6 @@ class PromptOptimizer:
data = self.data_utils.load_results(prompt_path)
if self.round == 1:
logger.info("\n⚡ RUNNING Round 1 PROMPT ⚡\n")
directory = self.prompt_utils.create_round_directory(prompt_path, self.round)
@ -91,10 +88,9 @@ class PromptOptimizer:
samples = self.data_utils.get_best_round()
logger.info(f"\n🚀Round {self.round+1} OPTIMIZATION STARTING 🚀\n")
logger.info(f"\n🚀Round {self.round + 1} OPTIMIZATION STARTING 🚀\n")
logger.info(f"\nSelecting prompt for round {samples['round']} and advancing to the iteration phase\n")
golden_answer = self.data_utils.list_to_markdown(qa)
best_answer = self.data_utils.list_to_markdown(samples["answers"])
@ -112,7 +108,7 @@ class PromptOptimizer:
modification = extract_content(response, "modification")
logger.info(f"Modification of {self.round+1} round: {modification}")
logger.info(f"Modification of {self.round + 1} round: {modification}")
prompt = extract_content(response, "prompt")
@ -122,7 +118,7 @@ class PromptOptimizer:
self.prompt = ""
logger.info("\n🎯 NEW PROMPT GENERATED 🎯\n")
logger.info(f"\nRound {self.round+1} Prompt: {self.prompt}\n")
logger.info(f"\nRound {self.round + 1} Prompt: {self.prompt}\n")
self.prompt_utils.write_prompt(directory, prompt=self.prompt)