aidev-bkplugin 2.2.0.post3__tar.gz → 2.2.0.post4__tar.gz

{aidev_bkplugin-2.2.0.post3 → aidev_bkplugin-2.2.0.post4}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: aidev-bkplugin
-Version: 2.2.0.post3
+Version: 2.2.0.post4
 Summary: AiDev Agent Bkplugin builtin functions
 Requires-Python: <3.12,>=3.11
-Requires-Dist: aidev-agent>=2.2.0post3
+Requires-Dist: aidev-agent>=2.2.0post4
 Requires-Dist: blueapps<5.0,>=4.15.1
 Requires-Dist: django<5.0,>=4.2.28
 Description-Content-Type: text/markdown

{aidev_bkplugin-2.2.0.post3 → aidev_bkplugin-2.2.0.post4}/aidev_bkplugin/services/agent_bkplugin.py

@@ -28,6 +28,7 @@ from aidev_agent.pydantic_models import ExecuteKwargs
 from aidev_agent.services.agent import AgentInstanceFactory
 from aidev_agent.services.common_agent import common_agent_factory
 from aidev_agent.services.event_handlers.agui_writer import AGUISessionWriter
+from aidev_agent.services.messages_handler import RetryableHeartbeatTimeoutError
 from aidev_agent.utils.local import request_local
 from django.contrib.auth import get_user_model
 
@@ -173,6 +174,13 @@ class BkpluginAgentRunner(ABC):
         """同步执行 Agent，返回最终 AI 回复字符串；异常时写回失败状态 (D-01)。"""
         try:
             return self._do_execute()
+        except RetryableHeartbeatTimeoutError:
+            logger.warning(
+                "[Bkplugin] skip session terminal update for retryable consumer heartbeat timeout session_code=%s",
+                self.session_code or "",
+                exc_info=True,
+            )
+            raise
         except Exception as e:
             logger.exception("[Bkplugin] execute error: %s", e)
             session_code = self.session_code or ""
@@ -212,7 +220,7 @@ class BkpluginAgentRunner(ABC):
         *,
         chat_context: list[dict] | None = None,
     ) -> None:
-        """Celery worker 入口：捕获异常并写 session 失败状态。"""
+        """Celery worker 入口：业务异常写失败；可重试消费异常保持 session 原状态。"""
         logger.info(
             "[Bkplugin] run_worker enter session_code=%s turn_id=%s thread=%s",
             session_code,
@@ -221,6 +229,14 @@ class BkpluginAgentRunner(ABC):
         )
         try:
             self.invoke_agent(session_code, execute_payload, chat_context=chat_context or [])
+        except RetryableHeartbeatTimeoutError:
+            logger.warning(
+                "[Bkplugin] skip session terminal update for retryable consumer heartbeat timeout "
+                "session_code=%s turn_id=%s",
+                session_code,
+                execute_payload.get("turn_id") or "",
+                exc_info=True,
+            )
         except Exception as e:
             logger.exception("[Bkplugin] worker error session_code=%s", session_code)
             manager = SessionManager(self.username or "")

{aidev_bkplugin-2.2.0.post3 → aidev_bkplugin-2.2.0.post4}/aidev_bkplugin/views/chat.py

@@ -397,9 +397,11 @@ class ChatCompletionViewSet(PluginViewSet):
         _preempted = False
         _client_disconnected = False
         _cancelled = False
+        _stream_completed = False
         try:
             for chunk in generator:
                 yield chunk
+            _stream_completed = True
         except GeneratorExit:
             _client_disconnected = True
             logger.info(f"[WRAP_STATUS] Client disconnected (GeneratorExit): session_code={session_code}")
@@ -421,7 +423,7 @@ class ChatCompletionViewSet(PluginViewSet):
                 if GeneratorStreamingHelper.is_cancelled(session_code):
                     _cancelled = True
                     logger.info(f"[WRAP_STATUS] Generator结束后检测到取消标志: session_code={session_code}")
-            if not _preempted and not _client_disconnected and not _cancelled:
+            if _stream_completed and not _preempted and not _client_disconnected and not _cancelled:
                 logger.info(f"[WRAP_STATUS] 流正常结束, 调用 set_streaming_finished: session_code={session_code}")
                 event_handler.set_streaming_finished()
 

{aidev_bkplugin-2.2.0.post3 → aidev_bkplugin-2.2.0.post4}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "aidev-bkplugin"
-version = "2.2.0post3"
+version = "2.2.0post4"
 description = "AiDev Agent Bkplugin builtin functions"
 readme = "readme.md"
 requires-python = "<3.12,>=3.11"
 dependencies = [
-    "aidev-agent>=2.2.0post3",
+    "aidev-agent>=2.2.0post4",
     "blueapps>=4.15.1, <5.0",
     "django>=4.2.28,<5.0",
 ]

{aidev_bkplugin-2.2.0.post3 → aidev_bkplugin-2.2.0.post4}/tests/services/test_bkplugin.py

@@ -23,6 +23,7 @@ sys.modules.setdefault("bk_plugin_framework.kit", _bk_plugin_framework.kit)
 sys.modules.setdefault("bk_plugin_framework.kit.decorators", _bk_plugin_framework.kit.decorators)
 
 from aidev_agent.enums import AgentType, ChatContentStatus, PromptRole, SessionsStatus  # noqa: E402
+from aidev_agent.services.messages_handler import RetryableHeartbeatTimeoutError  # noqa: E402
 from aidev_bkplugin.enums import PluginPollTaskState  # noqa: E402
 from aidev_bkplugin.services.agent_session import SessionManager  # noqa: E402
 
@@ -272,14 +273,10 @@ class TestBkpluginExecution:
         from aidev_bkplugin.services.agent_bkplugin import BkpluginChat
 
         agent = BkpluginChat(chat_history=[], execute_kwargs={}, username="alice")
-        monkeypatch.setattr(
-            agent, "invoke_agent", MagicMock(side_effect=RuntimeError("生产者心跳超时"))
-        )
+        monkeypatch.setattr(agent, "invoke_agent", MagicMock(side_effect=RuntimeError("生产者心跳超时")))
         mock_sm = MagicMock()
         mock_sm.retrieve_session.return_value = {"status": SessionsStatus.FINISHED.value}
-        monkeypatch.setattr(
-            "aidev_bkplugin.services.agent_bkplugin.SessionManager", lambda username: mock_sm
-        )
+        monkeypatch.setattr("aidev_bkplugin.services.agent_bkplugin.SessionManager", lambda username: mock_sm)
 
         agent.run_worker("sess-1", {"turn_id": "t1"})
 
@@ -291,16 +288,46 @@ class TestBkpluginExecution:
         from aidev_bkplugin.services.agent_bkplugin import BkpluginChat
 
         agent = BkpluginChat(chat_history=[], execute_kwargs={}, username="alice")
-        monkeypatch.setattr(
-            agent, "invoke_agent", MagicMock(side_effect=RuntimeError("生产者心跳超时"))
-        )
+        monkeypatch.setattr(agent, "invoke_agent", MagicMock(side_effect=RuntimeError("producer crashed")))
         mock_sm = MagicMock()
         mock_sm.retrieve_session.return_value = {"status": SessionsStatus.RUNNING.value}
+        monkeypatch.setattr("aidev_bkplugin.services.agent_bkplugin.SessionManager", lambda username: mock_sm)
+
+        agent.run_worker("sess-1", {"turn_id": "t1"})
+
+        mock_sm.save_stream_failure.assert_called_once_with("sess-1", ANY, turn_id="t1")
+
+    def test_run_worker_skips_session_update_for_retryable_heartbeat_timeout(self, monkeypatch):
+        """消费者心跳超时可重试，不应更新 session 终态。"""
+        from aidev_bkplugin.services.agent_bkplugin import BkpluginChat
+
+        agent = BkpluginChat(chat_history=[], execute_kwargs={}, username="alice")
         monkeypatch.setattr(
-            "aidev_bkplugin.services.agent_bkplugin.SessionManager", lambda username: mock_sm
+            agent,
+            "invoke_agent",
+            MagicMock(side_effect=RetryableHeartbeatTimeoutError("生产者心跳超时")),
         )
+        manager_factory = MagicMock()
+        monkeypatch.setattr("aidev_bkplugin.services.agent_bkplugin.SessionManager", manager_factory)
 
         agent.run_worker("sess-1", {"turn_id": "t1"})
 
-        mock_sm.save_stream_failure.assert_called_once_with("sess-1", ANY, turn_id="t1")
+        manager_factory.assert_not_called()
+
+    def test_execute_skips_session_update_for_retryable_heartbeat_timeout(self, monkeypatch):
+        """同步消费遇到可重试心跳超时，也不应覆盖 session 状态。"""
+        from aidev_bkplugin.services.agent_bkplugin import BkpluginChat
+
+        agent = BkpluginChat(chat_history=[], execute_kwargs={"session_code": "sess-1"}, username="alice")
+        monkeypatch.setattr(
+            agent,
+            "_do_execute",
+            MagicMock(side_effect=RetryableHeartbeatTimeoutError("生产者心跳超时")),
+        )
+        manager_factory = MagicMock()
+        monkeypatch.setattr("aidev_bkplugin.services.agent_bkplugin.SessionManager", manager_factory)
+
+        with pytest.raises(RetryableHeartbeatTimeoutError):
+            agent.execute()
 
+        manager_factory.assert_not_called()

aidev_bkplugin-2.2.0.post4/tests/views/test_chat_stream_status.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+"""Chat SSE 异常时的会话终态回归测试。"""
+
+from unittest.mock import MagicMock
+
+import pytest
+
+
+@pytest.mark.parametrize("error", [RuntimeError("生产者心跳超时"), SystemExit("worker abort")])
+def test_unexpected_stream_error_does_not_mark_session_finished(monkeypatch, error):
+    from aidev_agent.services.messages_handler.streaming_helper import GeneratorStreamingHelper
+    from aidev_bkplugin.views.chat import ChatCompletionViewSet
+
+    def failing_stream():
+        yield "chunk"
+        raise error
+
+    monkeypatch.setattr(GeneratorStreamingHelper, "is_cancelled", lambda _session_code: False)
+    event_handler = MagicMock()
+    stream = ChatCompletionViewSet()._wrap_streaming_with_status(
+        failing_stream(), event_handler, session_code="session-1"
+    )
+
+    assert next(stream) == "chunk"
+    with pytest.raises(type(error)):
+        next(stream)
+    event_handler.set_streaming_finished.assert_not_called()
+
+
+def test_normal_stream_marks_session_finished(monkeypatch):
+    from aidev_agent.services.messages_handler.streaming_helper import GeneratorStreamingHelper
+    from aidev_bkplugin.views.chat import ChatCompletionViewSet
+
+    monkeypatch.setattr(GeneratorStreamingHelper, "is_cancelled", lambda _session_code: False)
+    event_handler = MagicMock()
+
+    assert list(
+        ChatCompletionViewSet()._wrap_streaming_with_status(iter(["chunk"]), event_handler, session_code="session-1")
+    ) == ["chunk"]
+    event_handler.set_streaming_finished.assert_called_once_with()