mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-24 09:24:27 +01:00
6b9e3b21d3
* Add config as attribute to Agent, rename old config to ai_config * Code review: Pass ai_config --------- Co-authored-by: Nicholas Tindle <nick@ntindle.com> Co-authored-by: merwanehamadi <merwanehamadi@gmail.com>
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from autogpt.agent import Agent
|
|
from autogpt.config import AIConfig
|
|
from autogpt.config.config import Config
|
|
|
|
|
|
@pytest.fixture
|
|
def agent(config: Config):
|
|
ai_name = "Test AI"
|
|
memory = MagicMock()
|
|
next_action_count = 0
|
|
command_registry = MagicMock()
|
|
ai_config = AIConfig(ai_name=ai_name)
|
|
system_prompt = "System prompt"
|
|
triggering_prompt = "Triggering prompt"
|
|
workspace_directory = "workspace_directory"
|
|
|
|
agent = Agent(
|
|
ai_name=ai_name,
|
|
memory=memory,
|
|
next_action_count=next_action_count,
|
|
command_registry=command_registry,
|
|
ai_config=ai_config,
|
|
config=config,
|
|
system_prompt=system_prompt,
|
|
triggering_prompt=triggering_prompt,
|
|
workspace_directory=workspace_directory,
|
|
)
|
|
return agent
|
|
|
|
|
|
def test_agent_initialization(agent: Agent):
|
|
assert agent.ai_name == "Test AI"
|
|
assert agent.memory == agent.memory
|
|
assert agent.history.messages == []
|
|
assert agent.next_action_count == 0
|
|
assert agent.command_registry == agent.command_registry
|
|
assert agent.ai_config == agent.ai_config
|
|
assert agent.system_prompt == "System prompt"
|
|
assert agent.triggering_prompt == "Triggering prompt"
|
|
|
|
|
|
# More test methods can be added for specific agent interactions
|
|
# For example, mocking chat_with_ai and testing the agent's interaction loop
|