mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-19 06:54:22 +01:00
5de1025520
* Update Python version and benchmark file in benchmark.yml * Refactor main function and imports in cli.py * Update import statement in ai_config.py * Add set_temperature and set_memory_backend methods in config.py * Remove unused import in prompt.py * Add goal oriented tasks workflow * Added agent_utils to create agent * added pytest and vcrpy * added write file cassette * created goal oriented task write file with cassettes to not pay openai tokens * solve conflicts * add ability set azure because github workflow needs it off * solve conflicts in cli.py * black because linter fails * solve conflict * setup github action to v3 Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com> * fix conflicts Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com> * Plugins: debug line always printed in plugin load * add decorator to tests Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com> * move decorator higher up Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com> * init * more tests * passing tests * skip gitbranch decorator on ci * decorator skiponci * black * Update tests/utils.py decorator of skipping ci Co-authored-by: Nicholas Tindle <nicktindle@outlook.com> * black * I oopsed the name * black * finally * simple tests for agent and manager * ísort --------- Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com> Co-authored-by: Merwane Hamadi <merwanehamadi@gmail.com> Co-authored-by: Merwane Hamadi <merwane.hamadi@redica.com> Co-authored-by: Richard Beales <rich@richbeales.net> Co-authored-by: Nicholas Tindle <nick@ntindle.com> Co-authored-by: BillSchumacher <34168009+BillSchumacher@users.noreply.github.com> Co-authored-by: Nicholas Tindle <nicktindle@outlook.com>
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from autogpt.agent import Agent
|
|
from autogpt.chat import chat_with_ai
|
|
from autogpt.config import Config
|
|
from autogpt.speech import say_text
|
|
from autogpt.utils import clean_input
|
|
|
|
|
|
@pytest.fixture
|
|
def agent():
|
|
ai_name = "Test AI"
|
|
memory = MagicMock()
|
|
full_message_history = []
|
|
next_action_count = 0
|
|
command_registry = MagicMock()
|
|
config = Config()
|
|
system_prompt = "System prompt"
|
|
triggering_prompt = "Triggering prompt"
|
|
workspace_directory = "workspace_directory"
|
|
|
|
agent = Agent(
|
|
ai_name,
|
|
memory,
|
|
full_message_history,
|
|
next_action_count,
|
|
command_registry,
|
|
config,
|
|
system_prompt,
|
|
triggering_prompt,
|
|
workspace_directory,
|
|
)
|
|
return agent
|
|
|
|
|
|
def test_agent_initialization(agent):
|
|
assert agent.ai_name == "Test AI"
|
|
assert agent.memory == agent.memory
|
|
assert agent.full_message_history == []
|
|
assert agent.next_action_count == 0
|
|
assert agent.command_registry == agent.command_registry
|
|
assert agent.config == agent.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
|