Simplify to memory system api
This commit is contained in:
33
tests/test_memory_system_clients.py
Normal file
33
tests/test_memory_system_clients.py
Normal file
@ -0,0 +1,33 @@
|
||||
from memory_system_api.clients import EverOSMemorySystemClient
|
||||
|
||||
|
||||
def test_everos_assistant_payload_does_not_use_user_id_as_sender():
|
||||
client = EverOSMemorySystemClient()
|
||||
|
||||
payload = client.build_message_payload(
|
||||
user_id="tom",
|
||||
session_id="sess-1",
|
||||
role="assistant",
|
||||
content="我记住了",
|
||||
)
|
||||
|
||||
message = payload["messages"][0]
|
||||
assert message["role"] == "assistant"
|
||||
assert message["sender_id"] != "tom"
|
||||
assert message["sender_name"] != "tom"
|
||||
|
||||
|
||||
def test_everos_user_payload_uses_user_id_as_sender():
|
||||
client = EverOSMemorySystemClient()
|
||||
|
||||
payload = client.build_message_payload(
|
||||
user_id="tom",
|
||||
session_id="sess-1",
|
||||
role="user",
|
||||
content="我喜欢拿铁",
|
||||
)
|
||||
|
||||
message = payload["messages"][0]
|
||||
assert message["role"] == "user"
|
||||
assert message["sender_id"] == "tom"
|
||||
assert message["sender_name"] == "tom"
|
||||
Reference in New Issue
Block a user