letagents 0.12.5 → 0.12.9

package/dist/mcp/local-state.js

@@ -177,6 +177,71 @@ export function setStoredAgentIdentity(agentIdentity, identityKey) {
     });
     return agentIdentity;
 }
+export function getStoredAgentSession(sessionId) {
+    if (!sessionId) {
+        return null;
+    }
+    const state = readLocalState();
+    return state.agent_sessions?.[sessionId] ?? null;
+}
+export function getCurrentAgentSession(roomId) {
+    const state = readLocalState();
+    const sessionIds = state.current_agent_session_ids;
+    if (!sessionIds) {
+        return null;
+    }
+    if (roomId) {
+        const sessionId = sessionIds[roomId];
+        const session = sessionId ? (state.agent_sessions?.[sessionId] ?? null) : null;
+        return session && !session.ended_at ? session : null;
+    }
+    let best = null;
+    for (const id of Object.values(sessionIds)) {
+        const session = state.agent_sessions?.[id];
+        if (session && !session.ended_at && (!best || session.updated_at > best.updated_at)) {
+            best = session;
+        }
+    }
+    return best;
+}
+export function saveAgentSession(session, makeCurrent = true) {
+    updateLocalState((state) => {
+        state.agent_sessions = state.agent_sessions ?? {};
+        state.agent_sessions[session.session_id] = session;
+        if (makeCurrent) {
+            state.current_agent_session_ids = state.current_agent_session_ids ?? {};
+            state.current_agent_session_ids[session.room_id] = session.session_id;
+        }
+        return state;
+    });
+    return session;
+}
+export function endStoredAgentSession(sessionId, endedAt = new Date().toISOString()) {
+    let endedSession = null;
+    updateLocalState((state) => {
+        const session = state.agent_sessions?.[sessionId];
+        if (!session) {
+            return state;
+        }
+        endedSession = {
+            ...session,
+            ended_at: endedAt,
+            updated_at: endedAt,
+            last_seen_at: endedAt,
+        };
+        state.agent_sessions = state.agent_sessions ?? {};
+        state.agent_sessions[sessionId] = endedSession;
+        if (state.current_agent_session_ids) {
+            for (const [roomId, currentSessionId] of Object.entries(state.current_agent_session_ids)) {
+                if (currentSessionId === sessionId) {
+                    delete state.current_agent_session_ids[roomId];
+                }
+            }
+        }
+        return state;
+    });
+    return endedSession;
+}
 export function getStoredCurrentRoom() {
     const state = readLocalState();
     return state.current_room ?? null;