fix: use kw-only branch_id to satisfy seratch review and protocol compat

Author: HuxleyHu98

src/agents/extensions/memory/advanced_sqlite_session.py

@@ -143,22 +143,20 @@ async def add_items(
     async def get_items(
         self,
         limit: int | None = None,
-        wrapper: RunContextWrapper[Any] | str | None = None,
+        wrapper: RunContextWrapper[Any] | None = None,
+        *,
         branch_id: str | None = None,
     ) -> list[TResponseInputItem]:
         """Get items from current or specified branch.
 
         Args:
             limit: Maximum number of items to return. If None, uses session_settings.limit.
+            wrapper: Optional runtime wrapper for the current run context.
             branch_id: Branch to get items from. If None, uses current branch.
 
         Returns:
             List of conversation items from the specified branch.
         """
-        if isinstance(wrapper, str) and branch_id is None:
-            branch_id = wrapper
-            wrapper = None
-
         session_limit = resolve_session_limit(limit, self.session_settings)
 
         if branch_id is None:

tests/extensions/memory/test_advanced_sqlite_wrapper_compat.py

@@ -10,7 +10,7 @@
 pytestmark = pytest.mark.asyncio
 
 
-async def test_advanced_sqlite_get_items_preserves_branch_id_positional_argument() -> None:
+async def test_advanced_sqlite_get_items_branch_id_kwarg() -> None:
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "advanced.db"
         session = AdvancedSQLiteSession(session_id="test", db_path=db_path, create_tables=True)
@@ -29,7 +29,7 @@ async def test_advanced_sqlite_get_items_preserves_branch_id_positional_argument
         )
         await session.switch_to_branch("main")
 
-        branch_items = await session.get_items(50, "branch-a")
+        branch_items = await session.get_items(50, branch_id="branch-a")
         contents = [item.get("content") for item in branch_items if isinstance(item, dict)]
 
         assert "branch message" in contents