fix: fix session name typo in get_last_k_turns (#104)

Author: jona62

src/bedrock_agentcore/memory/session.py

@@ -622,7 +622,7 @@ def list_branches(self, actor_id: str, session_id: str) -> List[Branch]:
     def get_last_k_turns(
         self,
         actor_id: str,
-        sesssion_id: str,
+        session_id: str,
         k: int = 5,
         branch_name: Optional[str] = None,
         include_branches: bool = False,
@@ -639,7 +639,7 @@ def get_last_k_turns(
         try:
             events = self.list_events(
                 actor_id=actor_id,
-                session_id=sesssion_id,
+                session_id=session_id,
                 branch_name=branch_name,
                 include_parent_events=include_branches,
                 max_results=max_results,

tests/bedrock_agentcore/memory/test_session.py

@@ -942,7 +942,7 @@ def test_get_last_k_turns_success(self):
             with patch.object(manager, "list_events", return_value=mock_events):
                 result = manager.get_last_k_turns(
                     actor_id="user-123",
-                    sesssion_id="session-456",  # Note: typo in original method signature
+                    session_id="session-456",
                     k=2,
                 )
 
@@ -963,7 +963,7 @@ def test_get_last_k_turns_empty_events(self):
 
             # Mock empty list_events
             with patch.object(manager, "list_events", return_value=[]):
-                result = manager.get_last_k_turns(actor_id="user-123", sesssion_id="session-456", k=5)
+                result = manager.get_last_k_turns(actor_id="user-123", session_id="session-456", k=5)
 
                 assert result == []
 
@@ -987,7 +987,7 @@ def test_get_last_k_turns_client_error(self):
                 ),
             ):
                 with pytest.raises(ClientError):
-                    manager.get_last_k_turns(actor_id="user-123", sesssion_id="session-456", k=5)
+                    manager.get_last_k_turns(actor_id="user-123", session_id="session-456", k=5)
 
     def test_get_event_success(self):
         """Test get_event successful execution."""
@@ -1836,7 +1836,7 @@ def test_get_last_k_turns_turn_grouping(self):
                 )
             ]
             with patch.object(manager, "list_events", return_value=mock_events):
-                result = manager.get_last_k_turns(actor_id="user-123", sesssion_id="session-456", k=5)
+                result = manager.get_last_k_turns(actor_id="user-123", session_id="session-456", k=5)
 
                 # Should group into 2 turns
                 assert len(result) == 2
@@ -2027,7 +2027,7 @@ def test_get_last_k_turns_no_conversational_payload(self):
                 )
             ]
             with patch.object(manager, "list_events", return_value=mock_events):
-                result = manager.get_last_k_turns(actor_id="user-123", sesssion_id="session-456", k=5)
+                result = manager.get_last_k_turns(actor_id="user-123", session_id="session-456", k=5)
 
                 assert len(result) == 0  # No turns due to no conversational messages
 
@@ -2053,7 +2053,7 @@ def test_get_last_k_turns_break_on_k_limit(self):
                 )
 
             with patch.object(manager, "list_events", return_value=mock_events):
-                result = manager.get_last_k_turns(actor_id="user-123", sesssion_id="session-456", k=3)
+                result = manager.get_last_k_turns(actor_id="user-123", session_id="session-456", k=3)
 
                 # Should only return 3 turns even though there are 10 events
                 assert len(result) == 3
@@ -3016,7 +3016,7 @@ def test_get_last_k_turns_with_include_branches_true(self):
             with patch.object(manager, "list_events", return_value=mock_events) as mock_list_events:
                 result = manager.get_last_k_turns(
                     actor_id="user-123",
-                    sesssion_id="session-456",
+                    session_id="session-456",
                     k=2,
                     branch_name="test-branch",
                     include_branches=True,  # This should trigger include_parent_events=True