mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
fix: skip empty llm summaries (#8195)
This commit is contained in:
@@ -218,11 +218,15 @@ class LLMSummaryCompressor:
|
||||
# generate summary
|
||||
try:
|
||||
response = await self.provider.text_chat(contexts=llm_payload)
|
||||
summary_content = response.completion_text
|
||||
summary_content = (response.completion_text or "").strip()
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to generate summary: {e}")
|
||||
return messages
|
||||
|
||||
if not summary_content:
|
||||
logger.warning("LLM context compression returned an empty summary.")
|
||||
return messages
|
||||
|
||||
# build result
|
||||
result = []
|
||||
result.extend(system_messages)
|
||||
|
||||
@@ -94,6 +94,25 @@ class TestContextManager:
|
||||
|
||||
assert isinstance(manager.compressor, TruncateByTurnsCompressor)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_llm_compressor_keeps_history_when_summary_is_empty(self):
|
||||
from astrbot.core.agent.context.compressor import LLMSummaryCompressor
|
||||
|
||||
provider = MockProvider()
|
||||
provider.text_chat = AsyncMock(
|
||||
return_value=LLMResponse(role="assistant", completion_text=" ")
|
||||
)
|
||||
compressor = LLMSummaryCompressor(provider=provider, keep_recent=2) # type: ignore[arg-type]
|
||||
messages = self.create_messages(6)
|
||||
|
||||
with patch("astrbot.core.agent.context.compressor.logger") as mock_logger:
|
||||
result = await compressor(messages)
|
||||
|
||||
assert result == messages
|
||||
mock_logger.warning.assert_called_once_with(
|
||||
"LLM context compression returned an empty summary."
|
||||
)
|
||||
|
||||
# ==================== Empty and Edge Cases ====================
|
||||
|
||||
@pytest.mark.asyncio
|
||||
|
||||
Reference in New Issue
Block a user