Session Management

from agkit.agents import InMemoryMemory

# Create memory instance
memory = InMemoryMemory()

# Create session for user A
session_a = 'user-alice-session-1'
await memory.add(
    {
        'message': {
            'id': 'msg-1',
            'role': 'user',
            'content': 'Hello, I\'m Alice',
            'timestamp': datetime.now()
        },
        'state': {'user_id': 'alice'}
    },
    session_id=session_a
)

# Create session for user B
session_b = 'user-bob-session-1'
await memory.add(
    {
        'message': {
            'id': 'msg-1',
            'role': 'user',
            'content': 'Hi, I\'m Bob',
            'timestamp': datetime.now()
        },
        'state': {'user_id': 'bob'}
    },
    session_id=session_b
)

# Retrieve session-specific history
alice_history = await memory.list(session_id=session_a)
bob_history = await memory.list(session_id=session_b)

print('Alice:', alice_history[0]['message']['content'])  # "Hello, I'm Alice"
print('Bob:', bob_history[0]['message']['content'])  # "Hi, I'm Bob"

# User-based: one session per user
session_id = f'user-{user_id}'

# Conversation-based: multiple sessions per user
session_id = f'user-{user_id}-conv-{conversation_id}'

# Time-based: sessions with timestamps
session_id = f'user-{user_id}-{int(time.time() * 1000)}'

# Hierarchical: complex organization structure
session_id = f'org-{org_id}/team-{team_id}/user-{user_id}'

from agkit.agents import InMemoryMemory
import uuid

memory = InMemoryMemory()

# Agent handles request with session isolation
async def handle_user_message(user_id: str, message: str):
    session_id = f'user-{user_id}'
    
    # Add user message to session
    await memory.add(
        {
            'message': {
                'id': str(uuid.uuid4()),
                'role': 'user',
                'content': message,
                'timestamp': datetime.now()
            }
        },
        session_id=session_id
    )
    
    # Get session history for agent
    history = await memory.list(session_id=session_id)
    
    # Process with agent...
    response = await agent.run(
        messages=[m['message'] for m in history],
        session_id=session_id
    )
    
    return response