mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-04-08 03:00:28 -04:00
7cb4d4a903
Update Pydantic dependency of `autogpt`, `forge` and `benchmark` to `^2.7` [Pydantic Migration Guide](https://docs.pydantic.dev/2.7/migration/) - Migrate usages of now-deprecated functions to their replacements - Update `Field` definitions - Ellipsis `...` for required fields is deprecated - `Field` no longer supports extra `kwargs`, replace use of this feature with field metadata - Replace `Config` class for specifying model configuration with `model_config = ConfigDict(..)` - Removed `ModelContainer` in `BaseAgent`, component configuration dict is now directly serialized using Pydantic v2 helper functions - Forked `agent-protocol` and updated `packages/client/python` for Pydantic v2 support: https://github.com/Significant-Gravitas/agent-protocol --------- Co-authored-by: Reinier van der Leer <pwuts@agpt.co>
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
from forge.config.ai_profile import AIProfile
|
|
from forge.file_storage import FileStorageBackendName, get_storage
|
|
from forge.llm.providers import MultiProvider
|
|
|
|
from autogpt.agents.agent import Agent, AgentConfiguration, AgentSettings
|
|
from autogpt.app.config import AppConfig
|
|
|
|
|
|
@pytest.fixture
|
|
def dummy_agent(config: AppConfig, llm_provider: MultiProvider):
|
|
ai_profile = AIProfile(
|
|
ai_name="Dummy Agent",
|
|
ai_role="Dummy Role",
|
|
ai_goals=[
|
|
"Dummy Task",
|
|
],
|
|
)
|
|
|
|
agent_settings = AgentSettings(
|
|
name=Agent.default_settings.name,
|
|
description=Agent.default_settings.description,
|
|
ai_profile=ai_profile,
|
|
config=AgentConfiguration(
|
|
fast_llm=config.fast_llm,
|
|
smart_llm=config.smart_llm,
|
|
use_functions_api=config.openai_functions,
|
|
),
|
|
history=Agent.default_settings.history.model_copy(deep=True),
|
|
)
|
|
|
|
local = config.file_storage_backend == FileStorageBackendName.LOCAL
|
|
restrict_to_root = not local or config.restrict_to_workspace
|
|
file_storage = get_storage(
|
|
config.file_storage_backend,
|
|
root_path=Path("data"),
|
|
restrict_to_root=restrict_to_root,
|
|
)
|
|
file_storage.initialize()
|
|
|
|
agent = Agent(
|
|
settings=agent_settings,
|
|
llm_provider=llm_provider,
|
|
file_storage=file_storage,
|
|
app_config=config,
|
|
)
|
|
|
|
return agent
|