mirrors/hermes-agent
1
0
Fork 0
mirror of https://github.com/NousResearch/hermes-agent.git synced 2026-05-08 03:01:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

Fix WhatsApp long message splitting

Browse source
This commit is contained in:
Hedirman 2026-05-04 06:54:18 +08:00 • committed by Teknium
parent 4d32f40306
commit a9ebee5f02
3 changed files with 122 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

33
tests/gateway/test_whatsapp_formatting.py
View file

@ -145,6 +145,21 @@ class TestMessageLimits:
from gateway.platforms.whatsapp import WhatsAppAdapter
assert WhatsAppAdapter.MAX_MESSAGE_LENGTH == 4096
def test_chunk_limit_reserves_default_self_chat_prefix(self, monkeypatch):
adapter = _make_adapter()
monkeypatch.delenv("WHATSAPP_REPLY_PREFIX", raising=False)
monkeypatch.setenv("WHATSAPP_MODE", "self-chat")
assert adapter._outgoing_chunk_limit() == (
adapter.MAX_MESSAGE_LENGTH - len(adapter.DEFAULT_REPLY_PREFIX)
)
def test_chunk_limit_does_not_reserve_prefix_in_bot_mode(self, monkeypatch):
adapter = _make_adapter()
monkeypatch.setenv("WHATSAPP_MODE", "bot")
assert adapter._outgoing_chunk_limit() == adapter.MAX_MESSAGE_LENGTH
# ---------------------------------------------------------------------------
# send() chunking tests
@ -180,6 +195,24 @@ class TestSendChunking:
# Should have made multiple calls
assert adapter._http_session.post.call_count > 1
@pytest.mark.asyncio
async def test_chunks_leave_room_for_bridge_prefix(self, monkeypatch):
adapter = _make_adapter()
monkeypatch.delenv("WHATSAPP_REPLY_PREFIX", raising=False)
monkeypatch.setenv("WHATSAPP_MODE", "self-chat")
resp = MagicMock(status=200)
resp.json = AsyncMock(return_value={"messageId": "msg1"})
adapter._http_session.post = MagicMock(return_value=_AsyncCM(resp))
long_msg = "a " * 3000
await adapter.send("chat1", long_msg)
for call in adapter._http_session.post.call_args_list:
payload = call.kwargs.get("json") or call[1].get("json")
final_text = adapter.DEFAULT_REPLY_PREFIX + payload["message"]
assert len(final_text) <= adapter.MAX_MESSAGE_LENGTH
@pytest.mark.asyncio
async def test_empty_message_no_send(self):
adapter = _make_adapter()