kiku-agent 1.0.9 → 1.0.11

package/server/agent/session.js

@@ -1,244 +1,410 @@
-import { promises as fs } from 'node:fs';
-import path from 'node:path';
-import { EXPORT_DIR, SESSION_DIR as DATA_SESSION_DIR } from './appPaths.js';
-
-const SESSION_DIR = DATA_SESSION_DIR;
-const SESSION_FILE = path.join(SESSION_DIR, 'current_session.json');
-
-export class AgentSession {
-  constructor(options = {}) {
-    this.id = `sess_${Date.now()}_${Math.random().toString(36).slice(2, 7)}`;
-    this.createdAt = new Date();
-    this.thinkingMode = options.thinkingMode || 'hide';
-    this.messages = [];
-    this.stats = {
-      totalQueries: 0,
-      totalToolCalls: 0,
-      totalDurationMs: 0
-    };
-    this._storageWarningShown = false;
-  }
-
-  setThinkingMode(mode) {
-    if (mode === 'show' || mode === 'hide') {
-      this.thinkingMode = mode;
-      return true;
-    }
-    return false;
-  }
-
-  toggleThinkingMode() {
-    this.thinkingMode = this.thinkingMode === 'show' ? 'hide' : 'show';
-    return this.thinkingMode;
-  }
-
-  createUserMessage(text) {
-    const msg = {
-      id: `msg_${Date.now()}_u`,
-      role: 'user',
-      createdAt: new Date(),
-      parts: [
-        { type: 'text', content: text }
-      ]
-    };
-    this.messages.push(msg);
-    this.stats.totalQueries++;
-   
-    this.saveToDisk().catch(err => console.error('Session save failed:', err));
-    return msg;
-  }
-
-  createAssistantMessage() {
-    const msg = {
-      id: `msg_${Date.now()}_a`,
-      role: 'assistant',
-      createdAt: new Date(),
-      parts: []
-    };
-    this.messages.push(msg);
-    return msg;
-  }
-
-  addReasoningPart(message, reasoningText) {
-    let part = message.parts.find(p => p.type === 'reasoning');
-    if (!part) {
-      part = {
-        type: 'reasoning',
-        id: `rsn_${Date.now()}`,
-        content: '',
-        startedAt: Date.now(),
-        endedAt: null
-      };
-      message.parts.push(part);
-    }
-    part.content += reasoningText;
-    return part;
-  }
-
-  finalizeReasoningPart(message) {
-    const part = message.parts.find(p => p.type === 'reasoning');
-    if (part && !part.endedAt) {
-      part.endedAt = Date.now();
-      part.durationMs = part.endedAt - part.startedAt;
-    }
-  }
-
-  addToolPart(message, toolName, args) {
-    const part = {
-      type: 'tool',
-      id: `tool_${Date.now()}_${message.parts.length}`,
-      name: toolName,
-      args,
-      result: null,
-      startedAt: Date.now(),
-      endedAt: null,
-      durationMs: 0
-    };
-    message.parts.push(part);
-    this.stats.totalToolCalls++;
-    return part;
-  }
-
-  finalizeToolPart(message, toolId, result) {
-    const part = message.parts.find(p => p.id === toolId || (p.type === 'tool' && !p.endedAt));
-    if (part) {
-      part.result = result;
-      part.endedAt = Date.now();
-      part.durationMs = part.endedAt - part.startedAt;
-    }
-    return part;
-  }
-
-  addTextPart(message, textChunk) {
-    let part = message.parts.find(p => p.type === 'text');
-    if (!part) {
-      part = {
-        type: 'text',
-        content: ''
-      };
-      message.parts.push(part);
-    }
-    part.content += textChunk;
-    return part;
-  }
-
-  async saveToDisk() {
-    try {
-      await fs.mkdir(SESSION_DIR, { recursive: true });
-      const sessionData = {
-        id: this.id,
-        createdAt: this.createdAt,
-        thinkingMode: this.thinkingMode,
-        messages: this.messages,
-        stats: this.stats
-      };
-      await fs.writeFile(SESSION_FILE, JSON.stringify(sessionData, null, 2), 'utf-8');
-      return true;
-    } catch (error) {
-      if (!this._storageWarningShown) {
-        this._storageWarningShown = true;
-        console.error(`Kiku session is running without persistence (${error.code || 'unknown'}) in ${SESSION_DIR}. Set KIKU_DATA_DIR to a writable user folder; Administrator privileges are not required.`);
-      }
-      return false;
-    }
-  }
-
-  clear() {
-    this.messages = [];
-    this.stats.totalQueries = 0;
-    this.stats.totalToolCalls = 0;
-    this.stats.totalDurationMs = 0;
-  }
-
-  toSnapshot() {
-    return {
-      id: this.id,
-      createdAt: this.createdAt,
-      thinkingMode: this.thinkingMode,
-      messages: this.messages,
-      stats: { ...this.stats }
-    };
-  }
-
-  reset() {
-    this.id = `sess_${Date.now()}_${Math.random().toString(36).slice(2, 7)}`;
-    this.createdAt = new Date();
-    this.messages = [];
-    this.stats = {
-      totalQueries: 0,
-      totalToolCalls: 0,
-      totalDurationMs: 0
-    };
-  }
-
-  async loadFromDisk() {
-    try {
-      const data = await fs.readFile(SESSION_FILE, 'utf-8');
-      const sessionData = JSON.parse(data);
-      
-      this.id = sessionData.id;
-      this.createdAt = new Date(sessionData.createdAt);
-      this.thinkingMode = sessionData.thinkingMode || 'hide';
-      this.messages = sessionData.messages || [];
-      this.stats = sessionData.stats || {
-        totalQueries: 0,
-        totalToolCalls: 0,
-        totalDurationMs: 0
-      };
-      return true;
-    } catch (error) {
-     
-      return false;
-    }
-  }
-
-  async clearDisk() {
-    try {
-      await fs.unlink(SESSION_FILE);
-      return true;
-    } catch (error) {
-      return false;
-    }
-  }
-
-  async exportToMarkdown() {
-    const exportDir = EXPORT_DIR;
-    try {
-      await fs.mkdir(exportDir, { recursive: true });
-    } catch {}
-
-    const now = new Date();
-    const timestamp = now.toISOString().replace(/[:.]/g, '-').slice(0, 19);
-    const filename = `kiku-session-${timestamp}.md`;
-    const filePath = path.join(exportDir, filename);
-
-    let md = `# Kiku AI Agent - Session Transcript\n\n`;
-    md += `**Session ID:** \`${this.id}\`\n`;
-    md += `**Date:** ${now.toISOString().replace('T', ' ').slice(0, 19)} UTC\n`;
-    md += `**Total Queries:** ${this.stats.totalQueries} | **Tool Invocations:** ${this.stats.totalToolCalls}\n\n---\n\n`;
-
-    for (const msg of this.messages) {
-      if (msg.role === 'user') {
-        const textPart = msg.parts.find(p => p.type === 'text');
-        md += `## User\n${textPart?.content || ''}\n\n`;
-      } else if (msg.role === 'assistant') {
-        md += `## Kiku AI\n\n`;
-        for (const part of msg.parts) {
-          if (part.type === 'reasoning' && part.content) {
-            md += `<details><summary>Reasoning Process (${part.durationMs ? (part.durationMs/1000).toFixed(1) + 's' : 'Complete'})</summary>\n\n\`\`\`\n${part.content.trim()}\n\`\`\`\n</details>\n\n`;
-          } else if (part.type === 'tool') {
-            md += `> >> **Tool Execution: \`${part.name}\`** (${part.durationMs}ms)\n`;
-            md += `> - Arguments: \`${JSON.stringify(part.args)}\`\n`;
-            md += `> - Status: \`${part.result?.status || 'success'}\`\n\n`;
-          } else if (part.type === 'text') {
-            md += `${part.content}\n\n`;
-          }
-        }
-        md += `---\n\n`;
-      }
-    }
-
-    await fs.writeFile(filePath, md, 'utf-8');
-    return { filePath, filename };
-  }
-}
+import { promises as fs } from 'node:fs';
+import path from 'node:path';
+import { EXPORT_DIR, SESSION_DIR as DATA_SESSION_DIR } from './appPaths.js';
+
+const SESSION_DIR = DATA_SESSION_DIR;
+const SESSION_FILE = path.join(SESSION_DIR, 'current_session.json');
+
+export class AgentSession {
+  constructor(options = {}) {
+    this.id = `sess_${Date.now()}_${Math.random().toString(36).slice(2, 7)}`;
+    this.createdAt = new Date();
+    this.thinkingMode = options.thinkingMode || 'hide';
+    this.messages = [];
+    this.stats = {
+      totalQueries: 0,
+      totalToolCalls: 0,
+      totalDurationMs: 0,
+    };
+    this._storageWarningShown = false;
+  }
+
+  setThinkingMode(mode) {
+    if (mode === 'show' || mode === 'hide') {
+      this.thinkingMode = mode;
+      return true;
+    }
+    return false;
+  }
+
+  toggleThinkingMode() {
+    this.thinkingMode = this.thinkingMode === 'show' ? 'hide' : 'show';
+    return this.thinkingMode;
+  }
+
+  createUserMessage(text, meta = {}) {
+    this.stats.totalQueries++;
+    const msg = {
+      id: `msg_${Date.now()}_u`,
+      role: 'user',
+      queryNumber: meta.queryNumber || this.stats.totalQueries,
+      createdAt: new Date(),
+      parts: [
+        { type: 'text', content: String(text || '') }
+      ]
+    };
+    this.messages.push(msg);
+    this.saveToDisk().catch(() => {});
+    return msg;
+  }
+
+  createAssistantMessage(meta = {}) {
+    const msg = {
+      id: `msg_${Date.now()}_a`,
+      role: 'assistant',
+      queryNumber: meta.queryNumber || this.stats.totalQueries,
+      provider: meta.provider || null,
+      model: meta.model || null,
+      createdAt: new Date(),
+      durationMs: 0,
+      parts: []
+    };
+    this.messages.push(msg);
+    return msg;
+  }
+
+  addReasoningPart(message, reasoningText) {
+    if (!message || !Array.isArray(message.parts)) return null;
+    let part = message.parts.find(p => p.type === 'reasoning');
+    if (!part) {
+      part = {
+        type: 'reasoning',
+        id: `rsn_${Date.now()}`,
+        content: '',
+        startedAt: Date.now(),
+        endedAt: null,
+        durationMs: 0,
+      };
+      message.parts.push(part);
+    }
+    part.content += reasoningText;
+    return part;
+  }
+
+  finalizeReasoningPart(message) {
+    if (!message || !Array.isArray(message.parts)) return;
+    const part = message.parts.find(p => p.type === 'reasoning');
+    if (part && !part.endedAt) {
+      part.endedAt = Date.now();
+      part.durationMs = Math.max(0, part.endedAt - part.startedAt);
+    }
+  }
+
+  addToolPart(message, toolName, args) {
+    if (!message || !Array.isArray(message.parts)) return null;
+    const part = {
+      type: 'tool',
+      id: `tool_${Date.now()}_${message.parts.length}`,
+      name: toolName,
+      args: args || {},
+      result: null,
+      status: 'running',
+      startedAt: Date.now(),
+      endedAt: null,
+      durationMs: 0,
+    };
+    message.parts.push(part);
+    this.stats.totalToolCalls++;
+    return part;
+  }
+
+  finalizeToolPart(message, toolId, result) {
+    if (!message || !Array.isArray(message.parts)) return null;
+    const part = message.parts.find(p => p.id === toolId || (p.type === 'tool' && !p.endedAt));
+    if (part) {
+      part.result = result;
+      part.endedAt = Date.now();
+      part.durationMs = Math.max(0, part.endedAt - part.startedAt);
+      if (result?.status === 'error') {
+        part.status = 'error';
+      } else if (result?.status === 'not_found' || (result?.count === 0)) {
+        part.status = 'not_found';
+      } else {
+        part.status = 'ok';
+      }
+    }
+    return part;
+  }
+
+  addErrorPart(message, error) {
+    if (!message || !Array.isArray(message.parts)) return null;
+    const part = {
+      type: 'error',
+      id: `err_${Date.now()}`,
+      message: error?.message || String(error || 'Unknown error'),
+      code: error?.code || null,
+      createdAt: new Date(),
+    };
+    message.parts.push(part);
+    return part;
+  }
+
+  addTextPart(message, textChunk) {
+    if (!message || !Array.isArray(message.parts)) return null;
+    let part = message.parts.find(p => p.type === 'text');
+    if (!part) {
+      part = {
+        type: 'text',
+        content: ''
+      };
+      message.parts.push(part);
+    }
+    part.content += textChunk;
+    return part;
+  }
+
+  getToolStats() {
+    const stats = {};
+    for (const msg of this.messages) {
+      for (const part of msg.parts || []) {
+        if (part.type === 'tool') {
+          const name = part.name || 'unknown';
+          if (!stats[name]) {
+            stats[name] = { count: 0, totalDurationMs: 0, successCount: 0, errorCount: 0, notFoundCount: 0 };
+          }
+          stats[name].count++;
+          stats[name].totalDurationMs += part.durationMs || 0;
+          if (part.status === 'error') stats[name].errorCount++;
+          else if (part.status === 'not_found') stats[name].notFoundCount++;
+          else stats[name].successCount++;
+        }
+      }
+    }
+    return stats;
+  }
+
+  async saveToDisk() {
+    try {
+      await fs.mkdir(SESSION_DIR, { recursive: true });
+      const sessionData = {
+        id: this.id,
+        createdAt: this.createdAt,
+        thinkingMode: this.thinkingMode,
+        messages: this.messages,
+        stats: this.stats,
+      };
+      await fs.writeFile(SESSION_FILE, JSON.stringify(sessionData, null, 2), 'utf-8');
+      return true;
+    } catch (error) {
+      if (!this._storageWarningShown) {
+        this._storageWarningShown = true;
+        console.error(`Kiku session is running without persistence (${error.code || 'unknown'}) in ${SESSION_DIR}.`);
+      }
+      return false;
+    }
+  }
+
+  clear() {
+    this.messages = [];
+    this.stats.totalQueries = 0;
+    this.stats.totalToolCalls = 0;
+    this.stats.totalDurationMs = 0;
+  }
+
+  toSnapshot() {
+    return {
+      id: this.id,
+      createdAt: this.createdAt,
+      thinkingMode: this.thinkingMode,
+      messages: this.messages,
+      stats: { ...this.stats },
+    };
+  }
+
+  restoreFromSnapshot(sessionData) {
+    if (!sessionData) return false;
+    this.id = sessionData.id || `sess_${Date.now()}_${Math.random().toString(36).slice(2, 7)}`;
+    this.createdAt = sessionData.createdAt ? new Date(sessionData.createdAt) : new Date();
+    this.thinkingMode = sessionData.thinkingMode || 'hide';
+    this.messages = Array.isArray(sessionData.messages) ? sessionData.messages : [];
+    this.stats = sessionData.stats || {
+      totalQueries: 0,
+      totalToolCalls: 0,
+      totalDurationMs: 0
+    };
+    return true;
+  }
+
+  reset() {
+    this.id = `sess_${Date.now()}_${Math.random().toString(36).slice(2, 7)}`;
+    this.createdAt = new Date();
+    this.messages = [];
+    this.stats = {
+      totalQueries: 0,
+      totalToolCalls: 0,
+      totalDurationMs: 0
+    };
+  }
+
+  async loadFromDisk() {
+    try {
+      const data = await fs.readFile(SESSION_FILE, 'utf-8');
+      const sessionData = JSON.parse(data);
+      
+      this.id = sessionData.id;
+      this.createdAt = new Date(sessionData.createdAt);
+      this.thinkingMode = sessionData.thinkingMode || 'hide';
+      this.messages = sessionData.messages || [];
+      this.stats = sessionData.stats || {
+        totalQueries: 0,
+        totalToolCalls: 0,
+        totalDurationMs: 0
+      };
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  async clearDisk() {
+    try {
+      await fs.unlink(SESSION_FILE);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  getTranscriptMarkdown(options = {}) {
+    const now = new Date();
+    const createdStr = this.createdAt ? new Date(this.createdAt).toISOString().replace('T', ' ').slice(0, 19) : 'unknown';
+    const totalDurationSec = ((this.stats.totalDurationMs || 0) / 1000).toFixed(1);
+
+    let md = `# Kiku AI Agent — Session Transcript\n\n`;
+    md += `| Metadata | Value |\n`;
+    md += `| :--- | :--- |\n`;
+    md += `| **Session ID** | \`${this.id}\` |\n`;
+    md += `| **Created At** | ${createdStr} UTC |\n`;
+    md += `| **Total Queries** | ${this.stats.totalQueries} |\n`;
+    md += `| **Tool Invocations** | ${this.stats.totalToolCalls} |\n`;
+    md += `| **Total Duration** | ${totalDurationSec}s |\n`;
+    md += `| **Exported At** | ${now.toISOString().replace('T', ' ').slice(0, 19)} UTC |\n\n`;
+
+    const toolStats = this.getToolStats();
+    const toolEntries = Object.entries(toolStats);
+    if (toolEntries.length > 0) {
+      md += `### Tools Usage Summary\n\n`;
+      md += `| Tool Name | Invocations | OK | Not Found | Errors | Total Time |\n`;
+      md += `| :--- | :--- | :--- | :--- | :--- | :--- |\n`;
+      for (const [name, s] of toolEntries) {
+        const dur = (s.totalDurationMs / 1000).toFixed(2);
+        md += `| \`${name}\` | ${s.count} | ${s.successCount} | ${s.notFoundCount} | ${s.errorCount} | ${dur}s |\n`;
+      }
+      md += `\n---\n\n`;
+    }
+
+    let turnIndex = 1;
+    for (const msg of this.messages) {
+      if (msg.role === 'user') {
+        const textPart = msg.parts?.find(p => p.type === 'text');
+        const qNum = msg.queryNumber || turnIndex++;
+        md += `## Question ${qNum} (User)\n\n`;
+        md += `${textPart?.content || ''}\n\n`;
+      } else if (msg.role === 'assistant') {
+        const modelLabel = msg.model ? ` (${msg.model})` : '';
+        md += `### Kiku AI Response${modelLabel}\n\n`;
+
+        for (const part of msg.parts || []) {
+          if (part.type === 'reasoning' && part.content) {
+            const dur = part.durationMs ? `${(part.durationMs / 1000).toFixed(1)}s` : 'Complete';
+            md += `<details>\n<summary><b>Reasoning Process (${dur})</b></summary>\n\n\`\`\`\n${part.content.trim()}\n\`\`\`\n</details>\n\n`;
+          } else if (part.type === 'tool') {
+            const dur = part.durationMs !== undefined ? `${part.durationMs}ms` : '';
+            const statusTag = part.status === 'ok' ? '[OK]' : part.status === 'not_found' ? '[NOT FOUND]' : '[ERROR]';
+            md += `> **Tool Execution: \`${part.name}\`** ${statusTag} (${dur})\n`;
+            md += `> \n`;
+            md += `> Arguments:\n\`\`\`json\n${JSON.stringify(part.args, null, 2)}\n\`\`\`\n`;
+            if (part.result) {
+              md += `> Result:\n\`\`\`json\n${JSON.stringify(part.result, null, 2)}\n\`\`\`\n\n`;
+            }
+          } else if (part.type === 'error') {
+            md += `> **Error:** ${part.message}\n\n`;
+          } else if (part.type === 'text' && part.content) {
+            md += `${part.content}\n\n`;
+          }
+        }
+        md += `---\n\n`;
+      }
+    }
+
+    return md;
+  }
+
+  getTranscriptJson() {
+    return JSON.stringify({
+      id: this.id,
+      createdAt: this.createdAt,
+      stats: this.stats,
+      toolStats: this.getToolStats(),
+      messages: this.messages,
+    }, null, 2);
+  }
+
+  getTranscriptText() {
+    let out = `KIKU AGENT SESSION TRANSCRIPT\n`;
+    out += `Session ID: ${this.id}\n`;
+    out += `Created: ${this.createdAt}\n`;
+    out += `Queries: ${this.stats.totalQueries} | Tool Calls: ${this.stats.totalToolCalls}\n`;
+    out += `============================================================\n\n`;
+
+    for (const msg of this.messages) {
+      if (msg.role === 'user') {
+        const text = msg.parts?.find(p => p.type === 'text')?.content || '';
+        out += `[USER - Q${msg.queryNumber || '?'}]\n${text}\n\n`;
+      } else if (msg.role === 'assistant') {
+        out += `[KIKU AI${msg.model ? ` - ${msg.model}` : ''}]\n`;
+        for (const part of msg.parts || []) {
+          if (part.type === 'tool') {
+            out += `  -> Tool: ${part.name} (${part.status || 'done'})\n`;
+          } else if (part.type === 'text') {
+            out += `${part.content}\n`;
+          }
+        }
+        out += `\n------------------------------------------------------------\n\n`;
+      }
+    }
+    return out;
+  }
+
+  async exportToMarkdown(options = {}) {
+    const exportDir = EXPORT_DIR;
+    await fs.mkdir(exportDir, { recursive: true }).catch(() => {});
+
+    const now = new Date();
+    const timestamp = now.toISOString().replace(/[:.]/g, '-').slice(0, 19);
+    const filename = `kiku-session-${timestamp}.md`;
+    const filePath = path.join(exportDir, filename);
+
+    const md = this.getTranscriptMarkdown(options);
+    await fs.writeFile(filePath, md, 'utf-8');
+    return { filePath, filename, format: 'markdown', size: md.length };
+  }
+
+  async exportToJson() {
+    const exportDir = EXPORT_DIR;
+    await fs.mkdir(exportDir, { recursive: true }).catch(() => {});
+
+    const now = new Date();
+    const timestamp = now.toISOString().replace(/[:.]/g, '-').slice(0, 19);
+    const filename = `kiku-session-${timestamp}.json`;
+    const filePath = path.join(exportDir, filename);
+
+    const json = this.getTranscriptJson();
+    await fs.writeFile(filePath, json, 'utf-8');
+    return { filePath, filename, format: 'json', size: json.length };
+  }
+
+  async exportToText() {
+    const exportDir = EXPORT_DIR;
+    await fs.mkdir(exportDir, { recursive: true }).catch(() => {});
+
+    const now = new Date();
+    const timestamp = now.toISOString().replace(/[:.]/g, '-').slice(0, 19);
+    const filename = `kiku-session-${timestamp}.txt`;
+    const filePath = path.join(exportDir, filename);
+
+    const text = this.getTranscriptText();
+    await fs.writeFile(filePath, text, 'utf-8');
+    return { filePath, filename, format: 'text', size: text.length };
+  }
+}