mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-13 11:14:25 +01:00
3fe2246468
* Add categories to command registry * Fix tests * Clean up prompt generation * Rename Performance Evaluations to Best Practices * Move specification of response format from system prompt to Agent.construct_base_prompt * Clean up PromptGenerator class * Add debug logging to AIConfig autogeneration * Clarify prompting and add support for multiple thought processes to Agent * WIP: PlanningAgent * Disable message history by default on BaseAgent * Add CommandOutput and ThoughtProcessOutput type aliases * Fix interrupts in main.py * Use custom exceptions and clean up exception/error handling * Remove duplicate agent_history.py * Update PlanningAgent from upstream * WIP: Support for dynamic in-prompt context * WIP: response formats for PlanningAgent three-stage cycle * Remove browsing overlay & separate browsing from extraction code * Fix human feedback * Fix tests * Include history in Agent prompt generation * Code improvements in agent.py * Add ask_user command and revise system prompt
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
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.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_specific_agent(task, continuous_mode=False) -> Tuple[str, int]:
|
|
agent = bootstrap_agent(task, continuous_mode)
|
|
run_interaction_loop(agent)
|
|
|
|
|
|
def bootstrap_agent(task, continuous_mode) -> 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)
|
|
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,
|
|
ai_config=ai_config,
|
|
config=config,
|
|
triggering_prompt=DEFAULT_TRIGGERING_PROMPT,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# The first argument is the script name itself, second is the task
|
|
if len(sys.argv) != 2:
|
|
print("Usage: python script.py <task>")
|
|
sys.exit(1)
|
|
task = sys.argv[1]
|
|
run_specific_agent(task, continuous_mode=True)
|