fix(memory): #2618 make SQLiteSession satisfy Session protocol (#2619)

Co-authored-by: Kazuhiro Sera <seratch@openai.com>

Author: lavish0000

examples/memory/file_session.py

@@ -14,11 +14,14 @@
 from uuid import uuid4
 
 from agents.memory.session import Session
+from agents.memory.session_settings import SessionSettings
 
 
 class FileSession(Session):
     """Persist session items to a JSON file on disk."""
 
+    session_settings: SessionSettings | None = None
+
     def __init__(self, *, dir: str | Path | None = None, session_id: str | None = None) -> None:
         self._dir = Path(dir) if dir is not None else Path.cwd() / ".agents-sessions"
         self.session_id = session_id or ""

src/agents/extensions/memory/dapr_session.py

@@ -57,6 +57,8 @@
 class DaprSession(SessionABC):
     """Dapr State Store implementation of :pyclass:`agents.memory.session.Session`."""
 
+    session_settings: SessionSettings | None = None
+
     def __init__(
         self,
         session_id: str,

src/agents/extensions/memory/redis_session.py

@@ -42,6 +42,8 @@
 class RedisSession(SessionABC):
     """Redis implementation of :pyclass:`agents.memory.session.Session`."""
 
+    session_settings: SessionSettings | None = None
+
     def __init__(
         self,
         session_id: str,

src/agents/extensions/memory/sqlalchemy_session.py

@@ -56,6 +56,7 @@ class SQLAlchemySession(SessionABC):
     _metadata: MetaData
     _sessions: Table
     _messages: Table
+    session_settings: SessionSettings | None = None
 
     def __init__(
         self,

src/agents/memory/openai_conversations_session.py

@@ -21,6 +21,8 @@ async def start_openai_conversations_session(openai_client: AsyncOpenAI | None =
 
 
 class OpenAIConversationsSession(SessionABC):
+    session_settings: SessionSettings | None = None
+
     def __init__(
         self,
         *,

src/agents/memory/sqlite_session.py

@@ -19,6 +19,8 @@ class SQLiteSession(SessionABC):
     For persistent storage, provide a file path.
     """
 
+    session_settings: SessionSettings | None = None
+
     def __init__(
         self,
         session_id: str,

tests/test_session_protocol_typing.py

@@ -0,0 +1,13 @@
+from agents import Session, SQLiteSession
+
+
+def _accept_session(session: Session) -> None:
+    """Static typing helper: mypy should accept concrete sessions here."""
+
+
+def test_sqlite_session_satisfies_session_protocol() -> None:
+    session = SQLiteSession("session_protocol_test")
+
+    _accept_session(session)
+
+    assert isinstance(session, Session)

tests/utils/simple_session.py

@@ -4,11 +4,14 @@
 
 from agents.items import TResponseInputItem
 from agents.memory.session import Session
+from agents.memory.session_settings import SessionSettings
 
 
 class SimpleListSession(Session):
     """A minimal in-memory session implementation for tests."""
 
+    session_settings: SessionSettings | None = None
+
     def __init__(
         self,
         session_id: str = "test",