Skip to content

Commit 7b3bceb

Browse files
committed
prevent llm written code to be registered if it got runtime error
1 parent 17c3ed4 commit 7b3bceb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

amadeusgpt/programs/sandbox.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -512,7 +512,7 @@ def llm_step(self, user_query):
512512

513513
# task program that is written by llm is automatically registered to be used in the future
514514

515-
if qa_message['code'] is not None:
515+
if qa_message['code'] is not None and qa_message['error_message'] is None:
516516
TaskProgramLibrary.register_task_program(creator="llm")(qa_message['code'])
517517

518518
return qa_message

0 commit comments

Comments
 (0)