Skip to content

Commit d18c874

Browse files
author
Ro-ee Tal
committed
feat: implemented pruning conversation manager
1 parent 94b41b4 commit d18c874

File tree

7 files changed

+1441
-0
lines changed

7 files changed

+1441
-0
lines changed

src/strands/agent/conversation_manager/__init__.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,19 +8,31 @@
88
size while preserving conversation coherence
99
- SummarizingConversationManager: An implementation that summarizes older context instead
1010
of simply trimming it
11+
- PruningConversationManager: An implementation that selectively prunes messages using configurable strategies
1112
1213
Conversation managers help control memory usage and context length while maintaining relevant conversation state, which
1314
is critical for effective agent interactions.
1415
"""
1516

1617
from .conversation_manager import ConversationManager
1718
from .null_conversation_manager import NullConversationManager
19+
from .pruning_conversation_manager import MessageContext, PruningContext, PruningConversationManager, PruningStrategy
1820
from .sliding_window_conversation_manager import SlidingWindowConversationManager
21+
22+
# Import strategies
23+
from .strategies import (
24+
LargeToolResultPruningStrategy,
25+
)
1926
from .summarizing_conversation_manager import SummarizingConversationManager
2027

2128
__all__ = [
2229
"ConversationManager",
2330
"NullConversationManager",
2431
"SlidingWindowConversationManager",
2532
"SummarizingConversationManager",
33+
"PruningConversationManager",
34+
"PruningStrategy",
35+
"PruningContext",
36+
"MessageContext",
37+
"LargeToolResultPruningStrategy",
2638
]

0 commit comments

Comments
 (0)