fix(feishu): respect reply_to_message for group threads

This commit is contained in:
hanyuanling 2026-04-30 11:18:34 +08:00 committed by Xubin Ren
parent 26e953f0b9
commit d82f25e4d4
3 changed files with 167 additions and 28 deletions

View File

@ -1170,6 +1170,10 @@ class FeishuChannel(BaseChannel):
logger.error("Error replying to Feishu message {}: {}", parent_message_id, e) logger.error("Error replying to Feishu message {}: {}", parent_message_id, e)
return False return False
def _should_use_reply_in_thread(self, metadata: dict[str, Any]) -> bool:
"""Return whether a group reply should create a Feishu thread/topic."""
return metadata.get("chat_type", "group") == "group" and self.config.reply_to_message
def _send_message_sync( def _send_message_sync(
self, receive_id_type: str, receive_id: str, msg_type: str, content: str self, receive_id_type: str, receive_id: str, msg_type: str, content: str
) -> str | None: ) -> str | None:
@ -1211,13 +1215,15 @@ class FeishuChannel(BaseChannel):
receive_id_type: str, receive_id_type: str,
chat_id: str, chat_id: str,
reply_message_id: str | None = None, reply_message_id: str | None = None,
*,
reply_in_thread: bool = False,
) -> str | None: ) -> str | None:
"""Create a CardKit streaming card, send it to chat, return card_id. """Create a CardKit streaming card, send it to chat, return card_id.
When *reply_message_id* is provided the card is delivered via the When *reply_message_id* is provided the card is delivered via the
reply API (with reply_in_thread=True) so it lands inside the reply API. *reply_in_thread* controls whether Feishu creates a
originating thread / topic. Otherwise the plain create-message thread/topic for that reply. Otherwise the plain create-message API is
API is used. used.
""" """
from lark_oapi.api.cardkit.v1 import CreateCardRequest, CreateCardRequestBody from lark_oapi.api.cardkit.v1 import CreateCardRequest, CreateCardRequestBody
@ -1253,7 +1259,7 @@ class FeishuChannel(BaseChannel):
if reply_message_id: if reply_message_id:
sent = self._reply_message_sync( sent = self._reply_message_sync(
reply_message_id, "interactive", card_content, reply_message_id, "interactive", card_content,
reply_in_thread=True, reply_in_thread=reply_in_thread,
) )
else: else:
sent = self._send_message_sync( sent = self._send_message_sync(
@ -1409,11 +1415,10 @@ class FeishuChannel(BaseChannel):
{"config": {"wide_screen_mode": True}, "elements": chunk}, {"config": {"wide_screen_mode": True}, "elements": chunk},
ensure_ascii=False, ensure_ascii=False,
) )
# Fallback: reply via the Reply API for group chats. # Fallback: only create a group thread/topic when reply-to-message
# Target message_id — the Feishu API keeps the reply in # is enabled. Otherwise use the plain create-message API.
# the same topic automatically.
_f_msg = meta.get("message_id") _f_msg = meta.get("message_id")
fallback_msg_id = _f_msg if meta.get("chat_type", "group") == "group" else None fallback_msg_id = _f_msg if self._should_use_reply_in_thread(meta) else None
if fallback_msg_id: if fallback_msg_id:
await loop.run_in_executor( await loop.run_in_executor(
None, lambda: self._reply_message_sync( None, lambda: self._reply_message_sync(
@ -1438,16 +1443,18 @@ class FeishuChannel(BaseChannel):
now = time.monotonic() now = time.monotonic()
if buf.card_id is None: if buf.card_id is None:
# Send the streaming card as a reply for group chats so it # Send the streaming card as a group thread/topic reply only when
# lands inside the originating topic/thread. Always target # reply-to-message is enabled.
# message_id (the actual inbound message) — the Feishu Reply use_reply_in_thread = self._should_use_reply_in_thread(meta)
# API keeps the response in the same topic automatically. reply_msg_id = meta.get("message_id") if use_reply_in_thread else None
is_group = meta.get("chat_type", "group") == "group"
reply_msg_id = meta.get("message_id") if is_group else None
card_id = await loop.run_in_executor( card_id = await loop.run_in_executor(
None, None,
self._create_streaming_card_sync, lambda: self._create_streaming_card_sync(
rid_type, chat_id, reply_msg_id, rid_type,
chat_id,
reply_msg_id,
reply_in_thread=use_reply_in_thread,
),
) )
if card_id: if card_id:
buf.card_id = card_id buf.card_id = card_id
@ -1489,9 +1496,9 @@ class FeishuChannel(BaseChannel):
"\n\n" + self._format_tool_hint_delta(hint) + "\n\n", "\n\n" + self._format_tool_hint_delta(hint) + "\n\n",
) )
return return
# No active streaming card — send as a regular # No active streaming card — send as a regular interactive card
# interactive card with the same 🔧 prefix style. # with the same 🔧 prefix style. Only create a group thread/topic
# Use reply API for group chats so the hint stays in topic. # when reply-to-message is enabled.
card = json.dumps( card = json.dumps(
{"config": {"wide_screen_mode": True}, "elements": [ {"config": {"wide_screen_mode": True}, "elements": [
{"tag": "markdown", "content": self._format_tool_hint_delta(hint)}, {"tag": "markdown", "content": self._format_tool_hint_delta(hint)},
@ -1499,8 +1506,7 @@ class FeishuChannel(BaseChannel):
ensure_ascii=False, ensure_ascii=False,
) )
_th_msg_id = msg.metadata.get("message_id") _th_msg_id = msg.metadata.get("message_id")
_th_chat_type = msg.metadata.get("chat_type", "group") if _th_msg_id and self._should_use_reply_in_thread(msg.metadata):
if _th_msg_id and _th_chat_type == "group":
await loop.run_in_executor( await loop.run_in_executor(
None, lambda: self._reply_message_sync( None, lambda: self._reply_message_sync(
_th_msg_id, "interactive", card, _th_msg_id, "interactive", card,
@ -1531,18 +1537,16 @@ class FeishuChannel(BaseChannel):
def _do_send(m_type: str, content: str) -> None: def _do_send(m_type: str, content: str) -> None:
"""Send via reply (first message) or create (subsequent). """Send via reply (first message) or create (subsequent).
For group chats the reply API always uses reply_in_thread=True. Group chats only set reply_in_thread=True when
The Feishu API automatically keeps replies inside existing reply_to_message is enabled; otherwise a Reply API call for an
topics reply_in_thread only creates a *new* topic when the existing topic must not create a new topic.
target message is a plain (non-topic) message.
""" """
nonlocal first_send nonlocal first_send
if reply_message_id and first_send: if reply_message_id and first_send:
first_send = False first_send = False
chat_type = msg.metadata.get("chat_type", "group")
ok = self._reply_message_sync( ok = self._reply_message_sync(
reply_message_id, m_type, content, reply_message_id, m_type, content,
reply_in_thread=chat_type == "group", reply_in_thread=self._should_use_reply_in_thread(msg.metadata),
) )
if ok: if ok:
return return

View File

@ -580,6 +580,32 @@ async def test_reply_without_reply_in_thread_when_disabled() -> None:
channel._client.im.v1.message.create.assert_called_once() channel._client.im.v1.message.create.assert_called_once()
@pytest.mark.asyncio
async def test_topic_reply_does_not_force_reply_in_thread_when_disabled() -> None:
"""Topic replies must not create new Feishu topics when reply_to_message is False."""
channel = _make_feishu_channel(reply_to_message=False)
reply_resp = MagicMock()
reply_resp.success.return_value = True
channel._client.im.v1.message.reply.return_value = reply_resp
await channel.send(OutboundMessage(
channel="feishu",
chat_id="oc_abc",
content="hello",
metadata={
"message_id": "om_child456",
"chat_type": "group",
"thread_id": "om_root123",
},
))
channel._client.im.v1.message.reply.assert_called_once()
call_args = channel._client.im.v1.message.reply.call_args
request = call_args[0][0]
assert request.request_body.reply_in_thread is not True
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_reply_keeps_fallback_when_reply_fails() -> None: async def test_reply_keeps_fallback_when_reply_fails() -> None:
"""Even with reply_to_message=True, fallback to create on reply failure.""" """Even with reply_to_message=True, fallback to create on reply failure."""

View File

@ -10,13 +10,14 @@ from nanobot.bus.queue import MessageBus
from nanobot.channels.feishu import FeishuChannel, FeishuConfig, _FeishuStreamBuf from nanobot.channels.feishu import FeishuChannel, FeishuConfig, _FeishuStreamBuf
def _make_channel(streaming: bool = True) -> FeishuChannel: def _make_channel(streaming: bool = True, reply_to_message: bool = False) -> FeishuChannel:
config = FeishuConfig( config = FeishuConfig(
enabled=True, enabled=True,
app_id="cli_test", app_id="cli_test",
app_secret="secret", app_secret="secret",
allow_from=["*"], allow_from=["*"],
streaming=streaming, streaming=streaming,
reply_to_message=reply_to_message,
) )
ch = FeishuChannel(config, MessageBus()) ch = FeishuChannel(config, MessageBus())
ch._client = MagicMock() ch._client = MagicMock()
@ -148,6 +149,42 @@ class TestSendDelta:
ch._client.im.v1.message.create.assert_called_once() ch._client.im.v1.message.create.assert_called_once()
ch._client.cardkit.v1.card_element.content.assert_called_once() ch._client.cardkit.v1.card_element.content.assert_called_once()
@pytest.mark.asyncio
async def test_group_delta_uses_create_when_reply_disabled(self):
ch = _make_channel(reply_to_message=False)
ch._client.cardkit.v1.card.create.return_value = _mock_create_card_response("card_new")
ch._client.im.v1.message.create.return_value = _mock_send_response("om_new")
ch._client.cardkit.v1.card_element.content.return_value = _mock_content_response()
await ch.send_delta(
"oc_chat1",
"Hello ",
metadata={"message_id": "om_001", "chat_type": "group"},
)
ch._client.im.v1.message.create.assert_called_once()
ch._client.im.v1.message.reply.assert_not_called()
@pytest.mark.asyncio
async def test_group_delta_replies_in_thread_when_reply_enabled(self):
ch = _make_channel(reply_to_message=True)
ch._client.cardkit.v1.card.create.return_value = _mock_create_card_response("card_new")
reply_resp = MagicMock()
reply_resp.success.return_value = True
ch._client.im.v1.message.reply.return_value = reply_resp
ch._client.cardkit.v1.card_element.content.return_value = _mock_content_response()
await ch.send_delta(
"oc_chat1",
"Hello ",
metadata={"message_id": "om_001", "chat_type": "group"},
)
ch._client.im.v1.message.reply.assert_called_once()
ch._client.im.v1.message.create.assert_not_called()
request = ch._client.im.v1.message.reply.call_args[0][0]
assert request.request_body.reply_in_thread is True
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_second_delta_within_interval_skips_update(self): async def test_second_delta_within_interval_skips_update(self):
ch = _make_channel() ch = _make_channel()
@ -204,6 +241,44 @@ class TestSendDelta:
ch._client.cardkit.v1.card_element.content.assert_not_called() ch._client.cardkit.v1.card_element.content.assert_not_called()
ch._client.im.v1.message.create.assert_called_once() ch._client.im.v1.message.create.assert_called_once()
@pytest.mark.asyncio
async def test_stream_end_fallback_group_uses_create_when_reply_disabled(self):
ch = _make_channel(reply_to_message=False)
ch._stream_bufs["om_001"] = _FeishuStreamBuf(
text="Fallback content", card_id=None, sequence=0, last_edit=0.0,
)
ch._client.im.v1.message.create.return_value = _mock_send_response("om_fb")
await ch.send_delta(
"oc_chat1",
"",
metadata={"_stream_end": True, "message_id": "om_001", "chat_type": "group"},
)
ch._client.im.v1.message.create.assert_called_once()
ch._client.im.v1.message.reply.assert_not_called()
@pytest.mark.asyncio
async def test_stream_end_fallback_group_replies_when_reply_enabled(self):
ch = _make_channel(reply_to_message=True)
ch._stream_bufs["om_001"] = _FeishuStreamBuf(
text="Fallback content", card_id=None, sequence=0, last_edit=0.0,
)
reply_resp = MagicMock()
reply_resp.success.return_value = True
ch._client.im.v1.message.reply.return_value = reply_resp
await ch.send_delta(
"oc_chat1",
"",
metadata={"_stream_end": True, "message_id": "om_001", "chat_type": "group"},
)
ch._client.im.v1.message.reply.assert_called_once()
ch._client.im.v1.message.create.assert_not_called()
request = ch._client.im.v1.message.reply.call_args[0][0]
assert request.request_body.reply_in_thread is True
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_stream_end_fallback_when_final_update_fails(self): async def test_stream_end_fallback_when_final_update_fails(self):
"""If streaming mode was closed (e.g. Feishu timeout), fall back to a regular card.""" """If streaming mode was closed (e.g. Feishu timeout), fall back to a regular card."""
@ -316,6 +391,40 @@ class TestToolHintInlineStreaming:
assert "oc_chat1" not in ch._stream_bufs assert "oc_chat1" not in ch._stream_bufs
ch._client.im.v1.message.create.assert_called_once() ch._client.im.v1.message.create.assert_called_once()
@pytest.mark.asyncio
async def test_tool_hint_group_uses_create_when_reply_disabled(self):
ch = _make_channel(reply_to_message=False)
ch._client.im.v1.message.create.return_value = _mock_send_response("om_hint")
msg = OutboundMessage(
channel="feishu", chat_id="oc_chat1",
content='read_file("path")',
metadata={"_tool_hint": True, "message_id": "om_001", "chat_type": "group"},
)
await ch.send(msg)
ch._client.im.v1.message.create.assert_called_once()
ch._client.im.v1.message.reply.assert_not_called()
@pytest.mark.asyncio
async def test_tool_hint_group_replies_when_reply_enabled(self):
ch = _make_channel(reply_to_message=True)
reply_resp = MagicMock()
reply_resp.success.return_value = True
ch._client.im.v1.message.reply.return_value = reply_resp
msg = OutboundMessage(
channel="feishu", chat_id="oc_chat1",
content='read_file("path")',
metadata={"_tool_hint": True, "message_id": "om_001", "chat_type": "group"},
)
await ch.send(msg)
ch._client.im.v1.message.reply.assert_called_once()
ch._client.im.v1.message.create.assert_not_called()
request = ch._client.im.v1.message.reply.call_args[0][0]
assert request.request_body.reply_in_thread is True
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_consecutive_tool_hints_append(self): async def test_consecutive_tool_hints_append(self):
"""When multiple tool hints arrive consecutively, each appends to the card.""" """When multiple tool hints arrive consecutively, each appends to the card."""