feat: Enhance error handling and caching for Dify and WeCom interactions

tests/unit_tests/platform/test_telegram_adapter.py

@@ -53,6 +53,26 @@ def test_telegram_invalid_select_callback_is_rejected():
     assert _telegram_form_action_from_callback({'f': 1, 'x': 'invalid'}) is None
 
 
+def test_telegram_form_callback_cache_consumes_the_whole_form_group():
+    adapter = TelegramAdapter.model_construct()
+    adapter._form_action_titles = {}
+    adapter._cache_form_action_titles({'callback-a': 'A', 'callback-b': 'B'}, now=100.0)
+
+    assert adapter._take_form_action_title('callback-a', now=101.0) == 'A'
+    assert adapter._take_form_action_title('callback-a', now=101.0) is None
+    assert adapter._take_form_action_title('callback-b', now=101.0) is None
+    assert adapter._form_action_titles == {}
+
+
+def test_telegram_form_callback_cache_prunes_expired_entries():
+    adapter = TelegramAdapter.model_construct()
+    adapter._form_action_titles = {}
+    adapter._cache_form_action_titles({'callback-a': 'A'}, now=100.0)
+
+    assert adapter._take_form_action_title('callback-a', now=100.0 + adapter._FORM_ACTION_CACHE_TTL) is None
+    assert adapter._form_action_titles == {}
+
+
 @pytest.mark.asyncio
 async def test_telegram_select_field_sends_two_column_inline_keyboard():
     bot = MagicMock()

tests/unit_tests/platform/test_wecombot_template_card.py

@@ -9,6 +9,7 @@ logger_module.EventLogger = object
 sys.modules.setdefault('langbot.pkg.platform.logger', logger_module)
 
 from langbot.libs.wecom_ai_bot_api.api import (  # noqa: E402
+    WecomBotClient,
     build_button_interaction_payload,
     build_human_input_template_card_payload,
     build_button_interaction_update_card,
@@ -254,7 +255,7 @@ async def test_ws_push_form_pause_sends_text_prompt_without_empty_card():
 
 
 @pytest.mark.asyncio
-async def test_ws_stream_sends_delta_chunks_to_wecom():
+async def test_ws_stream_sends_cumulative_snapshots_to_wecom():
     client = WecomBotWsClient('bot-id', 'secret', object())
     client._stream_ids['msg-1'] = 'req-1|stream-1'
     client._stream_sessions['msg-1'] = {}
@@ -272,8 +273,29 @@ async def test_ws_stream_sends_delta_chunks_to_wecom():
 
     assert sent == [
         ('req-1', 'stream-1', '你', False),
-        ('req-1', 'stream-1', '好', False),
-        ('req-1', 'stream-1', '', True),
+        ('req-1', 'stream-1', '你好', False),
+        ('req-1', 'stream-1', '你好', True),
+    ]
+
+
+@pytest.mark.asyncio
+async def test_webhook_stream_queues_cumulative_snapshots_for_followups():
+    client = WecomBotClient('', '', '', object(), unified_mode=True)
+    session, _ = client.stream_sessions.create_or_get({'msgid': 'msg-1', 'chatid': '', 'from': {'userid': 'user-1'}})
+
+    assert await client.push_stream_chunk('msg-1', '你', is_final=False)
+    assert await client.push_stream_chunk('msg-1', '你好', is_final=False)
+    assert await client.push_stream_chunk('msg-1', '你好', is_final=True)
+
+    chunks = [
+        await client.stream_sessions.consume(session.stream_id),
+        await client.stream_sessions.consume(session.stream_id),
+        await client.stream_sessions.consume(session.stream_id),
+    ]
+    assert [(chunk.content, chunk.is_final) for chunk in chunks] == [
+        ('你', False),
+        ('你好', False),
+        ('你好', True),
     ]