aniva/Pantograph
aniva
/
Pantograph
1
1
Fork 0
Code Issues 13 Pull Requests 4 Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
experiments/minif2f/model/llm_agent.py
View File

@ -58,23 +58,27 @@ class LLMAgent(Agent):
new_state = None
for ii in range(self.n_trials):
print(f"===============trail {str(ii)}============")
s = select_tactic.run(
server=self.server,
state=state,
goal_id=goal_id,
informal_stmt=self.informal_stmt,
informal_proof=self.informal_proof,
feedback_turns=self.feedback_turns)
tactic, new_state = s.ret_value
for m in s.messages():
print(m["role"], ":", m["content"])
try:
s = select_tactic.run(
server=self.server,
state=state,
goal_id=goal_id,
informal_stmt=self.informal_stmt,
informal_proof=self.informal_proof,
feedback_turns=self.feedback_turns)
tactic, new_state = s.ret_value
for m in s.messages():
print(m["role"], ":", m["content"])
print("\n-- new state --\n", new_state)
if tactic:
if not isinstance(tactic, Tactic):
print(colored("[Tactic] Failed:", "red"), tactic)
return None
return tactic
print("\n-- new state --\n", new_state)
if tactic:
if not isinstance(tactic, Tactic):
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