@keyou007/agent-bot 0.1.14-alpha.1 → 0.1.14-alpha.3

package/src/runtime/types.ts

@@ -221,6 +221,7 @@ export interface AgentRuntime {
   steerTurn(sessionId: string, turnId: string, prompt: RuntimePrompt): Promise<void>;
   cancelTurn(sessionId: string, turnId: string): Promise<void>;
   interruptRemoteTurn?(remoteSessionId: string, turnId: string): Promise<void>;
+  archiveRemoteSession?(remoteSessionId: string): Promise<void>;
   closeSession(sessionId: string): Promise<void>;
   setTitle?(sessionId: string, title: string): Promise<void>;
   getGoal?(sessionId: string): Promise<RuntimeGoal | undefined>;

package/src/state/StateStore.ts

@@ -344,6 +344,31 @@ export class StateStore {
     return rows.map(mapChatContext);
   }
 
+  removeChatContext(contextKey: string): string[] {
+    const threadPrefix = `${contextKey}:thread_id:`;
+    const linked = this.db.prepare(`
+      SELECT DISTINCT local_session_id
+      FROM context_sessions
+      WHERE context_key = ? OR instr(context_key, ?) = 1
+    `).all(contextKey, threadPrefix) as Array<{ local_session_id: string }>;
+    const remove = this.db.transaction(() => {
+      this.db.prepare(`
+        DELETE FROM context_sessions
+        WHERE context_key = ? OR instr(context_key, ?) = 1
+      `).run(contextKey, threadPrefix);
+      this.db.prepare(`
+        DELETE FROM user_contexts
+        WHERE context_key = ? OR instr(context_key, ?) = 1
+      `).run(contextKey, threadPrefix);
+      this.db.prepare(`
+        DELETE FROM chat_contexts
+        WHERE context_key = ? OR instr(context_key, ?) = 1
+      `).run(contextKey, threadPrefix);
+    });
+    remove();
+    return linked.map((row) => row.local_session_id);
+  }
+
   enqueuePrompt(input: {
     promptId: string;
     localSessionId: string;
@@ -743,6 +768,25 @@ export class StateStore {
       );
   }
 
+  archiveSession(localSessionId: string): void {
+    const now = new Date().toISOString();
+    const archive = this.db.transaction(() => {
+      this.db.prepare(`
+        UPDATE sessions
+        SET status = 'closed', updated_at = ?
+        WHERE local_session_id = ?
+      `).run(now, localSessionId);
+      this.db.prepare(`
+        UPDATE user_contexts
+        SET current_session_id = CASE WHEN current_session_id = ? THEN NULL ELSE current_session_id END,
+            previous_session_id = CASE WHEN previous_session_id = ? THEN NULL ELSE previous_session_id END,
+            updated_at = ?
+        WHERE current_session_id = ? OR previous_session_id = ?
+      `).run(localSessionId, localSessionId, now, localSessionId, localSessionId);
+    });
+    archive();
+  }
+
   updateRuntimeSession(
     localSessionId: string,
     patch: Partial<
@@ -1214,6 +1258,18 @@ export class StateStore {
     return rows.map((row) => ({ ...mapSession(row), contextKey }));
   }
 
+  listSessionsForChat(contextKey: string): SessionRecord[] {
+    const threadPrefix = `${contextKey}:thread_id:`;
+    const rows = this.db.prepare(`
+      SELECT DISTINCT sessions.*
+      FROM context_sessions
+      JOIN sessions ON sessions.local_session_id = context_sessions.local_session_id
+      WHERE context_sessions.context_key = ? OR instr(context_sessions.context_key, ?) = 1
+      ORDER BY sessions.updated_at DESC
+    `).all(contextKey, threadPrefix) as SessionRow[];
+    return rows.map(mapSession);
+  }
+
   listAllSessions(): SessionRecord[] {
     const rows = this.db
       .prepare("SELECT * FROM sessions ORDER BY updated_at DESC, created_at DESC")