Disable ask_user for testing or benchmarking runs

This commit is contained in:
Reinier van der Leer
2023-08-24 16:20:35 +02:00
parent 4bcb39dc14
commit 055bbc10d4
4 changed files with 4 additions and 0 deletions

View File

@@ -25,6 +25,7 @@ def bootstrap_agent(task: str, continuous_mode: bool) -> Agent:
config.debug_mode = True
config.continuous_mode = continuous_mode
config.temperature = 0
config.noninteractive_mode = True
config.plain_output = True
config.memory_backend = "no_memory"
config.workspace_path = Workspace.init_workspace_directory(config)

View File

@@ -23,6 +23,7 @@ from autogpt.command_decorator import command
"required": True,
}
},
enabled=lambda config: not config.noninteractive_mode
)
def ask_user(question: str, agent: Agent) -> str:
resp = clean_input(agent.config, f"{agent.ai_config.ai_name} asks: '{question}': ")

View File

@@ -37,6 +37,7 @@ class Config(SystemSettings, arbitrary_types_allowed=True):
exit_key: str = "n"
debug_mode: bool = False
plain_output: bool = False
noninteractive_mode: bool = False
chat_messages_enabled: bool = True
# TTS configuration
speak_mode: bool = False

View File

@@ -57,6 +57,7 @@ def config(
config.plugins_dir = "tests/unit/data/test_plugins"
config.plugins_config_file = temp_plugins_config_file
config.noninteractive_mode = True
config.plain_output = True
# avoid circular dependency