rename to AdvancedSQLiteSession

Author: habema

…y/advanced_structured_session_example.py …ory/advanced_session_advanced_example.pyexamples/memory/advanced_structured_session_example.py renamed to examples/memory/advanced_session_advanced_example.py examples/memory/advanced_structured_session_example.py renamed to examples/memory/advanced_session_advanced_example.py

@@ -1,5 +1,5 @@
 """
-Advanced example demonstrating conversation branching with StructuredSQLiteSession.
+Advanced example demonstrating conversation branching with AdvancedSQLiteSession.
 
 This example shows how to use soft deletion for conversation editing/branching,
 allowing you to "undo" parts of a conversation and continue from any point.
@@ -8,7 +8,7 @@
 import asyncio
 
 from agents import Agent, Runner, function_tool
-from agents.extensions.memory import StructuredSQLiteSession
+from agents.extensions.memory import AdvancedSQLiteSession
 
 
 @function_tool
@@ -26,10 +26,10 @@ async def main():
         tools=[get_weather],
     )
 
-    # Create a structured session instance
-    session = StructuredSQLiteSession("conversation_advanced")
+    # Create a advanced session instance
+    session = AdvancedSQLiteSession("conversation_advanced")
 
-    print("=== Advanced Structured Session: Conversation Branching ===")
+    print("=== Advanced Session: Conversation Branching ===")
     print("This example demonstrates conversation editing and branching.\n")
 
     # Build initial conversation

…mory/basic_structured_session_example.py …memory/advanced_session_basic_example.pyexamples/memory/basic_structured_session_example.py renamed to examples/memory/advanced_session_basic_example.py examples/memory/basic_structured_session_example.py renamed to examples/memory/advanced_session_basic_example.py

@@ -1,14 +1,14 @@
 """
-Basic example demonstrating structured session memory functionality.
+Basic example demonstrating advanced session memory functionality.
 
-This example shows how to use StructuredSQLiteSession for conversation tracking
+This example shows how to use AdvancedSQLiteSession for conversation tracking
 with usage statistics and turn-based organization.
 """
 
 import asyncio
 
 from agents import Agent, Runner, function_tool
-from agents.extensions.memory import StructuredSQLiteSession
+from agents.extensions.memory import AdvancedSQLiteSession
 
 
 @function_tool
@@ -26,10 +26,10 @@ async def main():
         tools=[get_weather],
     )
 
-    # Create a structured session instance
-    session = StructuredSQLiteSession("conversation_basic")
+    # Create a advanced session instance
+    session = AdvancedSQLiteSession("conversation_basic")
 
-    print("=== Basic Structured Session Example ===")
+    print("=== Basic Advanced Session Example ===")
     print("The agent will remember previous messages with structured tracking.\n")
 
     # First turn

src/agents/extensions/memory/__init__.py

@@ -8,10 +8,10 @@
 """
 from __future__ import annotations
 
+from .advanced_sqlite_session import AdvancedSQLiteSession  # noqa: F401
 from .sqlalchemy_session import SQLAlchemySession  # noqa: F401
-from .structured_sqlite_session import StructuredSQLiteSession  # noqa: F401
 
 __all__: list[str] = [
     "SQLAlchemySession",
-    "StructuredSQLiteSession",
+    "AdvancedSQLiteSession",
 ]

…ions/memory/structured_sqlite_session.py …nsions/memory/advanced_sqlite_session.pysrc/agents/extensions/memory/structured_sqlite_session.py renamed to src/agents/extensions/memory/advanced_sqlite_session.py src/agents/extensions/memory/structured_sqlite_session.py renamed to src/agents/extensions/memory/advanced_sqlite_session.py

@@ -13,8 +13,14 @@
 from ...memory import SQLiteSession
 
 
-class StructuredSQLiteSession(SQLiteSession):
-    """SQLite session with simple structured metadata and soft deletion."""
+class AdvancedSQLiteSession(SQLiteSession):
+    """Enhanced SQLite session with turn tracking, soft deletion, and usage analytics.
+
+    Features:
+    - Turn-based conversation management with soft delete/reactivate
+    - Detailed usage tracking per turn with token breakdowns
+    - Message structure metadata and tool usage statistics
+    """
 
     ACTIVE = 1  # Message is active and visible in conversation
     INACTIVE = 0  # Message is soft-deleted (hidden but preserved)

…memory/test_structured_sqlite_session.py …s/memory/test_advanced_sqlite_session.pytests/extensions/memory/test_structured_sqlite_session.py renamed to tests/extensions/memory/test_advanced_sqlite_session.py tests/extensions/memory/test_structured_sqlite_session.py renamed to tests/extensions/memory/test_advanced_sqlite_session.py

@@ -1,4 +1,4 @@
-"""Tests for StructuredSQLiteSession functionality."""
+"""Tests for AdvancedSQLiteSession functionality."""
 
 import tempfile
 from pathlib import Path
@@ -10,7 +10,7 @@
 from openai.types.responses.response_usage import InputTokensDetails, OutputTokensDetails
 
 from agents import Agent, Runner, TResponseInputItem, function_tool
-from agents.extensions.memory import StructuredSQLiteSession
+from agents.extensions.memory import AdvancedSQLiteSession
 from agents.result import RunResult
 from agents.run_context import RunContextWrapper
 from agents.usage import Usage
@@ -77,12 +77,12 @@ def create_mock_run_result(
     )
 
 
-async def test_structured_session_basic_functionality(agent: Agent):
-    """Test basic StructuredSQLiteSession functionality."""
+async def test_advanced_session_basic_functionality(agent: Agent):
+    """Test basic AdvancedSQLiteSession functionality."""
     with tempfile.TemporaryDirectory() as temp_dir:
-        db_path = Path(temp_dir) / "test_structured.db"
-        session_id = "structured_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        db_path = Path(temp_dir) / "test_advanced.db"
+        session_id = "advanced_test"
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Test basic session operations work
         items: list[TResponseInputItem] = [
@@ -105,7 +105,7 @@ async def test_message_structure_tracking(agent: Agent):
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_structure.db"
         session_id = "structure_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Add various types of messages
         items: list[TResponseInputItem] = [
@@ -140,7 +140,7 @@ async def test_tool_usage_tracking(agent: Agent):
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_tools.db"
         session_id = "tools_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Add items with tool calls
         items: list[TResponseInputItem] = [
@@ -169,7 +169,7 @@ async def test_soft_deletion_functionality(agent: Agent):
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_deletion.db"
         session_id = "deletion_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Add multiple turns
         turn_1_items: list[TResponseInputItem] = [
@@ -224,7 +224,7 @@ async def test_usage_tracking_storage(agent: Agent, usage_data: Usage):
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_usage.db"
         session_id = "usage_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Simulate adding items for turn 1 to increment turn counter
         await session.add_items([{"role": "user", "content": "First turn"}])
@@ -285,9 +285,9 @@ async def test_runner_integration_with_usage_tracking(agent: Agent):
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_integration.db"
         session_id = "integration_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
-        async def store_session_usage(result: Any, session: StructuredSQLiteSession):
+        async def store_session_usage(result: Any, session: AdvancedSQLiteSession):
             """Helper function to store usage after runner completes."""
             try:
                 await session.store_run_usage(result)
@@ -335,7 +335,7 @@ async def test_sequence_ordering():
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_sequence.db"
         session_id = "sequence_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Add multiple items quickly to test sequence ordering
         items: list[TResponseInputItem] = [
@@ -362,7 +362,7 @@ async def test_conversation_structure_with_multiple_turns():
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_multi_turn.db"
         session_id = "multi_turn_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Turn 1
         turn_1: list[TResponseInputItem] = [
@@ -415,7 +415,7 @@ async def test_empty_session_operations():
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_empty.db"
         session_id = "empty_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Test getting items from empty session
         items = await session.get_items()
@@ -445,7 +445,7 @@ async def test_json_serialization_edge_cases(usage_data: Usage):
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_json.db"
         session_id = "json_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Test with normal usage data (need to add user message first to create turn)
         await session.add_items([{"role": "user", "content": "First test"}])
@@ -488,8 +488,8 @@ async def test_session_isolation():
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_isolation.db"
 
-        session1 = StructuredSQLiteSession("session_1", db_path)
-        session2 = StructuredSQLiteSession("session_2", db_path)
+        session1 = AdvancedSQLiteSession("session_1", db_path)
+        session2 = AdvancedSQLiteSession("session_2", db_path)
 
         # Add data to session 1
         await session1.add_items([{"role": "user", "content": "Session 1 message"}])
@@ -522,7 +522,7 @@ async def test_error_handling_in_usage_tracking(usage_data: Usage):
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_errors.db"
         session_id = "error_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Test normal operation
         run_result = create_mock_run_result(usage_data)
@@ -540,7 +540,7 @@ async def test_tool_name_extraction():
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_tool_names.db"
         session_id = "tool_names_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Add items with different ways of specifying tool names
         items: list[TResponseInputItem] = [
@@ -569,7 +569,7 @@ async def test_tool_execution_integration(agent: Agent):
     with tempfile.TemporaryDirectory() as temp_dir:
         db_path = Path(temp_dir) / "test_tool_integration.db"
         session_id = "tool_integration_test"
-        session = StructuredSQLiteSession(session_id, db_path)
+        session = AdvancedSQLiteSession(session_id, db_path)
 
         # Set up the fake model to trigger a tool call
         fake_model = cast(FakeModel, agent.model)