mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-28 10:24:26 +01:00
34352afd53
* Remove references to (broken) vector memory * Move workspace setup to `WorkspaceMixin.attach_fs` hook * Move directives into `BaseAgentSettings`
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
import pytest
|
|
|
|
from autogpt.agents.agent import Agent, AgentConfiguration, AgentSettings
|
|
from autogpt.config import AIProfile, Config
|
|
from autogpt.memory.vector import get_memory
|
|
from autogpt.models.command_registry import CommandRegistry
|
|
|
|
|
|
@pytest.fixture
|
|
def memory_json_file(config: Config):
|
|
was_memory_backend = config.memory_backend
|
|
|
|
config.memory_backend = "json_file"
|
|
memory = get_memory(config)
|
|
memory.clear()
|
|
yield memory
|
|
|
|
config.memory_backend = was_memory_backend
|
|
|
|
|
|
@pytest.fixture
|
|
def dummy_agent(config: Config, llm_provider, memory_json_file):
|
|
command_registry = CommandRegistry()
|
|
|
|
ai_profile = AIProfile(
|
|
ai_name="Dummy Agent",
|
|
ai_role="Dummy Role",
|
|
ai_goals=[
|
|
"Dummy Task",
|
|
],
|
|
)
|
|
|
|
agent_prompt_config = Agent.default_settings.prompt_config.copy(deep=True)
|
|
agent_prompt_config.use_functions_api = config.openai_functions
|
|
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,
|
|
plugins=config.plugins,
|
|
),
|
|
prompt_config=agent_prompt_config,
|
|
history=Agent.default_settings.history.copy(deep=True),
|
|
)
|
|
|
|
agent = Agent(
|
|
settings=agent_settings,
|
|
llm_provider=llm_provider,
|
|
command_registry=command_registry,
|
|
legacy_config=config,
|
|
)
|
|
|
|
return agent
|