refactor: 梳理规则体系并统一记忆与部署流程

backend/tests/unit/core/agentscope/events/test_store.py

@@ -59,16 +59,6 @@ def _patch_repositories(
     monkeypatch.setattr(store_module, "MessageRepository", _FakeMessageRepository)
     monkeypatch.setattr(store_module, "AgentChatSessionStatus", _SessionStatus)
 
-    async def _fake_stage_bit_map(self, *, session: object) -> dict[str, int]:
-        del self, session
-        return {"router": 16, "worker": 17, "memory": 18}
-
-    monkeypatch.setattr(
-        store_module.SqlAlchemyEventStore,
-        "_load_stage_visibility_bit_map",
-        _fake_stage_bit_map,
-    )
-
 
 @pytest.mark.asyncio
 async def test_store_persists_worker_output_with_answer_as_content(
@@ -113,7 +103,7 @@ async def test_store_persists_worker_output_with_answer_as_content(
     assert metadata["agent_output"]["answer"] == "worker-answer"
     assert metadata["agent_output"]["ui_hints"]["intent"] == "message"
     assert append_kwargs["cost"] == Decimal("0.123")
-    assert append_kwargs["visibility_mask"] == ((1 << 0) | (1 << 17))
+    assert append_kwargs["visibility_mask"] == ((1 << 0) | (1 << 1))
     assert captured["message_delta"] == 1
     assert captured["token_delta"] == 8
 
@@ -153,3 +143,65 @@ async def test_store_persists_tool_output_with_summary_as_content(
         == "status=success batch=1 success=1 failed=0 ids=[event-1]"
     )
     assert append_kwargs["visibility_mask"] == (1 << 0)
+
+
+@pytest.mark.asyncio
+async def test_store_persists_router_step_output_for_cost_tracking(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    captured: dict[str, object] = {}
+    fake_chat_session = SimpleNamespace(state_snapshot={}, message_count=10)
+    _patch_repositories(monkeypatch, captured, fake_chat_session)
+
+    store = store_module.SqlAlchemyEventStore(session_factory=lambda: _FakeSessionCtx())
+    await store.persist(
+        {
+            "type": "STEP_FINISHED",
+            "threadId": "00000000-0000-0000-0000-000000000001",
+            "runId": "run-router-1",
+            "stepName": "router",
+            "_router_persist": {
+                "router_output": {
+                    "normalized_task_input": {
+                        "user_text": "安排明天会议",
+                        "context_summary": "",
+                    },
+                    "key_entities": [],
+                    "constraints": [],
+                    "task_typing": {"primary": "scheduling"},
+                    "execution_mode": "tool_assisted",
+                    "result_typing": {"primary": "execution_report"},
+                    "ui": {
+                        "ui_mode": "none",
+                        "ui_decision_reason": "单任务",
+                    },
+                },
+                "response_metadata": {
+                    "model": "doubao-seed-1-6-250615",
+                    "inputTokens": 12,
+                    "outputTokens": 8,
+                    "cost": "0.01",
+                    "latencyMs": 320,
+                },
+            },
+        }
+    )
+
+    append_kwargs = cast(dict[str, Any], captured["append_kwargs"])
+    assert append_kwargs["seq"] == 11
+    assert append_kwargs["content"] == ""
+    assert append_kwargs["model_code"] == "doubao-seed-1-6-250615"
+    assert append_kwargs["input_tokens"] == 12
+    assert append_kwargs["output_tokens"] == 8
+    assert append_kwargs["latency_ms"] == 320
+    assert append_kwargs["cost"] == Decimal("0.01")
+    assert append_kwargs["visibility_mask"] == 0
+
+    metadata = cast(dict[str, Any], append_kwargs["metadata"])
+    assert sorted(metadata.keys()) == ["agent_type", "router_agent_output", "run_id"]
+    assert metadata["agent_type"] == "router"
+    assert metadata["router_agent_output"]["execution_mode"] == "tool_assisted"
+
+    assert captured["message_delta"] == 1
+    assert captured["token_delta"] == 20
+    assert captured["cost_delta"] == Decimal("0.01")