apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-27 17:56:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

use memory index to get the last user message

Browse source
This commit is contained in:
seehi 2024-09-10 20:05:56 +08:00
parent 5d21d255e4
commit d077cd0b2f
4 changed files with 16 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/metagpt/roles/di/test_role_zero.py
View file

@ -101,7 +101,7 @@ class TestRoleZero:
mock_role_zero.rc.memory.get.assert_called_once_with(really_k)
if k != 0:
mock_role_zero._should_use_longterm_memory.assert_called_once_with(k=really_k, k_memories=memories)
mock_role_zero._should_use_longterm_memory.assert_called_once_with(k=really_k)
if should_use_ltm:
mock_role_zero.longterm_memory.fetch.assert_called_once_with("user")
@ -174,18 +174,6 @@ class TestRoleZero:
mock_role_zero.rc.memory.get_by_position.assert_any_call(-(mock_role_zero.memory_k + 1))
mock_role_zero.rc.memory.get_by_position.assert_any_call(-(mock_role_zero.memory_k + 2))
@pytest.mark.parametrize(
"memories,expected",
[
([UserMessage(content="user")], True),
([AIMessage(content="ai")], False),
([], False),
],
)
def test_is_last_message_from_user(self, mock_role_zero: RoleZero, memories, expected):
result = mock_role_zero._is_last_message_from_user(memories)
assert result == expected
@pytest.mark.parametrize(
"memories,expected",
[