47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
from autogpt.agent.agent import Agent
|
|
from autogpt.config import AIConfig
|
|
from autogpt.llm import create_chat_completion
|
|
|
|
|
|
def test_get_self_feedback(mocker):
|
|
# Define a sample thoughts dictionary
|
|
thoughts = {
|
|
"reasoning": "Sample reasoning.",
|
|
"plan": "Sample plan.",
|
|
"thoughts": "Sample thoughts.",
|
|
}
|
|
|
|
# Define a fake response for the create_chat_completion function
|
|
fake_response = (
|
|
"The AI Agent has demonstrated a reasonable thought process, but there is room for improvement. "
|
|
"For example, the reasoning could be elaborated to better justify the plan, and the plan itself "
|
|
"could be more detailed to ensure its effectiveness. In addition, the AI Agent should focus more "
|
|
"on its core role and prioritize thoughts that align with that role."
|
|
)
|
|
|
|
# Mock the create_chat_completion function
|
|
mock_create_chat_completion = mocker.patch(
|
|
"autogpt.agent.agent.create_chat_completion", wraps=create_chat_completion
|
|
)
|
|
mock_create_chat_completion.return_value = fake_response
|
|
|
|
# Create a MagicMock object to replace the Agent instance
|
|
agent_mock = mocker.MagicMock(spec=Agent)
|
|
|
|
# Mock the config attribute of the Agent instance
|
|
agent_mock.config = AIConfig()
|
|
|
|
# Call the get_self_feedback method
|
|
feedback = Agent.get_self_feedback(
|
|
agent_mock,
|
|
thoughts,
|
|
"gpt-3.5-turbo",
|
|
)
|
|
|
|
# Check if the response is a non-empty string
|
|
assert isinstance(feedback, str) and len(feedback) > 0
|
|
|
|
# Check if certain keywords from input thoughts are present in the feedback response
|
|
for keyword in ["reasoning", "plan", "thoughts"]:
|
|
assert keyword in feedback
|