feature: 更新总结和code execute的功能

- 会话总结增加了角色判断,保证会话总结的是一次完整的用户对话
- code execute 增加了稳定性,避免循环运行tool和代码执行

Author: raychen911

tests/agents/core/test_history_processor.py

@@ -116,6 +116,18 @@ def test_invocation_mode_filters_by_id(self, invocation_context):
         assert len(events) == 1
         assert events[0].content.parts[0].text == "current"
 
+    def test_invocation_mode_includes_summary_events(self, invocation_context):
+        proc = HistoryProcessor(timeline_filter_mode=TimelineFilterMode.INVOCATION)
+        summary_event = _make_event("system", "Previous conversation summary", invocation_id="summary")
+        summary_event.set_summary_event(True)
+        current_event = _make_event("user", "current", invocation_id="inv-1")
+
+        events = proc.filter_events(invocation_context, [summary_event, current_event])
+
+        assert len(events) == 2
+        assert events[0].is_summary_event()
+        assert events[1].content.parts[0].text == "current"
+
 
 # ---------------------------------------------------------------------------
 # HistoryProcessor.filter_events - Branch filtering

tests/sessions/test_base_session_service.py

@@ -170,7 +170,8 @@ def test_filter_by_num_recent_events(self):
         svc = ConcreteSessionService(session_config=config)
         session = _make_session()
         for i in range(10):
-            session.events.append(_make_event(text=f"msg{i}"))
+            author = "user" if i == 7 else "agent"
+            session.events.append(_make_event(author=author, text=f"msg{i}"))
         svc.filter_events(session)
         assert len(session.events) == 10
         visible_events = [event for event in session.events if event.is_model_visible()]
@@ -185,7 +186,7 @@ def test_filter_by_event_ttl(self):
         old_event.timestamp = time.time() - 100
         session.events.append(old_event)
 
-        new_event = _make_event(text="new")
+        new_event = _make_event(author="user", text="new")
         new_event.timestamp = time.time()
         session.events.append(new_event)
 
@@ -215,6 +216,23 @@ def test_filter_ttl_removes_all_old(self):
         assert len(session.events) == 5
         assert all(not event.is_model_visible() for event in session.events)
 
+    def test_filter_by_num_recent_events_preserves_summary_anchor(self):
+        config = SessionServiceConfig(num_recent_events=3)
+        svc = ConcreteSessionService(session_config=config)
+        session = _make_session()
+
+        summary_event = _make_event(author="system", text="summary")
+        summary_event.set_summary_event(True)
+        session.events.append(summary_event)
+        for i in range(5):
+            session.events.append(_make_event(text=f"agent{i}"))
+
+        svc.filter_events(session)
+
+        visible_events = [event for event in session.events if event.is_model_visible()]
+        assert len(visible_events) == 1
+        assert visible_events[0].is_summary_event()
+
 
 class TestBaseSessionServiceSetSummarizerManager:
     """Test set_summarizer_manager method."""

tests/sessions/test_session.py

@@ -35,15 +35,18 @@ def test_add_event(self):
         assert session.events[0].author == "user"
         assert session.last_update_time == event.timestamp
 
-    def test_is_user_message(self):
-        """Test checking if an event is a user message."""
+    def test_is_anchor_message(self):
+        """Test checking if an event can anchor visible conversation history."""
         session = Session(id="test-session", app_name="test-app", user_id="test-user", save_key="test-key")
 
         user_event = Event(author="user", content=Content(parts=[Part.from_text(text="Hello")]))
         agent_event = Event(author="agent-1", content=Content(parts=[Part.from_text(text="Hi")]))
+        summary_event = Event(author="system", content=Content(parts=[Part.from_text(text="Summary")]))
+        summary_event.set_summary_event(True)
 
-        assert session._is_user_message(user_event) is True
-        assert session._is_user_message(agent_event) is False
+        assert session._is_anchor_message(user_event) is True
+        assert session._is_anchor_message(agent_event) is False
+        assert session._is_anchor_message(summary_event) is True
 
     def test_apply_event_filtering_no_config(self):
         """Test event filtering with no configuration."""

tests/sessions/test_session_summarizer.py

@@ -344,6 +344,107 @@ async def mock_generate(request, stream=False, ctx=None):
         visible_events = [event for event in result_events if event.is_model_visible()]
         assert len(visible_events) == 4  # 1 summary + 3 recent
         assert any(event.is_summary_event() for event in result_events)
+        summary_event = next(event for event in result_events if event.is_summary_event())
+        assert summary_event.author == "system"
+        assert summary_event.content.role == "user"
+
+    async def test_summary_filters_visible_events_and_starts_from_user(self):
+        model = _make_model_mock()
+        llm_response = MagicMock()
+        llm_response.content = Content(parts=[Part.from_text(text="summary text")])
+        captured_prompts = []
+
+        async def mock_generate(request, stream=False, ctx=None):
+            captured_prompts.append(request.contents[0].parts[0].text)
+            yield llm_response
+
+        model.generate_async = mock_generate
+        summarizer = SessionSummarizer(model=model, start_by_user_turn=True)
+        hidden_user = _make_event(author="user", text="hidden question")
+        hidden_user.set_model_visible(False)
+        first_user = _make_event(author="user", text="visible question")
+        old_answer = _make_event(author="agent", text="visible answer")
+        recent_user = _make_event(author="user", text="recent question")
+        events = [
+            _make_event(author="system", text="system preamble"),
+            hidden_user,
+            first_user,
+            old_answer,
+            recent_user,
+        ]
+
+        summary_text, result_events = await summarizer.create_session_summary_by_events(
+            events, "s1", keep_recent_count=1)
+
+        assert summary_text == "summary text"
+        assert result_events is events
+        assert captured_prompts
+        assert "visible question" in captured_prompts[0]
+        assert "visible answer" in captured_prompts[0]
+        assert "system preamble" not in captured_prompts[0]
+        assert "hidden question" not in captured_prompts[0]
+        assert "recent question" not in captured_prompts[0]
+        assert first_user.is_model_visible() is False
+        assert old_answer.is_model_visible() is False
+        assert recent_user.is_model_visible() is True
+        assert any(event.is_summary_event() for event in result_events)
+
+    async def test_summary_can_start_from_existing_summary_event(self):
+        model = _make_model_mock()
+        llm_response = MagicMock()
+        llm_response.content = Content(parts=[Part.from_text(text="summary text")])
+        captured_prompts = []
+
+        async def mock_generate(request, stream=False, ctx=None):
+            captured_prompts.append(request.contents[0].parts[0].text)
+            yield llm_response
+
+        model.generate_async = mock_generate
+        summarizer = SessionSummarizer(model=model, start_by_user_turn=True)
+        existing_summary = _make_event(author="system", text="previous summary")
+        existing_summary.set_summary_event(True)
+        events = [
+            _make_event(author="system", text="system preamble"),
+            existing_summary,
+            _make_event(author="agent", text="old answer"),
+            _make_event(author="user", text="recent question"),
+        ]
+
+        summary_text, result_events = await summarizer.create_session_summary_by_events(
+            events, "s1", keep_recent_count=1)
+
+        assert summary_text == "summary text"
+        assert "previous summary" in captured_prompts[0]
+        assert "old answer" in captured_prompts[0]
+        assert "system preamble" not in captured_prompts[0]
+        assert result_events[1].is_summary_event()
+
+    async def test_summary_falls_back_to_first_visible_event_and_ignores_large_keep_recent(self):
+        model = _make_model_mock()
+        llm_response = MagicMock()
+        llm_response.content = Content(parts=[Part.from_text(text="summary text")])
+        captured_prompts = []
+
+        async def mock_generate(request, stream=False, ctx=None):
+            captured_prompts.append(request.contents[0].parts[0].text)
+            yield llm_response
+
+        model.generate_async = mock_generate
+        summarizer = SessionSummarizer(model=model, start_by_user_turn=True)
+        events = [
+            _make_event(author="agent", text="agent message 1"),
+            _make_event(author="agent", text="agent message 2"),
+        ]
+
+        summary_text, result_events = await summarizer.create_session_summary_by_events(
+            events, "s1", keep_recent_count=10)
+
+        assert summary_text == "summary text"
+        assert "agent message 1" in captured_prompts[0]
+        assert "agent message 2" in captured_prompts[0]
+        visible_events = [event for event in result_events if event.is_model_visible()]
+        assert len(visible_events) == 1
+        assert visible_events[0].is_summary_event()
 
     async def test_summary_without_keep_recent(self):
         model = _make_model_mock()
@@ -362,6 +463,8 @@ async def mock_generate(request, stream=False, ctx=None):
         assert len(result_events) == 6  # preserve all original events + 1 summary
         visible_events = [event for event in result_events if event.is_model_visible()]
         assert len(visible_events) == 1  # only summary event remains model-visible
+        assert visible_events[0].is_summary_event()
+        assert visible_events[0].content.role == "user"
 
     async def test_summary_no_events(self):
         model = _make_model_mock()
@@ -407,6 +510,69 @@ async def mock_generate(request, stream=False, ctx=None):
         assert len(visible_events) == 3  # 1 summary + 2 recent
         assert any(event.is_summary_event() for event in session.events)
 
+    async def test_summary_uses_visible_events_starting_from_user(self):
+        model = _make_model_mock()
+        llm_response = MagicMock()
+        llm_response.content = Content(parts=[Part.from_text(text="session summary")])
+        captured_prompts = []
+
+        async def mock_generate(request, stream=False, ctx=None):
+            captured_prompts.append(request.contents[0].parts[0].text)
+            yield llm_response
+
+        model.generate_async = mock_generate
+        summarizer = SessionSummarizer(model=model, keep_recent_count=1, start_by_user_turn=True)
+        hidden_user = _make_event(author="user", text="hidden question")
+        hidden_user.set_model_visible(False)
+        first_user = _make_event(author="user", text="visible question")
+        old_answer = _make_event(author="agent", text="visible answer")
+        recent_user = _make_event(author="user", text="recent question")
+        session = _make_session(events=[
+            _make_event(author="system", text="system preamble"),
+            hidden_user,
+            first_user,
+            old_answer,
+            recent_user,
+        ])
+
+        result = await summarizer.create_session_summary(session)
+
+        assert result == "session summary"
+        assert captured_prompts
+        assert "visible question" in captured_prompts[0]
+        assert "visible answer" in captured_prompts[0]
+        assert "system preamble" not in captured_prompts[0]
+        assert "hidden question" not in captured_prompts[0]
+        assert "recent question" not in captured_prompts[0]
+        assert first_user.is_model_visible() is False
+        assert old_answer.is_model_visible() is False
+        assert recent_user.is_model_visible() is True
+        assert any(event.is_summary_event() for event in session.events)
+
+    async def test_summary_without_visible_user_falls_back_to_first_visible_event(self):
+        model = _make_model_mock()
+        llm_response = MagicMock()
+        llm_response.content = Content(parts=[Part.from_text(text="session summary")])
+
+        async def mock_generate(request, stream=False, ctx=None):
+            yield llm_response
+
+        model.generate_async = mock_generate
+        summarizer = SessionSummarizer(model=model, keep_recent_count=10, start_by_user_turn=True)
+        events = [
+            _make_event(author="system", text="system preamble"),
+            _make_event(author="agent", text="agent answer"),
+        ]
+        session = _make_session(events=events)
+
+        result = await summarizer.create_session_summary(session)
+
+        assert result == "session summary"
+        assert len(session.events) == 3
+        visible_events = [event for event in session.events if event.is_model_visible()]
+        assert len(visible_events) == 1
+        assert visible_events[0].is_summary_event()
+
     async def test_summary_no_update_on_failure(self):
         model = _make_model_mock()
 

tests/sessions/test_utils.py

@@ -18,17 +18,25 @@
 
 import pytest
 
+from trpc_agent_sdk.events import Event
 from trpc_agent_sdk.sessions._utils import (
     StateStorageEntry,
     app_state_key,
     extract_state_delta,
+    find_events_for_summary,
     merge_state,
     session_key,
     user_state_key,
 )
+from trpc_agent_sdk.types import Content
+from trpc_agent_sdk.types import Part
 from trpc_agent_sdk.types import State
 
 
+def _make_event(author: str = "agent", text: str = "hello") -> Event:
+    return Event(author=author, content=Content(parts=[Part.from_text(text=text)]))
+
+
 class TestStateStorageEntry:
     """Test StateStorageEntry dataclass."""
 
@@ -160,6 +168,95 @@ def test_need_copy_false(self):
         assert "new_key" in original_session_state
 
 
+class TestFindEventsForSummary:
+    """Test event selection for summary generation."""
+
+    def test_defaults_start_from_user_and_keep_recent(self):
+        events = [
+            _make_event(author="system", text="system preamble"),
+            _make_event(author="user", text="question"),
+            _make_event(author="agent", text="answer"),
+            _make_event(author="user", text="recent question"),
+        ]
+
+        selected_events, start_index = find_events_for_summary(events, keep_recent_count=1)
+
+        assert selected_events == events[1:3]
+        assert start_index == 1
+
+    def test_start_from_summary_event(self):
+        summary_event = _make_event(author="system", text="previous summary")
+        summary_event.set_summary_event(True)
+        events = [
+            _make_event(author="system", text="system preamble"),
+            summary_event,
+            _make_event(author="agent", text="answer"),
+            _make_event(author="user", text="recent question"),
+        ]
+
+        selected_events, start_index = find_events_for_summary(events, keep_recent_count=1)
+
+        assert selected_events == events[1:3]
+        assert start_index == 1
+
+    def test_fallback_to_first_visible_event(self):
+        events = [
+            _make_event(author="agent", text="answer 1"),
+            _make_event(author="agent", text="answer 2"),
+        ]
+
+        selected_events, start_index = find_events_for_summary(events, keep_recent_count=1)
+
+        assert selected_events == events[:1]
+        assert start_index == 0
+
+    def test_filters_model_invisible_events(self):
+        hidden_user = _make_event(author="user", text="hidden")
+        hidden_user.set_model_visible(False)
+        events = [
+            hidden_user,
+            _make_event(author="user", text="visible question"),
+            _make_event(author="agent", text="visible answer"),
+        ]
+
+        selected_events, start_index = find_events_for_summary(events, keep_recent_count=10)
+
+        assert selected_events == events[1:]
+        assert start_index == 1
+
+    def test_ignores_keep_recent_when_it_would_empty_selection(self):
+        events = [
+            _make_event(author="user", text="question"),
+            _make_event(author="agent", text="answer"),
+        ]
+
+        selected_events, start_index = find_events_for_summary(events, keep_recent_count=10)
+
+        assert selected_events == events
+        assert start_index == 0
+
+    def test_none_keep_recent_selects_all_matching_events(self):
+        events = [
+            _make_event(author="system", text="system preamble"),
+            _make_event(author="user", text="question"),
+            _make_event(author="agent", text="answer"),
+        ]
+
+        selected_events, start_index = find_events_for_summary(events, keep_recent_count=None)
+
+        assert selected_events == events[1:]
+        assert start_index == 1
+
+    def test_no_visible_events(self):
+        event = _make_event(author="user", text="hidden")
+        event.set_model_visible(False)
+
+        selected_events, start_index = find_events_for_summary([event])
+
+        assert selected_events == []
+        assert start_index == -1
+
+
 class TestKeyFunctions:
     """Test key generation functions."""