feat: Handle exceptions in tactic generation

This commit is contained in:
Leni Aniva 2024-10-11 22:51:07 -07:00
parent 9fc035d466
commit 8196493258
Signed by: aniva
GPG Key ID: 4D9B1C8D10EA4C50
1 changed files with 20 additions and 16 deletions

View File

@ -58,6 +58,7 @@ class LLMAgent(Agent):
new_state = None
for ii in range(self.n_trials):
print(f"===============trail {str(ii)}============")
try:
s = select_tactic.run(
server=self.server,
state=state,
@ -75,6 +76,9 @@ class LLMAgent(Agent):
print(colored("[Tactic] Failed:", "red"), tactic)
return None
return tactic
except Exception as e:
print(colored(str(e), "red"))
return None
return None
else:
self.goal_tactic_id_map[key] = i + 1