fix: restore protocol-compatible session signatures with wrapper support

Author: HuxleyHu98

examples/memory/file_session.py

@@ -47,7 +47,6 @@ async def get_session_id(self) -> str:
     async def get_items(
         self,
         limit: int | None = None,
-        *,
         wrapper: RunContextWrapper[Any] | None = None,
     ) -> list[Any]:
         session_id = await self._ensure_session_id()
@@ -59,7 +58,6 @@ async def get_items(
     async def add_items(
         self,
         items: list[Any],
-        *,
         wrapper: RunContextWrapper[Any] | None = None,
     ) -> None:
         if not items:

src/agents/extensions/memory/advanced_sqlite_session.py

@@ -140,9 +140,8 @@ async def add_items(
     async def get_items(
         self,
         limit: int | None = None,
+        wrapper: RunContextWrapper[Any] | str | None = None,
         branch_id: str | None = None,
-        *,
-        wrapper: RunContextWrapper[Any] | None = None,
     ) -> list[TResponseInputItem]:
         """Get items from current or specified branch.
 
@@ -153,6 +152,10 @@ async def get_items(
         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: