Skip to content

Commit 0027298

Browse files
jona62Jonathan James
andauthored
fix: swap event_timestamp with branch in add_turns (#99)
Co-authored-by: Jonathan James <[email protected]>
1 parent 107fd53 commit 0027298

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/bedrock_agentcore/memory/session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -889,7 +889,7 @@ def add_turns(
889889
event_timestamp: Optional[datetime] = None,
890890
) -> Event:
891891
"""Delegates to manager.add_turns."""
892-
return self._manager.add_turns(self._actor_id, self._session_id, messages, event_timestamp, branch)
892+
return self._manager.add_turns(self._actor_id, self._session_id, messages, branch, event_timestamp)
893893

894894
def fork_conversation(
895895
self,

tests/bedrock_agentcore/memory/test_session.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1867,8 +1867,8 @@ def test_session_delegation_with_optional_parameters(self):
18671867
"user-123",
18681868
"session-456",
18691869
[ConversationalMessage("Hello", MessageRole.USER)],
1870-
custom_timestamp,
18711870
branch,
1871+
custom_timestamp,
18721872
)
18731873

18741874
def test_comprehensive_error_coverage(self):
@@ -2110,13 +2110,13 @@ def test_session_add_turns_parameter_order(self):
21102110
event_timestamp=custom_timestamp,
21112111
)
21122112

2113-
# Verify the exact parameter order: actor_id, session_id, messages, event_timestamp, branch
2113+
# Verify the exact parameter order: actor_id, session_id, messages, branch, event_timestamp
21142114
mock_add_turns.assert_called_once_with(
21152115
"user-123",
21162116
"session-456",
21172117
[ConversationalMessage("Hello", MessageRole.USER)],
2118-
custom_timestamp,
21192118
branch,
2119+
custom_timestamp,
21202120
)
21212121

21222122

0 commit comments

Comments
 (0)