From 714f0fc676bc6c7723b3f16bdb7c3cb3eb54bbd0 Mon Sep 17 00:00:00 2001 From: Reinier van der Leer Date: Thu, 24 Aug 2023 15:12:10 +0200 Subject: [PATCH] Fix logging output in benchmark --- agbenchmark/benchmarks.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/agbenchmark/benchmarks.py b/agbenchmark/benchmarks.py index 50caf90b..ff8ab877 100644 --- a/agbenchmark/benchmarks.py +++ b/agbenchmark/benchmarks.py @@ -1,40 +1,44 @@ import sys from pathlib import Path -from typing import Tuple from autogpt.agents import Agent from autogpt.app.main import run_interaction_loop from autogpt.commands import COMMAND_CATEGORIES -from autogpt.config import AIConfig, Config, ConfigBuilder +from autogpt.config import AIConfig, ConfigBuilder +from autogpt.logs.config import configure_logging 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 PROJECT_DIR = Path().resolve() +LOG_DIR = Path(__file__).parent / "logs" -def run_specific_agent(task, continuous_mode=False) -> Tuple[str, int]: +def run_specific_agent(task: str, continuous_mode: bool = False) -> None: agent = bootstrap_agent(task, continuous_mode) run_interaction_loop(agent) -def bootstrap_agent(task, continuous_mode) -> Agent: +def bootstrap_agent(task: str, continuous_mode: bool) -> Agent: config = ConfigBuilder.build_config_from_env(workdir=PROJECT_DIR) config.debug_mode = True config.continuous_mode = continuous_mode config.temperature = 0 config.plain_output = True - command_registry = CommandRegistry.with_command_modules(COMMAND_CATEGORIES, config) config.memory_backend = "no_memory" config.workspace_path = Workspace.init_workspace_directory(config) config.file_logger_path = Workspace.build_file_logger_path(config.workspace_path) + + configure_logging(config, LOG_DIR) + + command_registry = CommandRegistry.with_command_modules(COMMAND_CATEGORIES, config) + ai_config = AIConfig( ai_name="Auto-GPT", ai_role="a multi-purpose AI assistant.", ai_goals=[task], ) - ai_config.command_registry = command_registry return Agent( memory=get_memory(config), command_registry=command_registry,