mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 22:14:28 +01:00
a74548d3cd
This incremental re-architecture unifies Agent code and plugins, so everything is component-based. ## Breaking changes - Removed command categories and `DISABLED_COMMAND_CATEGORIES` environment variable. Use `DISABLED_COMMANDS` environment variable to disable individual commands. - Changed `command` decorator; old-style commands are no longer supported. Implement `CommandProvider` on components instead. - Removed `CommandRegistry`, now all commands are provided by components implementing `CommandProvider`. - Removed `prompt_config` from `AgentSettings`. - Removed plugin support: old plugins will no longer be loaded and executed. - Removed `PromptScratchpad`, it was used by plugins and is no longer needed. - Changed `ThoughtProcessOutput` from tuple to pydantic `BaseModel`. ## Other changes - Created `AgentComponent`, protocols and logic to execute them. - `BaseAgent` and `Agent` is now composed of components. - Moved some logic from `BaseAgent` to `Agent`. - Moved agent features and commands to components. - Removed check if the same operation is about to be executed twice in a row. - Removed file logging from `FileManagerComponent` (formerly `AgentFileManagerMixin`) - Updated tests - Added docs See [Introduction](https://github.com/kcze/AutoGPT/blob/kpczerwinski/open-440-modular-agents/docs/content/AutoGPT/component%20agent/introduction.md) for more information.
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
import pytest
|
|
from git.exc import GitCommandError
|
|
from git.repo.base import Repo
|
|
|
|
from autogpt.agents.agent import Agent
|
|
from autogpt.commands.git_operations import GitOperationsComponent
|
|
from autogpt.file_storage.base import FileStorage
|
|
from autogpt.utils.exceptions import CommandExecutionError
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_clone_from(mocker):
|
|
return mocker.patch.object(Repo, "clone_from")
|
|
|
|
|
|
@pytest.fixture
|
|
def git_ops_component(agent: Agent):
|
|
return agent.git_ops
|
|
|
|
|
|
def test_clone_auto_gpt_repository(
|
|
git_ops_component: GitOperationsComponent,
|
|
storage: FileStorage,
|
|
mock_clone_from,
|
|
agent: Agent,
|
|
):
|
|
mock_clone_from.return_value = None
|
|
|
|
repo = "github.com/Significant-Gravitas/Auto-GPT.git"
|
|
scheme = "https://"
|
|
url = scheme + repo
|
|
clone_path = storage.get_path("auto-gpt-repo")
|
|
|
|
expected_output = f"Cloned {url} to {clone_path}"
|
|
|
|
clone_result = git_ops_component.clone_repository(url, clone_path)
|
|
|
|
assert clone_result == expected_output
|
|
mock_clone_from.assert_called_once_with(
|
|
url=f"{scheme}{agent.legacy_config.github_username}:{agent.legacy_config.github_api_key}@{repo}", # noqa: E501
|
|
to_path=clone_path,
|
|
)
|
|
|
|
|
|
def test_clone_repository_error(
|
|
git_ops_component: GitOperationsComponent,
|
|
storage: FileStorage,
|
|
mock_clone_from,
|
|
agent: Agent,
|
|
):
|
|
url = "https://github.com/this-repository/does-not-exist.git"
|
|
clone_path = storage.get_path("does-not-exist")
|
|
|
|
mock_clone_from.side_effect = GitCommandError(
|
|
"clone", "fatal: repository not found", ""
|
|
)
|
|
|
|
with pytest.raises(CommandExecutionError):
|
|
git_ops_component.clone_repository(url, clone_path)
|