34 lines
913 B
Python
34 lines
913 B
Python
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"
|