mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-02-03 19:35:15 -05:00
08612cc3bf
- Move `autogpt/tests/vcr_cassettes` submodule to `forge/tests/vcr_cassettes` - Remove not needed markers from `pyproject.toml`: `"requires_openai_api_key", "requires_huggingface_api_key"` - Update relevant GitHub workflows Moved relevant tests from `autogpt/tests` to appropiate directories: - Component tests to their respective component dirs - `autogpt/tests/unit/test_web_search.py` → `forge/components/web/test_search.py` - `autogpt/tests/unit/test_git_commands.py` → `forge/components/git_operations/test_git_operations.py` - `autogpt/tests/unit/test_file_operations.py` → `forge/components/file_manager/test_file_manager.py` - `autogpt/tests/integration/test_image_gen.py` → `forge/components/image_gen/test_image_gen.py` - `autogpt/tests/integration/test_web_selenium.py` → `forge/components/web/test_selenium.py` - `autogpt/tests/integration/test_execute_code.py` → `forge/components/code_executor/test_code_executor.py` - `autogpt/tests/unit/test_s3_file_storage.py` → `forge/file_storage/test_s3_file_storage.py` - `autogpt/tests/unit/test_gcs_file_storage.py` → `forge/file_storage/test_gcs_file_storage.py` - `autogpt/tests/unit/test_local_file_storage.py` → `forge/file_storage/test_local_file_storage.py` - `autogpt/tests/unit/test_json.py` → `forge/json/test_parsing.py` - `autogpt/tests/unit/test_logs.py` → `forge/logging/test_utils.py` - `autogpt/tests/unit/test_url_validation.py` → `forge/utils/test_url_validator.py` - `autogpt/tests/unit/test_text_file_parsers.py` → `forge/utils/test_file_operations.py` - (Re)moved dependencies from `autogpt/pyproject.toml` that were only used in these test files. Also: - Added `load_env_vars` fixture to `forge/conftest.py` - Fixed a type error in `forge/components/web/test_search.py` - Merged `autogpt/.gitattributes` into root `.gitattributes` --------- Co-authored-by: Reinier van der Leer <pwuts@agpt.co>
108 lines
2.6 KiB
Python
108 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import uuid
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from forge.config.ai_profile import AIProfile
|
|
from forge.file_storage.local import (
|
|
FileStorage,
|
|
FileStorageConfiguration,
|
|
LocalFileStorage,
|
|
)
|
|
from forge.llm.providers import MultiProvider
|
|
from forge.logging.config import configure_logging
|
|
|
|
from autogpt.agents.agent import Agent, AgentConfiguration, AgentSettings
|
|
from autogpt.app.config import AppConfig, ConfigBuilder
|
|
from autogpt.app.main import _configure_llm_provider
|
|
|
|
pytest_plugins = [
|
|
"tests.integration.agent_factory",
|
|
]
|
|
|
|
|
|
@pytest.fixture()
|
|
def tmp_project_root(tmp_path: Path) -> Path:
|
|
return tmp_path
|
|
|
|
|
|
@pytest.fixture()
|
|
def app_data_dir(tmp_project_root: Path) -> Path:
|
|
dir = tmp_project_root / "data"
|
|
dir.mkdir(parents=True, exist_ok=True)
|
|
return dir
|
|
|
|
|
|
@pytest.fixture()
|
|
def storage(app_data_dir: Path) -> FileStorage:
|
|
storage = LocalFileStorage(
|
|
FileStorageConfiguration(root=app_data_dir, restrict_to_root=False)
|
|
)
|
|
storage.initialize()
|
|
return storage
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def config(
|
|
tmp_project_root: Path,
|
|
app_data_dir: Path,
|
|
):
|
|
if not os.environ.get("OPENAI_API_KEY"):
|
|
os.environ["OPENAI_API_KEY"] = "sk-dummy"
|
|
config = ConfigBuilder.build_config_from_env(project_root=tmp_project_root)
|
|
|
|
config.app_data_dir = app_data_dir
|
|
|
|
config.noninteractive_mode = True
|
|
|
|
yield config
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def setup_logger():
|
|
configure_logging(
|
|
debug=True,
|
|
log_dir=Path(__file__).parent / "logs",
|
|
plain_console_output=True,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def llm_provider(config: AppConfig) -> MultiProvider:
|
|
return _configure_llm_provider(config)
|
|
|
|
|
|
@pytest.fixture
|
|
def agent(
|
|
config: AppConfig, llm_provider: MultiProvider, storage: FileStorage
|
|
) -> Agent:
|
|
ai_profile = AIProfile(
|
|
ai_name="Base",
|
|
ai_role="A base AI",
|
|
ai_goals=[],
|
|
)
|
|
|
|
agent_settings = AgentSettings(
|
|
name=Agent.default_settings.name,
|
|
description=Agent.default_settings.description,
|
|
agent_id=f"AutoGPT-test-agent-{str(uuid.uuid4())[:8]}",
|
|
ai_profile=ai_profile,
|
|
config=AgentConfiguration(
|
|
fast_llm=config.fast_llm,
|
|
smart_llm=config.smart_llm,
|
|
allow_fs_access=not config.restrict_to_workspace,
|
|
use_functions_api=config.openai_functions,
|
|
),
|
|
history=Agent.default_settings.history.model_copy(deep=True),
|
|
)
|
|
|
|
agent = Agent(
|
|
settings=agent_settings,
|
|
llm_provider=llm_provider,
|
|
file_storage=storage,
|
|
app_config=config,
|
|
)
|
|
return agent
|