fix: satisfy lint for wrapper propagation changes

Author: HuxleyHu98

examples/memory/file_session.py

@@ -14,8 +14,8 @@
 from uuid import uuid4
 
 from agents.memory.session import Session
-from agents.run_context import RunContextWrapper
 from agents.memory.session_settings import SessionSettings
+from agents.run_context import RunContextWrapper
 
 
 class FileSession(Session):

src/agents/extensions/memory/advanced_sqlite_session.py

@@ -8,15 +8,14 @@
 from pathlib import Path
 from typing import Any, Union, cast
 
-from ...run_context import RunContextWrapper
-
 from agents.result import RunResult
 from agents.usage import Usage
 
 from ..._tool_identity import is_reserved_synthetic_tool_namespace, tool_qualified_name
 from ...items import TResponseInputItem
 from ...memory import SQLiteSession
 from ...memory.session_settings import SessionSettings, resolve_session_limit
+from ...run_context import RunContextWrapper
 
 
 class AdvancedSQLiteSession(SQLiteSession):

src/agents/extensions/memory/async_sqlite_session.py

@@ -7,13 +7,12 @@
 from pathlib import Path
 from typing import Any, cast
 
-from ...run_context import RunContextWrapper
-
 import aiosqlite
 
 from ...items import TResponseInputItem
 from ...memory import SessionABC
 from ...memory.session_settings import SessionSettings
+from ...run_context import RunContextWrapper
 
 
 class AsyncSQLiteSession(SessionABC):

src/agents/extensions/memory/encrypt_session.py

@@ -31,8 +31,6 @@
 import json
 from typing import Any, cast
 
-from ...run_context import RunContextWrapper
-
 from cryptography.fernet import Fernet, InvalidToken
 from cryptography.hazmat.primitives import hashes
 from cryptography.hazmat.primitives.kdf.hkdf import HKDF
@@ -41,6 +39,7 @@
 from ...items import TResponseInputItem
 from ...memory.session import SessionABC
 from ...memory.session_settings import SessionSettings
+from ...run_context import RunContextWrapper
 
 
 class EncryptedEnvelope(TypedDict):
@@ -194,10 +193,14 @@ async def add_items(
         wrapper: RunContextWrapper[Any] | None = None,
     ) -> None:
         wrapped: list[EncryptedEnvelope] = [self._wrap(it) for it in items]
+        wrapped_items = cast(list[TResponseInputItem], wrapped)
         if wrapper is not None:
-            await self.underlying_session.add_items(cast(list[TResponseInputItem], wrapped), wrapper=wrapper)
+            await self.underlying_session.add_items(
+                wrapped_items,
+                wrapper=wrapper,
+            )
         else:
-            await self.underlying_session.add_items(cast(list[TResponseInputItem], wrapped))
+            await self.underlying_session.add_items(wrapped_items)
 
     async def pop_item(self) -> TResponseInputItem | None:
         while True:

src/agents/extensions/memory/sqlalchemy_session.py

@@ -27,8 +27,6 @@
 import json
 from typing import Any
 
-from ...run_context import RunContextWrapper
-
 from sqlalchemy import (
     TIMESTAMP,
     Column,
@@ -50,6 +48,7 @@
 from ...items import TResponseInputItem
 from ...memory.session import SessionABC
 from ...memory.session_settings import SessionSettings, resolve_session_limit
+from ...run_context import RunContextWrapper
 
 
 class SQLAlchemySession(SessionABC):

tests/utils/simple_session.py

@@ -3,9 +3,9 @@
 from typing import Any, cast
 
 from agents.items import TResponseInputItem
-from agents.run_context import RunContextWrapper
 from agents.memory.session import Session
 from agents.memory.session_settings import SessionSettings
+from agents.run_context import RunContextWrapper
 
 
 class SimpleListSession(Session):