@dexto/core 1.7.1 → 1.8.0

package/dist/session/session-manager.js

@@ -1,8 +1,9 @@
-import "../chunk-PTJYTZNU.js";
+import "../chunk-C6A6W6XS.js";
 import { randomUUID } from "crypto";
 import { ChatSession } from "./chat-session.js";
 import { DextoLogComponent } from "../logger/v2/types.js";
 import { SessionError } from "./errors.js";
+import { normalizeTokenUsageForAccounting } from "../llm/usage-metadata.js";
 import { ZodError } from "zod";
 import {
   SessionPromptContributorSchema
@@ -22,7 +23,6 @@ class SessionManager {
   sessions = /* @__PURE__ */ new Map();
   maxSessions;
   sessionTTL;
-  static MESSAGE_QUEUE_KEY_PREFIX = "session-message-queue:";
   initialized = false;
   cleanupInterval;
   initializationPromise;
@@ -31,7 +31,6 @@ class SessionManager {
   // Per-session mutex for any SessionData read-modify-write path.
   sessionDataLocks = /* @__PURE__ */ new Map();
   logger;
-  static FORK_HISTORY_BATCH_SIZE = 500;
   static FORK_ID_GENERATION_MAX_ATTEMPTS = 5;
   static FORK_TITLE_PREFIX = "Fork: ";
   static FORK_PARENT_ID_PREVIEW_LENGTH = 8;
@@ -74,23 +73,21 @@ class SessionManager {
    */
   async restoreSessionsFromStorage() {
     try {
-      const sessionKeys = await this.services.storageManager.getDatabase().list("session:");
-      this.logger.debug(`Found ${sessionKeys.length} persisted sessions to restore`);
-      for (const sessionKey of sessionKeys) {
-        const sessionId = sessionKey.replace("session:", "");
-        const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+      const sessionIds = await this.services.sessionStore.listSessionIds();
+      this.logger.debug(`Found ${sessionIds.length} persisted sessions to restore`);
+      for (const sessionId of sessionIds) {
+        const sessionData = await this.services.sessionStore.getSession({ sessionId });
         if (sessionData) {
           const now = Date.now();
           const lastActivity = sessionData.lastActivity;
           if (now - lastActivity <= this.sessionTTL) {
             this.logger.debug(`Session ${sessionId} restored from storage`);
           } else {
-            await Promise.all([
-              this.services.storageManager.getDatabase().delete(sessionKey),
-              this.services.storageManager.getCache().delete(sessionKey),
-              this.deleteSessionInteractionState(sessionId)
-            ]);
-            this.logger.debug(`Expired session ${sessionId} cleaned up during restore`);
+            await this.services.sessionStore.evictSession({ sessionId });
+            this.evictSessionInteractionState(sessionId);
+            this.logger.debug(
+              `Expired session ${sessionId} evicted during restore; durable history preserved`
+            );
           }
         }
       }
@@ -102,18 +99,19 @@ class SessionManager {
   }
   async clearPersistedQueuedMessages(reason) {
     try {
-      const queueKeys = await this.services.storageManager.getDatabase().list(SessionManager.MESSAGE_QUEUE_KEY_PREFIX);
-      if (queueKeys.length === 0) {
+      const steerSessionIds = await this.services.steerQueueStore.listSessionIds();
+      const followUpSessionIds = await this.services.followUpQueueStore.listSessionIds();
+      const sessionIds = Array.from(/* @__PURE__ */ new Set([...steerSessionIds, ...followUpSessionIds]));
+      if (sessionIds.length === 0) {
         return;
       }
       await Promise.all(
-        queueKeys.map(
-          (key) => this.services.messageQueueStore.delete(
-            key.slice(SessionManager.MESSAGE_QUEUE_KEY_PREFIX.length)
-          )
-        )
+        sessionIds.flatMap((sessionId) => [
+          this.services.steerQueueStore.delete({ sessionId }),
+          this.services.followUpQueueStore.delete({ sessionId })
+        ])
       );
-      const message = `${reason === "startup" ? "Cleared stale queued follow-up state from previous agent run" : "Cleared queued follow-up state during agent shutdown"} (${queueKeys.length} session bucket(s))`;
+      const message = `${reason === "startup" ? "Cleared stale pending input state from previous agent run" : "Cleared pending input state during agent shutdown"} (${sessionIds.length} session bucket(s))`;
       if (reason === "startup") {
         this.logger.info(message);
       } else {
@@ -174,21 +172,17 @@ class SessionManager {
    */
   async forkSession(parentSessionId) {
     await this.ensureInitialized();
-    const database = this.services.storageManager.getDatabase();
-    const cache = this.services.storageManager.getCache();
-    const parentSessionKey = `session:${parentSessionId}`;
-    const parentMessagesKey = `messages:${parentSessionId}`;
-    const parentSessionData = await database.get(parentSessionKey);
+    const parentSessionData = await this.services.sessionStore.getSession({
+      sessionId: parentSessionId
+    });
     if (!parentSessionData) {
       throw SessionError.notFound(parentSessionId);
     }
-    const activeSessionKeys = await database.list("session:");
-    if (activeSessionKeys.length >= this.maxSessions) {
-      throw SessionError.maxSessionsExceeded(activeSessionKeys.length, this.maxSessions);
+    const activeSessionIds = await this.services.sessionStore.listSessionIds();
+    if (activeSessionIds.length >= this.maxSessions) {
+      throw SessionError.maxSessionsExceeded(activeSessionIds.length, this.maxSessions);
     }
     const childSessionId = await this.generateForkSessionId();
-    const childSessionKey = `session:${childSessionId}`;
-    const childMessagesKey = `messages:${childSessionId}`;
     const now = Date.now();
     const childTitle = this.buildForkTitle(parentSessionData, parentSessionId);
     const childSessionData = {
@@ -215,16 +209,19 @@ class SessionManager {
       }
     };
     try {
-      await database.set(childSessionKey, childSessionData);
-      await this.copySessionHistory(parentMessagesKey, childMessagesKey);
+      await this.services.sessionStore.saveSession({
+        sessionId: childSessionId,
+        session: childSessionData,
+        ttlSeconds: this.sessionTTL / 1e3
+      });
+      await this.copySessionHistory(parentSessionId, childSessionId);
       const childSession = await this.createSession(childSessionId);
       this.logger.info(`Forked session '${parentSessionId}' into child '${childSessionId}'`);
       return childSession;
     } catch (error) {
       await Promise.allSettled([
-        database.delete(childSessionKey),
-        database.delete(childMessagesKey),
-        cache.delete(childSessionKey)
+        this.services.sessionStore.deleteSession({ sessionId: childSessionId }),
+        this.services.conversationStore.clearMessages({ sessionId: childSessionId })
       ]);
       const inMemorySession = this.sessions.get(childSessionId);
       if (inMemorySession) {
@@ -245,13 +242,14 @@ class SessionManager {
     return `${prefix}${baseTitle}`;
   }
   async generateForkSessionId() {
-    const database = this.services.storageManager.getDatabase();
     for (let attempt = 0; attempt < SessionManager.FORK_ID_GENERATION_MAX_ATTEMPTS; attempt++) {
       const candidateId = randomUUID();
       if (this.sessions.has(candidateId) || this.pendingCreations.has(candidateId)) {
         continue;
       }
-      const existing = await database.get(`session:${candidateId}`);
+      const existing = await this.services.sessionStore.getSession({
+        sessionId: candidateId
+      });
       if (!existing) {
         return candidateId;
       }
@@ -261,25 +259,15 @@ class SessionManager {
       "failed to generate unique child session ID"
     );
   }
-  async copySessionHistory(parentMessagesKey, childMessagesKey) {
-    const database = this.services.storageManager.getDatabase();
-    let offset = 0;
-    while (true) {
-      const batch = await database.getRange(
-        parentMessagesKey,
-        offset,
-        SessionManager.FORK_HISTORY_BATCH_SIZE
-      );
-      if (batch.length === 0) {
-        return;
-      }
-      for (const message of batch) {
-        await database.append(childMessagesKey, message);
-      }
-      offset += batch.length;
-      if (batch.length < SessionManager.FORK_HISTORY_BATCH_SIZE) {
-        return;
-      }
+  async copySessionHistory(parentSessionId, childSessionId) {
+    const messages = await this.services.conversationStore.listMessages({
+      sessionId: parentSessionId
+    });
+    for (const message of messages) {
+      await this.services.conversationStore.saveMessage({
+        sessionId: childSessionId,
+        message
+      });
     }
   }
   /**
@@ -288,8 +276,7 @@ class SessionManager {
    */
   async createSessionInternal(id) {
     await this.cleanupExpiredSessions();
-    const sessionKey = `session:${id}`;
-    const existingMetadata = await this.services.storageManager.getDatabase().get(sessionKey);
+    const existingMetadata = await this.services.sessionStore.getSession({ sessionId: id });
     if (existingMetadata) {
       await this.updateSessionActivity(id);
       const runtimeConfig = this.services.stateManager.getRuntimeConfig();
@@ -299,7 +286,7 @@ class SessionManager {
         agentId,
         sessionId: id
       });
-      const sessionData2 = await this.services.storageManager.getDatabase().get(sessionKey);
+      const sessionData2 = await this.services.sessionStore.getSession({ sessionId: id });
       if (sessionData2?.llmOverride) {
         const { resolveApiKeyForProvider } = await import("../utils/api-key-resolver.js");
         const apiKey = resolveApiKeyForProvider(sessionData2.llmOverride.provider);
@@ -324,7 +311,7 @@ class SessionManager {
       this.logger.info(`Restored session from storage: ${id}`);
       return session2;
     }
-    const activeSessionKeys = await this.services.storageManager.getDatabase().list("session:");
+    const activeSessionKeys = await this.services.sessionStore.listSessionIds();
     if (activeSessionKeys.length >= this.maxSessions) {
       throw SessionError.maxSessionsExceeded(activeSessionKeys.length, this.maxSessions);
     }
@@ -338,7 +325,11 @@ class SessionManager {
       ...workspace?.id !== void 0 && { workspaceId: workspace.id }
     };
     try {
-      await this.services.storageManager.getDatabase().set(sessionKey, sessionData);
+      await this.services.sessionStore.saveSession({
+        sessionId: id,
+        session: sessionData,
+        ttlSeconds: this.sessionTTL / 1e3
+      });
     } catch (error) {
       this.logger.error(`Failed to store session metadata for ${id}:`, {
         error: error instanceof Error ? error.message : String(error)
@@ -357,15 +348,13 @@ class SessionManager {
       session = new ChatSession(this.getChatSessionServices(), id, sessionLogger);
       await session.init();
       this.sessions.set(id, session);
-      await this.services.storageManager.getCache().set(sessionKey, sessionData, this.sessionTTL / 1e3);
       this.logger.info(`Created new session: ${id}`);
       return session;
     } catch (error) {
       this.logger.error(
         `Failed to initialize session ${id}: ${error instanceof Error ? error.message : String(error)}`
       );
-      await this.services.storageManager.getDatabase().delete(sessionKey);
-      await this.services.storageManager.getCache().delete(sessionKey);
+      await this.services.sessionStore.deleteSession({ sessionId: id });
       const reason = error instanceof Error ? error.message : "unknown error";
       throw SessionError.initializationFailed(id, reason);
     }
@@ -386,8 +375,7 @@ class SessionManager {
       return this.sessions.get(sessionId);
     }
     if (restoreFromStorage) {
-      const sessionKey = `session:${sessionId}`;
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (sessionData) {
         const runtimeConfig = this.services.stateManager.getRuntimeConfig();
         const agentId = runtimeConfig.agentCard?.name ?? runtimeConfig.agentId;
@@ -439,8 +427,7 @@ class SessionManager {
       await session.cleanup();
       this.sessions.delete(sessionId);
     }
-    const sessionKey = `session:${sessionId}`;
-    await this.services.storageManager.getCache().delete(sessionKey);
+    await this.services.sessionStore.evictSession({ sessionId });
     this.evictSessionInteractionState(sessionId);
     this.logger.debug(
       `Ended session (removed from memory, chat history preserved): ${sessionId}`
@@ -459,12 +446,9 @@ class SessionManager {
       await session.cleanup();
       this.sessions.delete(sessionId);
     }
-    const sessionKey = `session:${sessionId}`;
-    await this.services.storageManager.getDatabase().delete(sessionKey);
-    await this.services.storageManager.getCache().delete(sessionKey);
+    await this.services.sessionStore.deleteSession({ sessionId });
     await this.deleteSessionInteractionState(sessionId);
-    const messagesKey = `messages:${sessionId}`;
-    await this.services.storageManager.getDatabase().delete(messagesKey);
+    await this.services.conversationStore.clearMessages({ sessionId });
     this.logger.debug(`Deleted session and conversation history: ${sessionId}`);
   }
   /**
@@ -480,7 +464,7 @@ class SessionManager {
       throw SessionError.notFound(sessionId);
     }
     await session.reset();
-    await session.clearMessageQueue();
+    await session.clearPendingInput();
     await Promise.all([
       this.services.toolManager.deleteSessionState(sessionId),
       this.services.approvalManager.deleteSessionState(sessionId)
@@ -489,15 +473,15 @@ class SessionManager {
       this.services.stateManager.clearSessionOverride(sessionId);
       await session.switchLLM(this.services.stateManager.getRuntimeConfig().llm);
     }
-    await this.runWithSessionDataLock(sessionId, async (sessionKey) => {
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    await this.runWithSessionDataLock(sessionId, async () => {
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (!sessionData) {
         return;
       }
       sessionData.messageCount = 0;
       sessionData.lastActivity = Date.now();
       delete sessionData.llmOverride;
-      await this.persistSessionData(sessionKey, sessionData);
+      await this.persistSessionData(sessionId, sessionData);
     });
     this.logger.debug(`Reset session conversation: ${sessionId}`);
   }
@@ -508,8 +492,7 @@ class SessionManager {
    */
   async listSessions() {
     await this.ensureInitialized();
-    const sessionKeys = await this.services.storageManager.getDatabase().list("session:");
-    return sessionKeys.map((key) => key.replace("session:", ""));
+    return await this.services.sessionStore.listSessionIds();
   }
   /**
    * Gets metadata for a specific session.
@@ -519,8 +502,7 @@ class SessionManager {
    */
   async getSessionMetadata(sessionId) {
     await this.ensureInitialized();
-    const sessionKey = `session:${sessionId}`;
-    const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    const sessionData = await this.services.sessionStore.getSession({ sessionId });
     if (!sessionData) return void 0;
     return {
       createdAt: sessionData.createdAt,
@@ -541,8 +523,7 @@ class SessionManager {
   }
   async getSessionSystemPromptContributors(sessionId) {
     await this.ensureInitialized();
-    const sessionKey = `session:${sessionId}`;
-    const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    const sessionData = await this.services.sessionStore.getSession({ sessionId });
     if (!sessionData) {
       throw SessionError.notFound(sessionId);
     }
@@ -550,8 +531,8 @@ class SessionManager {
   }
   async upsertSessionSystemPromptContributor(sessionId, contributor) {
     await this.ensureInitialized();
-    return await this.runWithSessionDataLock(sessionId, async (sessionKey) => {
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    return await this.runWithSessionDataLock(sessionId, async () => {
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (!sessionData) {
         throw SessionError.notFound(sessionId);
       }
@@ -563,14 +544,14 @@ class SessionManager {
       sessionData.metadata = sessionData.metadata || {};
       sessionData.metadata.systemPromptContributors = next;
       sessionData.lastActivity = Date.now();
-      await this.persistSessionData(sessionKey, sessionData);
+      await this.persistSessionData(sessionId, sessionData);
       return replaced;
     });
   }
   async removeSessionSystemPromptContributor(sessionId, contributorId) {
     await this.ensureInitialized();
-    return await this.runWithSessionDataLock(sessionId, async (sessionKey) => {
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    return await this.runWithSessionDataLock(sessionId, async () => {
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (!sessionData) {
         throw SessionError.notFound(sessionId);
       }
@@ -583,7 +564,7 @@ class SessionManager {
       sessionData.metadata = sessionData.metadata || {};
       sessionData.metadata.systemPromptContributors = next;
       sessionData.lastActivity = Date.now();
-      await this.persistSessionData(sessionKey, sessionData);
+      await this.persistSessionData(sessionId, sessionData);
       return true;
     });
   }
@@ -601,8 +582,8 @@ class SessionManager {
   }
   async markUntrackedChatGPTLoginUsage(sessionId) {
     await this.ensureInitialized();
-    await this.runWithSessionDataLock(sessionId, async (sessionKey) => {
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    await this.runWithSessionDataLock(sessionId, async () => {
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (!sessionData || sessionData.usageTracking?.hasUntrackedChatGPTLoginUsage) {
         return;
       }
@@ -610,7 +591,7 @@ class SessionManager {
         ...sessionData.usageTracking ?? {},
         hasUntrackedChatGPTLoginUsage: true
       };
-      await this.persistSessionData(sessionKey, sessionData);
+      await this.persistSessionData(sessionId, sessionData);
     });
   }
   /**
@@ -626,13 +607,13 @@ class SessionManager {
    * Updates the last activity timestamp for a session.
    */
   async updateSessionActivity(sessionId) {
-    await this.runWithSessionDataLock(sessionId, async (sessionKey) => {
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    await this.runWithSessionDataLock(sessionId, async () => {
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (!sessionData) {
         return;
       }
       sessionData.lastActivity = Date.now();
-      await this.persistSessionData(sessionKey, sessionData);
+      await this.persistSessionData(sessionId, sessionData);
     });
   }
   /**
@@ -640,14 +621,14 @@ class SessionManager {
    */
   async incrementMessageCount(sessionId) {
     await this.ensureInitialized();
-    await this.runWithSessionDataLock(sessionId, async (sessionKey) => {
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    await this.runWithSessionDataLock(sessionId, async () => {
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (!sessionData) {
         return;
       }
       sessionData.messageCount++;
       sessionData.lastActivity = Date.now();
-      await this.persistSessionData(sessionKey, sessionData);
+      await this.persistSessionData(sessionId, sessionData);
     });
   }
   /**
@@ -663,11 +644,13 @@ class SessionManager {
    */
   async accumulateTokenUsage(sessionId, usage, cost, modelInfo) {
     await this.ensureInitialized();
-    await this.runWithSessionDataLock(sessionId, async (sessionKey) => {
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    await this.runWithSessionDataLock(sessionId, async () => {
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
+      const normalizedUsage = normalizeTokenUsageForAccounting(usage);
+      const finiteCost = typeof cost === "number" && Number.isFinite(cost) ? cost : void 0;
       if (!sessionData) return;
       if (modelInfo) {
-        this.updateModelStats(sessionData, usage, cost, modelInfo);
+        this.updateModelStats(sessionData, normalizedUsage, finiteCost, modelInfo);
       }
       if (!sessionData.tokenUsage) {
         sessionData.tokenUsage = {
@@ -679,12 +662,12 @@ class SessionManager {
           totalTokens: 0
         };
       }
-      this.accumulateTokensInto(sessionData.tokenUsage, usage);
-      if (cost !== void 0) {
-        sessionData.estimatedCost = (sessionData.estimatedCost ?? 0) + cost;
+      this.accumulateTokensInto(sessionData.tokenUsage, normalizedUsage);
+      if (finiteCost !== void 0) {
+        sessionData.estimatedCost = (sessionData.estimatedCost ?? 0) + finiteCost;
       }
       sessionData.lastActivity = Date.now();
-      await this.persistSessionData(sessionKey, sessionData);
+      await this.persistSessionData(sessionId, sessionData);
     });
   }
   /**
@@ -745,8 +728,8 @@ class SessionManager {
   async setSessionTitle(sessionId, title, opts = {}) {
     await this.ensureInitialized();
     const normalized = title.trim().slice(0, 80);
-    await this.runWithSessionDataLock(sessionId, async (sessionKey) => {
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    await this.runWithSessionDataLock(sessionId, async () => {
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (!sessionData) {
         throw SessionError.notFound(sessionId);
       }
@@ -756,7 +739,7 @@ class SessionManager {
       sessionData.metadata = sessionData.metadata || {};
       sessionData.metadata.title = normalized;
       sessionData.lastActivity = Date.now();
-      await this.persistSessionData(sessionKey, sessionData);
+      await this.persistSessionData(sessionId, sessionData);
     });
   }
   /**
@@ -764,8 +747,7 @@ class SessionManager {
    */
   async getSessionTitle(sessionId) {
     await this.ensureInitialized();
-    const sessionKey = `session:${sessionId}`;
-    const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    const sessionData = await this.services.sessionStore.getSession({ sessionId });
     return sessionData?.metadata?.title;
   }
   /**
@@ -776,8 +758,7 @@ class SessionManager {
     const now = Date.now();
     const expiredSessions = [];
     for (const [sessionId, _session] of this.sessions.entries()) {
-      const sessionKey = `session:${sessionId}`;
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (sessionData && now - sessionData.lastActivity > this.sessionTTL) {
         expiredSessions.push(sessionId);
       }
@@ -887,8 +868,8 @@ class SessionManager {
     return configWithoutApiKey;
   }
   async setPersistedSessionLLMOverride(sessionId, llmOverride) {
-    await this.runWithSessionDataLock(sessionId, async (sessionKey) => {
-      const sessionData = await this.services.storageManager.getDatabase().get(sessionKey);
+    await this.runWithSessionDataLock(sessionId, async () => {
+      const sessionData = await this.services.sessionStore.getSession({ sessionId });
       if (!sessionData) {
         return;
       }
@@ -897,7 +878,7 @@ class SessionManager {
       } else {
         delete sessionData.llmOverride;
       }
-      await this.persistSessionData(sessionKey, sessionData);
+      await this.persistSessionData(sessionId, sessionData);
     });
   }
   async deleteSessionInteractionState(sessionId) {
@@ -905,7 +886,8 @@ class SessionManager {
     await Promise.all([
       this.services.toolManager.deleteSessionState(sessionId),
       this.services.approvalManager.deleteSessionState(sessionId),
-      this.services.messageQueueStore.delete(sessionId)
+      this.services.steerQueueStore.delete({ sessionId }),
+      this.services.followUpQueueStore.delete({ sessionId })
     ]);
   }
   evictSessionInteractionState(sessionId) {
@@ -916,7 +898,7 @@ class SessionManager {
     const sessionKey = `session:${sessionId}`;
     const previousLock = this.sessionDataLocks.get(sessionKey) ?? Promise.resolve();
     const currentResult = previousLock.catch(() => {
-    }).then(() => fn(sessionKey));
+    }).then(() => fn());
     const currentLock = currentResult.then(
       () => void 0,
       () => void 0
@@ -930,9 +912,12 @@ class SessionManager {
       }
     }
   }
-  async persistSessionData(sessionKey, sessionData) {
-    await this.services.storageManager.getDatabase().set(sessionKey, sessionData);
-    await this.services.storageManager.getCache().set(sessionKey, sessionData, this.sessionTTL / 1e3);
+  async persistSessionData(sessionId, sessionData) {
+    await this.services.sessionStore.saveSession({
+      sessionId,
+      session: sessionData,
+      ttlSeconds: this.sessionTTL / 1e3
+    });
   }
   /**
    * Get session statistics for monitoring and debugging.
@@ -956,8 +941,7 @@ class SessionManager {
    */
   async getSessionData(sessionId) {
     await this.ensureInitialized();
-    const sessionKey = `session:${sessionId}`;
-    return await this.services.storageManager.getDatabase().get(sessionKey);
+    return await this.services.sessionStore.getSession({ sessionId });
   }
   /**
    * Cleanup all sessions and resources.

package/dist/session/title-generator.cjs

@@ -23,10 +23,8 @@ __export(title_generator_exports, {
 });
 module.exports = __toCommonJS(title_generator_exports);
 var import_factory = require("../llm/services/factory.js");
-var import_events = require("../events/index.js");
-var import_memory = require("./history/memory.js");
-var import_message_queue = require("./message-queue.js");
-async function generateSessionTitle(config, toolManager, systemPromptManager, resourceManager, userText, logger, opts = {}) {
+var import_ai = require("ai");
+async function generateSessionTitle(config, userText, logger, opts = {}) {
   const timeoutMs = opts.timeoutMs;
   const controller = timeoutMs !== void 0 ? new AbortController() : void 0;
   let timer;
@@ -34,24 +32,12 @@ async function generateSessionTitle(config, toolManager, systemPromptManager, re
     timer = setTimeout(() => controller.abort(), timeoutMs);
   }
   try {
-    const history = new import_memory.MemoryHistoryProvider(logger);
-    const bus = new import_events.SessionEventBus();
-    const sessionId = `titlegen-${Math.random().toString(36).slice(2)}`;
-    const options = {
-      messageQueue: import_message_queue.MessageQueueService.createEphemeral(bus, logger, sessionId)
-    };
-    const tempService = (0, import_factory.createLLMService)(
+    const providerContext = opts.providerContext ?? {};
+    const model = opts.languageModelFactory?.({
       config,
-      toolManager,
-      systemPromptManager,
-      history,
-      bus,
-      sessionId,
-      resourceManager,
-      logger,
-      options,
-      opts.languageModelFactory
-    );
+      context: providerContext,
+      createDefaultLanguageModel: () => (0, import_factory.createVercelModel)(config, providerContext)
+    }) ?? (0, import_factory.createVercelModel)(config, providerContext);
     const instruction = [
       "Generate a short conversation title from the following user message.",
       "Rules: 3\u20138 words; no surrounding punctuation, emojis, or PII; return only the title.",
@@ -59,11 +45,13 @@ async function generateSessionTitle(config, toolManager, systemPromptManager, re
       "Message:",
       sanitizeUserText(userText, 512)
     ].join("\n");
-    const streamResult = await tempService.stream(
-      instruction,
-      controller ? { signal: controller.signal } : void 0
-    );
-    const processed = postProcessTitle(streamResult.text);
+    const result = await (0, import_ai.generateText)({
+      model,
+      prompt: instruction,
+      maxOutputTokens: 32,
+      ...controller ? { abortSignal: controller.signal } : {}
+    });
+    const processed = postProcessTitle(result.text);
     if (!processed) {
       return { error: "LLM returned empty title" };
     }

package/dist/session/title-generator.d.ts

@@ -1,9 +1,6 @@
 import type { ValidatedLLMConfig } from '../llm/schemas.js';
-import type { ToolManager } from '../tools/tool-manager.js';
-import type { SystemPromptManager } from '../systemPrompt/manager.js';
-import type { ResourceManager } from '../resources/index.js';
 import type { Logger } from '../logger/v2/types.js';
-import type { LanguageModelFactory } from '../llm/services/types.js';
+import type { DextoProviderContext, LanguageModelFactory } from '../llm/services/types.js';
 export interface GenerateSessionTitleResult {
     title?: string;
     error?: string;
@@ -13,9 +10,10 @@ export interface GenerateSessionTitleResult {
  * Generate a concise title for a chat based on the first user message.
  * Runs a lightweight, isolated LLM completion that does not touch real history.
  */
-export declare function generateSessionTitle(config: ValidatedLLMConfig, toolManager: ToolManager, systemPromptManager: SystemPromptManager, resourceManager: ResourceManager, userText: string, logger: Logger, opts?: {
-    timeoutMs?: number;
+export declare function generateSessionTitle(config: ValidatedLLMConfig, userText: string, logger: Logger, opts?: {
     languageModelFactory?: LanguageModelFactory;
+    providerContext?: DextoProviderContext;
+    timeoutMs?: number;
 }): Promise<GenerateSessionTitleResult>;
 /**
  * Heuristic fallback when the LLM-based title fails.

package/dist/session/title-generator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"title-generator.d.ts","sourceRoot":"","sources":["../../src/session/title-generator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAC5D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,KAAK,EAA2B,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAM9F,MAAM,WAAW,0BAA0B;IACvC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,OAAO,CAAC;CACtB;AAED;;;GAGG;AACH,wBAAsB,oBAAoB,CACtC,MAAM,EAAE,kBAAkB,EAC1B,WAAW,EAAE,WAAW,EACxB,mBAAmB,EAAE,mBAAmB,EACxC,eAAe,EAAE,eAAe,EAChC,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,IAAI,GAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,oBAAoB,CAAC,EAAE,oBAAoB,CAAA;CAAO,GAC/E,OAAO,CAAC,0BAA0B,CAAC,CAyDrC;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CA8BzE"}
+{"version":3,"file":"title-generator.d.ts","sourceRoot":"","sources":["../../src/session/title-generator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAC5D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,KAAK,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAI3F,MAAM,WAAW,0BAA0B;IACvC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,OAAO,CAAC;CACtB;AAED;;;GAGG;AACH,wBAAsB,oBAAoB,CACtC,MAAM,EAAE,kBAAkB,EAC1B,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,IAAI,GAAE;IACF,oBAAoB,CAAC,EAAE,oBAAoB,CAAC;IAC5C,eAAe,CAAC,EAAE,oBAAoB,CAAC;IACvC,SAAS,CAAC,EAAE,MAAM,CAAC;CACjB,GACP,OAAO,CAAC,0BAA0B,CAAC,CAgDrC;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CA8BzE"}