aboutsummaryrefslogtreecommitdiff
path: root/benchmarks.py
diff options
context:
space:
mode:
Diffstat (limited to 'benchmarks.py')
-rw-r--r--benchmarks.py52
1 files changed, 0 insertions, 52 deletions
diff --git a/benchmarks.py b/benchmarks.py
deleted file mode 100644
index cb592be8a..000000000
--- a/benchmarks.py
+++ /dev/null
@@ -1,52 +0,0 @@
-from autogpt.agent import Agent
-from autogpt.config import AIConfig, Config, ConfigBuilder
-from autogpt.main import COMMAND_CATEGORIES
-from autogpt.memory.vector import get_memory
-from autogpt.models.command_registry import CommandRegistry
-from autogpt.prompts.prompt import DEFAULT_TRIGGERING_PROMPT
-from autogpt.workspace import Workspace
-
-
-def run_task(task) -> None:
- agent = bootstrap_agent(task)
- agent.start_interaction_loop()
-
-
-def bootstrap_agent(task):
- config = ConfigBuilder.build_config_from_env()
- config.continuous_mode = False
- config.temperature = 0
- config.plain_output = True
- command_registry = get_command_registry(config)
- config.memory_backend = "no_memory"
- workspace_directory = Workspace.get_workspace_directory(config)
- workspace_directory_path = Workspace.make_workspace(workspace_directory)
- Workspace.build_file_logger_path(config, workspace_directory_path)
- ai_config = AIConfig(
- ai_name="Auto-GPT",
- ai_role="a multi-purpose AI assistant.",
- ai_goals=[task.user_input],
- )
- ai_config.command_registry = command_registry
- system_prompt = ai_config.construct_full_prompt(config)
- return Agent(
- ai_name="Auto-GPT",
- memory=get_memory(config),
- command_registry=command_registry,
- ai_config=ai_config,
- config=config,
- next_action_count=0,
- system_prompt=system_prompt,
- triggering_prompt=DEFAULT_TRIGGERING_PROMPT,
- workspace_directory=str(workspace_directory_path),
- )
-
-
-def get_command_registry(config: Config):
- command_registry = CommandRegistry()
- enabled_command_categories = [
- x for x in COMMAND_CATEGORIES if x not in config.disabled_command_categories
- ]
- for command_category in enabled_command_categories:
- command_registry.import_commands(command_category)
- return command_registry