Function main

examples/memory/advanced_sqlite_session_example.py:22–274

()

async def main():
    # Create an agent
    agent = Agent(
        name="Assistant",
        instructions="Reply very concisely.",
        tools=[get_weather],
    )

    # Create an advanced session instance
    session = AdvancedSQLiteSession(
        session_id="conversation_comprehensive",
        create_tables=True,
    )

    print("=== AdvancedSQLiteSession Comprehensive Example ===")
    print("This example demonstrates both basic and advanced session features.\n")

    # === PART 1: Basic Session Functionality ===
    print("=== PART 1: Basic Session Memory ===")
    print("The agent will remember previous messages with structured tracking.\n")

    # First turn
    print("First turn:")
    print("User: What city is the Golden Gate Bridge in?")
    result = await Runner.run(
        agent,
        "What city is the Golden Gate Bridge in?",
        session=session,
    )
    print(f"Assistant: {result.final_output}")
    print(f"Usage: {result.context_wrapper.usage.total_tokens} tokens")

    # Store usage data automatically
    await session.store_run_usage(result)
    print()

    # Second turn - continuing the conversation
    print("Second turn:")
    print("User: What's the weather in that city?")
    result = await Runner.run(
        agent,
        "What's the weather in that city?",
        session=session,
    )
    print(f"Assistant: {result.final_output}")
    print(f"Usage: {result.context_wrapper.usage.total_tokens} tokens")

    # Store usage data automatically
    await session.store_run_usage(result)
    print()

    # Third turn
    print("Third turn:")
    print("User: What's the population of that city?")
    result = await Runner.run(
        agent,
        "What's the population of that city?",
        session=session,