Update docs

Signed-off-by: Aidan Jensen <[email protected]>

Author: artificial-aidan

docs/ref/extensions/memory.md

@@ -0,0 +1,7 @@
+# `Memory Extensions`
+
+::: agents.extensions.memory
+
+    options:
+        members:
+            - PostgreSQLSession

docs/sessions.md

@@ -141,6 +141,27 @@ result = await Runner.run(
 )
 ```
 
+### PostgreSQL memory
+
+```python
+from agents.extensions.memory import PostgreSQLSession
+from psycopg_pool import AsyncConnectionPool
+
+# From a connection string (creates a new connection pool)
+session = PostgreSQLSession.from_connection_string("user_123", "postgresql://user:pass@host/db")
+
+# From existing connection pool
+pool = AsyncConnectionPool(connection_string)
+session = PostgreSQLSession("user_123", pool)
+
+# Use the session
+result = await Runner.run(
+    agent,
+    "Hello",
+    session=session
+)
+```
+
 ### Multiple sessions
 
 ```python
@@ -221,7 +242,7 @@ Use meaningful session IDs that help you organize conversations:
 
 -   Use in-memory SQLite (`SQLiteSession("session_id")`) for temporary conversations
 -   Use file-based SQLite (`SQLiteSession("session_id", "path/to/db.sqlite")`) for persistent conversations
--   Consider implementing custom session backends for production systems (Redis, PostgreSQL, etc.)
+-   Consider implementing custom session backends for production systems (Redis, etc.)
 
 ### Session management
 
@@ -317,3 +338,4 @@ For detailed API documentation, see:
 
 -   [`Session`][agents.memory.Session] - Protocol interface
 -   [`SQLiteSession`][agents.memory.SQLiteSession] - SQLite implementation
+-   [`PostgreSQLSession`][agents.extensions.memory.PostgrSQLSession] - PostgreSQL implementation

mkdocs.yml

@@ -141,6 +141,7 @@ plugins:
                     - ref/extensions/handoff_filters.md
                     - ref/extensions/handoff_prompt.md
                     - ref/extensions/litellm.md
+                    - ref/extensions/memory.md
 
         - locale: ja
           name: 日本語

src/agents/extensions/memory/__init__.py

@@ -0,0 +1,3 @@
+from .postgres_session import PostgreSQLSession
+
+__all__ = ["PostgreSQLSession"]

tests/test_postgresql_session.py

@@ -7,10 +7,10 @@
 from psycopg.rows import TupleRow
 from psycopg_pool import AsyncConnectionPool
 
-from agents.extensions.memory.postgres_session import (
-    MessageRow,
+from agents.extensions.memory import (
     PostgreSQLSession,
 )
+from agents.extensions.memory.postgres_session import MessageRow
 from agents.items import TResponseInputItem