mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-14 04:02:26 +00:00
test: update send_message_tool mocks for force_document kwarg
This commit is contained in:
parent
d34f03c32a
commit
e82f3b0c41
1 changed files with 3 additions and 1 deletions
|
|
@ -140,6 +140,7 @@ class TestSendMessageTool:
|
||||||
"hello",
|
"hello",
|
||||||
thread_id="17585",
|
thread_id="17585",
|
||||||
media_files=[],
|
media_files=[],
|
||||||
|
force_document=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_display_label_target_resolves_via_channel_directory(self, tmp_path):
|
def test_display_label_target_resolves_via_channel_directory(self, tmp_path):
|
||||||
|
|
@ -178,6 +179,7 @@ class TestSendMessageTool:
|
||||||
"hello",
|
"hello",
|
||||||
thread_id="17585",
|
thread_id="17585",
|
||||||
media_files=[],
|
media_files=[],
|
||||||
|
force_document=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_mirror_receives_current_session_user_id(self):
|
def test_mirror_receives_current_session_user_id(self):
|
||||||
|
|
@ -483,7 +485,7 @@ class TestSendToPlatformChunking:
|
||||||
|
|
||||||
sent_calls = []
|
sent_calls = []
|
||||||
|
|
||||||
async def fake_send(token, chat_id, message, media_files=None, thread_id=None, disable_link_previews=False):
|
async def fake_send(token, chat_id, message, media_files=None, thread_id=None, disable_link_previews=False, force_document=False):
|
||||||
sent_calls.append(media_files or [])
|
sent_calls.append(media_files or [])
|
||||||
return {"success": True, "platform": "telegram", "chat_id": chat_id, "message_id": str(len(sent_calls))}
|
return {"success": True, "platform": "telegram", "chat_id": chat_id, "message_id": str(len(sent_calls))}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue