session.py

import uuid

class Session:
    def __init__(self, project_name):
        self.session_id = str(uuid.uuid4())
        self.project_name = project_name
        self.variables = {}
        self.auth_tokens = {}
        self.last_intent = None
        self.awaiting_variable = None
        self.chat_history = []

class SessionStore:
    def __init__(self):
        self.sessions = {}

    def create_session(self, project_name):
        session = Session(project_name)
        self.sessions[session.session_id] = session
        return session

    def get_session(self, session_id):
        return self.sessions.get(session_id)

    def remove_session(self, session_id):
        if session_id in self.sessions:
            del self.sessions[session_id]