mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-19 15:04:26 +01:00
2c53530e99
* Fix and clean up path processing in logs module * Fix path processing throughout the project * Fix plugins test * Fix borky pytest vs mkdir(exist_ok=True) * Update docs and gitignore for new workspace location * Fix borky pytest vol.2 * ok james
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
from pathlib import Path
|
|
|
|
from autogpt.agents import Agent
|
|
from autogpt.config import AIConfig, Config, ConfigBuilder
|
|
from autogpt.main import COMMAND_CATEGORIES, run_interaction_loop
|
|
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()
|
|
|
|
|
|
def run_task(task) -> None:
|
|
agent = bootstrap_agent(task)
|
|
run_interaction_loop(agent)
|
|
|
|
|
|
def bootstrap_agent(task):
|
|
config = ConfigBuilder.build_config_from_env(workdir=PROJECT_DIR)
|
|
config.continuous_mode = False
|
|
config.temperature = 0
|
|
config.plain_output = True
|
|
command_registry = get_command_registry(config)
|
|
config.memory_backend = "no_memory"
|
|
Workspace.set_workspace_directory(config)
|
|
Workspace.build_file_logger_path(config, config.workspace_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
|
|
return Agent(
|
|
memory=get_memory(config),
|
|
command_registry=command_registry,
|
|
ai_config=ai_config,
|
|
config=config,
|
|
triggering_prompt=DEFAULT_TRIGGERING_PROMPT,
|
|
workspace_directory=str(config.workspace_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
|