From a727e17156568b8519ec0f1daa66f5aaf867c506 Mon Sep 17 00:00:00 2001 From: cpacker Date: Thu, 19 Oct 2023 12:52:04 -0700 Subject: [PATCH] allow bypassing message check --- main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 76cc6ed9d7..cb3a12b8f0 100644 --- a/main.py +++ b/main.py @@ -25,6 +25,7 @@ flags.DEFINE_string("model", default=constants.DEFAULT_MEMGPT_MODEL, required=False, help="Specify the LLM model") flags.DEFINE_boolean("first", default=False, required=False, help="Use -first to send the first message in the sequence") flags.DEFINE_boolean("debug", default=False, required=False, help="Use -debug to enable debugging output") +flags.DEFINE_boolean("no_verify", default=False, required=False, help="Bypass message verification") flags.DEFINE_string("archival_storage_faiss_path", default="", required=False, help="Specify archival storage with FAISS index to load (a folder with a .index and .json describing documents to be loaded)") flags.DEFINE_string("archival_storage_files", default="", required=False, help="Specify files to pre-load into archival memory (glob pattern)") flags.DEFINE_string("archival_storage_files_compute_embeddings", default="", required=False, help="Specify files to pre-load into archival memory (glob pattern), and compute embeddings over them") @@ -243,7 +244,7 @@ async def main(): skip_next_user_input = False with console.status("[bold cyan]Thinking...") as status: - new_messages, heartbeat_request, function_failed, token_warning = await memgpt_agent.step(user_message, first_message=False) + new_messages, heartbeat_request, function_failed, token_warning = await memgpt_agent.step(user_message, first_message=False, skip_verify=FLAGS.no_verify) # Skip user inputs if there's a memory warning, function execution failed, or the agent asked for control if token_warning: