aboutsummaryrefslogtreecommitdiff
path: root/tests/integration/agent_factory.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/integration/agent_factory.py')
-rw-r--r--tests/integration/agent_factory.py47
1 files changed, 0 insertions, 47 deletions
diff --git a/tests/integration/agent_factory.py b/tests/integration/agent_factory.py
deleted file mode 100644
index 664c6cbb4..000000000
--- a/tests/integration/agent_factory.py
+++ /dev/null
@@ -1,47 +0,0 @@
-import pytest
-
-from autogpt.agent import Agent
-from autogpt.config import AIConfig, Config
-from autogpt.memory.vector import get_memory
-from autogpt.models.command_registry import CommandRegistry
-from autogpt.workspace import Workspace
-
-
-@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, memory_json_file, workspace: Workspace):
- command_registry = CommandRegistry()
-
- ai_config = AIConfig(
- ai_name="Dummy Agent",
- ai_role="Dummy Role",
- ai_goals=[
- "Dummy Task",
- ],
- )
- ai_config.command_registry = command_registry
-
- agent = Agent(
- ai_name="Dummy Agent",
- memory=memory_json_file,
- command_registry=command_registry,
- ai_config=ai_config,
- config=config,
- next_action_count=0,
- system_prompt="dummy_prompt",
- triggering_prompt="dummy triggering prompt",
- workspace_directory=workspace.root,
- )
-
- return agent