from justapi import JustAPIApp

app = JustAPIApp()
app.enable_sessions(cookie_name="session_id")

@app.post("/sessions")
def create_session():
    session_id = app.create_session({"user_id": 1, "role": "admin"})
    return {"session_id": session_id}

@app.get("/sessions/{session_id}")
def get_session(session_id: str):
    data = app.get_session(session_id)
    if data is None:
        raise HTTPException(status_code=404, detail="Session not found")
    return data

@app.patch("/sessions/{session_id}")
def update_session(session_id: str, fields: dict):
    app.update_session(session_id, **fields)
    return {"status": "updated"}

@app.delete("/sessions/{session_id}")
def delete_session(session_id: str):
    app.delete_session(session_id)
    return {"status": "deleted"}

from justapi import Session

@app.get("/agent")
def agent_endpoint(request):
    session = Session(request)
    session.set("context", {"topic": "weather"})
    context = session.get("context")
    return {"context": context}