morpheus-cli 0.3.3 → 0.3.6

package/dist/runtime/memory/sqlite.js

@@ -95,37 +95,37 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      */
     ensureTable() {
         try {
-            this.db.exec(`
-        CREATE TABLE IF NOT EXISTS messages (
-          id INTEGER PRIMARY KEY AUTOINCREMENT,
-          session_id TEXT NOT NULL,
-          type TEXT NOT NULL,
-          content TEXT NOT NULL,
-          created_at INTEGER NOT NULL,
-          input_tokens INTEGER,
-          output_tokens INTEGER,
-          total_tokens INTEGER,
-          cache_read_tokens INTEGER,
-          provider TEXT,
-          model TEXT
-        );
-        
-        CREATE INDEX IF NOT EXISTS idx_messages_session_id 
-        ON messages(session_id);
-
-        CREATE TABLE IF NOT EXISTS sessions (
-          id TEXT PRIMARY KEY,
-          title TEXT,
-          status TEXT CHECK (
-            status IN ('active', 'paused', 'archived', 'deleted')
-          ) NOT NULL DEFAULT 'paused',
-          started_at INTEGER NOT NULL,
-          ended_at INTEGER,
-          archived_at INTEGER,
-          deleted_at INTEGER,
-          embedding_status TEXT CHECK (embedding_status IN ('none', 'pending', 'embedded', 'failed')) NOT NULL DEFAULT 'none'
-        );
-
+            this.db.exec(`
+        CREATE TABLE IF NOT EXISTS messages (
+          id INTEGER PRIMARY KEY AUTOINCREMENT,
+          session_id TEXT NOT NULL,
+          type TEXT NOT NULL,
+          content TEXT NOT NULL,
+          created_at INTEGER NOT NULL,
+          input_tokens INTEGER,
+          output_tokens INTEGER,
+          total_tokens INTEGER,
+          cache_read_tokens INTEGER,
+          provider TEXT,
+          model TEXT
+        );
+        
+        CREATE INDEX IF NOT EXISTS idx_messages_session_id 
+        ON messages(session_id);
+
+        CREATE TABLE IF NOT EXISTS sessions (
+          id TEXT PRIMARY KEY,
+          title TEXT,
+          status TEXT CHECK (
+            status IN ('active', 'paused', 'archived', 'deleted')
+          ) NOT NULL DEFAULT 'paused',
+          started_at INTEGER NOT NULL,
+          ended_at INTEGER,
+          archived_at INTEGER,
+          deleted_at INTEGER,
+          embedding_status TEXT CHECK (embedding_status IN ('none', 'pending', 'embedded', 'failed')) NOT NULL DEFAULT 'none'
+        );
+
       `);
             this.migrateTable();
         }
@@ -174,10 +174,10 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
     async getMessages() {
         try {
             // Fetch new columns
-            const stmt = this.db.prepare(`SELECT type, content, input_tokens, output_tokens, total_tokens, cache_read_tokens, provider, model
-         FROM messages
-         WHERE session_id = ?
-         ORDER BY id DESC
+            const stmt = this.db.prepare(`SELECT type, content, input_tokens, output_tokens, total_tokens, cache_read_tokens, provider, model
+         FROM messages
+         WHERE session_id = ?
+         ORDER BY id DESC
          LIMIT ?`);
             const rows = stmt.all(this.sessionId, this.limit);
             return rows.map((row) => {
@@ -341,21 +341,21 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      */
     async setSessionTitleIfNeeded() {
         // Verificar se a sessão já tem título
-        const session = this.db.prepare(`
-      SELECT title FROM sessions
-      WHERE id = ?
+        const session = this.db.prepare(`
+      SELECT title FROM sessions
+      WHERE id = ?
     `).get(this.sessionId);
         if (session && session.title) {
             // A sessão já tem título, não precisa fazer nada
             return;
         }
         // Obter a mensagem mais antiga do tipo "human" da sessão
-        const oldestHumanMessage = this.db.prepare(`
-      SELECT content
-      FROM messages
-      WHERE session_id = ? AND type = 'human'
-      ORDER BY created_at ASC
-      LIMIT 1
+        const oldestHumanMessage = this.db.prepare(`
+      SELECT content
+      FROM messages
+      WHERE session_id = ? AND type = 'human'
+      ORDER BY created_at ASC
+      LIMIT 1
     `).get(this.sessionId);
         if (oldestHumanMessage) {
             // Pegar os primeiros 50 caracteres como título
@@ -412,16 +412,16 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      */
     async getUsageStatsByProviderAndModel() {
         try {
-            const stmt = this.db.prepare(`SELECT 
-          provider,
-          COALESCE(model, 'unknown') as model,
-          SUM(input_tokens) as totalInputTokens,
-          SUM(output_tokens) as totalOutputTokens,
-          SUM(total_tokens) as totalTokens,
-          COUNT(*) as messageCount
-        FROM messages
-        WHERE provider IS NOT NULL
-        GROUP BY provider, COALESCE(model, 'unknown')
+            const stmt = this.db.prepare(`SELECT 
+          provider,
+          COALESCE(model, 'unknown') as model,
+          SUM(input_tokens) as totalInputTokens,
+          SUM(output_tokens) as totalOutputTokens,
+          SUM(total_tokens) as totalTokens,
+          COUNT(*) as messageCount
+        FROM messages
+        WHERE provider IS NOT NULL
+        GROUP BY provider, COALESCE(model, 'unknown')
         ORDER BY provider, model`);
             const rows = stmt.all();
             return rows.map((row) => ({
@@ -485,26 +485,26 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
         // Transação para garantir consistência
         const tx = this.db.transaction(() => {
             // Pegar a sessão atualmente ativa
-            const activeSession = this.db.prepare(`
-        SELECT id FROM sessions
-        WHERE status = 'active'
+            const activeSession = this.db.prepare(`
+        SELECT id FROM sessions
+        WHERE status = 'active'
       `).get();
             // Se houver uma sessão ativa, mudar seu status para 'paused'
             if (activeSession) {
-                this.db.prepare(`
-          UPDATE sessions
-          SET status = 'paused'
-          WHERE id = ?
+                this.db.prepare(`
+          UPDATE sessions
+          SET status = 'paused'
+          WHERE id = ?
         `).run(activeSession.id);
             }
             // Criar uma nova sessão ativa
             const newId = randomUUID();
-            this.db.prepare(`
-        INSERT INTO sessions (
-          id,
-          started_at,
-          status
-        ) VALUES (?, ?, 'active')
+            this.db.prepare(`
+        INSERT INTO sessions (
+          id,
+          started_at,
+          status
+        ) VALUES (?, ?, 'active')
       `).run(newId, now);
             // Atualizar o ID da sessão atual desta instância
             this.sessionId = newId;
@@ -521,9 +521,9 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      */
     async archiveSession(sessionId) {
         // Validar sessão existe e está em active ou paused
-        const session = this.db.prepare(`
-      SELECT id, status FROM sessions
-      WHERE id = ?
+        const session = this.db.prepare(`
+      SELECT id, status FROM sessions
+      WHERE id = ?
     `).get(sessionId);
         if (!session) {
             throw new Error(`Sessão com ID ${sessionId} não encontrada.`);
@@ -535,20 +535,20 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
         // Transação para garantir consistência
         const tx = this.db.transaction(() => {
             // Marcar sessão como: status = 'archived', ended_at = now, archived_at = now, embedding_status = 'pending'
-            this.db.prepare(`
-        UPDATE sessions
-        SET status = 'archived',
-            ended_at = ?,
-            archived_at = ?,
-            embedding_status = 'pending'
-        WHERE id = ?
+            this.db.prepare(`
+        UPDATE sessions
+        SET status = 'archived',
+            ended_at = ?,
+            archived_at = ?,
+            embedding_status = 'pending'
+        WHERE id = ?
       `).run(now, now, sessionId);
             // Exportar mensagens → texto
-            const messages = this.db.prepare(`
-        SELECT type, content
-        FROM messages
-        WHERE session_id = ?
-        ORDER BY created_at ASC
+            const messages = this.db.prepare(`
+        SELECT type, content
+        FROM messages
+        WHERE session_id = ?
+        ORDER BY created_at ASC
       `).all(sessionId);
             if (messages.length > 0) {
                 const sessionText = messages
@@ -558,22 +558,22 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
                 if (this.dbSati) {
                     const chunks = this.chunkText(sessionText);
                     for (let i = 0; i < chunks.length; i++) {
-                        this.dbSati.prepare(`
-              INSERT INTO session_chunks (
-                id,
-                session_id,
-                chunk_index,
-                content,
-                created_at
-              ) VALUES (?, ?, ?, ?, ?)
+                        this.dbSati.prepare(`
+              INSERT INTO session_chunks (
+                id,
+                session_id,
+                chunk_index,
+                content,
+                created_at
+              ) VALUES (?, ?, ?, ?, ?)
             `).run(randomUUID(), sessionId, i, chunks[i], now);
                     }
                     this.display.log(`🧩 ${chunks.length} chunks criados para sessão ${sessionId}`, { source: 'Sati' });
                 }
                 // Remover mensagens da sessão após criar os chunks
-                this.db.prepare(`
-          DELETE FROM messages
-          WHERE session_id = ?
+                this.db.prepare(`
+          DELETE FROM messages
+          WHERE session_id = ?
         `).run(sessionId);
             }
         });
@@ -587,9 +587,9 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      */
     async deleteSession(sessionId) {
         // Validar sessão existe
-        const session = this.db.prepare(`
-      SELECT id, status FROM sessions
-      WHERE id = ?
+        const session = this.db.prepare(`
+      SELECT id, status FROM sessions
+      WHERE id = ?
     `).get(sessionId);
         if (!session) {
             throw new Error(`Sessão com ID ${sessionId} não encontrada.`);
@@ -602,33 +602,33 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
         // Transação: deletar mensagens da sessão, marcar sessão como: status = 'deleted', deleted_at = now
         const tx = this.db.transaction(() => {
             // Deletar mensagens da sessão
-            this.db.prepare(`
-        DELETE FROM messages
-        WHERE session_id = ?
+            this.db.prepare(`
+        DELETE FROM messages
+        WHERE session_id = ?
       `).run(sessionId);
             // Marcar sessão como: status = 'deleted', deleted_at = now
-            this.db.prepare(`
-        UPDATE sessions
-        SET status = 'deleted',
-            deleted_at = ?
-        WHERE id = ?
+            this.db.prepare(`
+        UPDATE sessions
+        SET status = 'deleted',
+            deleted_at = ?
+        WHERE id = ?
       `).run(now, sessionId);
         });
         tx(); // Executar a transação
         // Se a sessão era active, verificar se há outra para ativar
         if (session.status === 'active') {
-            const nextSession = this.db.prepare(`
-        SELECT id FROM sessions
-        WHERE status = 'paused'
-        ORDER BY started_at DESC
-        LIMIT 1
+            const nextSession = this.db.prepare(`
+        SELECT id FROM sessions
+        WHERE status = 'paused'
+        ORDER BY started_at DESC
+        LIMIT 1
       `).get();
             if (nextSession) {
                 // Promover a próxima sessão a ativa
-                this.db.prepare(`
-          UPDATE sessions
-          SET status = 'active'
-          WHERE id = ?
+                this.db.prepare(`
+          UPDATE sessions
+          SET status = 'active'
+          WHERE id = ?
         `).run(nextSession.id);
             }
             else {
@@ -644,9 +644,9 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      */
     async renameSession(sessionId, title) {
         // Validar sessão existe e status ∈ (paused, active)
-        const session = this.db.prepare(`
-      SELECT id, status FROM sessions
-      WHERE id = ?
+        const session = this.db.prepare(`
+      SELECT id, status FROM sessions
+      WHERE id = ?
     `).get(sessionId);
         if (!session) {
             throw new Error(`Sessão com ID ${sessionId} não encontrada.`);
@@ -657,10 +657,10 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
         // Transação para garantir consistência
         const tx = this.db.transaction(() => {
             // Atualizar o título da sessão
-            this.db.prepare(`
-        UPDATE sessions
-        SET title = ?
-        WHERE id = ?
+            this.db.prepare(`
+        UPDATE sessions
+        SET title = ?
+        WHERE id = ?
       `).run(title, sessionId);
         });
         tx(); // Executar a transação
@@ -673,9 +673,9 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      */
     async switchSession(targetSessionId) {
         // Validar sessão alvo: existe e status ∈ (paused, active)
-        const targetSession = this.db.prepare(`
-      SELECT id, status FROM sessions
-      WHERE id = ?
+        const targetSession = this.db.prepare(`
+      SELECT id, status FROM sessions
+      WHERE id = ?
     `).get(targetSessionId);
         if (!targetSession) {
             throw new Error(`Sessão alvo com ID ${targetSessionId} não encontrada.`);
@@ -690,23 +690,23 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
         // Transação: sessão atual active → paused, sessão alvo → active
         const tx = this.db.transaction(() => {
             // Pegar a sessão atualmente ativa
-            const currentActiveSession = this.db.prepare(`
-        SELECT id FROM sessions
-        WHERE status = 'active'
+            const currentActiveSession = this.db.prepare(`
+        SELECT id FROM sessions
+        WHERE status = 'active'
       `).get();
             // Se houver uma sessão ativa, mudar seu status para 'paused'
             if (currentActiveSession) {
-                this.db.prepare(`
-          UPDATE sessions
-          SET status = 'paused'
-          WHERE id = ?
+                this.db.prepare(`
+          UPDATE sessions
+          SET status = 'paused'
+          WHERE id = ?
         `).run(currentActiveSession.id);
             }
             // Mudar o status da sessão alvo para 'active'
-            this.db.prepare(`
-        UPDATE sessions
-        SET status = 'active'
-        WHERE id = ?
+            this.db.prepare(`
+        UPDATE sessions
+        SET status = 'active'
+        WHERE id = ?
       `).run(targetSessionId);
         });
         tx(); // Executar a transação
@@ -718,9 +718,9 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      */
     async getCurrentSessionOrCreate() {
         // Buscar sessão com status = 'active'
-        const activeSession = this.db.prepare(`
-      SELECT id FROM sessions
-      WHERE status = 'active'
+        const activeSession = this.db.prepare(`
+      SELECT id FROM sessions
+      WHERE status = 'active'
     `).get();
         if (activeSession) {
             // Se existir, retornar seu id
@@ -734,21 +734,21 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
     }
     async createFreshSession() {
         // Validar que não existe sessão 'active'
-        const activeSession = this.db.prepare(`
-      SELECT id FROM sessions
-      WHERE status = 'active'
+        const activeSession = this.db.prepare(`
+      SELECT id FROM sessions
+      WHERE status = 'active'
     `).get();
         if (activeSession) {
             throw new Error('Já existe uma sessão ativa. Não é possível criar uma nova sessão ativa.');
         }
         const now = Date.now();
         const newId = randomUUID();
-        this.db.prepare(`
-    INSERT INTO sessions (
-      id,
-      started_at,
-      status
-    ) VALUES (?, ?, 'active')
+        this.db.prepare(`
+    INSERT INTO sessions (
+      id,
+      started_at,
+      status
+    ) VALUES (?, ?, 'active')
   `).run(newId, now);
         return newId;
     }
@@ -782,11 +782,11 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      * Returns an array of session objects containing id, title, status, and started_at.
      */
     async listSessions() {
-        const sessions = this.db.prepare(`
-      SELECT id, title, status, started_at
-      FROM sessions
-      WHERE status IN ('active', 'paused')
-      ORDER BY started_at DESC
+        const sessions = this.db.prepare(`
+      SELECT id, title, status, started_at
+      FROM sessions
+      WHERE status IN ('active', 'paused')
+      ORDER BY started_at DESC
     `).all();
         return sessions;
     }